cmd/seb benchmark: generate random batch data once

micvbang · Jul 10, 2024 · e02105a · e02105a
1 parent 76ee866
commit e02105a
Showing 1 changed file with 13 additions and 13 deletions.
diff --git a/cmd/seb/app/benchmark.go b/cmd/seb/app/benchmark.go
@@ -268,22 +268,22 @@ func generateBatches(log logger.Logger, batches chan<- recordBatch, numBatches i
 	randSource := rand.New(rand.NewSource(1))
 	batchSize := recordsPerBatch * recordSize
 
-	for range numBatches {
-		records := make([]byte, batchSize)
-		n, err := randSource.Read(records)
-		if err != nil {
-			log.Fatalf("failed to generate random data: %s", err)
-		}
+	records := make([]byte, batchSize)
+	n, err := randSource.Read(records)
+	if err != nil {
+		log.Fatalf("failed to generate random data: %s", err)
+	}
 
-		if n != len(records) {
-			log.Fatalf("expected to generate %d bytes, got %d", len(records), n)
-		}
+	if n != len(records) {
+		log.Fatalf("expected to generate %d bytes, got %d", len(records), n)
+	}
 
-		recordSizes := make([]uint32, recordsPerBatch)
-		for i := range recordsPerBatch {
-			recordSizes[i] = uint32(recordSize)
-		}
+	recordSizes := make([]uint32, recordsPerBatch)
+	for i := range recordsPerBatch {
+		recordSizes[i] = uint32(recordSize)
+	}
 
+	for range numBatches {
 		batches <- recordBatch{
 			recordSizes: recordSizes,
 			records:     records,