package sensor import ( "context" "fmt" "sync" "time" "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 } // ReadHumidity measure the humidity func (s *DHT11) ReadHumidity() (*types.Humidity, 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, _, err := dht.Read() if err != nil { return nil, fmt.Errorf("Read error: %v", err) } humidity := &types.Humidity{ HumidityID: uuid.NewV4().String(), HumidityValue: humidityValue, HumidityFromDate: time.Now(), HumidityTillDate: time.Now(), SensorID: s.SensorID, } return humidity, nil } // ReadHumidityWriteIntoChannel and write values into a channel func (s *DHT11) ReadHumidityWriteIntoChannel(humidityChannel chan<- *types.Humidity, errorChannel chan<- error, wg *sync.WaitGroup) { if wg != nil { defer wg.Done() } humidity, err := s.ReadHumidity() if err != nil { errorChannel <- err return } humidityChannel <- humidity } // ReadHumidityContinously into a channel until context closed func (s *DHT11) ReadHumidityContinously(ctx context.Context, humidityChannel chan<- *types.Humidity, errorChannel chan<- error) { for { select { case <-ctx.Done(): errorChannel <- fmt.Errorf("%v: Context closed: %v", s.SensorName, ctx.Err()) return default: s.ReadHumidityWriteIntoChannel(humidityChannel, errorChannel, nil) } } } // ReadTemperature measure the temperature func (s *DHT11) ReadTemperature() (*types.Temperature, 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) } _, temperatureValue, err := dht.Read() if err != nil { return nil, fmt.Errorf("Read error: %v", err) } temperature := &types.Temperature{ TemperatureID: uuid.NewV4().String(), TemperatureValue: temperatureValue, TemperatureFromDate: time.Now(), TemperatureTillDate: time.Now(), SensorID: s.SensorID, } return temperature, nil } // ReadTemperatureWriteIntoChannel and write values into a channel func (s *DHT11) ReadTemperatureWriteIntoChannel(temperatureChannel chan<- *types.Temperature, errorChannel chan<- error, wg *sync.WaitGroup) { if wg != nil { defer wg.Done() } temperature, err := s.ReadTemperature() if err != nil { errorChannel <- err return } temperatureChannel <- temperature } // ReadTemperatureContinously into a channel until context closed func (s *DHT11) ReadTemperatureContinously(ctx context.Context, temperatureChannel chan<- *types.Temperature, errorChannel chan<- error) { for { select { case <-ctx.Done(): errorChannel <- fmt.Errorf("%v: Context closed: %v", s.SensorName, ctx.Err()) return default: s.ReadTemperatureWriteIntoChannel(temperatureChannel, errorChannel, nil) } } }