Add option for adding more words to the list.

CHANGELOG

@@ -1,3 +1,7 @@
+Version 0.9.2
+
+ - Add "more words" option.
+
 Version 0.9.1
 
  - Add 'nastyview' subcommand for 'normal'.

feeds/nastyview.feed

@@ -1,5 +1,5 @@
 [
 normal create collection=c numberOfShards=3 replicationFactor=2 drop=true
-normal nastyViewData collection=c size=10G numberFields=10 parallelism=40 keySize=32
+normal nastyViewData collection=c size=10M numberFields=10 parallelism=1 keySize=32
 normal createView viewDefFile=feeds/nastyview.json view=v drop=true
 ]

main.go

@@ -10,6 +10,7 @@ import (
 	"strconv"
 
 	"github.com/arangodb/feed/pkg/config"
+	"github.com/arangodb/feed/pkg/datagen"
 	"github.com/arangodb/feed/pkg/feedlang"
 	"github.com/arangodb/feed/pkg/operations"
 	"github.com/prometheus/client_golang/prometheus/promhttp"
@@ -25,7 +26,8 @@ var (
 		Short: "The 'feed' tool feeds ArangoDB with generated data, quickly.",
 		RunE:  mainExecute,
 	}
-	ProgName string
+	ProgName  string
+	MoreWords int
 )
 
 func init() {
@@ -39,6 +41,7 @@ func init() {
 	flags.StringVar(&config.Protocol, "protocol", "vst", "Protocol (http1, http2, vst)")
 	flags.StringVar(&config.JSONOutput, "jsonOutputFile", "feed.json", "Filename for JSON result output.")
 	flags.IntVar(&config.MetricsPort, "metricsPort", 8888, "Metrics port (0 for no metrics)")
+	flags.IntVar(&MoreWords, "moreWords", 0, "Number of additional random terms in long word list")
 }
 
 func produceJSONResult(prog feedlang.Program, fileName string) {
@@ -70,6 +73,8 @@ func mainExecute(cmd *cobra.Command, _ []string) error {
 
 	fmt.Printf("Hello world, this is 'feed' version %s !\n\n", Version)
 
+	datagen.ExtendLongWordList(MoreWords)
+
 	// Expose metrics:
 	if config.MetricsPort != 0 {
 		fmt.Printf("Exposing Prometheus metrics on port %d under /metrics...\n\n",
@@ -105,7 +110,9 @@ func mainExecute(cmd *cobra.Command, _ []string) error {
 		fmt.Printf("Error in parse: %v\n", err)
 		os.Exit(3)
 	}
+
 	err = prog.Execute()
+
 	if err != nil {
 		fmt.Printf("Error in execution: %v\n", err)
 		os.Exit(4)

pkg/datagen/longwordlist.go

@@ -1,5 +1,9 @@
 package datagen
 
+import (
+	"math/rand"
+)
+
 var LongWordList = []string{
 	"Abbott",
 	"AbbVie",
@@ -670,3 +674,10 @@ var LongWordList = []string{
 	"Zions",
 	"Zoetis",
 }
+
+func ExtendLongWordList(nr int) {
+	source := rand.New(rand.NewSource(int64(12345) + rand.Int63()))
+	for i := 0; i < nr; i++ {
+		LongWordList = append(LongWordList, MakeRandomString(rand.Intn(10)+5, source))
+	}
+}