45 lines
1.5 KiB
Go
45 lines
1.5 KiB
Go
package websocket
|
||
|
||
import (
|
||
"catface/app/global/consts"
|
||
"catface/app/global/variable"
|
||
controllerWs "catface/app/http/controller/websocket"
|
||
"catface/app/http/validator/core/data_transfer"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
"go.uber.org/zap"
|
||
)
|
||
|
||
type Connect struct {
|
||
Token string `form:"token" json:"token" binding:"required,min=10"`
|
||
}
|
||
|
||
// 验证器语法，参见 Register.go文件，有详细说明
|
||
// 注意：websocket 连接建立之前如果有错误，只能在服务端同构日志输出方式记录（因为使用response.Fail等函数，客户端是收不到任何信息的）
|
||
|
||
func (c Connect) CheckParams(context *gin.Context) {
|
||
|
||
// 1. 首先检查是否开启websocket服务配置（在配置项中开启）
|
||
if variable.ConfigYml.GetInt("Websocket.Start") != 1 {
|
||
variable.ZapLog.Error(consts.WsServerNotStartMsg)
|
||
return
|
||
}
|
||
//2.基本的验证规则没有通过
|
||
if err := context.ShouldBind(&c); err != nil {
|
||
variable.ZapLog.Error("客户端上线参数不合格", zap.Error(err))
|
||
return
|
||
}
|
||
extraAddBindDataContext := data_transfer.DataAddContext(c, consts.ValidatorPrefix, context)
|
||
if extraAddBindDataContext == nil {
|
||
variable.ZapLog.Error("websocket-Connect 表单验证器json化失败")
|
||
context.Abort()
|
||
return
|
||
} else {
|
||
if serviceWs, ok := (&controllerWs.Ws{}).OnOpen(extraAddBindDataContext); ok == false {
|
||
variable.ZapLog.Error(consts.WsOpenFailMsg)
|
||
} else {
|
||
(&controllerWs.Ws{}).OnMessage(serviceWs, extraAddBindDataContext) // 注意这里传递的service_ws必须是调用open返回的，必须保证的ws对象的一致性
|
||
}
|
||
}
|
||
}
|