fix(tracker): more flexible channel return

joshuar · Dec 17, 2023 · 5b890f1 · 5b890f1
1 parent 2c53790
commit 5b890f1
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/internal/tracker/sensor.go b/internal/tracker/sensor.go
@@ -78,14 +78,17 @@ func marshalClass[C ComparableStringer](class C) string {
 	}
 }
 
-func MergeSensorCh(ctx context.Context, sensorCh ...<-chan Sensor) <-chan Sensor {
+func MergeSensorCh(ctx context.Context, sensorCh ...<-chan Sensor) chan Sensor {
 	var wg sync.WaitGroup
 	out := make(chan Sensor)
 
 	// Start an output goroutine for each input channel in sensorCh.  output
 	// copies values from c to out until c is closed, then calls wg.Done.
 	output := func(c <-chan Sensor) {
 		defer wg.Done()
+		if c == nil {
+			return
+		}
 		for n := range c {
 			select {
 			case out <- n: