Initialize LineUp app server

modules/client.go

@@ -0,0 +1,88 @@
+package modules
+
+import (
+	"encoding/json"
+	"log"
+	"net/http"
+	"time"
+
+	"github.com/gin-gonic/gin"
+	"github.com/gorilla/websocket"
+	"github.com/lineup/app-server/internal"
+	"github.com/lineup/app-server/internal/wkapi"
+)
+
+var upgrader = websocket.Upgrader{
+	CheckOrigin: func(r *http.Request) bool { return true },
+}
+
+// ClientModule manages user WebSocket connections for real-time push.
+type ClientModule struct {
+	hub *internal.Hub
+	wk  *wkapi.Client
+}
+
+// NewClientModule creates a new ClientModule.
+func NewClientModule(hub *internal.Hub, wk *wkapi.Client) *ClientModule {
+	return &ClientModule{hub: hub, wk: wk}
+}
+
+// RegisterRoutes registers WebSocket routes.
+func (m *ClientModule) RegisterRoutes(r *gin.Engine) {
+	r.GET("/ws", m.handleWS)
+}
+
+func (m *ClientModule) handleWS(c *gin.Context) {
+	token := c.Query("token")
+	uid := c.Query("uid")
+	if token == "" || uid == "" {
+		c.JSON(http.StatusBadRequest, gin.H{"error": "参数错误: token和uid必填"})
+		return
+	}
+
+	// Verify token with WuKongIM (register if not exists, or validate)
+	if err := m.wk.RegisterToken(uid, token); err != nil {
+		c.JSON(http.StatusUnauthorized, gin.H{"error": "Token验证失败: " + err.Error()})
+		return
+	}
+
+	conn, err := upgrader.Upgrade(c.Writer, c.Request, nil)
+	if err != nil {
+		log.Printf("WebSocket升级失败: %v", err)
+		return
+	}
+
+	m.hub.Register(uid, conn)
+	log.Printf("客户端连接: uid=%s", uid)
+
+	// Send a welcome message
+	welcome, _ := json.Marshal(map[string]interface{}{
+		"event": "connected",
+		"uid":   uid,
+	})
+	conn.WriteMessage(websocket.TextMessage, welcome)
+
+	// Read loop (keep connection alive, handle pings)
+	go m.readLoop(uid, conn)
+}
+
+func (m *ClientModule) readLoop(uid string, conn *websocket.Conn) {
+	defer func() {
+		m.hub.Unregister(uid)
+		conn.Close()
+		log.Printf("客户端断开: uid=%s", uid)
+	}()
+
+	conn.SetReadDeadline(time.Now().Add(60 * time.Second))
+	conn.SetPongHandler(func(string) error {
+		conn.SetReadDeadline(time.Now().Add(60 * time.Second))
+		return nil
+	})
+
+	for {
+		_, _, err := conn.ReadMessage()
+		if err != nil {
+			break
+		}
+	}
+}