Compare commits
1 Commits
| Author | SHA1 | Date |
|---|---|---|
 suguo.yao
|
8800620f7e |
2
Makefile
2
Makefile
|
|
@ -7,4 +7,6 @@ arm:
|
|||
release: arm
|
||||
ssh pi@192.168.0.21 "sudo systemctl stop ble.service"
|
||||
scp ./ble-ibeacon pi@192.168.0.21:~/
|
||||
|
||||
aaa:
|
||||
ssh pi@192.168.0.21 "sudo systemctl start ble.service"
|
||||
255
main.go
255
main.go
|
|
@ -1,139 +1,164 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"encoding/binary"
|
||||
"encoding/hex"
|
||||
"errors"
|
||||
"flag"
|
||||
"fmt"
|
||||
"log"
|
||||
"strconv"
|
||||
"os"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/influxdata/influxdb/client/v2"
|
||||
"github.com/paypal/gatt"
|
||||
"github.com/paypal/gatt/examples/option"
|
||||
"myschools.me/suguo/ble-ibeacon/influx"
|
||||
)
|
||||
|
||||
type IBeancon struct {
|
||||
Uuid string
|
||||
Major uint16
|
||||
Minor uint16
|
||||
}
|
||||
var done = make(chan struct{})
|
||||
|
||||
func NewiBeacon(data []byte) (*IBeancon, error) {
|
||||
if len(data) < 25 || binary.BigEndian.Uint32(data) != 0x4c000215 {
|
||||
return nil, errors.New("not an iBeacon")
|
||||
}
|
||||
beacon := new(IBeancon)
|
||||
beacon.Uuid = strings.ToUpper(hex.EncodeToString(data[4:8]) + hex.EncodeToString(data[8:10]) + hex.EncodeToString(data[10:12]) + hex.EncodeToString(data[12:14]) + hex.EncodeToString(data[14:20]))
|
||||
beacon.Major = binary.BigEndian.Uint16(data[20:22])
|
||||
beacon.Minor = binary.BigEndian.Uint16(data[22:24])
|
||||
return beacon, nil
|
||||
}
|
||||
|
||||
var (
|
||||
chn chan *client.Point
|
||||
)
|
||||
|
||||
func init() {
|
||||
chn = make(chan *client.Point, 500)
|
||||
}
|
||||
|
||||
func onPerhipheralDiscovered(p gatt.Peripheral, a *gatt.Advertisement, rssi int) {
|
||||
// if p.ID() != "78:2C:29:16:F2:78" {
|
||||
// return
|
||||
// }
|
||||
// if p.ID() != "90:98:38:F5:DA:60" {
|
||||
// return
|
||||
// }
|
||||
// if p.ID() != "E4:78:41:19:10:98" {
|
||||
// return
|
||||
// }
|
||||
// if !(p.ID() == "C6:1A:05:02:06:C0" || p.ID() == "90:98:38:F5:DA:60") {
|
||||
// return
|
||||
// }
|
||||
// fmt.Println("NAME:", a.LocalName, "T:", time.Now(), "MAC:", p.ID(), "RSSI:", rssi, "LEN:", len(a.ManufacturerData))
|
||||
|
||||
tags := make(map[string]string)
|
||||
tags["mac"] = p.ID()
|
||||
|
||||
fields := make(map[string]interface{})
|
||||
fields["name"] = a.LocalName
|
||||
fields["rssi"] = rssi
|
||||
fields["pl"] = strconv.Itoa(len(a.ManufacturerData)) //包长度
|
||||
fields["power"] = a.TxPowerLevel
|
||||
|
||||
point, err := client.NewPoint(
|
||||
"band",
|
||||
tags,
|
||||
fields,
|
||||
time.Now(),
|
||||
)
|
||||
if err != nil {
|
||||
log.Fatalln(err.Error())
|
||||
}
|
||||
chn <- point
|
||||
}
|
||||
|
||||
func onStateChanged(device gatt.Device, s gatt.State) {
|
||||
func onStateChanged(d gatt.Device, s gatt.State) {
|
||||
fmt.Println("State:", s)
|
||||
switch s {
|
||||
case gatt.StatePoweredOn:
|
||||
device.Scan([]gatt.UUID{}, true)
|
||||
fmt.Println("Scanning...")
|
||||
d.Scan([]gatt.UUID{}, false)
|
||||
return
|
||||
default:
|
||||
device.StopScanning()
|
||||
d.StopScanning()
|
||||
}
|
||||
}
|
||||
|
||||
func onPeriphDiscovered(p gatt.Peripheral, a *gatt.Advertisement, rssi int) {
|
||||
id := strings.ToUpper(flag.Args()[0])
|
||||
if strings.ToUpper(p.ID()) != id {
|
||||
return
|
||||
}
|
||||
|
||||
// Stop scanning once we've got the peripheral we're looking for.
|
||||
p.Device().StopScanning()
|
||||
|
||||
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("")
|
||||
|
||||
p.Device().Connect(p)
|
||||
}
|
||||
|
||||
func 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)
|
||||
}
|
||||
|
||||
// Discovery services
|
||||
ss, err := p.DiscoverServices(nil)
|
||||
if err != nil {
|
||||
fmt.Printf("Failed to discover services, err: %s\n", err)
|
||||
return
|
||||
}
|
||||
|
||||
for _, s := range ss {
|
||||
msg := "Service: " + s.UUID().String()
|
||||
if len(s.Name()) > 0 {
|
||||
msg += " (" + s.Name() + ")"
|
||||
}
|
||||
fmt.Println(msg)
|
||||
|
||||
// Discovery characteristics
|
||||
cs, err := p.DiscoverCharacteristics(nil, s)
|
||||
if err != nil {
|
||||
fmt.Printf("Failed to discover characteristics, err: %s\n", err)
|
||||
continue
|
||||
}
|
||||
|
||||
for _, c := range cs {
|
||||
msg := " Characteristic " + c.UUID().String()
|
||||
if len(c.Name()) > 0 {
|
||||
msg += " (" + c.Name() + ")"
|
||||
}
|
||||
msg += "\n properties " + c.Properties().String()
|
||||
fmt.Println(msg)
|
||||
|
||||
// Read the characteristic, if possible.
|
||||
if (c.Properties() & gatt.CharRead) != 0 {
|
||||
b, err := p.ReadCharacteristic(c)
|
||||
if err != nil {
|
||||
fmt.Printf("Failed to read characteristic, err: %s\n", err)
|
||||
continue
|
||||
}
|
||||
fmt.Printf(" value %x | %q\n", b, b)
|
||||
}
|
||||
|
||||
// Discovery descriptors
|
||||
ds, err := p.DiscoverDescriptors(nil, c)
|
||||
if err != nil {
|
||||
fmt.Printf("Failed to discover descriptors, err: %s\n", err)
|
||||
continue
|
||||
}
|
||||
|
||||
for _, d := range ds {
|
||||
msg := " Descriptor " + d.UUID().String()
|
||||
if len(d.Name()) > 0 {
|
||||
msg += " (" + d.Name() + ")"
|
||||
}
|
||||
fmt.Println(msg)
|
||||
|
||||
// Read descriptor (could fail, if it's not readable)
|
||||
b, err := p.ReadDescriptor(d)
|
||||
if err != nil {
|
||||
fmt.Printf("Failed to read descriptor, err: %s\n", err)
|
||||
continue
|
||||
}
|
||||
fmt.Printf(" value %x | %q\n", b, b)
|
||||
}
|
||||
|
||||
// Subscribe the characteristic, if possible.
|
||||
if (c.Properties() & (gatt.CharNotify | gatt.CharIndicate)) != 0 {
|
||||
f := func(c *gatt.Characteristic, b []byte, err error) {
|
||||
fmt.Printf("notified: % X | %q\n", b, b)
|
||||
}
|
||||
if err := p.SetNotifyValue(c, f); err != nil {
|
||||
fmt.Printf("Failed to subscribe characteristic, err: %s\n", err)
|
||||
continue
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
fmt.Println()
|
||||
}
|
||||
|
||||
fmt.Printf("Waiting for 5 seconds to get some notifiations, if any.\n")
|
||||
time.Sleep(5 * time.Second)
|
||||
}
|
||||
|
||||
func onPeriphDisconnected(p gatt.Peripheral, err error) {
|
||||
fmt.Println("Disconnected")
|
||||
close(done)
|
||||
}
|
||||
|
||||
func main() {
|
||||
influx.Init(&influx.Config{
|
||||
Address: "http://127.0.0.1:8086",
|
||||
DBName: "wristband",
|
||||
Username: "abc",
|
||||
Password: "abc",
|
||||
Timeout: 3,
|
||||
})
|
||||
|
||||
// if err := influx.CreateDB("wristband", 10); err != nil {
|
||||
// log.Fatalln("influx.CreateDB", err.Error())
|
||||
// }
|
||||
|
||||
go func() {
|
||||
db, err := influx.New()
|
||||
if err != nil {
|
||||
log.Fatalln(err.Error())
|
||||
}
|
||||
|
||||
for {
|
||||
bp, err := client.NewBatchPoints(client.BatchPointsConfig{
|
||||
Database: "wristband",
|
||||
Precision: "ms",
|
||||
})
|
||||
if err != nil {
|
||||
log.Fatalln(err.Error())
|
||||
}
|
||||
for i := 1; i < 50; i++ {
|
||||
if d, ok := <-chn; ok {
|
||||
bp.AddPoint(d)
|
||||
}
|
||||
}
|
||||
if err := db.Write(bp); err != nil {
|
||||
log.Println("db.Write: ", err.Error())
|
||||
}
|
||||
}
|
||||
}()
|
||||
|
||||
dev, err := gatt.NewDevice(option.DefaultClientOptions...)
|
||||
if err != nil {
|
||||
log.Fatalf("Failed to open device, err:%s\n", err)
|
||||
flag.Parse()
|
||||
if len(flag.Args()) != 1 {
|
||||
log.Fatalf("usage: %s [options] peripheral-id\n", os.Args[0])
|
||||
}
|
||||
dev.Handle(
|
||||
gatt.PeripheralDiscovered(onPerhipheralDiscovered),
|
||||
gatt.CentralConnected(func(c gatt.Central) { fmt.Println("Connect: ", c.ID()) }),
|
||||
gatt.CentralDisconnected(func(c gatt.Central) { fmt.Println("Disconnect: ", c.ID()) }),
|
||||
|
||||
d, err := gatt.NewDevice(option.DefaultClientOptions...)
|
||||
if err != nil {
|
||||
log.Fatalf("Failed to open device, err: %s\n", err)
|
||||
return
|
||||
}
|
||||
|
||||
// Register handlers.
|
||||
d.Handle(
|
||||
gatt.PeripheralDiscovered(onPeriphDiscovered),
|
||||
gatt.PeripheralConnected(onPeriphConnected),
|
||||
gatt.PeripheralDisconnected(onPeriphDisconnected),
|
||||
)
|
||||
dev.Init(onStateChanged)
|
||||
select {}
|
||||
|
||||
d.Init(onStateChanged)
|
||||
<-done
|
||||
fmt.Println("Done")
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in New Issue