Fix statshouse-client-test to use specified number of tags

Author: razmser

cmd/statshouse-client-test/main.go

@@ -9,6 +9,7 @@ import (
 
 type argv struct {
 	n, m      int
+	tagsNum   int
 	maxStrLen int
 	viewCode  bool
 	keepTemp  bool
@@ -22,6 +23,7 @@ func main() {
 	var args argv
 	flag.IntVar(&args.n, "n", 100, "number of iterations")
 	flag.IntVar(&args.m, "m", 300, "number of series to send per iteration")
+	flag.IntVar(&args.tagsNum, "tags", 16, "number of tags to use")
 	flag.IntVar(&args.maxStrLen, "max-str-len", 32, "maximum string length")
 	flag.BoolVar(&args.viewCode, "view-code", false, "open generated source files in Visual Studio Code")
 	flag.BoolVar(&args.keepTemp, "keep-temp", false, "do not remove generated temporary files")

cmd/statshouse-client-test/test_data.go

@@ -13,7 +13,6 @@ import (
 	"strings"
 	"time"
 
-	"github.com/vkcom/statshouse/internal/format"
 	"pgregory.net/rand"
 )
 
@@ -54,9 +53,9 @@ func newTestData(args argv) testData {
 			m.Timestamp = r.Uint32n(now)
 		}
 		m.Name = string(str[:rand.Intn(len(str)-1)+1])
-		m.Tags = make([][2]string, r.Intn(format.MaxTags))
-		for i, j := r.Intn(format.MaxTags), 0; j < len(m.Tags); i, j = i+1, j+1 {
-			m.Tags[j] = [2]string{strconv.Itoa(i % format.MaxTags), string(str[:rand.Intn(len(str)-1)+1])}
+		m.Tags = make([][2]string, r.Intn(args.tagsNum))
+		for i, j := r.Intn(args.tagsNum), 0; j < len(m.Tags); i, j = i+1, j+1 {
+			m.Tags[j] = [2]string{strconv.Itoa(i % args.tagsNum), string(str[:rand.Intn(len(str)-1)+1])}
 		}
 		m.Kind = r.Intn(3)
 		switch m.Kind {