接口全览
//聊一聊接口
func Handler(c *gin.Context) {
fmt.Println("进入函数")
//获取发送者uid和被发送者uid
token := module.WsGetTokenFromHeader(c.Request.Header)
fmt.Println(token)
stuNum := dao.Redis.Get(token).Val()
toStuNum := c.Query("toStuNum")
if stuNum == toStuNum {
c.JSON(http.StatusOK, gin.H{
"status": 203,
"msg": "自己无法与自己建立沟通",
})
return
}
//升级websocket协议
conn, err := (&websocket.Upgrader{
CheckOrigin: func(r *http.Request) bool {
return true
},
//将获取的参数放进这个数组
Subprotocols: []string{token},
}).Upgrade(c.Writer, c.Request, nil) //升级ws协议
if err != nil {
http.NotFound(c.Writer, c.Request)
return
}
//创建一个用户实例
client := &models.Client{
ID: createID(stuNum, toStuNum), //1->2
SendID: createID(toStuNum, stuNum), //2->1
FromUsername: stuNum,
ToUsername: toStuNum,
Socket: conn,
Send: make(chan []byte),
}
//用户注册到用户管理上
mux.Lock()
models.Manager.Register <- client
mux.Unlock()
go client.Read()
go client.Write()
go models.BatchRead(toStuNum, stuNum)
}
接口切片
如果前端使用Sec-WebSocket-Protocol字段传递token的话会引发websocket刚连接上就断开的问题
届时需要在Upgrader添加
//将获取的参数放进这个数组
Subprotocols: []string{token},
以建立连接
websocket升级
conn, err := (&websocket.Upgrader{
CheckOrigin: func(r *http.Request) bool {
return true
},
//将获取的参数放进这个数组
Subprotocols: []string{token},
}).Upgrade(c.Writer, c.Request, nil) //升级ws协议
if err != nil {
http.NotFound(c.Writer, c.Request)
return
}
