1282 lines
41 KiB
Go
1282 lines
41 KiB
Go
/**
|
||
* package controller
|
||
* 作用：处理核心业务逻辑，包括 WebSocket 连接管理、消息转发、Redis 集群通信以及 API 接口实现。
|
||
* 该包是连接网络层（Gin）与数据层（Gorm/Redis）的桥梁。
|
||
*/
|
||
package controller
|
||
|
||
import (
|
||
"context"
|
||
"encoding/json"
|
||
"fmt"
|
||
"log"
|
||
"net/http"
|
||
"os"
|
||
"strconv"
|
||
"sync"
|
||
"time"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
"github.com/go-redis/redis/v8"
|
||
"github.com/gorilla/websocket"
|
||
"gorm.io/gorm"
|
||
"xk-websocket/models"
|
||
"xk-websocket/utils"
|
||
)
|
||
|
||
var upgrader = websocket.Upgrader{
|
||
ReadBufferSize: 1024,
|
||
WriteBufferSize: 1024,
|
||
CheckOrigin: func(r *http.Request) bool {
|
||
return true
|
||
},
|
||
}
|
||
|
||
/**
|
||
* WebSocketController
|
||
* 结构体定义：WebSocket 服务的主控制器。
|
||
* 包含：
|
||
* - 客户端连接池 (Clients)
|
||
* - 读写锁 (ClientsMux) 保护连接池安全
|
||
* - Redis 客户端 (RedisCli) 用于集群通信
|
||
* - 数据库连接 (DB)
|
||
*/
|
||
type WebSocketController struct {
|
||
Clients map[string]*websocket.Conn
|
||
ClientsMux sync.RWMutex
|
||
WriteMutex sync.Mutex // 添加写锁
|
||
RedisCli *redis.Client
|
||
NodeID string
|
||
Port string
|
||
RedisCtx context.Context
|
||
Logger *log.Logger
|
||
DB *gorm.DB // 数据库连接
|
||
}
|
||
|
||
/**
|
||
* NewWebSocketController
|
||
* 功能：创建并初始化 WebSocketController 实例。
|
||
* @param db *gorm.DB 数据库连接实例
|
||
*/
|
||
func NewWebSocketController(db *gorm.DB) *WebSocketController {
|
||
return &WebSocketController{
|
||
Clients: make(map[string]*websocket.Conn),
|
||
RedisCtx: context.Background(),
|
||
DB: db, // 初始化数据库连接
|
||
}
|
||
}
|
||
|
||
/**
|
||
* ConfigureSystem
|
||
* 功能：加载系统配置，如 NodeID 和端口，并设置日志格式。
|
||
* 也会启动节点注册协程。
|
||
*/
|
||
func (c *WebSocketController) ConfigureSystem() {
|
||
c.NodeID = utils.GetEnv("NODE_ID", "local")
|
||
|
||
// 只有当端口未设置时才从环境变量获取
|
||
if c.Port == "" {
|
||
c.Port = utils.GetEnv("PORT", "12080")
|
||
log.Printf("📡📡📡📡📡📡📡📡📡📡📡📡📡📡📡📡 使用环境变量设置端口: %s", c.Port)
|
||
}
|
||
|
||
log.SetPrefix(fmt.Sprintf("[Node:%s] ", c.NodeID))
|
||
log.SetFlags(log.LstdFlags | log.Lmicroseconds)
|
||
c.configureLogger()
|
||
|
||
// 必须先初始化 Redis 客户端
|
||
c.InitRedisClient()
|
||
|
||
// 注册节点到Redis
|
||
go c.registerNode() // 现在 RedisCli 已初始化
|
||
}
|
||
|
||
/**
|
||
* registerNode
|
||
* 功能：定时将当前节点的 IP 和端口注册到 Redis 中，以便其他服务发现。
|
||
* 这是一个心跳机制，每 20 秒执行一次。
|
||
*/
|
||
func (c *WebSocketController) registerNode() {
|
||
// 添加空值检查
|
||
if c.RedisCli == nil {
|
||
log.Println("⚠️ Redis客户端未初始化，节点注册已取消")
|
||
return
|
||
}
|
||
|
||
for {
|
||
key := fmt.Sprintf("websocket:nodes:%s", c.NodeID)
|
||
value := fmt.Sprintf("%s:%s", utils.GetOutboundIP(), c.Port)
|
||
|
||
if err := c.RedisCli.Set(c.RedisCtx, key, value, 30*time.Second).Err(); err != nil {
|
||
log.Printf("⚠️ 节点注册失败: %v", err)
|
||
} else {
|
||
log.Printf("📌📌📌📌📌📌📌📌📌📌📌📌📌📌📌📌 节点已注册: %s = %s", key, value)
|
||
}
|
||
|
||
time.Sleep(20 * time.Second)
|
||
}
|
||
}
|
||
|
||
/**
|
||
* configureLogger
|
||
* 功能：初始化日志目录和写入器配置。
|
||
*/
|
||
func (c *WebSocketController) configureLogger() {
|
||
logDir := utils.GetEnv("LOG_DIR", "./logs")
|
||
if err := os.MkdirAll(logDir, 0755); err != nil {
|
||
log.Fatalf("❌❌❌❌❌❌❌❌ 创建日志目录失败: %v", err)
|
||
}
|
||
|
||
logFreq := utils.GetEnv("LOG_ROTATE_FREQ", "daily")
|
||
if c.Logger == nil {
|
||
c.Logger = log.New(os.Stdout, "", log.LstdFlags|log.Lmicroseconds)
|
||
}
|
||
|
||
c.Logger.SetPrefix(fmt.Sprintf("[Node:%s] ", c.NodeID))
|
||
log.SetOutput(c.getLogWriter(logDir, logFreq))
|
||
}
|
||
|
||
/**
|
||
* getLogWriter
|
||
* 功能：工厂方法，获取 DailyFileWriter 实例。
|
||
*/
|
||
func (c *WebSocketController) getLogWriter(logDir, freq string) *utils.DailyFileWriter {
|
||
return utils.NewDailyFileWriter(logDir, freq, c.Logger, c.NodeID)
|
||
}
|
||
|
||
/**
|
||
* InitRedisClient
|
||
* 功能：建立 Redis 连接并验证连通性。
|
||
*/
|
||
func (c *WebSocketController) InitRedisClient() {
|
||
redisAddr := utils.GetEnv("REDIS_ADDR", "localhost:6379")
|
||
redisPassword := utils.GetEnv("REDIS_PASSWORD", "")
|
||
c.RedisCli = redis.NewClient(&redis.Options{
|
||
Addr: redisAddr,
|
||
Password: redisPassword,
|
||
DB: 0,
|
||
})
|
||
|
||
if err := c.checkRedisConnection(); err != nil {
|
||
log.Fatalf("❌❌❌❌❌❌❌❌ Redis连接失败: %v", err)
|
||
}
|
||
}
|
||
|
||
// 检查Redis连接
|
||
func (c *WebSocketController) checkRedisConnection() error {
|
||
_, err := c.RedisCli.Ping(c.RedisCtx).Result()
|
||
return err
|
||
}
|
||
|
||
/**
|
||
* PrintStartupInfo
|
||
* 功能：在控制台打印详细的服务启动信息，包括节点ID、端口、Redis状态及支持的消息类型。
|
||
*/
|
||
func (c *WebSocketController) PrintStartupInfo() {
|
||
hostname, _ := os.Hostname()
|
||
log.Printf("🚀🚀🚀🚀🚀🚀🚀🚀 WebSocket服务启动: NodeID=%s", c.NodeID)
|
||
log.Printf("🌐🌐🌐🌐🌐🌐🌐🌐 监听端口: %s", c.Port)
|
||
log.Printf("📡📡📡📡📡📡📡📡 Redis地址: %s", utils.GetEnv("REDIS_ADDR", "localhost:6379"))
|
||
log.Printf("💻💻💻💻💻💻💻💻 主机: %s", hostname)
|
||
log.Printf("🕒🕒🕒🕒🕒🕒🕒🕒🕒🕒🕒🕒🕒🕒🕒🕒🕒🕒🕒🕒🕒🕒🕒🕒🕒🕒🕒 启动时间: %s", time.Now().Format("2006-01-02 15:04:05"))
|
||
log.Printf("🔔🔔🔔🔔🔔🔔🔔🔔 支持消息类型: \n 0:文本\n 1:图片\n 2:音频\n 3:视频\n 4:处方\n 5:病例\n 6:视频通话\n 7:语音通话\n 8:文件消息")
|
||
log.Printf("🔑🔑🔑🔑🔑🔑🔑🔑 用户绑定键: \n %s\n %s", models.ClientUserKey, models.UserClientKey)
|
||
log.Printf("📬📬📬📬📬📬📬📬 新增接口: POST /send-to-user (直接通过用户ID发送消息)")
|
||
log.Printf("🔗🔗🔗🔗🔗🔗🔗🔗 集群节点注册: %s:%s", utils.GetOutboundIP(), c.Port)
|
||
log.Println("🔗🔗🔗🔗🔗🔗🔗🔗 等待客户端连接...")
|
||
}
|
||
|
||
/**
|
||
* HealthHandler
|
||
* 功能：API 接口，用于健康检查。
|
||
* 路径：GET /api/health
|
||
*/
|
||
func (c *WebSocketController) HealthHandler(ctx *gin.Context) {
|
||
ctx.JSON(http.StatusOK, gin.H{
|
||
"status": "ok",
|
||
"node": c.NodeID,
|
||
"port": c.Port,
|
||
"time": time.Now().Format(time.RFC3339),
|
||
})
|
||
}
|
||
|
||
/**
|
||
* IsUserOnline
|
||
* 功能：API 接口，检查指定用户当前是否在线（通过 Redis 查询）。
|
||
* 路径：GET /api/check-user-online
|
||
*/
|
||
func (c *WebSocketController) IsUserOnline(ctx *gin.Context) {
|
||
// 优先尝试查询参数，再尝试路径参数
|
||
userID := ctx.Query("user_id")
|
||
if userID == "" {
|
||
userID = ctx.Param("user_id")
|
||
}
|
||
|
||
if userID == "" {
|
||
ctx.JSON(http.StatusBadRequest, gin.H{
|
||
"code": -1,
|
||
"status": "error",
|
||
"message": "缺少user_id参数",
|
||
})
|
||
return
|
||
}
|
||
|
||
// 使用与绑定逻辑一致的键名格式
|
||
userClientsKey := fmt.Sprintf("%s:%s", models.UserClientKey, userID)
|
||
|
||
// 检查键是否存在并处理错误
|
||
exists, err := c.RedisCli.Exists(c.RedisCtx, userClientsKey).Result()
|
||
if err != nil {
|
||
log.Printf("⚠️ 检查用户在线状态失败: %v | UserID=%s", err, userID)
|
||
ctx.JSON(http.StatusInternalServerError, gin.H{
|
||
"code": -2,
|
||
"status": "error",
|
||
"message": "服务内部错误",
|
||
"result": false,
|
||
})
|
||
return
|
||
}
|
||
|
||
online := exists > 0
|
||
ctx.JSON(http.StatusOK, gin.H{
|
||
"code": 0,
|
||
"status": "success",
|
||
"message": "请求成功",
|
||
"result": online,
|
||
})
|
||
}
|
||
|
||
/**
|
||
* HandleWebSocket
|
||
* 功能：处理 WebSocket 握手升级，并开启消息监听循环。
|
||
* 路径：GET /ws
|
||
*/
|
||
func (c *WebSocketController) HandleWebSocket(ctx *gin.Context) {
|
||
start := time.Now()
|
||
clientIP := ctx.ClientIP()
|
||
log.Printf("👤👤👤👤👤👤👤👤 客户端连接中: IP=%s", clientIP)
|
||
|
||
conn, err := upgrader.Upgrade(ctx.Writer, ctx.Request, nil)
|
||
if err != nil {
|
||
log.Printf("⚠️ WebSocket升级失败: %v | ClientIP=%s", err, clientIP)
|
||
ctx.JSON(http.StatusBadRequest, gin.H{"error": "无法升级为WebSocket连接"})
|
||
return
|
||
}
|
||
defer conn.Close()
|
||
|
||
clientID := utils.GenerateClientID(c.NodeID)
|
||
log.Printf("✅ 客户端已连接: \n ClientID=%s \n IP=%s \n Duration=%s",
|
||
clientID, clientIP, time.Since(start))
|
||
|
||
c.addClient(clientID, conn)
|
||
defer c.removeClient(clientID)
|
||
|
||
if err := conn.WriteJSON(gin.H{"clientId": clientID}); err != nil {
|
||
log.Printf("⚠️ 发送客户端ID失败: %v | ClientID=%s", err, clientID)
|
||
return
|
||
}
|
||
|
||
conn.SetPongHandler(func(string) error {
|
||
conn.SetReadDeadline(time.Now().Add(60 * time.Second))
|
||
return nil
|
||
})
|
||
|
||
for {
|
||
messageType, message, err := conn.ReadMessage()
|
||
if err != nil {
|
||
if websocket.IsUnexpectedCloseError(err, websocket.CloseGoingAway) {
|
||
log.Printf("❌❌❌❌❌❌❌❌ 连接意外断开: %v | ClientID=%s", err, clientID)
|
||
} else {
|
||
log.Printf("⚠️ 连接正常关闭: ClientID=%s", clientID)
|
||
}
|
||
break
|
||
}
|
||
|
||
if messageType == websocket.TextMessage {
|
||
go c.handleClientMessage(clientID, message)
|
||
}
|
||
}
|
||
}
|
||
|
||
/**
|
||
* handleClientMessage
|
||
* 功能：处理从客户端接收到的消息，支持绑定、发消息、通话信令等操作。
|
||
* @param senderID string 发送者的客户端ID
|
||
* @param message []byte 原始消息体
|
||
*/
|
||
func (c *WebSocketController) handleClientMessage(senderID string, message []byte) {
|
||
log.Printf("📥📥📥📥📥📥📥📥 收到客户端消息: \n SenderID=%s \n Size=%d bytes", senderID, len(message))
|
||
|
||
var payload models.SendMessagePayload
|
||
if err := json.Unmarshal(message, &payload); err == nil && payload.RequestType != "" {
|
||
log.Printf("📦📦📦📦📦📦📦📦 解析JSON消息成功: Type=%s", payload.RequestType)
|
||
|
||
if payload.RequestType == "bind" && payload.SenderUserID != "" {
|
||
log.Printf("🔗🔗🔗🔗🔗🔗🔗🔗 处理绑定请求: \n ClientID=%s \n UserID=%s", senderID, payload.SenderUserID)
|
||
if err := c.bindClientToUser(senderID, payload.SenderUserID); err != nil {
|
||
log.Printf("❌❌❌❌❌❌❌❌ 绑定失败: %v", err)
|
||
} else {
|
||
log.Printf("✅ 绑定成功: \n ClientID=%s \n UserID=%s", senderID, payload.SenderUserID)
|
||
}
|
||
return
|
||
}
|
||
|
||
if payload.RequestType == "send_message" {
|
||
if payload.TargetClientID != "" {
|
||
log.Printf("📨📨📨📨📨📨📨📨 处理客户端发起的发送请求: \n TargetID=%s \n MsgType=%d",
|
||
payload.TargetClientID, payload.MessageType)
|
||
|
||
clientMsg := models.ClientReceivedMessage{
|
||
SenderID: senderID,
|
||
ReceiverID: payload.TargetClientID,
|
||
SenderUserID: payload.SenderUserID,
|
||
ReceiverUserID: payload.ReceiverUserID,
|
||
MessageType: payload.MessageType,
|
||
Content: payload.MessageContent,
|
||
SendTime: time.Now().Format(time.RFC3339),
|
||
}
|
||
|
||
c.sendMessageToClient(payload.TargetClientID, clientMsg)
|
||
return
|
||
}
|
||
|
||
if payload.ReceiverUserID != "" {
|
||
log.Printf("📨📨📨📨📨📨📨📨 处理客户端发起的用户发送请求: \n ReceiverUserID=%s \n MsgType=%d",
|
||
payload.ReceiverUserID, payload.MessageType)
|
||
|
||
clientMsg := models.ClientReceivedMessage{
|
||
SenderID: senderID,
|
||
ReceiverID: "",
|
||
SenderUserID: payload.SenderUserID,
|
||
ReceiverUserID: payload.ReceiverUserID,
|
||
MessageType: payload.MessageType,
|
||
Content: payload.MessageContent,
|
||
SendTime: time.Now().Format(time.RFC3339),
|
||
}
|
||
|
||
c.sendMessageToUser(payload.ReceiverUserID, clientMsg)
|
||
return
|
||
}
|
||
}
|
||
}
|
||
|
||
var clientMsg models.ClientReceivedMessage
|
||
if err := json.Unmarshal(message, &clientMsg); err == nil && clientMsg.ReceiverID != "" {
|
||
log.Printf("📨📨📨📨📨📨📨📨 处理客户端封装消息: \n TargetID=%s \n MsgType=%d",
|
||
clientMsg.ReceiverID, clientMsg.MessageType)
|
||
|
||
clientMsg.SenderID = senderID
|
||
c.sendMessageToClient(clientMsg.ReceiverID, clientMsg)
|
||
return
|
||
}
|
||
|
||
// 处理通话信令
|
||
var callSignal models.CallSignal
|
||
if err := json.Unmarshal(message, &callSignal); err == nil && callSignal.CallStatus != "" {
|
||
// 获取发送者用户ID
|
||
senderUserID, _ := c.getUserIDByClientID(senderID)
|
||
|
||
log.Printf("📞📞 处理通话信令: \n CallStatus=%s \n CallID=%s \n CallType=%d \n SenderUserID=%s \n CallerID=%s \n CalleeID=%s",
|
||
callSignal.CallStatus, callSignal.CallID, callSignal.CallType, senderUserID, callSignal.CallerID, callSignal.CalleeID)
|
||
|
||
// 处理不同类型的通话动作
|
||
switch callSignal.CallStatus {
|
||
case "invite":
|
||
c.handleCallInvite(senderID, senderUserID, callSignal)
|
||
case "accepted":
|
||
c.handleCallAccept(senderID, senderUserID, callSignal)
|
||
case "rejected":
|
||
c.handleCallReject(senderID, senderUserID, callSignal)
|
||
case "ended":
|
||
c.handleCallEnd(senderID, senderUserID, callSignal)
|
||
case "offer":
|
||
c.handleCallOffer(senderID, senderUserID, callSignal)
|
||
case "answer":
|
||
c.handleCallAnswer(senderID, senderUserID, callSignal)
|
||
case "candidate":
|
||
c.handleCallCandidate(senderID, senderUserID, callSignal)
|
||
case "hangup":
|
||
c.handleCallHangup(senderID, senderUserID, callSignal)
|
||
case "disconnected":
|
||
c.handleCallDisconnected(senderID, senderUserID, callSignal)
|
||
case "terminated":
|
||
c.handleCallTerminated(senderID, senderUserID, callSignal)
|
||
case "no-answer":
|
||
c.handleCallNoAnswer(senderID, senderUserID, callSignal)
|
||
case "busy":
|
||
c.handleCallBusy(senderID, senderUserID, callSignal)
|
||
case "failed":
|
||
c.handleCallFailed(senderID, senderUserID, callSignal)
|
||
default:
|
||
log.Printf("⚠️ 未知的通话动作: %s", callSignal.CallStatus)
|
||
}
|
||
return
|
||
}
|
||
|
||
log.Printf("⚠️ 无法识别的消息格式: \n Size=%d bytes \n Message=%s", len(message), string(message))
|
||
}
|
||
|
||
/**
|
||
* handleCallInvite
|
||
* 功能：处理WebRTC通话邀请信令。
|
||
*/
|
||
func (c *WebSocketController) handleCallInvite(senderID, senderUserID string, signal models.CallSignal) {
|
||
log.Printf("📞 处理通话邀请: \n CallID=%s \n CallType=%d \n From=%s \n To=%s",
|
||
signal.CallID, signal.CallType, senderUserID, signal.CalleeID)
|
||
|
||
// 创建通话消息
|
||
callMsg := models.ClientReceivedMessage{
|
||
SenderID: senderID,
|
||
ReceiverID: "",
|
||
SenderUserID: senderUserID,
|
||
ReceiverUserID: signal.CalleeID,
|
||
MessageType: signal.CallType,
|
||
Content: signal.Data,
|
||
SendTime: time.Now().Format(time.RFC3339),
|
||
CallID: signal.CallID,
|
||
CallStatus: "invite",
|
||
}
|
||
|
||
// 发送邀请给被叫方
|
||
if err := c.sendMessageToUser(signal.CalleeID, callMsg); err != nil {
|
||
log.Printf("❌ 发送通话邀请失败: %v | CalleeID=%s", err, signal.CalleeID)
|
||
}
|
||
}
|
||
|
||
/**
|
||
* handleCallAccept
|
||
* 功能：处理被叫方接受通话的信令。
|
||
*/
|
||
func (c *WebSocketController) handleCallAccept(senderID, senderUserID string, signal models.CallSignal) {
|
||
log.Printf("✅ 处理通话接受: \n CallID=%s \n From=%s \n To=%s", signal.CallID, senderUserID, signal.CallerID)
|
||
|
||
// 创建通话接受消息
|
||
acceptMsg := models.ClientReceivedMessage{
|
||
SenderID: senderID,
|
||
ReceiverID: "",
|
||
SenderUserID: senderUserID,
|
||
ReceiverUserID: signal.CallerID,
|
||
MessageType: signal.CallType,
|
||
Content: signal.Data,
|
||
SendTime: time.Now().Format(time.RFC3339),
|
||
CallID: signal.CallID,
|
||
CallStatus: "accepted",
|
||
}
|
||
|
||
// 发送接受消息给主叫方
|
||
if err := c.sendMessageToUser(signal.CallerID, acceptMsg); err != nil {
|
||
log.Printf("❌ 发送通话接受失败: %v | CallerID=%s", err, signal.CallerID)
|
||
}
|
||
}
|
||
|
||
/**
|
||
* handleCallReject
|
||
* 功能：处理被叫方拒绝通话的信令。
|
||
*/
|
||
func (c *WebSocketController) handleCallReject(senderID, senderUserID string, signal models.CallSignal) {
|
||
log.Printf("❌ 处理通话拒绝: \n CallID=%s \n From=%s \n To=%s", signal.CallID, senderUserID, signal.CallerID)
|
||
|
||
// 创建通话拒绝消息
|
||
rejectMsg := models.ClientReceivedMessage{
|
||
SenderID: senderID,
|
||
ReceiverID: "",
|
||
SenderUserID: senderUserID,
|
||
ReceiverUserID: signal.CallerID,
|
||
MessageType: signal.CallType,
|
||
Content: signal.Data,
|
||
SendTime: time.Now().Format(time.RFC3339),
|
||
CallID: signal.CallID,
|
||
CallStatus: "rejected",
|
||
}
|
||
|
||
// 发送拒绝消息给主叫方
|
||
if err := c.sendMessageToUser(signal.CallerID, rejectMsg); err != nil {
|
||
log.Printf("❌ 发送通话拒绝失败: %v | CallerID=%s", err, signal.CallerID)
|
||
}
|
||
}
|
||
|
||
/**
|
||
* handleCallEnd
|
||
* 功能：处理通话结束/挂断信令。
|
||
*/
|
||
func (c *WebSocketController) handleCallEnd(senderID, senderUserID string, signal models.CallSignal) {
|
||
log.Printf("📞 处理通话结束: \n CallID=%s \n From=%s \n To=%s", signal.CallID, senderUserID, signal.CalleeID)
|
||
|
||
// 创建通话结束消息
|
||
endMsg := models.ClientReceivedMessage{
|
||
SenderID: senderID,
|
||
ReceiverID: "",
|
||
SenderUserID: senderUserID,
|
||
ReceiverUserID: signal.CalleeID,
|
||
MessageType: signal.CallType,
|
||
Content: signal.Data,
|
||
SendTime: time.Now().Format(time.RFC3339),
|
||
CallID: signal.CallID,
|
||
CallStatus: "ended",
|
||
}
|
||
|
||
// 发送结束消息给对端
|
||
if err := c.sendMessageToUser(signal.CalleeID, endMsg); err != nil {
|
||
log.Printf("❌ 发送通话结束失败: %v | CalleeID=%s", err, signal.CalleeID)
|
||
}
|
||
}
|
||
|
||
/**
|
||
* handleCallOffer
|
||
* 功能：处理 WebRTC 的 Offer 信令（SDP交换）。
|
||
*/
|
||
func (c *WebSocketController) handleCallOffer(senderID, senderUserID string, signal models.CallSignal) {
|
||
log.Printf("📞 处理Offer信令: \n CallID=%s \n From=%s \n To=%s", signal.CallID, senderUserID, signal.CalleeID)
|
||
|
||
// 创建Offer消息
|
||
offerMsg := models.ClientReceivedMessage{
|
||
SenderID: senderID,
|
||
ReceiverID: "",
|
||
SenderUserID: senderUserID,
|
||
ReceiverUserID: signal.CalleeID,
|
||
MessageType: signal.CallType,
|
||
Content: signal.Data,
|
||
SendTime: time.Now().Format(time.RFC3339),
|
||
CallID: signal.CallID,
|
||
CallStatus: "offer",
|
||
}
|
||
|
||
// 发送Offer给对端
|
||
if err := c.sendMessageToUser(signal.CalleeID, offerMsg); err != nil {
|
||
log.Printf("❌ 发送Offer失败: %v | CalleeID=%s", err, signal.CalleeID)
|
||
}
|
||
}
|
||
|
||
/**
|
||
* handleCallAnswer
|
||
* 功能：处理 WebRTC 的 Answer 信令（SDP交换）。
|
||
*/
|
||
func (c *WebSocketController) handleCallAnswer(senderID, senderUserID string, signal models.CallSignal) {
|
||
log.Printf("📞 处理Answer信令: \n CallID=%s \n From=%s \n To=%s", signal.CallID, senderUserID, signal.CallerID)
|
||
|
||
// 创建Answer消息
|
||
answerMsg := models.ClientReceivedMessage{
|
||
SenderID: senderID,
|
||
ReceiverID: "",
|
||
SenderUserID: senderUserID,
|
||
ReceiverUserID: signal.CallerID,
|
||
MessageType: signal.CallType,
|
||
Content: signal.Data,
|
||
SendTime: time.Now().Format(time.RFC3339),
|
||
CallID: signal.CallID,
|
||
CallStatus: "answer",
|
||
}
|
||
|
||
// 发送Answer给对端
|
||
if err := c.sendMessageToUser(signal.CallerID, answerMsg); err != nil {
|
||
log.Printf("❌ 发送Answer失败: %v | CallerID=%s", err, signal.CallerID)
|
||
}
|
||
}
|
||
|
||
/**
|
||
* handleCallCandidate
|
||
* 功能：处理 ICE Candidate 信令（网络协商）。
|
||
*/
|
||
func (c *WebSocketController) handleCallCandidate(senderID, senderUserID string, signal models.CallSignal) {
|
||
log.Printf("📶 处理通话候选: \n CallID=%s From=%s \n To=%s", signal.CallID, senderUserID, signal.CalleeID)
|
||
|
||
// 创建候选消息
|
||
candidateMsg := models.ClientReceivedMessage{
|
||
SenderID: senderID,
|
||
ReceiverID: "",
|
||
SenderUserID: senderUserID,
|
||
ReceiverUserID: signal.CalleeID,
|
||
MessageType: signal.CallType,
|
||
Content: signal.Data,
|
||
SendTime: time.Now().Format(time.RFC3339),
|
||
CallID: signal.CallID,
|
||
CallStatus: "candidate",
|
||
}
|
||
|
||
// 发送候选消息给对端
|
||
if err := c.sendMessageToUser(signal.CalleeID, candidateMsg); err != nil {
|
||
log.Printf("❌ 发送候选失败: %v | CalleeID=%s", err, signal.CalleeID)
|
||
}
|
||
}
|
||
|
||
/**
|
||
* handleCallHangup
|
||
* 功能：处理主动挂断通话。
|
||
*/
|
||
func (c *WebSocketController) handleCallHangup(senderID, senderUserID string, signal models.CallSignal) {
|
||
log.Printf("📞 处理通话挂断: \n CallID=%s \n From=%s \n To=%s", signal.CallID, senderUserID, signal.CalleeID)
|
||
|
||
// 创建挂断消息
|
||
hangupMsg := models.ClientReceivedMessage{
|
||
SenderID: senderID,
|
||
ReceiverID: "",
|
||
SenderUserID: senderUserID,
|
||
ReceiverUserID: signal.CalleeID,
|
||
MessageType: signal.CallType,
|
||
Content: signal.Data,
|
||
SendTime: time.Now().Format(time.RFC3339),
|
||
CallID: signal.CallID,
|
||
CallStatus: "hangup",
|
||
}
|
||
|
||
// 发送挂断消息给对端
|
||
if err := c.sendMessageToUser(signal.CalleeID, hangupMsg); err != nil {
|
||
log.Printf("❌ 发送通话挂断失败: %v | CalleeID=%s", err, signal.CalleeID)
|
||
}
|
||
}
|
||
|
||
/**
|
||
* handleCallDisconnected
|
||
* 功能：处理连接意外断开。
|
||
*/
|
||
func (c *WebSocketController) handleCallDisconnected(senderID, senderUserID string, signal models.CallSignal) {
|
||
log.Printf("📞 处理通话掉线: \n CallID=%s \n From=%s \n To=%s", signal.CallID, senderUserID, signal.CalleeID)
|
||
|
||
// 创建掉线消息
|
||
disconnectedMsg := models.ClientReceivedMessage{
|
||
SenderID: senderID,
|
||
ReceiverID: "",
|
||
SenderUserID: senderUserID,
|
||
ReceiverUserID: signal.CalleeID,
|
||
MessageType: signal.CallType,
|
||
Content: signal.Data,
|
||
SendTime: time.Now().Format(time.RFC3339),
|
||
CallID: signal.CallID,
|
||
CallStatus: "disconnected",
|
||
}
|
||
|
||
// 发送掉线消息给对端
|
||
if err := c.sendMessageToUser(signal.CalleeID, disconnectedMsg); err != nil {
|
||
log.Printf("❌ 发送通话掉线失败: %v | CalleeID=%s", err, signal.CalleeID)
|
||
}
|
||
}
|
||
|
||
/**
|
||
* handleCallTerminated
|
||
* 功能：处理通话终止信号。
|
||
*/
|
||
func (c *WebSocketController) handleCallTerminated(senderID, senderUserID string, signal models.CallSignal) {
|
||
log.Printf("📞 处理通话终止: \n CallID=%s \n From=%s \n To=%s", signal.CallID, senderUserID, signal.CalleeID)
|
||
|
||
// 创建终止消息
|
||
terminatedMsg := models.ClientReceivedMessage{
|
||
SenderID: senderID,
|
||
ReceiverID: "",
|
||
SenderUserID: senderUserID,
|
||
ReceiverUserID: signal.CalleeID,
|
||
MessageType: signal.CallType,
|
||
Content: signal.Data,
|
||
SendTime: time.Now().Format(time.RFC3339),
|
||
CallID: signal.CallID,
|
||
CallStatus: "terminated",
|
||
}
|
||
|
||
// 发送终止消息给对端
|
||
if err := c.sendMessageToUser(signal.CalleeID, terminatedMsg); err != nil {
|
||
log.Printf("❌ 发送通话终止失败: %v | CalleeID=%s", err, signal.CalleeID)
|
||
}
|
||
}
|
||
|
||
/**
|
||
* handleCallNoAnswer
|
||
* 功能：处理无人接听状态。
|
||
*/
|
||
func (c *WebSocketController) handleCallNoAnswer(senderID, senderUserID string, signal models.CallSignal) {
|
||
log.Printf("📞 处理无人接听: \n CallID=%s \n From=%s \n To=%s", signal.CallID, senderUserID, signal.CallerID)
|
||
|
||
// 创建无人接听消息
|
||
noAnswerMsg := models.ClientReceivedMessage{
|
||
SenderID: senderID,
|
||
ReceiverID: "",
|
||
SenderUserID: senderUserID,
|
||
ReceiverUserID: signal.CallerID,
|
||
MessageType: signal.CallType,
|
||
Content: signal.Data,
|
||
SendTime: time.Now().Format(time.RFC3339),
|
||
CallID: signal.CallID,
|
||
CallStatus: "no-answer",
|
||
}
|
||
|
||
// 发送无人接听消息
|
||
if err := c.sendMessageToUser(signal.CallerID, noAnswerMsg); err != nil {
|
||
log.Printf("❌ 发送无人接听失败: %v | CallerID=%s", err, signal.CallerID)
|
||
}
|
||
}
|
||
|
||
/**
|
||
* handleCallBusy
|
||
* 功能：处理忙线状态。
|
||
*/
|
||
func (c *WebSocketController) handleCallBusy(senderID, senderUserID string, signal models.CallSignal) {
|
||
log.Printf("📞 处理忙线状态: \n CallID=%s \n From=%s \n To=%s", signal.CallID, senderUserID, signal.CallerID)
|
||
|
||
// 创建忙线消息
|
||
busyMsg := models.ClientReceivedMessage{
|
||
SenderID: senderID,
|
||
ReceiverID: "",
|
||
SenderUserID: senderUserID,
|
||
ReceiverUserID: signal.CallerID,
|
||
MessageType: signal.CallType,
|
||
Content: signal.Data,
|
||
SendTime: time.Now().Format(time.RFC3339),
|
||
CallID: signal.CallID,
|
||
CallStatus: "busy",
|
||
}
|
||
|
||
// 发送忙线消息
|
||
if err := c.sendMessageToUser(signal.CallerID, busyMsg); err != nil {
|
||
log.Printf("❌ 发送忙线状态失败: %v | CallerID=%s", err, signal.CallerID)
|
||
}
|
||
}
|
||
|
||
/**
|
||
* handleCallFailed
|
||
* 功能：处理通话建立失败状态。
|
||
*/
|
||
func (c *WebSocketController) handleCallFailed(senderID, senderUserID string, signal models.CallSignal) {
|
||
log.Printf("📞 处理通话失败: \n CallID=%s \n From=%s \n To=%s", signal.CallID, senderUserID, signal.CalleeID)
|
||
|
||
// 创建失败消息
|
||
failedMsg := models.ClientReceivedMessage{
|
||
SenderID: senderID,
|
||
ReceiverID: "",
|
||
SenderUserID: senderUserID,
|
||
ReceiverUserID: signal.CalleeID,
|
||
MessageType: signal.CallType,
|
||
Content: signal.Data,
|
||
SendTime: time.Now().Format(time.RFC3339),
|
||
CallID: signal.CallID,
|
||
CallStatus: "failed",
|
||
}
|
||
|
||
// 发送失败消息
|
||
if err := c.sendMessageToUser(signal.CalleeID, failedMsg); err != nil {
|
||
log.Printf("❌ 发送通话失败消息失败: %v | CalleeID=%s", err, signal.CalleeID)
|
||
}
|
||
}
|
||
|
||
/**
|
||
* SendMessageHandler
|
||
* 功能：API 接口，用于服务器端主动发送消息（HTTP -> WebSocket）。
|
||
* 路径：POST /api/send
|
||
*/
|
||
func (c *WebSocketController) SendMessageHandler(ctx *gin.Context) {
|
||
start := time.Now()
|
||
|
||
var payload models.SendMessagePayload
|
||
if err := ctx.ShouldBindJSON(&payload); err != nil {
|
||
log.Printf("⚠️ 无效的JSON请求格式: %v", err)
|
||
ctx.JSON(http.StatusBadRequest, gin.H{"error": "无效的JSON格式"})
|
||
return
|
||
}
|
||
|
||
if payload.RequestType == "" || (payload.TargetClientID == "" && payload.ReceiverUserID == "") ||
|
||
payload.SenderUserID == "" || payload.ReceiverUserID == "" {
|
||
log.Printf("⚠️ 缺少必要参数: \n request_type=%s \n target_client_id=%s \n receiver_user_id=%s \n sender_user_id=%s \n receiver_user_id=%s",
|
||
payload.RequestType, payload.TargetClientID, payload.ReceiverUserID, payload.SenderUserID, payload.ReceiverUserID)
|
||
ctx.JSON(http.StatusBadRequest, gin.H{"error": "缺少必要参数"})
|
||
return
|
||
}
|
||
|
||
if payload.RequestType == "send_message" {
|
||
log.Printf("📤📤📤📤📤📤📤📤 处理API发送请求: \n SenderUser=%s \n MsgType=%d",
|
||
payload.SenderUserID, payload.MessageType)
|
||
|
||
clientMsg := models.ClientReceivedMessage{
|
||
SenderID: "system",
|
||
RoomID: payload.RoomId,
|
||
ReceiverID: payload.TargetClientID,
|
||
SenderUserID: payload.SenderUserID,
|
||
ReceiverUserID: payload.ReceiverUserID,
|
||
MessageType: payload.MessageType,
|
||
Content: payload.MessageContent,
|
||
SendTime: time.Now().Format(time.RFC3339),
|
||
}
|
||
|
||
var result string
|
||
var sendErr error
|
||
|
||
if payload.TargetClientID != "" {
|
||
log.Printf("🎯🎯🎯🎯🎯🎯🎯🎯 目标类型: ClientID | Target=%s", payload.TargetClientID)
|
||
sendErr = c.sendMessageToClient(payload.TargetClientID, clientMsg)
|
||
result = fmt.Sprintf("消息已发送到ClientID: %s", payload.TargetClientID)
|
||
} else if payload.ReceiverUserID != "" {
|
||
log.Printf("🎯🎯🎯🎯🎯🎯🎯🎯 目标类型: UserID | ReceiverUser=%s", payload.ReceiverUserID)
|
||
sendErr = c.sendMessageToUser(payload.ReceiverUserID, clientMsg)
|
||
result = fmt.Sprintf("消息已发送到UserID: %s", payload.ReceiverUserID)
|
||
} else {
|
||
ctx.JSON(http.StatusBadRequest, gin.H{"error": "必须指定目标客户端ID或用户ID"})
|
||
return
|
||
}
|
||
|
||
if sendErr != nil {
|
||
ctx.JSON(http.StatusInternalServerError, gin.H{"error": sendErr.Error()})
|
||
return
|
||
}
|
||
|
||
ctx.JSON(http.StatusOK, gin.H{"status": "success", "message": result})
|
||
log.Printf("✅ API请求完成: Duration=%s", time.Since(start))
|
||
return
|
||
}
|
||
|
||
ctx.JSON(http.StatusBadRequest, gin.H{"error": "不支持的request_type"})
|
||
}
|
||
|
||
/**
|
||
* BindHandler
|
||
* 功能：API 接口，用于强制绑定 ClientID 和 UserID。
|
||
* 路径：POST /api/bind
|
||
*/
|
||
func (c *WebSocketController) BindHandler(ctx *gin.Context) {
|
||
start := time.Now()
|
||
|
||
var req models.BindRequest
|
||
if err := ctx.ShouldBindJSON(&req); err != nil {
|
||
log.Printf("⚠️ 无效的绑定请求格式: %v", err)
|
||
ctx.JSON(http.StatusBadRequest, gin.H{"error": "无效的JSON格式"})
|
||
return
|
||
}
|
||
|
||
if req.UserID == "" || req.ClientID == "" {
|
||
log.Printf("⚠️ 缺少必要参数: \n user_id=%s \n client_id=%s", req.UserID, req.ClientID)
|
||
ctx.JSON(http.StatusBadRequest, gin.H{"error": "user_id和client_id不能为空"})
|
||
return
|
||
}
|
||
|
||
log.Printf("🔗🔗🔗🔗🔗🔗🔗🔗 处理用户绑定请求: \n UserID=%s \n ClientID=%s", req.UserID, req.ClientID)
|
||
|
||
if err := c.bindClientToUser(req.ClientID, req.UserID); err != nil {
|
||
log.Printf("❌❌❌❌❌❌❌❌ 绑定失败: %v", err)
|
||
ctx.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
||
return
|
||
}
|
||
|
||
ctx.JSON(http.StatusOK, gin.H{
|
||
"status": "success",
|
||
"message": fmt.Sprintf("绑定成功: ClientID %s -> UserID %s", req.ClientID, req.UserID),
|
||
})
|
||
|
||
log.Printf("✅ 绑定请求完成: \n UserID=%s \n ClientID=%s \n Duration=%s",
|
||
req.UserID, req.ClientID, time.Since(start))
|
||
}
|
||
|
||
/**
|
||
* SendToUserHandler
|
||
* 功能：API 接口，向指定用户的所有在线设备发送消息，并持久化到数据库。
|
||
* 路径：POST /api/send-to-user
|
||
*/
|
||
// SendToUserHandler 通过用户ID发送消息处理器（支持通话信令）
|
||
func (c *WebSocketController) SendToUserHandler(ctx *gin.Context) {
|
||
start := time.Now()
|
||
|
||
var payload models.SendToUserPayload
|
||
if err := ctx.ShouldBindJSON(&payload); err != nil {
|
||
log.Printf("⚠️ 无效的JSON请求格式: %v", err)
|
||
ctx.JSON(http.StatusBadRequest, gin.H{"error": "无效的JSON格式"})
|
||
return
|
||
}
|
||
|
||
if payload.ReceiverUserID == "" || payload.SenderUserID == "" {
|
||
log.Printf("⚠️ 缺少必要参数: \n receiver_user_id=%s \n sender_user_id=%s",
|
||
payload.ReceiverUserID, payload.SenderUserID)
|
||
ctx.JSON(http.StatusBadRequest, gin.H{"error": "缺少接收方或发送方用户ID"})
|
||
return
|
||
}
|
||
|
||
log.Printf("📤📤📤📤📤📤📤📤 处理API用户发送请求: \n SenderUser=%s → ReceiverUser=%s \n MsgType=%d",
|
||
payload.SenderUserID, payload.ReceiverUserID, payload.MessageType)
|
||
|
||
// 创建消息记录
|
||
chatMsg := models.XkChatMessage{
|
||
RoomId: payload.RoomId,
|
||
SenderUserID: payload.SenderUserID,
|
||
ReceiverUserID: payload.ReceiverUserID,
|
||
Duration: payload.Duration,
|
||
MessageType: payload.MessageType,
|
||
MessageContent: payload.MessageContent,
|
||
CallID: payload.CallID,
|
||
CallStatus: payload.CallStatus,
|
||
}
|
||
|
||
var savedID = 0
|
||
|
||
// 保存到数据库
|
||
if err := c.DB.Create(&chatMsg).Error; err != nil {
|
||
log.Printf("❌ 保存消息到数据库失败: %v", err)
|
||
} else {
|
||
// 获取消息id
|
||
log.Printf("💾 消息保存到数据库成功: ID=%d", chatMsg.ID)
|
||
savedID = int(chatMsg.ID)
|
||
}
|
||
|
||
// 创建基础消息结构
|
||
clientMsg := models.ClientReceivedMessage{
|
||
SenderID: "system",
|
||
ReceiverID: "",
|
||
ID: savedID,
|
||
RoomID: payload.RoomId,
|
||
SenderUserID: payload.SenderUserID,
|
||
ReceiverUserID: payload.ReceiverUserID,
|
||
Duration: payload.Duration,
|
||
MessageType: payload.MessageType,
|
||
Content: payload.MessageContent,
|
||
SendTime: time.Now().Format(time.RFC3339),
|
||
}
|
||
|
||
// 如果是通话信令消息，添加通话相关字段
|
||
if payload.MessageType == 6 || payload.MessageType == 7 {
|
||
clientMsg.CallStatus = payload.CallStatus
|
||
clientMsg.CallID = payload.CallID
|
||
|
||
log.Printf("📞📞 处理通话信令消息: \n CallStatus=%s \n CallID=%s",
|
||
payload.CallStatus, payload.CallID)
|
||
}
|
||
|
||
// 发送消息给目标用户
|
||
if err := c.sendMessageToUser(payload.ReceiverUserID, clientMsg); err != nil {
|
||
log.Printf("❌ 发送消息失败: %v | ReceiverUserID=%s", err, payload.ReceiverUserID)
|
||
ctx.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
||
return
|
||
}
|
||
|
||
ctx.JSON(http.StatusOK, gin.H{
|
||
"status": "success",
|
||
"message": fmt.Sprintf("消息已发送到用户: %s", payload.ReceiverUserID),
|
||
})
|
||
|
||
log.Printf("✅ API用户发送请求完成: Duration=%s", time.Since(start))
|
||
}
|
||
|
||
/**
|
||
* GetMessagesHandler
|
||
* 功能：API 接口，分页查询历史聊天记录。
|
||
* 路径：GET /api/messages
|
||
*/
|
||
func (c *WebSocketController) GetMessagesHandler(ctx *gin.Context) {
|
||
// 解析查询参数
|
||
roomId := ctx.Query("room_id")
|
||
senderUserId := ctx.Query("sender_user_id")
|
||
receiverUserId := ctx.Query("receiver_user_id")
|
||
|
||
page, _ := strconv.Atoi(ctx.DefaultQuery("page", "1"))
|
||
pageSize, _ := strconv.Atoi(ctx.DefaultQuery("page_size", "15"))
|
||
|
||
if page < 1 {
|
||
page = 1
|
||
}
|
||
if pageSize < 1 || pageSize > 100 {
|
||
pageSize = 15
|
||
}
|
||
|
||
offset := (page - 1) * pageSize
|
||
|
||
// 构建查询
|
||
query := c.DB.Model(&models.XkChatMessage{})
|
||
if roomId != "" {
|
||
query = query.Where("room_id = ?", roomId)
|
||
}
|
||
if senderUserId != "" {
|
||
query = query.Where("sender_user_id = ?", senderUserId)
|
||
}
|
||
if receiverUserId != "" {
|
||
query = query.Where("receiver_user_id = ?", receiverUserId)
|
||
}
|
||
|
||
// 获取总数
|
||
var total int64
|
||
query.Count(&total)
|
||
|
||
// 获取分页数据
|
||
var messages []models.XkChatMessage
|
||
if err := query.Order("created_at DESC").Offset(offset).Limit(pageSize).Find(&messages).Error; err != nil {
|
||
ctx.JSON(http.StatusInternalServerError, gin.H{"error": "获取消息失败"})
|
||
return
|
||
}
|
||
|
||
ctx.JSON(http.StatusOK, gin.H{
|
||
"status": "success",
|
||
"data": messages,
|
||
"total": total,
|
||
"page": page,
|
||
"hasNext": total > int64(offset+pageSize),
|
||
})
|
||
}
|
||
|
||
/**
|
||
* SyncMessagesHandler
|
||
* 功能：API 接口，获取全部聊天记录（用于同步）。
|
||
* 路径：GET /api/messages/sync
|
||
*/
|
||
func (c *WebSocketController) SyncMessagesHandler(ctx *gin.Context) {
|
||
roomId := ctx.Query("room_id")
|
||
senderUserId := ctx.Query("sender_user_id")
|
||
receiverUserId := ctx.Query("receiver_user_id")
|
||
|
||
query := c.DB.Model(&models.XkChatMessage{})
|
||
if roomId != "" {
|
||
query = query.Where("room_id = ?", roomId)
|
||
}
|
||
if senderUserId != "" {
|
||
query = query.Where("sender_user_id = ?", senderUserId)
|
||
}
|
||
if receiverUserId != "" {
|
||
query = query.Where("receiver_user_id = ?", receiverUserId)
|
||
}
|
||
|
||
var messages []models.XkChatMessage
|
||
if err := query.Order("created_at ASC").Find(&messages).Error; err != nil {
|
||
ctx.JSON(http.StatusInternalServerError, gin.H{"error": "获取消息失败"})
|
||
return
|
||
}
|
||
|
||
ctx.JSON(http.StatusOK, gin.H{
|
||
"status": "success",
|
||
"data": messages,
|
||
})
|
||
}
|
||
|
||
/**
|
||
* addClient
|
||
* 功能：将新的 WebSocket 连接添加到本地连接池。
|
||
* 注意：使用 Mutex 锁保证线程安全。
|
||
*/
|
||
func (c *WebSocketController) addClient(clientID string, conn *websocket.Conn) {
|
||
c.ClientsMux.Lock()
|
||
defer c.ClientsMux.Unlock()
|
||
c.Clients[clientID] = conn
|
||
log.Printf("📊📊📊📊📊📊📊📊 当前连接数: %d", len(c.Clients))
|
||
}
|
||
|
||
/**
|
||
* removeClient
|
||
* 功能：从本地连接池移除客户端，并清理 Redis 中的绑定关系。
|
||
*/
|
||
func (c *WebSocketController) removeClient(clientID string) {
|
||
c.ClientsMux.Lock()
|
||
defer c.ClientsMux.Unlock()
|
||
|
||
if _, exists := c.Clients[clientID]; exists {
|
||
delete(c.Clients, clientID)
|
||
log.Printf("🔌🔌🔌🔌🔌🔌🔌🔌 客户端已断开: ClientID=%s | 剩余连接数: %d", clientID, len(c.Clients))
|
||
|
||
// 清理用户绑定关系
|
||
c.cleanupUserBinding(clientID)
|
||
}
|
||
}
|
||
|
||
/**
|
||
* cleanupUserBinding
|
||
* 功能：从 Redis 中清除用户与客户端ID的映射关系。
|
||
*/
|
||
func (c *WebSocketController) cleanupUserBinding(clientID string) {
|
||
// 获取用户ID
|
||
userID, err := c.getUserIDByClientID(clientID)
|
||
if err != nil {
|
||
return
|
||
}
|
||
|
||
// 从用户-客户端映射中移除
|
||
userClientsKey := fmt.Sprintf("%s:%s", models.UserClientKey, userID)
|
||
c.RedisCli.SRem(c.RedisCtx, userClientsKey, clientID)
|
||
|
||
// 从客户端-用户映射中移除
|
||
clientUserKey := fmt.Sprintf("%s:%s", models.ClientUserKey, clientID)
|
||
c.RedisCli.Del(c.RedisCtx, clientUserKey)
|
||
|
||
log.Printf("🧹🧹🧹🧹🧹🧹🧹🧹 清理绑定关系: ClientID=%s | UserID=%s", clientID, userID)
|
||
}
|
||
|
||
/**
|
||
* bindClientToUser
|
||
* 功能：在 Redis 中建立 UserID 和 ClientID 的双向映射。
|
||
* 映射1：client_user_mapping:clientID -> userID (string)
|
||
* 映射2：user_client_mapping:userID -> [clientID1, clientID2] (set)
|
||
*/
|
||
func (c *WebSocketController) bindClientToUser(clientID, userID string) error {
|
||
// 设置客户端->用户映射
|
||
clientUserKey := fmt.Sprintf("%s:%s", models.ClientUserKey, clientID)
|
||
if err := c.RedisCli.Set(c.RedisCtx, clientUserKey, userID, 0).Err(); err != nil {
|
||
return fmt.Errorf("设置客户端用户映射失败: %v", err)
|
||
}
|
||
|
||
// 添加用户->客户端集合映射
|
||
userClientsKey := fmt.Sprintf("%s:%s", models.UserClientKey, userID)
|
||
if err := c.RedisCli.SAdd(c.RedisCtx, userClientsKey, clientID).Err(); err != nil {
|
||
return fmt.Errorf("添加用户客户端集合失败: %v", err)
|
||
}
|
||
|
||
log.Printf("🔗🔗🔗🔗🔗🔗🔗🔗 绑定成功: \n ClientID=%s \n UserID=%s", clientID, userID)
|
||
return nil
|
||
}
|
||
|
||
// 根据客户端ID获取用户ID
|
||
func (c *WebSocketController) getUserIDByClientID(clientID string) (string, error) {
|
||
clientUserKey := fmt.Sprintf("%s:%s", models.ClientUserKey, clientID)
|
||
userID, err := c.RedisCli.Get(c.RedisCtx, clientUserKey).Result()
|
||
if err != nil {
|
||
return "", fmt.Errorf("获取用户ID失败: %v", err)
|
||
}
|
||
return userID, nil
|
||
}
|
||
|
||
// 根据用户ID获取客户端ID列表
|
||
func (c *WebSocketController) getClientIDsByUserID(userID string) ([]string, error) {
|
||
userClientsKey := fmt.Sprintf("%s:%s", models.UserClientKey, userID)
|
||
clientIDs, err := c.RedisCli.SMembers(c.RedisCtx, userClientsKey).Result()
|
||
if err != nil {
|
||
return nil, fmt.Errorf("获取客户端ID列表失败: %v", err)
|
||
}
|
||
return clientIDs, nil
|
||
}
|
||
|
||
/**
|
||
* sendMessageToClient
|
||
* 功能：向指定 ClientID 发送消息。
|
||
* 逻辑：如果 Client 在本节点，直接通过 WebSocket 发送；如果不在，尝试通过 Redis 广播转发。
|
||
*/
|
||
func (c *WebSocketController) sendMessageToClient(clientID string, message models.ClientReceivedMessage) error {
|
||
c.ClientsMux.RLock()
|
||
conn, exists := c.Clients[clientID]
|
||
c.ClientsMux.RUnlock()
|
||
|
||
if !exists {
|
||
// 尝试通过Redis转发到其他节点
|
||
return c.forwardMessageToOtherNodes(clientID, message)
|
||
}
|
||
|
||
c.WriteMutex.Lock()
|
||
defer c.WriteMutex.Unlock()
|
||
|
||
if err := conn.WriteJSON(message); err != nil {
|
||
log.Printf("❌ 发送消息失败: %v | ClientID=%s", err, clientID)
|
||
c.removeClient(clientID)
|
||
return err
|
||
}
|
||
|
||
log.Printf("✅ 消息已发送: \n ClientID=%s \n MsgType=%d \n Content=%s",
|
||
clientID, message.MessageType, message.Content)
|
||
return nil
|
||
}
|
||
|
||
/**
|
||
* sendMessageToUser
|
||
* 功能：向指定用户的所有设备发送消息。
|
||
*/
|
||
func (c *WebSocketController) sendMessageToUser(userID string, message models.ClientReceivedMessage) error {
|
||
clientIDs, err := c.getClientIDsByUserID(userID)
|
||
if err != nil {
|
||
log.Printf("⚠️ 获取用户客户端列表失败: %v | UserID=%s", err, userID)
|
||
return err
|
||
}
|
||
|
||
if len(clientIDs) == 0 {
|
||
log.Printf("⚠️ 用户无在线客户端: UserID=%s", userID)
|
||
//return fmt.Errorf("用户 %s 无在线客户端", userID)
|
||
} else {
|
||
log.Printf("📤📤📤📤📤📤📤📤 发送消息给用户: \n UserID=%s \n ClientCount=%d \n MsgType=%d",
|
||
userID, len(clientIDs), message.MessageType)
|
||
|
||
var lastErr error
|
||
successCount := 0
|
||
|
||
for _, clientID := range clientIDs {
|
||
if err := c.sendMessageToClient(clientID, message); err != nil {
|
||
log.Printf("⚠️ 发送到客户端失败: %v | ClientID=%s", err, clientID)
|
||
lastErr = err
|
||
} else {
|
||
successCount++
|
||
}
|
||
}
|
||
|
||
if successCount == 0 {
|
||
return fmt.Errorf("所有客户端发送失败，最后错误: %v", lastErr)
|
||
}
|
||
|
||
log.Printf("✅ 用户消息发送完成: \n UserID=%s \n 成功=%d/%d", userID, successCount, len(clientIDs))
|
||
}
|
||
return nil
|
||
}
|
||
|
||
/**
|
||
* forwardMessageToOtherNodes
|
||
* 功能：当目标客户端不在本机时，通过 Redis Pub/Sub 将消息广播到集群其他节点。
|
||
*/
|
||
func (c *WebSocketController) forwardMessageToOtherNodes(clientID string, message models.ClientReceivedMessage) error {
|
||
redisMsg := models.RedisMessage{
|
||
SenderNodeID: c.NodeID,
|
||
ClientID: clientID,
|
||
Message: "",
|
||
}
|
||
|
||
msgBytes, err := json.Marshal(message)
|
||
if err != nil {
|
||
return fmt.Errorf("序列化消息失败: %v", err)
|
||
}
|
||
redisMsg.Message = string(msgBytes)
|
||
|
||
redisMsgBytes, err := json.Marshal(redisMsg)
|
||
if err != nil {
|
||
return fmt.Errorf("序列化Redis消息失败: %v", err)
|
||
}
|
||
|
||
channel := fmt.Sprintf("websocket:forward:%s", clientID)
|
||
if err := c.RedisCli.Publish(c.RedisCtx, channel, redisMsgBytes).Err(); err != nil {
|
||
return fmt.Errorf("发布Redis消息失败: %v", err)
|
||
}
|
||
|
||
log.Printf("📡📡📡📡📡📡📡📡 消息已转发到其他节点: \n ClientID=%s \n Channel=%s", clientID, channel)
|
||
return nil
|
||
}
|
||
|
||
/**
|
||
* SubscribeToRedis
|
||
* 功能：订阅 Redis 广播频道，监听来自其他节点的消息转发请求。
|
||
*/
|
||
func (c *WebSocketController) SubscribeToRedis() {
|
||
go func() {
|
||
pattern := "websocket:forward:*"
|
||
pubsub := c.RedisCli.PSubscribe(c.RedisCtx, pattern)
|
||
defer pubsub.Close()
|
||
|
||
log.Printf("📡📡📡📡📡📡📡📡 Redis消息监听已启动: Pattern=%s", pattern)
|
||
|
||
for msg := range pubsub.Channel() {
|
||
var redisMsg models.RedisMessage
|
||
if err := json.Unmarshal([]byte(msg.Payload), &redisMsg); err != nil {
|
||
log.Printf("⚠️ 解析Redis消息失败: %v", err)
|
||
continue
|
||
}
|
||
|
||
// 忽略自己发送的消息
|
||
if redisMsg.SenderNodeID == c.NodeID {
|
||
continue
|
||
}
|
||
|
||
// 检查目标客户端是否在本节点
|
||
c.ClientsMux.RLock()
|
||
conn, exists := c.Clients[redisMsg.ClientID]
|
||
c.ClientsMux.RUnlock()
|
||
|
||
if !exists {
|
||
continue
|
||
}
|
||
|
||
var clientMsg models.ClientReceivedMessage
|
||
if err := json.Unmarshal([]byte(redisMsg.Message), &clientMsg); err != nil {
|
||
log.Printf("⚠️ 解析客户端消息失败: %v", err)
|
||
continue
|
||
}
|
||
|
||
c.WriteMutex.Lock()
|
||
if err := conn.WriteJSON(clientMsg); err != nil {
|
||
log.Printf("❌ 转发消息失败: %v | ClientID=%s", err, redisMsg.ClientID)
|
||
c.removeClient(redisMsg.ClientID)
|
||
} else {
|
||
log.Printf("✅ 转发消息成功: \n ClientID=%s \n FromNode=%s",
|
||
redisMsg.ClientID, redisMsg.SenderNodeID)
|
||
}
|
||
c.WriteMutex.Unlock()
|
||
}
|
||
}()
|
||
}
|