初始

api/saft_hat/hat_status.go

@@ -0,0 +1,133 @@
+package saft_hat
+
+import (
+	"bytes"
+	"context"
+	"encoding/json"
+	"fmt"
+	"github.com/gogf/gf/v2/frame/g"
+	"io/ioutil"
+	"net/http"
+	"strconv"
+	"strings"
+	"time"
+)
+
+// 获取设备状态的请求结构
+type DeviceStatusReq struct {
+	g.Meta `path:"/device/status" method:"post" tags:"安全帽相关" summary:"获取安全帽状态"`
+	Data   []string `json:"data" dc:"设备号"`
+}
+
+// 获取设备状态
+func (h Hat) DeviceStatus(ctx context.Context, req *DeviceStatusReq) (res *DeviceStatusRes, err error) {
+	return StatusCheck(ctx, req.Data)
+}
+
+type DeviceStatusRes struct {
+	Code int                `json:"code"` // 响应代码
+	Data []DeviceStatusInfo `json:"data"` // 设备数据
+	Msg  string             `json:"msg"`  // 响应消息
+}
+
+// 定义数组中每个对象的结构
+type DeviceStatusInfo struct {
+	IMEI   string `json:"imei"`
+	Status int    `json:"status"`
+}
+
+// 检查设备的状态情况
+func StatusCheck(ctx context.Context, devNums []string) (res *DeviceStatusRes, err error) {
+	res = new(DeviceStatusRes)
+	data := strings.Join(devNums, ",")
+	timestamp := time.Now().Unix()
+
+	// 准备生成签名的参数
+	params := map[string]string{
+		"appid":     appid,
+		"data":      data,
+		"secret":    secret,
+		"timestamp": strconv.FormatInt(timestamp, 10),
+	}
+
+	// 生成签名
+	sign := GenerateSignature(params)
+
+	// 构造请求体
+	payload := map[string]interface{}{
+		"appid":     appid,
+		"data":      data,
+		"secret":    secret,
+		"sign":      sign,
+		"timestamp": timestamp,
+	}
+	payloadBytes, err := json.Marshal(payload)
+	if err != nil {
+		return res, err
+	}
+
+	// 发送POST请求
+	resp, err := http.Post("https://www.loctp.com/api/crm/v1/getStatus", "application/json", bytes.NewBuffer(payloadBytes))
+
+	if err != nil {
+		return res, err
+	}
+	defer resp.Body.Close()
+
+	// 检查响应
+	respBody, err := ioutil.ReadAll(resp.Body)
+	fmt.Println(string(respBody))
+	if err != nil {
+		return res, err
+	}
+
+	err = json.Unmarshal(respBody, res)
+	if err != nil {
+		return res, err
+	}
+
+	return res, nil
+}
+
+// 心跳检测
+func HeartCheck() {
+	ticker := time.NewTicker(60 * time.Second)
+	defer ticker.Stop() // 确保ticker被适当释放
+
+	for {
+		select {
+		case <-ticker.C:
+			var devices []Device
+			err := g.Model("device").Fields("dev_num").Scan(&devices)
+			if err != nil {
+				fmt.Println("获取设备列表出错:", err)
+				continue // 如果获取失败，则跳过本次循环
+			}
+
+			if len(devices) == 0 {
+				fmt.Println("没有找到任何设备")
+				continue // 如果没有设备，则跳过本次循环
+			}
+
+			// 构建devNums切片
+			var devNums []string
+			for _, device := range devices {
+				devNums = append(devNums, device.DevNum)
+			}
+			res, err := StatusCheck(context.Background(), devNums)
+			if err != nil {
+				fmt.Println("检查设备状态出错:", err)
+				continue // 如果检查失败，则跳过本次循环
+			}
+
+			for _, deviceStatus := range res.Data {
+				// 更新 device 表中的 status 字段
+				_, err := g.Model("device").Data("status", deviceStatus.Status).Where("dev_num", deviceStatus.IMEI).Update()
+				if err != nil {
+					fmt.Printf("更新设备 %s 状态出错: %v\n", deviceStatus.IMEI, err)
+					continue // 如果更新失败，则跳过本次设备的更新
+				}
+			}
+		}
+	}
+}