Handle default values for InfluxDB Service Configs (#593)

influxdata · Jun 2, 2016 · 6379768 · 6379768
1 parent 8ab770d
commit 6379768
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/cmd/kapacitord/run/config.go b/cmd/kapacitord/run/config.go
@@ -115,6 +115,11 @@ func (c *Config) PostInit() {
 	} else if len(c.InfluxDB) == 1 && c.InfluxDB[0].Name == "" {
 		c.InfluxDB[0].Name = "default"
 	}
+	// Set default Values
+	for i, influx := range c.InfluxDB {
+		influx.SetDefaultValues()
+		c.InfluxDB[i] = influx
+	}
 }
 
 // NewDemoConfig returns the config that runs when no config is specified.

diff --git a/services/influxdb/config.go b/services/influxdb/config.go
@@ -63,6 +63,18 @@ func NewConfig() Config {
 	}
 }
 
+func (c *Config) SetDefaultValues() {
+	if c.UDPBuffer == 0 {
+		c.UDPBuffer = udp.DefaultBuffer
+	}
+	if c.StartUpTimeout == toml.Duration(0) {
+		c.StartUpTimeout = toml.Duration(DefaultStartUpTimeout)
+	}
+	if c.SubscriptionProtocol == "" {
+		c.SubscriptionProtocol = DefaultSubscriptionProtocol
+	}
+}
+
 func (c Config) Validate() error {
 	if c.Name == "" {
 		return errors.New("influxdb cluster must be given a name")