[dbnode] Add ability to force enable cold writes from config (#3002)

m3db · Dec 11, 2020 · 19fdb08 · 19fdb08
1 parent b5a22ba
commit 19fdb08
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 0 deletions.
diff --git a/src/cmd/services/m3dbnode/config/config.go b/src/cmd/services/m3dbnode/config/config.go
@@ -207,6 +207,10 @@ type DBConfiguration struct {
 
 	// Debug configuration.
 	Debug config.DebugConfiguration `yaml:"debug"`
+
+	// ForceColdWritesEnabled will force enable cold writes for all namespaces
+	// if set.
+	ForceColdWritesEnabled *bool `yaml:"forceColdWritesEnabled"`
 }
 
 // LoggingOrDefault returns the logging configuration or defaults.

diff --git a/src/cmd/services/m3dbnode/config/config_test.go b/src/cmd/services/m3dbnode/config/config_test.go
@@ -729,6 +729,7 @@ func TestConfiguration(t *testing.T) {
   debug:
     mutexProfileFraction: 0
     blockProfileRate: 0
+  forceColdWritesEnabled: null
 coordinator: null
 `
 

diff --git a/src/dbnode/server/server.go b/src/dbnode/server/server.go
@@ -651,6 +651,11 @@ func Run(runOpts RunOptions) {
 	}()
 	opts = opts.SetIndexClaimsManager(icm)
 
+	if value := cfg.ForceColdWritesEnabled; value != nil {
+		// Allow forcing cold writes to be enabled by config.
+		opts = opts.SetForceColdWritesEnabled(*value)
+	}
+
 	forceColdWrites := opts.ForceColdWritesEnabled()
 	var envCfgResults environment.ConfigureResults
 	if len(envConfig.Statics) == 0 {