package sensor import ( "context" "fmt" "sync" "github.com/go-flucky/flucky/pkg/internal/format" "github.com/go-flucky/flucky/pkg/types" "github.com/go-flucky/go-dht" uuid "github.com/satori/go.uuid" ) // DHT11 is a sensor to measure humidity and temperature. type DHT11 struct { *types.Sensor } // GetSensorModel returns the sensor model func (s *DHT11) GetSensorModel() types.SensorModel { return s.Sensor.SensorModel } // Read measured values func (s *DHT11) Read() ([]*types.MeasuredValue, error) { err := dht.HostInit() if err != nil { return nil, fmt.Errorf("HostInit error: %v", err) } gpio, err := types.GPIOToString(*s.GPIONumber) if err != nil { return nil, err } dht, err := dht.NewDHT(gpio, dht.Celsius, "") if err != nil { return nil, fmt.Errorf("NewDHT error: %v", err) } humidityValue, temperatureValue, err := dht.Read() if err != nil { return nil, fmt.Errorf("Read error: %v", err) } measuredValues := []*types.MeasuredValue{ &types.MeasuredValue{ ID: uuid.NewV4().String(), Value: float64(humidityValue), ValueType: types.MeasuredValueTypeHumidity, FromDate: format.FormatedTime(), TillDate: format.FormatedTime(), SensorID: s.SensorID, }, &types.MeasuredValue{ ID: uuid.NewV4().String(), Value: float64(temperatureValue), ValueType: types.MeasuredValueTypeTemperature, FromDate: format.FormatedTime(), TillDate: format.FormatedTime(), SensorID: s.SensorID, }, } return measuredValues, nil } // ReadChannel reads the measured values from the sensor and writes them to a // channel. func (s *DHT11) ReadChannel(measuredValuesChannel chan<- []*types.MeasuredValue, errorChannel chan<- error, wg *sync.WaitGroup) { if wg != nil { defer wg.Done() } measuredValues, err := s.Read() if err != nil { errorChannel <- err return } measuredValuesChannel <- measuredValues } // ReadContinously reads the measured values continously from the sensor and // writes them to a channel. func (s *DHT11) ReadContinously(ctx context.Context, measuredValuesChannel chan<- []*types.MeasuredValue, errorChannel chan<- error) { for { select { case <-ctx.Done(): errorChannel <- fmt.Errorf("%v: Context closed: %v", s.SensorName, ctx.Err()) return default: s.ReadChannel(measuredValuesChannel, errorChannel, nil) } } }