allow specifying influx org in config. fixes #10

tedpearson · Aug 8, 2024 · 819e464 · 819e464
1 parent 60a0c50
commit 819e464
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/config.example.yaml b/config.example.yaml
@@ -8,5 +8,6 @@ smarthub:
 influxdb:
   host: https://localhost:8428
   auth_token: johndoe:influx_password
+  org: ""
   database: db_name
   insecure: false
diff --git a/internal/app/main.go b/internal/app/main.go
@@ -15,6 +15,7 @@ import (
 type InfluxConfig struct {
 	Host      string `yaml:"host"`
 	AuthToken string `yaml:"auth_token"`
+	Org       string `yaml:"org"`
 	Database  string `yaml:"database"`
 	Insecure  bool   `yaml:"insecure"`
 }

diff --git a/internal/app/metrics.go b/internal/app/metrics.go
@@ -17,7 +17,7 @@ func WriteMetrics(records []ElectricUsage, config InfluxConfig) error {
 		opts.SetTLSConfig(&tls.Config{InsecureSkipVerify: true})
 	}
 	client := influxdb2.NewClientWithOptions(config.Host, config.AuthToken, opts)
-	writeApi := client.WriteAPIBlocking("", config.Database)
+	writeApi := client.WriteAPIBlocking(config.Org, config.Database)
 	for _, record := range records {
 		minutes := record.EndTime.Sub(record.StartTime).Minutes()
 		points := make([]*write.Point, 0, int(minutes))