feat: add trend enabled config

penguin-statistics · Mar 2, 2023 · 9a55ae0 · 9a55ae0
1 parent d0c8d5b
commit 9a55ae0
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/internal/app/appconfig/spec.go b/internal/app/appconfig/spec.go
@@ -109,6 +109,9 @@ type ConfigSpec struct {
 	// WorkerTrendInterval describes the interval in-between different batches
 	WorkerTrendInterval time.Duration `required:"true" split_words:"true" default:"6h"`
 
+	// WorkerTrendEnabled describes whether to enable the trend worker
+	WorkerTrendEnabled bool `required:"true" split_words:"true" default:"true"`
+
 	// WorkerSeparation describes the separation time in-between different microtasks
 	WorkerSeparation time.Duration `required:"true" split_words:"true" default:"3s"`
 

diff --git a/internal/workers/calcwkr/calcwkr.go b/internal/workers/calcwkr/calcwkr.go
@@ -96,7 +96,13 @@ func Start(conf *appconfig.Config, deps WorkerDeps) {
 		}
 		w.checkConfig()
 		w.doMainCalc(conf.MatrixWorkerSourceCategories)
-		w.doTrendCalc(conf.MatrixWorkerSourceCategories)
+		if conf.WorkerTrendEnabled {
+			w.doTrendCalc(conf.MatrixWorkerSourceCategories)
+		} else {
+			log.Info().
+				Str("evt.name", "worker.calcwkr.trend.disabled").
+				Msg("worker trend is disabled due to configuration")
+		}
 	} else {
 		log.Info().
 			Str("evt.name", "worker.calcwkr.disabled").