Fix MB/GB total count

vkuznet · Sep 22, 2022 · 7d065b6 · 7d065b6
1 parent 7379fd3
commit 7d065b6
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/httpgo.go b/httpgo.go
@@ -65,6 +65,12 @@ func genRecords(size string) ([]Record, error) {
 	if err != nil {
 		return nil, err
 	}
+	if suffix == "MB" {
+		total *= 1024
+	}
+	if suffix == "BB" {
+		total *= 1024 * 1024
+	}
 	records = genNRecords(total)
 	return records, nil
 }