109 lines
2.4 KiB
Go
109 lines
2.4 KiB
Go
package sensor
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/volker-raschek/flucky/pkg/internal/errutils"
|
|
|
|
"github.com/volker-raschek/flucky/pkg/types"
|
|
)
|
|
|
|
type HumiditySensor interface {
|
|
ReadHumidity() (*types.Humidity, error)
|
|
}
|
|
|
|
type TemperatureSensor interface {
|
|
ReadTemperature() (*types.Temperature, error)
|
|
}
|
|
|
|
func ReadHumidities(humiditySensors []HumiditySensor) ([]*types.Humidity, error) {
|
|
|
|
wg := new(sync.WaitGroup)
|
|
wg.Add(len(humiditySensors))
|
|
|
|
errChannel := make(chan error, len(humiditySensors))
|
|
humidityChannel := make(chan *types.Humidity, len(humiditySensors))
|
|
|
|
for _, humiditySensor := range humiditySensors {
|
|
go func(hs HumiditySensor) {
|
|
defer wg.Done()
|
|
humidity, err := hs.ReadHumidity()
|
|
if err != nil {
|
|
errChannel <- err
|
|
humidityChannel <- nil
|
|
}
|
|
errChannel <- nil
|
|
humidityChannel <- humidity
|
|
}(humiditySensor)
|
|
}
|
|
|
|
wg.Wait()
|
|
|
|
errorList := errutils.CollectErrors(errChannel)
|
|
if err := errutils.FormatErrors(errorList); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
humidityList := collectHumidities(humidityChannel)
|
|
return humidityList, nil
|
|
|
|
}
|
|
|
|
func ReadTemperatures(temperatureSensors []TemperatureSensor) ([]*types.Temperature, error) {
|
|
wg := new(sync.WaitGroup)
|
|
wg.Add(len(temperatureSensors))
|
|
errChannel := make(chan error, len(temperatureSensors))
|
|
temperatureChannel := make(chan *types.Temperature, len(temperatureSensors))
|
|
for _, temperatureSensor := range temperatureSensors {
|
|
go func(ts TemperatureSensor) {
|
|
defer wg.Done()
|
|
temperature, err := ts.ReadTemperature()
|
|
if err != nil {
|
|
errChannel <- err
|
|
}
|
|
temperatureChannel <- temperature
|
|
}(temperatureSensor)
|
|
}
|
|
wg.Wait()
|
|
errorList := errutils.CollectErrors(errChannel)
|
|
if err := errutils.FormatErrors(errorList); err != nil {
|
|
return nil, err
|
|
}
|
|
temperatureList := collectTemperatures(temperatureChannel)
|
|
return temperatureList, nil
|
|
}
|
|
|
|
func collectHumidities(humChan <-chan *types.Humidity) []*types.Humidity {
|
|
humidityList := make([]*types.Humidity, 0)
|
|
|
|
for {
|
|
select {
|
|
case hum, more := <-humChan:
|
|
if more {
|
|
humidityList = append(humidityList, hum)
|
|
continue
|
|
}
|
|
return nil
|
|
default:
|
|
return humidityList
|
|
}
|
|
}
|
|
}
|
|
|
|
func collectTemperatures(tempChan <-chan *types.Temperature) []*types.Temperature {
|
|
temperatureList := make([]*types.Temperature, 0)
|
|
|
|
for {
|
|
select {
|
|
case temp, more := <-tempChan:
|
|
if more {
|
|
temperatureList = append(temperatureList, temp)
|
|
continue
|
|
}
|
|
return nil
|
|
default:
|
|
return temperatureList
|
|
}
|
|
}
|
|
}
|