Decorator for ble service added

ble_handler.go

@@ -0,0 +1,98 @@
+package main
+
+import (
+	"fmt"
+	"log"
+
+	"blegw/workers"
+
+	"github.com/currantlabs/gatt"
+	"github.com/currantlabs/gatt/examples/option"
+)
+
+var done = make(chan struct{})
+
+type BleService struct {
+	workers []workers.Worker
+}
+
+// AddWorker - add new worker
+func (s BleService) AddWorker(worker workers.Worker) {
+	if s.workers == nil {
+		s.workers = []workers.Worker{}
+	}
+	s.workers = append(s.workers, worker)
+}
+
+func (s BleService) Init() error {
+	d, err := gatt.NewDevice(option.DefaultClientOptions...)
+	if err != nil {
+		log.Fatalf("Failed to open device, err: %s\n", err)
+		return err
+	}
+
+	// Register handlers.
+	d.Handle(
+		gatt.PeripheralDiscovered(s.onPeriphDiscovered),
+		gatt.PeripheralConnected(s.onPeriphConnected),
+		gatt.PeripheralDisconnected(s.onPeriphDisconnected),
+	)
+
+	d.Init(s.onStateChanged)
+	<-done
+	fmt.Println("Done")
+
+	return nil
+}
+
+func (s BleService) onStateChanged(d gatt.Device, state gatt.State) {
+	fmt.Println("State:", state)
+	switch state {
+	case gatt.StatePoweredOn:
+		fmt.Println("scanning...")
+		d.Scan([]gatt.UUID{}, false)
+		return
+	default:
+		d.StopScanning()
+	}
+}
+
+func (s BleService) onPeriphDiscovered(p gatt.Peripheral, a *gatt.Advertisement, rssi int) {
+	fmt.Printf("\nPeripheral ID:%s, NAME:(%s)\n", p.ID(), p.Name())
+	fmt.Println("  Local Name        =", a.LocalName)
+	fmt.Println("  TX Power Level    =", a.TxPowerLevel)
+	fmt.Println("  Manufacturer Data =", a.ManufacturerData)
+	fmt.Println("  Service Data      =", a.ServiceData)
+	fmt.Println("  RSSI      =", rssi)
+
+	isConnect := false
+	for _, worker := range s.workers {
+		ret := worker.OnDiscovery(p, a, rssi)
+		if ret && !isConnect {
+			isConnect = true
+		}
+	}
+
+	if isConnect {
+		p.Device().Connect(p)
+	}
+}
+
+func (s BleService) onPeriphConnected(p gatt.Peripheral, err error) {
+	fmt.Println("Connected")
+	defer p.Device().CancelConnection(p)
+
+	if err := p.SetMTU(500); err != nil {
+		fmt.Printf("Failed to set MTU, err: %s\n", err)
+	}
+	fmt.Printf("Set mtu done.\n")
+
+	for _, worker := range s.workers {
+		worker.OnConnect(p, err)
+	}
+}
+
+func (s BleService) onPeriphDisconnected(p gatt.Peripheral, err error) {
+	fmt.Println("Disconnected")
+	close(done)
+}