Exemplo n.º 1
0
func apiSyncPut(topic string, b *testing.B) {
	var data []*message.Ins
	batch := 200
	for i := 0; i < batch; i++ {
		d := message.NewData([]byte(utils.RandString(200)))
		data = append(data, message.NewByData(d))
	}

	client, err := api.New(":12345")
	if err != nil {
		b.Fatal(err)
	}
	defer client.Close()

	if b.N < batch {
		_, err := client.Put(topic, data[:b.N])
		if err != nil {
			b.Error(err)
		}
		return
	}
	for i := 0; i < b.N; i += batch {
		_, err := client.Put(topic, data)
		if err != nil {
			b.Error(err)
			return
		}
	}
}
Exemplo n.º 2
0
func BenchmarkApiSyncGet(b *testing.B) {
	topic := "api-put"
	apiSyncPut(topic, b)
	b.ResetTimer()
	client, err := api.New(":12345")
	if err != nil {
		b.Fatal(err)
	}
	defer client.Close()
	offset := int64(0)
	bench := 200
	total := b.N
	for total > 0 {
		read := min(total, bench)
		total -= read

		if err := client.Get(topic, offset, read); err != nil {
			b.Fatal(err)
		}
		for reply := range client.ReplyChan() {
			read -= len(reply.Msgs)
			if read == 0 {
				offset = reply.Offset
				break
			}
		}
	}
}