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) {
 
 	//设置读取消息大小上线