nsqd: fix E_BAD_BODY for MPUB benchmarks

nsqio · May 1, 2015 · ed5ce72 · ed5ce72
1 parent e417b83
commit ed5ce72
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nsqd/protocol_v2_test.go b/nsqd/protocol_v2_test.go
@@ -1400,7 +1400,7 @@ func benchmarkProtocolV2Pub(b *testing.B, size int) {
 	opts.MemQueueSize = int64(b.N)
 	tcpAddr, _, nsqd := mustStartNSQD(opts)
 	msg := make([]byte, size)
-	batchSize := 200
+	batchSize := int(opts.MaxBodySize) / size
 	batch := make([][]byte, batchSize)
 	for i := range batch {
 		batch[i] = msg