ws

business/partner/purchase/im/im_server.go

@@ -3,6 +3,7 @@ package im
 import (
 	"errors"
 	"fmt"
+	"net/http"
 	"time"
 
 	"git.rosy.net.cn/baseapi/utils"
@@ -32,6 +33,39 @@ func Init() {
 	fmt.Printf("服务器启动成功，端口号：%s\n", CommonSetting.HttpPort)
 }
 
+func Run(w http.ResponseWriter, r *http.Request) {
+	conn, err := (&websocket.Upgrader{
+		ReadBufferSize:  1024,
+		WriteBufferSize: 1024,
+		// 允许所有CORS跨域请求
+		CheckOrigin: func(r *http.Request) bool {
+			return true
+		},
+	}).Upgrade(w, r, nil)
+	if err != nil {
+		globals.SugarLogger.Debugf("upgrade error: %v", err)
+		http.NotFound(w, r)
+		return
+	}
+
+	//设置读取消息大小上线
+	conn.SetReadLimit(maxMessageSize)
+
+	clientID := r.FormValue("clientId")
+	clientSocket := NewClient(clientID, conn)
+
+	//读取客户端消息
+	clientSocket.Read()
+
+	if err = ConnRender(conn, renderData{ClientId: clientID}); err != nil {
+		_ = conn.Close()
+		return
+	}
+
+	// 用户连接事件
+	Manager.Connect <- clientSocket
+}
+
 func StartWebSocket(conn *websocket.Conn, clientID string, err error) {
 
 	//设置读取消息大小上线

controllers/im.go

@@ -2,7 +2,6 @@ package controllers
 
 import (
 	"encoding/json"
-	"git.rosy.net.cn/jx-callback/globals"
 
 	"git.rosy.net.cn/jx-callback/business/partner/purchase/im"
 	"github.com/astaxie/beego/server/web"
@@ -17,26 +16,18 @@ type IMController struct {
 // @Success 200 {object} controllers.CallResult
 // @Failure 200 {object} controllers.CallResult
 // @router /StartWebSocket [get]
-func (c *IMController) StartWebSocket() {
+//func (c *IMController) StartWebSocket() {
-	ws, err := upgrader.Upgrade(c.Ctx.ResponseWriter, c.Ctx.Request, nil)
+//	ws, err := upgrader.Upgrade(c.Ctx.ResponseWriter, c.Ctx.Request, nil)
-	//ws, err := upgrader.Upgrade(c.Ctx.ResponseWriter, &http.Request{
+//	if err != nil {
-	//	Method: "GET",
+//		globals.SugarLogger.Errorf("upgrade error: %v", err)
-	//	Header: http.Header{
+//		return
-	//		"Upgrade":               []string{"websocket"},
+//	}
-	//		"Connection":            []string{"upgrade"},
+//	defer ws.Close()
-	//		"Sec-Websocket-Key":     []string{"dGhlIHNhbXBsZSBub25jZQ=="},
+//
-	//		"Sec-Websocket-Version": []string{"13"},
+//	clientID := c.GetString("clientID")
-	//	}}, nil)
+//	globals.SugarLogger.Debugf("clientID=%s", clientID)
-	if err != nil {
+//	im.StartWebSocket(ws, clientID, err)
-		globals.SugarLogger.Errorf("upgrade error: %v", err)
+//}
-		return
-	}
-	defer ws.Close()
-
-	clientID := c.GetString("clientID")
-	globals.SugarLogger.Debugf("clientID=%s", clientID)
-	im.StartWebSocket(ws, clientID, err)
-}
 
 // @Title IM获取门店用户聊天列表
 // @Description IM获取门店用户聊天列表

main.go

@@ -97,7 +97,7 @@ func Init() {
 	auto_delivery.Init() // 初始化骑手列表
 
 	go im.Init() //初始化ws连接
-	//go http.HandleFunc("/v2/im/StartWebSocket", im.StartWebSocket)
+	go http.HandleFunc("/v2/im/StartWebSocket", im.Run)
 
 	//test
 	//mux := http.NewServeMux()

routers/commentsRouter_controllers.go

@@ -4356,14 +4356,14 @@ func init() {
 			MethodParams:     param.Make(),
 			Filters:          nil,
 			Params:           nil})
-	web.GlobalControllerRouter["git.rosy.net.cn/jx-callback/controllers:IMController"] = append(web.GlobalControllerRouter["git.rosy.net.cn/jx-callback/controllers:IMController"],
+	//web.GlobalControllerRouter["git.rosy.net.cn/jx-callback/controllers:IMController"] = append(web.GlobalControllerRouter["git.rosy.net.cn/jx-callback/controllers:IMController"],
-		web.ControllerComments{
+	//	web.ControllerComments{
-			Method:           "StartWebSocket",
+	//		Method:           "StartWebSocket",
-			Router:           `/StartWebSocket`,
+	//		Router:           `/StartWebSocket`,
-			AllowHTTPMethods: []string{"get"},
+	//		AllowHTTPMethods: []string{"get"},
-			MethodParams:     param.Make(),
+	//		MethodParams:     param.Make(),
-			Filters:          nil,
+	//		Filters:          nil,
-			Params:           nil})
+	//		Params:           nil})
 
 	//web.GlobalControllerRouter["git.rosy.net.cn/jx-callback/controllers:FnController"] = append(web.GlobalControllerRouter["git.rosy.net.cn/jx-callback/controllers:FnController"],
 	//	web.ControllerComments{