mirror of
https://gitee.com/dromara/mayfly-go
synced 2025-11-04 00:10:25 +08:00
42 lines
824 B
Go
42 lines
824 B
Go
package api
|
|
|
|
import (
|
|
"mayfly-go/pkg/biz"
|
|
"mayfly-go/pkg/global"
|
|
"mayfly-go/pkg/req"
|
|
"mayfly-go/pkg/ws"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/gorilla/websocket"
|
|
)
|
|
|
|
type System struct {
|
|
}
|
|
|
|
// 连接websocket
|
|
func (s *System) ConnectWs(g *gin.Context) {
|
|
wsConn, err := ws.Upgrader.Upgrade(g.Writer, g.Request, nil)
|
|
defer func() {
|
|
if err := recover(); err != nil {
|
|
global.Log.Error(err.(error).Error())
|
|
if wsConn != nil {
|
|
wsConn.WriteMessage(websocket.TextMessage, []byte(err.(error).Error()))
|
|
wsConn.Close()
|
|
}
|
|
}
|
|
}()
|
|
|
|
if err != nil {
|
|
panic(biz.NewBizErr("升级websocket失败"))
|
|
}
|
|
// 权限校验
|
|
rc := req.NewCtxWithGin(g)
|
|
if err = req.PermissionHandler(rc); err != nil {
|
|
panic(biz.NewBizErr("没有权限"))
|
|
}
|
|
|
|
// 登录账号信息
|
|
la := rc.LoginAccount
|
|
ws.Put(la.Id, wsConn)
|
|
}
|