90 lines
2.1 KiB
Go
90 lines
2.1 KiB
Go
package video_hat
|
||
|
||
import (
|
||
"encoding/json"
|
||
"github.com/gorilla/websocket"
|
||
"github.com/tiger1103/gfast/v3/api/constant"
|
||
"github.com/tiger1103/gfast/v3/api/video_hat/ws2"
|
||
"log"
|
||
"time"
|
||
)
|
||
|
||
// 项目启动就登录视频安全帽的 WS
|
||
func WsConnection() {
|
||
// 首先加载 WS 路径映射
|
||
InitWsMap()
|
||
|
||
// 无限循环尝试连接，直到成功
|
||
for {
|
||
if connectWs() {
|
||
break // 如果连接成功，则退出循环
|
||
}
|
||
log.Println("Reconnecting in 5 seconds...")
|
||
time.Sleep(5 * time.Second) // 等待5秒后重新连接WS
|
||
}
|
||
}
|
||
|
||
func connectWs() bool {
|
||
// 创建 WebSocket 连接
|
||
dialer := websocket.Dialer{
|
||
HandshakeTimeout: 10 * time.Second,
|
||
}
|
||
|
||
// 尝试连接 WebSocket
|
||
var err error
|
||
constant.Conn, _, err = dialer.Dial(constant.Url, nil)
|
||
if err != nil {
|
||
log.Println("Dial:", err)
|
||
return false
|
||
}
|
||
defer constant.Conn.Close()
|
||
|
||
// 连接成功之后发送登录信息
|
||
ws2.Login(constant.Conn)
|
||
|
||
// 设置定时任务，每间隔60秒发送参数获取实时、状态等心跳包
|
||
go ws2.SendHeartbeatTime(constant.Conn)
|
||
|
||
// 处理从 WebSocket 接收的消息
|
||
if handlerFun(constant.Conn) {
|
||
return true
|
||
}
|
||
|
||
return false
|
||
}
|
||
|
||
func handlerFun(Conn *websocket.Conn) bool {
|
||
// 读取 Conn 的数据成 JSON 字符串
|
||
for {
|
||
_, message, err := Conn.ReadMessage()
|
||
|
||
if err != nil {
|
||
log.Println("ReadMessage:", err)
|
||
return false // 如果读取过程中出现错误，则返回 false
|
||
}
|
||
|
||
// 创建一个 map 来存储解析后的 JSON
|
||
var msgMap map[string]interface{}
|
||
|
||
// 将读取到的字节切片解析为 JSON
|
||
err = json.Unmarshal(message, &msgMap)
|
||
if err != nil {
|
||
log.Println("Unmarshal:", err)
|
||
continue // 如果解析失败，则跳过此次循环
|
||
}
|
||
|
||
if cmd, ok := msgMap["cmd"].(string); ok {
|
||
if handler, exists := commandHandlers[cmd]; exists {
|
||
jsonString, err := json.Marshal(msgMap)
|
||
if err != nil {
|
||
log.Println("Marshal:", err)
|
||
continue // 如果序列化失败，则跳过此次循环
|
||
}
|
||
handler(string(jsonString)) // 调用处理函数
|
||
}
|
||
}
|
||
}
|
||
|
||
return true // 正常退出循环，返回 true
|
||
}
|