重构

common/socket/ServerEvent.go

@@ -6,9 +6,8 @@ import (
 	"sync/atomic"
 	"time"
 
-	"blazing/common/data/entity"
 	"blazing/common/data/share"
-	"blazing/cool"
+	"blazing/common/data/socket"
 
 	"github.com/gogf/gf/v2/os/glog"
 	"github.com/panjf2000/gnet/v2"
@@ -42,17 +41,17 @@ func (s *Server) Stop() error {
 func (s *Server) OnClose(c gnet.Conn, _ error) (action gnet.Action) {
 	atomic.AddInt64(&s.connected, -1)
 	//logging.Infof("conn[%v] disconnected", c.RemoteAddr().String())
-	v, ok := c.Context().(*entity.ClientData)
+	v, ok := c.Context().(*socket.ClientData)
 	if !ok {
 		return
 
 	}
 	t := v.GetPlayer()
-	if t != nil {
-		glog.Debug(context.Background(), t.UserID, "断开连接")
+	if v != nil {
+		glog.Debug(context.Background(), t, "断开连接")
 		t.IsLogin = false
-		cool.Mainplayer.Delete(t.UserID)
-		share.ShareManager.DeleteUserOnline(t.UserID) //设置用户登录服务器
+		socket.Mainplayer.Delete(t.Info.UserID)
+		share.ShareManager.DeleteUserOnline(t.Info.UserID) //设置用户登录服务器
 	}
 
 	//}
@@ -73,7 +72,7 @@ func (s *Server) OnBoot(eng gnet.Engine) gnet.Action {
 
 func (s *Server) OnOpen(conn gnet.Conn) (out []byte, action gnet.Action) {
 	if conn.Context() == nil {
-		conn.SetContext(entity.NewClientData()) //注入data
+		conn.SetContext(socket.NewClientData()) //注入data
 	}
 
 	atomic.AddInt64(&s.connected, 1)
@@ -86,7 +85,7 @@ func (s *Server) OnTraffic(c gnet.Conn) (action gnet.Action) {
 		return gnet.Close
 	}
 
-	ws := c.Context().(*entity.ClientData).Getwsmsg()
+	ws := c.Context().(*socket.ClientData).Getwsmsg()
 	tt, len1 := ws.ReadBufferBytes(c)
 	if tt == gnet.Close {
 
@@ -165,7 +164,7 @@ const CROSS_DOMAIN = "<?xml version=\"1.0\"?><!DOCTYPE cross-domain-policy><cros
 const TEXT = "<policy-file-request/>\x00"
 
 func handle(c gnet.Conn) {
-	clientdata := c.Context().(*entity.ClientData)
+	clientdata := c.Context().(*socket.ClientData)
 	if clientdata.GetIsCrossDomain() {
 		return