140 lines
3.1 KiB
Go
140 lines
3.1 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"encoding/hex"
|
|
"errors"
|
|
"fmt"
|
|
"log"
|
|
"strconv"
|
|
"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
|
|
}
|
|
|
|
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) {
|
|
switch s {
|
|
case gatt.StatePoweredOn:
|
|
device.Scan([]gatt.UUID{}, true)
|
|
return
|
|
default:
|
|
device.StopScanning()
|
|
}
|
|
}
|
|
|
|
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)
|
|
}
|
|
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()) }),
|
|
)
|
|
dev.Init(onStateChanged)
|
|
select {}
|
|
}
|