Enabled capability to change default workload

Author: daniel-percona

README.md

@@ -132,6 +132,10 @@ The tool comes with a built-in default workload useful for immediate testing and
 ./bin/genMongoLoad
 ```
 
+**Note about default workload:** genMongoLoad comes pre-configured with a [default collection](./resources/collections/default.json) and [default queries](./resources/queries/default.json). If you do not provide any parameters and leave the configuration setting default_workload: true, this default workload will be used.
+
+If you wish to use a different default workload, you can replace these two files with your own default.json files in the same paths. This allows you to define a different collection and set of queries as the default workload.
+
 **Note on config file usage:** If you do not specify the config file name (above example), genMongoLoad will use the [config.yaml](./config.yaml) by default. You can create separate configuration files if you wish and then pass it as an argument:
 
 ```bash

config.yaml

@@ -43,9 +43,8 @@ debug: false
 # Workload Definition
 # ==============================================================================
 
-# If true, loads the built-in "Flights" example (resources/collections/default.json).
+# If true, genMongoLoad will only run the default workload default (resources/collections/default.json).
 # If false, loads custom JSON files from the paths below.
-# Override via env: GENMONGOLOAD_DEFAULT_WORKLOAD
 default_workload: true
 
 # Directories containing your custom JSON schema and query definitions.

internal/workloads/generator.go

@@ -11,15 +11,15 @@ import (
 
 // GenerateDocument creates a single document.
 func GenerateDocument(col config.CollectionDefinition, cfg *config.AppConfig) map[string]interface{} {
-	if cfg.DefaultWorkload {
+	if cfg.DefaultWorkload && col.Name == "flights" {
 		return GenerateDefaultDocument(col)
 	}
 	return generateGenericDocument(col)
 }
 
 // GenerateFallbackUpdate creates an update document when no configured query is found.
 func GenerateFallbackUpdate(col config.CollectionDefinition, cfg *config.AppConfig, rng *rand.Rand) map[string]interface{} {
-	if cfg.DefaultWorkload {
+	if cfg.DefaultWorkload && col.Name == "flights" {
 		return GenerateDefaultUpdate(rng)
 	}
 	return generateGenericUpdate(col, rng)