260 lines
7.6 KiB
Go
260 lines
7.6 KiB
Go
|
package saft_hat
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"encoding/json"
|
|||
|
|
"errors"
|
|||
|
|
"fmt"
|
|||
|
|
"github.com/gogf/gf/v2/encoding/gjson"
|
|||
|
|
"github.com/gogf/gf/v2/frame/g"
|
|||
|
|
"github.com/gogf/gf/v2/os/glog"
|
|||
|
|
"github.com/gorilla/websocket"
|
|||
|
|
"io/ioutil"
|
|||
|
|
"log"
|
|||
|
|
"net/http"
|
|||
|
|
"strconv"
|
|||
|
|
"time"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func StartWs() {
|
|||
|
|
// 设置 WebSocket 路由
|
|||
|
|
http.HandleFunc("/ws", HandleConnections)
|
|||
|
|
|
|||
|
|
// 开启一个新的协程，处理消息广播
|
|||
|
|
go HandleMessages()
|
|||
|
|
|
|||
|
|
// 启动 WebSocket 服务器
|
|||
|
|
log.Println("http server started on :8222")
|
|||
|
|
err := http.ListenAndServe(":8222", nil)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Fatal("ListenAndServe: ", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// WebSocket 部分
|
|||
|
|
var clients = make(map[*websocket.Conn]bool) // 连接的客户端
|
|||
|
|
var broadcast = make(chan []byte) // 广播通道
|
|||
|
|
var upgrader = websocket.Upgrader{
|
|||
|
|
CheckOrigin: func(r *http.Request) bool {
|
|||
|
|
return true
|
|||
|
|
},
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 处理连接
|
|||
|
|
func HandleConnections(w http.ResponseWriter, r *http.Request) {
|
|||
|
|
ws, err := upgrader.Upgrade(w, r, nil)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Fatal(err)
|
|||
|
|
}
|
|||
|
|
defer ws.Close()
|
|||
|
|
|
|||
|
|
// 注册新客户端
|
|||
|
|
clients[ws] = true
|
|||
|
|
|
|||
|
|
// 无限循环，保持连接活跃，但不处理任何消息
|
|||
|
|
for {
|
|||
|
|
if _, _, err := ws.ReadMessage(); err != nil {
|
|||
|
|
log.Printf("error: %v", err)
|
|||
|
|
delete(clients, ws)
|
|||
|
|
break
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 处理消息
|
|||
|
|
func HandleMessages() {
|
|||
|
|
for {
|
|||
|
|
// 从广播通道中获取消息
|
|||
|
|
msg := <-broadcast
|
|||
|
|
|
|||
|
|
// 发送消息到所有连接的客户端
|
|||
|
|
for client := range clients {
|
|||
|
|
err := client.WriteMessage(websocket.TextMessage, msg)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Printf("websocket write error: %v", err)
|
|||
|
|
client.Close()
|
|||
|
|
delete(clients, client)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// WS 推送的数据
|
|||
|
|
type BroadcastLocation struct {
|
|||
|
|
DevNum string `json:"devNum"`
|
|||
|
|
Time string `json:"time"`
|
|||
|
|
Latitude float64 `json:"latitude"`
|
|||
|
|
Longitude float64 `json:"longitude"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Webhook 定位数据上传【远程发送给我们平台的数据】
|
|||
|
|
type WebhookRequest struct {
|
|||
|
|
Temperature float64 `json:"temperature"` // 设备采集温度数据
|
|||
|
|
Humidity float64 `json:"humidity"` // 设备采集湿度数据
|
|||
|
|
Posture int `json:"posture"` // 姿态：1正常 -1脱帽 -2倒地
|
|||
|
|
BatteryTemp float64 `json:"batteryTemp"` // 电池温度
|
|||
|
|
FixedBy string `json:"fixedBy"` // 定位方式:GPS/BD、WIFI、BT
|
|||
|
|
UtcDateTime int64 `json:"utcDateTime"` // 设备内部上传时间戳
|
|||
|
|
IMEI string `json:"IMEI"` // 设备IMEI
|
|||
|
|
BatteryLevel float64 `json:"batteryLevel"` // 电池电量
|
|||
|
|
Charging int `json:"charging"` // 是否正在充电 1充电 0没充电
|
|||
|
|
BluetoothMac string `json:"bluetoothMac"` // 蓝牙Mac地址
|
|||
|
|
Type string `json:"type"` // 设备上传类型和时间
|
|||
|
|
Latitude float64 `json:"latitude"` // WGS-84：是国际标准，GPS坐标（Google Earth使用、或者GPS模块）纬度坐标
|
|||
|
|
Longitude float64 `json:"longitude"` // WGS-84：是国际标准，GPS坐标（Google Earth使用、或者GPS模块）经度坐标
|
|||
|
|
Altitude int `json:"altitude"` // 海拔高度
|
|||
|
|
BT string `json:"bt"` // 蓝牙定位相关：{个数}:{地址,信号}
|
|||
|
|
LBS []string `json:"LBS"` // 多基站定位信息
|
|||
|
|
MAC []string `json:"MAC"` // WiFi - MAC地址信号
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 定位数据上传请求
|
|||
|
|
type DataReq struct {
|
|||
|
|
g.Meta `path:"/device/data" method:"post" tags:"安全帽相关" summary:"接收安全帽数据(不需要前端调用)"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type DataRes struct {
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (h Hat) Data(ctx context.Context, req *DataReq) (res *DataRes, err error) {
|
|||
|
|
res = new(DataRes)
|
|||
|
|
r := g.RequestFromCtx(ctx)
|
|||
|
|
body, err := ioutil.ReadAll(r.Body)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Printf("读取请求头失败: %v", err)
|
|||
|
|
return res, err
|
|||
|
|
}
|
|||
|
|
defer r.Body.Close() // 延迟关闭请求体
|
|||
|
|
|
|||
|
|
// 从请求头中获取时间戳和签名
|
|||
|
|
timestamp := r.GetHeader("timestamp")
|
|||
|
|
signature := r.GetHeader("signature")
|
|||
|
|
|
|||
|
|
// 验证签名的有效性
|
|||
|
|
if !VerifySignature(string(body)+timestamp, signature, secret) {
|
|||
|
|
glog.Errorf(ctx, "签名验证失败")
|
|||
|
|
return res, errors.New("验证签名失败")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 解析请求体中的JSON数据到预定义的结构体
|
|||
|
|
var webhookData WebhookRequest
|
|||
|
|
if err := gjson.DecodeTo(body, &webhookData); err != nil {
|
|||
|
|
return res, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 拿到数据之后，先取出 IMEI 设备号，根据 IMEI 查询，不存在则先插入数据到 device 表
|
|||
|
|
device := Device{
|
|||
|
|
DevNum: webhookData.IMEI,
|
|||
|
|
Temperature: webhookData.Temperature,
|
|||
|
|
Humidity: webhookData.Humidity,
|
|||
|
|
Posture: webhookData.Posture,
|
|||
|
|
FixedBy: webhookData.FixedBy,
|
|||
|
|
BatteryLevel: webhookData.BatteryLevel,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 查询设备是否存在，存在则更新安全帽的状态信息
|
|||
|
|
count, err := g.Model(&device).Where("dev_num", device.DevNum).Count()
|
|||
|
|
if err != nil {
|
|||
|
|
glog.Errorf(ctx, "查询设备是否存在出错：%v", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if count > 0 {
|
|||
|
|
// 判断电池是否处于低电量状态
|
|||
|
|
var isLowBattery int
|
|||
|
|
|
|||
|
|
// 电量小于 0.2 则设置为低电量
|
|||
|
|
if device.BatteryLevel < 0.2 {
|
|||
|
|
isLowBattery = 1
|
|||
|
|
} else {
|
|||
|
|
isLowBattery = 0
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 更新设备的数据
|
|||
|
|
_, err = g.Model("device").Data(g.Map{
|
|||
|
|
"temperature": device.Temperature,
|
|||
|
|
"humidity": device.Humidity,
|
|||
|
|
"posture": device.Posture,
|
|||
|
|
"battery_temp": device.BatteryTemp,
|
|||
|
|
"fixed_by": device.FixedBy,
|
|||
|
|
"battery_level": device.BatteryLevel * 100,
|
|||
|
|
"is_low_battery": isLowBattery,
|
|||
|
|
"update_time": time.Now(),
|
|||
|
|
}).Where("dev_num", device.DevNum).Update()
|
|||
|
|
|
|||
|
|
if err != nil {
|
|||
|
|
glog.Errorf(ctx, "更新设备数据出错：%v", err)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 将接收到的数据转换为 Location 结构体
|
|||
|
|
location := Location{
|
|||
|
|
DevNum: webhookData.IMEI, // 设备编号为 IMEI
|
|||
|
|
Time: time.Now().Format("2006-01-02 15:04:05"), // 当前时间
|
|||
|
|
Latitude: webhookData.Latitude, // 将纬度转换为指针类型
|
|||
|
|
Longitude: webhookData.Longitude, // 将经度转换为指针类型
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 构建插入数据的参数
|
|||
|
|
data := g.Map{
|
|||
|
|
"dev_num": location.DevNum,
|
|||
|
|
"time": location.Time,
|
|||
|
|
"latitude": location.Latitude,
|
|||
|
|
"longitude": location.Longitude,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 检查经纬度是否同时为 0，如果是则不执行插入操作，也不需要调用 Redis
|
|||
|
|
if location.Latitude != 0 || location.Longitude != 0 {
|
|||
|
|
// 执行插入操作
|
|||
|
|
_, err = g.Model("location").Data(data).Insert()
|
|||
|
|
// 获取 Redis 客户端
|
|||
|
|
redis := g.Redis("helmetRedis")
|
|||
|
|
|
|||
|
|
// 构造 Redis 的 key 和 value
|
|||
|
|
key := "safety_helmet:" + location.DevNum
|
|||
|
|
value := strconv.FormatFloat(location.Latitude, 'f', -1, 64) + "," + strconv.FormatFloat(location.Longitude, 'f', -1, 64)
|
|||
|
|
|
|||
|
|
// 插入数据之后，写入 Redis 的发布订阅
|
|||
|
|
_, err = redis.Publish(ctx, key, value)
|
|||
|
|
if err != nil {
|
|||
|
|
glog.Info(ctx, "发布订阅出错")
|
|||
|
|
return res, nil
|
|||
|
|
}
|
|||
|
|
// 插入数据之后，写入 Redis 的键值对
|
|||
|
|
_, err = redis.Set(ctx, key, value)
|
|||
|
|
if err != nil {
|
|||
|
|
glog.Info(ctx, "设置到Redis出错")
|
|||
|
|
return res, nil
|
|||
|
|
}
|
|||
|
|
if err != nil {
|
|||
|
|
// 处理可能的错误
|
|||
|
|
fmt.Println("Insert error:", err)
|
|||
|
|
}
|
|||
|
|
} else {
|
|||
|
|
fmt.Println("Latitude and Longitude are both zero, insertion skipped.")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 返回响应对象
|
|||
|
|
return &DataRes{}, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 发送数据到 WS 客户端
|
|||
|
|
func sendToWs(location Location, err error) {
|
|||
|
|
// 创建 WS 需要发送的数据对象
|
|||
|
|
broadcastLocation := BroadcastLocation{
|
|||
|
|
DevNum: location.DevNum,
|
|||
|
|
Time: location.Time,
|
|||
|
|
Latitude: location.Latitude,
|
|||
|
|
Longitude: location.Longitude,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 转换为 JSON 字符串
|
|||
|
|
locationJSON, err := json.Marshal(broadcastLocation)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Printf("location json marshal error: %v", err)
|
|||
|
|
} else {
|
|||
|
|
// 发送转换后的JSON字符串到广播通道
|
|||
|
|
broadcast <- locationJSON
|
|||
|
|
}
|
|||
|
|
}
|