This commit is contained in:
richboo111
2023-04-27 11:12:24 +08:00
parent 78633ef914
commit d775e408eb
2 changed files with 30 additions and 31 deletions

View File

@@ -2,9 +2,6 @@ package controllers
import ( import (
"encoding/json" "encoding/json"
"net/http"
"git.rosy.net.cn/jx-callback/globals"
"git.rosy.net.cn/jx-callback/business/partner/purchase/im" "git.rosy.net.cn/jx-callback/business/partner/purchase/im"
"github.com/astaxie/beego/server/web" "github.com/astaxie/beego/server/web"
@@ -22,34 +19,37 @@ var maxMessageSize = int64(8192) // 最大的消息大小
// @Failure 200 {object} controllers.CallResult // @Failure 200 {object} controllers.CallResult
// @router /StartWebSocket [get] // @router /StartWebSocket [get]
func (c *IMController) StartWebSocket() { func (c *IMController) StartWebSocket() {
upgrader.CheckOrigin = func(r *http.Request) bool {
return true
}
conn, err := upgrader.Upgrade(c.Ctx.ResponseWriter, c.Ctx.Request, nil)
if err != nil {
globals.SugarLogger.Errorf("upgrade error: %v", err)
return
}
defer conn.Close()
clientID := c.GetString("clientID") im.Run(c.Ctx.ResponseWriter, c.Ctx.Request)
globals.SugarLogger.Debugf("clientID=%s", clientID) //
//upgrader.CheckOrigin = func(r *http.Request) bool {
//设置读取消息大小上线 // return true
conn.SetReadLimit(maxMessageSize) //}
//conn, err := upgrader.Upgrade(c.Ctx.ResponseWriter, c.Ctx.Request, nil)
clientSocket := im.NewClient(clientID, conn) //if err != nil {
// globals.SugarLogger.Errorf("upgrade error: %v", err)
//读取客户端消息 // return
clientSocket.Read() //}
//defer conn.Close()
if err = im.ConnRender(conn, im.RetData{Data: clientID}); err != nil { //
_ = conn.Close() //clientID := c.GetString("clientID")
return //globals.SugarLogger.Debugf("clientID=%s", clientID)
} //
////设置读取消息大小上线
// 用户连接事件 //conn.SetReadLimit(maxMessageSize)
im.Manager.Connect <- clientSocket //
//clientSocket := im.NewClient(clientID, conn)
//
////读取客户端消息
//clientSocket.Read()
//
//if err = im.ConnRender(conn, im.RetData{Data: clientID}); err != nil {
// _ = conn.Close()
// return
//}
//
//// 用户连接事件
//im.Manager.Connect <- clientSocket
} }

View File

@@ -1 +0,0 @@
package controllers