27 lines
590 B
Go
27 lines
590 B
Go
package handler
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"kefu-sys/server/internal/middleware"
|
|
"kefu-sys/server/internal/ws"
|
|
)
|
|
|
|
type WsHandler struct{}
|
|
|
|
func NewWsHandler() *WsHandler { return &WsHandler{} }
|
|
|
|
func (h *WsHandler) Connect(c *gin.Context) {
|
|
userID := middleware.GetUserID(c)
|
|
tenantID := middleware.GetTenantID(c)
|
|
|
|
client, err := ws.UpgradeAgent(c.Writer, c.Request, userID, tenantID, middleware.GetRole(c))
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"code": 500, "message": "升级连接失败"})
|
|
return
|
|
}
|
|
|
|
ws.HandleWebSocket(client)
|
|
}
|