cn/miflorad
1
0
Fork 0
mirror of https://github.com/cmur2/miflorad.git synced 2025-01-24 03:27:05 +01:00
Code Releases Activity

common: remove unneeded ble.Profile parameter from ble impl

Browse Source
This commit is contained in:
cn 2018-12-22 12:19:43 +01:00
parent f7f951e2a1
commit 5be425040d
3 changed files with 43 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
cmd/miflorad/main.go
View File

@ -20,8 +20,8 @@ import (
) )
var ( var (
scanTimeout = flag.Duration("scantimeout", 6*time.Second, "timeout after that a scan per peripheral will be aborted") scanTimeout = flag.Duration("scantimeout", 10*time.Second, "timeout after that a scan per peripheral will be aborted")
interval = flag.Duration("interval", 15*time.Second, "metrics collection interval") interval = flag.Duration("interval", 25*time.Second, "metrics collection interval")
readRetries = flag.Int("readretries", 2, "number of times reading will be attempted per peripheral") readRetries = flag.Int("readretries", 2, "number of times reading will be attempted per peripheral")
) )
@ -40,7 +40,8 @@ var (
func checkTooShortInterval() error { func checkTooShortInterval() error {
numPeripherals := int64(len(flag.Args())) numPeripherals := int64(len(flag.Args()))
if (*scanTimeout).Nanoseconds()*int64(*readRetries)*numPeripherals >= (*interval).Nanoseconds() { numReadRetries := int64(*readRetries)
if (*scanTimeout).Nanoseconds()*numReadRetries*numPeripherals >= (*interval).Nanoseconds() {
return errors.New(fmt.Sprintf( return errors.New(fmt.Sprintf(
"The interval of %s is too short given the scan timeout of %s for %d peripheral(s) with %d retries each! Exiting...\n", "The interval of %s is too short given the scan timeout of %s for %d peripheral(s) with %d retries each! Exiting...\n",
*interval, *scanTimeout, numPeripherals, *readRetries)) *interval, *scanTimeout, numPeripherals, *readRetries))
@ -49,8 +50,10 @@ func checkTooShortInterval() error {
} }
func readData(peripheral *peripheral, client ble.Client) error { func readData(peripheral *peripheral, client ble.Client) error {
// re-request meta data (for battery level) if last check more than 24 hours ago
// Source: https://github.com/open-homeautomation/miflora/blob/ffd95c3e616df8843cc8bff99c9b60765b124092/miflora/miflora_poller.py#L92
if time.Since(peripheral.lastMetaDataFetch) >= 24*time.Hour { if time.Since(peripheral.lastMetaDataFetch) >= 24*time.Hour {
metaData, err := impl.RequestVersionBattery(client, client.Profile()) metaData, err := impl.RequestVersionBattery(client)
if err != nil { if err != nil {
return errors.Wrap(err, "can't request version battery") return errors.Wrap(err, "can't request version battery")
} }
@ -59,13 +62,13 @@ func readData(peripheral *peripheral, client ble.Client) error {
} }
if peripheral.metaData.RequiresModeChangeBeforeRead() { if peripheral.metaData.RequiresModeChangeBeforeRead() {
err2 := impl.RequestModeChange(client, client.Profile()) err2 := impl.RequestModeChange(client)
if err2 != nil { if err2 != nil {
return errors.Wrap(err2, "can't request mode change") return errors.Wrap(err2, "can't request mode change")
} }
} }
sensorData, err3 := impl.RequestSensorData(client, client.Profile()) sensorData, err3 := impl.RequestSensorData(client)
if err3 != nil { if err3 != nil {
return errors.Wrap(err3, "can't request sensor data") return errors.Wrap(err3, "can't request sensor data")
} }
@ -78,8 +81,15 @@ func readData(peripheral *peripheral, client ble.Client) error {
func connectPeripheral(peripheral *peripheral) error { func connectPeripheral(peripheral *peripheral) error {
fmt.Fprintf(os.Stderr, "Scanning for %s...\n", peripheral.id) fmt.Fprintf(os.Stderr, "Scanning for %s...\n", peripheral.id)
// only way to get back the found advertisement, must be buffered!
foundAdvertisementChannel := make(chan ble.Advertisement, 1)
filter := func(adv ble.Advertisement) bool { filter := func(adv ble.Advertisement) bool {
return strings.ToUpper(adv.Addr().String()) == strings.ToUpper(peripheral.id) if strings.ToUpper(adv.Addr().String()) == strings.ToUpper(peripheral.id) {
foundAdvertisementChannel <- adv
return true
}
return false
} }
timeConnectStart := time.Now() timeConnectStart := time.Now()
@ -94,6 +104,9 @@ func connectPeripheral(peripheral *peripheral) error {
fmt.Println(timeConnectTook) fmt.Println(timeConnectTook)
// fmt.Fprintf(os.Stdout, "%s.miflora.%s.connect_time %.2f %d\n", prefix, id, timeConnectTook, time.Now().Unix()) // fmt.Fprintf(os.Stdout, "%s.miflora.%s.connect_time %.2f %d\n", prefix, id, timeConnectTook, time.Now().Unix())
// foundAdvertisement := <-foundAdvertisementChannel
// fmt.Fprintf(os.Stdout, "%s.miflora.%s.rssi %d %d\n", prefix, id, foundAdvertisement.RSSI(), time.Now().Unix())
// Source: https://github.com/go-ble/ble/blob/master/examples/basic/explorer/main.go#L53 // Source: https://github.com/go-ble/ble/blob/master/examples/basic/explorer/main.go#L53
// Make sure we had the chance to print out the message. // Make sure we had the chance to print out the message.
done := make(chan struct{}) done := make(chan struct{})
@ -138,6 +151,7 @@ func readPeripheral(peripheral *peripheral) error {
func readAllPeripherals(quit chan struct{}) { func readAllPeripherals(quit chan struct{}) {
for _, peripheral := range allPeripherals { for _, peripheral := range allPeripherals {
// check for quit signal (non-blocking) and terminate
select { select {
case <-quit: case <-quit:
return return
@ -180,14 +194,16 @@ func main() {
go func() { go func() {
fmt.Fprintf(os.Stderr, "Starting miflorad loop with %s interval...\n", *interval) fmt.Fprintf(os.Stderr, "Starting miflorad loop with %s interval...\n", *interval)
// populate all peripherals data structure
allPeripherals = make([]*peripheral, len(flag.Args())) allPeripherals = make([]*peripheral, len(flag.Args()))
for i, peripheralID := range flag.Args() { for i, peripheralID := range flag.Args() {
allPeripherals[i] = &peripheral{ allPeripherals[i] = &peripheral{
id: peripheralID, id: peripheralID,
lastMetaDataFetch: time.Unix(0, 0), lastMetaDataFetch: time.Unix(0, 0), // force immediate 1st request
} }
} }
// main loop
readAllPeripherals(quit) readAllPeripherals(quit)
for range intervalTicker.C { for range intervalTicker.C {
readAllPeripherals(quit) readAllPeripherals(quit)

13
cmd/munin-miflora/main.go
View File

@ -17,11 +17,11 @@ import (
const discoveryTimeout = 10 * time.Second const discoveryTimeout = 10 * time.Second
func readData(client ble.Client, profile *ble.Profile) { func readData(client ble.Client) {
prefix := flag.Args()[0] prefix := flag.Args()[0]
id := common.MifloraGetAlphaNumericID(flag.Args()[1]) id := common.MifloraGetAlphaNumericID(flag.Args()[1])
metaData, err := impl.RequestVersionBattery(client, profile) metaData, err := impl.RequestVersionBattery(client)
if err != nil { if err != nil {
fmt.Fprintf(os.Stderr, "Failed to request version battery, err: %s\n", err) fmt.Fprintf(os.Stderr, "Failed to request version battery, err: %s\n", err)
fmt.Fprintf(os.Stdout, "%s.miflora.%s.failed 1 %d\n", prefix, id, time.Now().Unix()) fmt.Fprintf(os.Stdout, "%s.miflora.%s.failed 1 %d\n", prefix, id, time.Now().Unix())
@ -34,7 +34,7 @@ func readData(client ble.Client, profile *ble.Profile) {
fmt.Fprintf(os.Stdout, "%s.miflora.%s.firmware_version %d %d\n", prefix, id, metaData.NumericFirmwareVersion(), time.Now().Unix()) fmt.Fprintf(os.Stdout, "%s.miflora.%s.firmware_version %d %d\n", prefix, id, metaData.NumericFirmwareVersion(), time.Now().Unix())
if metaData.RequiresModeChangeBeforeRead() { if metaData.RequiresModeChangeBeforeRead() {
err2 := impl.RequestModeChange(client, profile) err2 := impl.RequestModeChange(client)
if err2 != nil { if err2 != nil {
fmt.Fprintf(os.Stderr, "Failed to request mode change, err: %s\n", err2) fmt.Fprintf(os.Stderr, "Failed to request mode change, err: %s\n", err2)
fmt.Fprintf(os.Stdout, "%s.miflora.%s.failed 1 %d\n", prefix, id, time.Now().Unix()) fmt.Fprintf(os.Stdout, "%s.miflora.%s.failed 1 %d\n", prefix, id, time.Now().Unix())
@ -42,7 +42,7 @@ func readData(client ble.Client, profile *ble.Profile) {
} }
} }
sensorData, err3 := impl.RequestSensorData(client, profile) sensorData, err3 := impl.RequestSensorData(client)
if err3 != nil { if err3 != nil {
fmt.Fprintf(os.Stderr, "Failed to request sensor data, err: %s\n", err3) fmt.Fprintf(os.Stderr, "Failed to request sensor data, err: %s\n", err3)
fmt.Fprintf(os.Stdout, "%s.miflora.%s.failed 1 %d\n", prefix, id, time.Now().Unix()) fmt.Fprintf(os.Stdout, "%s.miflora.%s.failed 1 %d\n", prefix, id, time.Now().Unix())
@ -118,14 +118,13 @@ func main() {
timeReadoutStart := time.Now() timeReadoutStart := time.Now()
profile, err := client.DiscoverProfile(true) if _, err := client.DiscoverProfile(true); err != nil {
if err != nil {
fmt.Fprintf(os.Stderr, "Failed to discover profile, err: %s\n", err) fmt.Fprintf(os.Stderr, "Failed to discover profile, err: %s\n", err)
fmt.Fprintf(os.Stdout, "%s.miflora.%s.failed 1 %d\n", prefix, id, time.Now().Unix()) fmt.Fprintf(os.Stdout, "%s.miflora.%s.failed 1 %d\n", prefix, id, time.Now().Unix())
os.Exit(1) os.Exit(1)
} }
readData(client, profile) readData(client)
timeReadoutTook := time.Since(timeReadoutStart).Seconds() timeReadoutTook := time.Since(timeReadoutStart).Seconds()
fmt.Fprintf(os.Stdout, "%s.miflora.%s.readout_time %.2f %d\n", prefix, id, timeReadoutTook, time.Now().Unix()) fmt.Fprintf(os.Stdout, "%s.miflora.%s.readout_time %.2f %d\n", prefix, id, timeReadoutTook, time.Now().Unix())

24
common/ble/impl.go
View File

@ -7,7 +7,8 @@ import (
"github.com/pkg/errors" "github.com/pkg/errors"
) )
func FindServiceByUUID(services []*ble.Service, u ble.UUID) *ble.Service { func FindServiceByUUID(services []*ble.Service, uuid string) *ble.Service {
u := ble.MustParse(uuid)
for _, service := range services { for _, service := range services {
if service.UUID.Equal(u) { if service.UUID.Equal(u) {
return service return service
@ -16,7 +17,8 @@ func FindServiceByUUID(services []*ble.Service, u ble.UUID) *ble.Service {
return nil return nil
} }
func FindCharacteristicByUUID(characteristics []*ble.Characteristic, u ble.UUID) *ble.Characteristic { func FindCharacteristicByUUID(characteristics []*ble.Characteristic, uuid string) *ble.Characteristic {
u := ble.MustParse(uuid)
for _, characteristic := range characteristics { for _, characteristic := range characteristics {
if characteristic.UUID.Equal(u) { if characteristic.UUID.Equal(u) {
return characteristic return characteristic
@ -25,13 +27,13 @@ func FindCharacteristicByUUID(characteristics []*ble.Characteristic, u ble.UUID)
return nil return nil
} }
func RequestVersionBattery(client ble.Client, profile *ble.Profile) (common.VersionBatteryResponse, error) { func RequestVersionBattery(client ble.Client) (common.VersionBatteryResponse, error) {
mifloraService := FindServiceByUUID(profile.Services, ble.MustParse(common.MifloraServiceUUID)) mifloraService := FindServiceByUUID(client.Profile().Services, common.MifloraServiceUUID)
if mifloraService == nil { if mifloraService == nil {
return common.VersionBatteryResponse{}, errors.New("Failed to get the miflora service") return common.VersionBatteryResponse{}, errors.New("Failed to get the miflora service")
} }
mifloraVersionBatteryChar := FindCharacteristicByUUID(mifloraService.Characteristics, ble.MustParse(common.MifloraCharVersionBatteryUUID)) mifloraVersionBatteryChar := FindCharacteristicByUUID(mifloraService.Characteristics, common.MifloraCharVersionBatteryUUID)
if mifloraVersionBatteryChar == nil { if mifloraVersionBatteryChar == nil {
return common.VersionBatteryResponse{}, errors.New("Failed to get the version battery characteristic") return common.VersionBatteryResponse{}, errors.New("Failed to get the version battery characteristic")
} }
@ -44,13 +46,13 @@ func RequestVersionBattery(client ble.Client, profile *ble.Profile) (common.Vers
return common.ParseVersionBattery(bytes), nil return common.ParseVersionBattery(bytes), nil
} }
func RequestModeChange(client ble.Client, profile *ble.Profile) error { func RequestModeChange(client ble.Client) error {
mifloraService := FindServiceByUUID(profile.Services, ble.MustParse(common.MifloraServiceUUID)) mifloraService := FindServiceByUUID(client.Profile().Services, common.MifloraServiceUUID)
if mifloraService == nil { if mifloraService == nil {
return errors.New("Failed to get the miflora service") return errors.New("Failed to get the miflora service")
} }
mifloraModeChangeChar := FindCharacteristicByUUID(mifloraService.Characteristics, ble.MustParse(common.MifloraCharModeChangeUUID)) mifloraModeChangeChar := FindCharacteristicByUUID(mifloraService.Characteristics, common.MifloraCharModeChangeUUID)
if mifloraModeChangeChar == nil { if mifloraModeChangeChar == nil {
return errors.New("Failed to discover the mode change characteristic") return errors.New("Failed to discover the mode change characteristic")
} }
@ -63,13 +65,13 @@ func RequestModeChange(client ble.Client, profile *ble.Profile) error {
return nil return nil
} }
func RequestSensorData(client ble.Client, profile *ble.Profile) (common.SensorDataResponse, error) { func RequestSensorData(client ble.Client) (common.SensorDataResponse, error) {
mifloraService := FindServiceByUUID(profile.Services, ble.MustParse(common.MifloraServiceUUID)) mifloraService := FindServiceByUUID(client.Profile().Services, common.MifloraServiceUUID)
if mifloraService == nil { if mifloraService == nil {
return common.SensorDataResponse{}, errors.New("Failed to get the miflora service") return common.SensorDataResponse{}, errors.New("Failed to get the miflora service")
} }
mifloraSensorDataChar := FindCharacteristicByUUID(mifloraService.Characteristics, ble.MustParse(common.MifloraCharReadSensorDataUUID)) mifloraSensorDataChar := FindCharacteristicByUUID(mifloraService.Characteristics, common.MifloraCharReadSensorDataUUID)
if mifloraSensorDataChar == nil { if mifloraSensorDataChar == nil {
return common.SensorDataResponse{}, errors.New("Failed to discover the sensor data characteristic") return common.SensorDataResponse{}, errors.New("Failed to discover the sensor data characteristic")
} }