init

2024-12-23 18:34:46 +08:00
commit cbfcc91eec
24 changed files with 1702 additions and 0 deletions
--- a/ws/client.go
+++ b/ws/client.go
@@ -0,0 +1,116 @@
+package ws
+
+import (
+	"encoding/json"
+	"github.com/gorilla/websocket"
+	"log"
+	"time"
+)
+
+const (
+	writeWait      = 10 * time.Second
+	pongWait       = 60 * time.Second
+	pingPeriod     = (pongWait * 9) / 10
+	maxMessageSize = 512
+)
+
+// Client WebSocket 客户端连接
+type Client struct {
+	Hub      *Hub
+	Conn     *websocket.Conn
+	Send     chan *WsMessage
+	Imei     string
+	DeviceId int
+}
+
+func (c *Client) ReadPump() {
+	defer func() {
+		c.Hub.Unregister <- c
+		c.Conn.Close()
+	}()
+
+	c.Conn.SetReadLimit(maxMessageSize)
+	c.Conn.SetReadDeadline(time.Now().Add(pongWait))
+	c.Conn.SetPongHandler(func(string) error {
+		c.Conn.SetReadDeadline(time.Now().Add(pongWait))
+		return nil
+	})
+
+	for {
+		_, message, err := c.Conn.ReadMessage()
+		if err != nil {
+			if websocket.IsUnexpectedCloseError(err, websocket.CloseGoingAway, websocket.CloseAbnormalClosure) {
+				log.Printf("读取错误: %v", err)
+			}
+			return
+		}
+
+		// 检查是否是字符串 "PING"
+		if string(message) == "PING" {
+			c.Conn.SetWriteDeadline(time.Now().Add(writeWait))
+			err := c.Conn.WriteMessage(websocket.TextMessage, []byte("PONG"))
+			if err != nil {
+				log.Printf("发送 PONG 响应失败: %v", err)
+				return
+			}
+			c.Conn.SetReadDeadline(time.Now().Add(pongWait))
+			continue
+		}
+
+		// 处理 JSON 消息
+		var msg struct {
+			Type string `json:"Type"`
+		}
+		if err := json.Unmarshal(message, &msg); err == nil && msg.Type == "ping" {
+			c.Conn.SetWriteDeadline(time.Now().Add(writeWait))
+			response := struct {
+				Type string `json:"Type"`
+			}{
+				Type: "pong",
+			}
+			responseData, _ := json.Marshal(response)
+			err := c.Conn.WriteMessage(websocket.TextMessage, responseData)
+			if err != nil {
+				log.Printf("发送 pong 响应失败: %v", err)
+				return
+			}
+			c.Conn.SetReadDeadline(time.Now().Add(pongWait))
+		}
+	}
+}
+
+func (c *Client) WritePump() {
+	ticker := time.NewTicker(pingPeriod)
+	defer func() {
+		ticker.Stop()
+		c.Conn.Close()
+	}()
+
+	for {
+		select {
+		case message, ok := <-c.Send:
+			if !ok {
+				c.Conn.WriteMessage(websocket.CloseMessage, []byte{})
+				return
+			}
+
+			c.Conn.SetWriteDeadline(time.Now().Add(writeWait))
+			log.Printf("准备发送消息到 WebSocket: IMEI=%s, Data=%s", message.IMEI, message.Data)
+
+			// 直接发送原始消息字符串
+			err := c.Conn.WriteMessage(websocket.TextMessage, []byte(message.Data))
+			if err != nil {
+				log.Printf("发送消息失败: %v", err)
+				return
+			}
+			log.Printf("消息发送成功: IMEI=%s", message.IMEI)
+
+		case <-ticker.C:
+			c.Conn.SetWriteDeadline(time.Now().Add(writeWait))
+			if err := c.Conn.WriteMessage(websocket.PingMessage, nil); err != nil {
+				log.Printf("发送 ping 失败: %v", err)
+				return
+			}
+		}
+	}
+}