分离websocket

pkg/websocket/socket_new.go

@@ -0,0 +1,177 @@
+package websocket
+
+import (
+	"github.com/gogf/gf/v2/container/gmap"
+	"github.com/gogf/gf/v2/frame/g"
+	"github.com/gogf/gf/v2/os/gctx"
+	"github.com/gogf/gf/v2/os/glog"
+	"github.com/gogf/gf/v2/util/gconv"
+	"github.com/google/uuid"
+	"github.com/gorilla/websocket"
+	"sync"
+	"github.com/gogf/gf/v2/net/ghttp"
+)
+
+type SocketV1 struct{}
+
+var (
+	ctx = gctx.New()
+	//Conn map[uuid.UUID]*WebsocketData
+	lock sync.Mutex
+
+	m = gmap.New(true)
+)
+
+type WebsocketData struct {
+	Ws   *websocket.Conn
+	Uuid uuid.UUID
+	Uid  int64
+	Data g.Var
+}
+
+func NewV1() *SocketV1 {
+	return &SocketV1{}
+}
+
+type SocketInterface interface {
+	OnConnect(*websocket.Conn)
+	OnMessage(*WebsocketData, []byte, int)
+	Send(uuid.UUID, []byte) (err error)
+	SendAll(data []byte)
+	OnClose(conn *websocket.Conn)
+}
+
+func (s *SocketV1) Load(serv *ghttp.Server, prefix string) {
+	//websocket服务启动
+	serv.Group(prefix, func(group *ghttp.RouterGroup) {
+
+		var websocketCfg = websocket.Upgrader{
+			ReadBufferSize:  1024,
+			WriteBufferSize: 1024,
+		}
+		group.Bind(
+			func(r *ghttp.Request) {
+				ws, err := websocketCfg.Upgrade(r.Response.Writer, r.Request, nil)
+				if err != nil {
+					glog.Error(ctx, err)
+					r.Exit()
+				}
+
+				//ws联机触发器
+				NewV1().OnConnect(ws)
+			},
+		)
+
+	})
+}
+
+// OnConnect
+//
+//	@Description:
+//	@receiver s
+//	@param conn
+func (s *SocketV1) OnConnect(conn *websocket.Conn) {
+	//lock.Lock()
+	//defer lock.Unlock()
+
+	defer conn.Close()
+	id, _ := uuid.NewUUID()
+	ip := conn.RemoteAddr().String()
+
+	data := &WebsocketData{
+		Uuid: id,
+		Ws:   conn,
+		Data: g.Var{},
+	}
+	m.Set(id, data)
+
+	//defer delete(Conn, id)
+
+	//to := fmt.Sprintf("创建连接：%v,ip=%v", id, ip)
+	//s.Send(id, []byte(to))
+
+	for {
+		//进入当前连接线程拥堵
+		msgType, msg, err := conn.ReadMessage()
+		if err != nil {
+			//客户端断开返回错误，断开当前连接
+			break
+		}
+		s.OnMessage(m.Get(id).(*WebsocketData), msg, msgType)
+	}
+	//关闭连接触发
+	s.OnClose(id, conn)
+	g.Log().Debugf(ctx, "断开连接:uuid=%v,ip=%v", id, ip)
+}
+
+// OnMessage
+//
+//	@Description:
+//	@receiver s
+//	@param msg
+//	@param msgType
+func (s *SocketV1) OnMessage(conn *WebsocketData, req []byte, msgType int) {
+	//g.Log().Debugf(ctx, "收到消息：%v,type=%v,conn=%v", string(req), msgType, conn)
+	//s.Send(conn.Uuid, msg)
+	//s.SendAll(msg)
+	msgStr := string(req)
+	msg := msgStr[8:]
+	cmd := gconv.Int(msgStr[:8])
+	//GetRouter(cmd, conn.Uid, msg)
+	handler, exist := handlers[cmd]
+	if exist {
+		//匹配上路由器
+		handler(conn.Uid, []byte(msg))
+	} else {
+		//fmt.Println("未注册的路由器ID:", cmd)
+		s.Send(conn.Uuid, []byte("未注册的协议号:"+msgStr[:8]))
+		s.OnClose(conn.Uuid, conn.Ws)
+		return
+	}
+
+}
+
+// Send
+//
+//	@Description:
+//	@receiver s
+//	@param uid
+//	@param data
+//	@return err
+func (s *SocketV1) Send(id uuid.UUID, data []byte) (err error) {
+	if !m.Contains(id) {
+		return
+	}
+
+	conn := m.Get(id).(*WebsocketData)
+	conn.Ws.WriteMessage(1, data)
+
+	return
+}
+
+// 批量发送
+func (s *SocketV1) SendAll(data []byte) {
+	m.Iterator(func(k interface{}, v interface{}) bool {
+		//fmt.Printf("%v:%v ", k, v)
+		conn := v.(*WebsocketData)
+		conn.Ws.WriteMessage(1, data)
+		return true
+	})
+}
+
+// OnClose
+//
+//	@Description:
+//	@receiver s
+//	@param conn
+func (s *SocketV1) OnClose(id uuid.UUID, conn *websocket.Conn) {
+	// 在此处编写断开连接后的处理逻辑
+	g.Log().Debugf(ctx, "WebSocket connection from %s has been closed.", conn.RemoteAddr())
+
+	// 可能的后续操作：
+	// 1. 更新连接状态或从连接池移除
+	// 2. 发送通知或清理关联资源
+	// 3. 执行特定于业务的断开处理
+	m.Remove(id)
+	conn.Close()
+}