fix: actually separate collector by device for multi-device stations

tris · Jul 24, 2023 · b42749d · b42749d
1 parent e9a928b
commit b42749d
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/exporter/exporter.go b/exporter/exporter.go
@@ -72,6 +72,14 @@ func (we *WeatherExporter) ScrapeHandler(w http.ResponseWriter, r *http.Request)
 			client = weatherflow.NewClient(apiToken, nil, prefixedLogger(apiToken, log.Printf))
 			we.clients[apiToken] = client
 			client.Start(func(msg weatherflow.Message) {
+				deviceID, ok := msg.GetDeviceID()
+				if !ok {
+					return
+				}
+				collector, ok := collectorsForToken[deviceID]
+				if !ok {
+					return
+				}
 				collector.update(msg, apiToken)
 			})
 		}

diff --git a/go.mod b/go.mod
@@ -4,7 +4,7 @@ go 1.20
 
 require (
 	github.com/prometheus/client_golang v1.15.1
-	github.com/tris/weatherflow v0.3.0
+	github.com/tris/weatherflow v0.3.1
 )
 
 require (