"feat(socket): 添加跨域请求处理并集成enum依赖，优化TCP连接数据注入"

common/socket/codec/CrossDomain.go

@@ -1,33 +0,0 @@
-package codec
-
-import (
-	"log"
-
-	"github.com/panjf2000/gnet/v2"
-)
-
-// CROSS_DOMAIN 定义跨域策略文件内容
-const CROSS_DOMAIN = "<?xml version=\"1.0\"?><!DOCTYPE cross-domain-policy><cross-domain-policy><allow-access-from domain=\"*\" to-ports=\"*\" /></cross-domain-policy>\x00"
-
-// TEXT 定义跨域请求的文本格式
-const TEXT = "<policy-file-request/>\x00"
-
-// Handle 处理网络连接
-func Handle(conn gnet.Conn) error {
-	// 读取数据并检查是否为跨域请求
-	data, err := conn.Peek(len(TEXT))
-	if err != nil {
-		log.Printf("Error reading cross-domain request: %v", err)
-		return err
-	}
-
-	if string(data) == TEXT { //判断是否是跨域请求
-		log.Printf("Received cross-domain request from %s", conn.RemoteAddr())
-		// 处理跨域请求
-		conn.Write([]byte(CROSS_DOMAIN))
-		conn.Discard(len(TEXT))
-		return nil
-	}
-
-	return nil
-}

common/socket/codec/SocketCodec.go

@@ -10,4 +10,5 @@ type SocketCodec interface {
 	Encode([]byte) ([]byte, error)
 	Decode(gnet.Conn) ([]byte, error)
 	
+	
 }

common/socket/codec/SocketCodec_Tomee.go

@@ -1,13 +1,21 @@
 package codec
 
 import (
+	"blazing/common/data/entity"
 	"encoding/binary"
 	"errors"
 	"io"
+	"log"
 
 	"github.com/panjf2000/gnet/v2"
 )
 
+// CROSS_DOMAIN 定义跨域策略文件内容
+const CROSS_DOMAIN = "<?xml version=\"1.0\"?><!DOCTYPE cross-domain-policy><cross-domain-policy><allow-access-from domain=\"*\" to-ports=\"*\" /></cross-domain-policy>\x00"
+
+// TEXT 定义跨域请求的文本格式
+const TEXT = "<policy-file-request/>\x00"
+
 var ErrIncompletePacket = errors.New("incomplete packet")
 
 // TomeeSocketCodec 协议格式:
@@ -23,26 +31,54 @@ var ErrIncompletePacket = errors.New("incomplete packet")
 // * |  ... ...  |
 // * +-----------+
 type TomeeSocketCodec struct{}
-var _ SocketCodec = (*TomeeSocketCodec)(nil) 
 
+var _ SocketCodec = (*TomeeSocketCodec)(nil)
 
 func NewTomeeSocketCodec() *TomeeSocketCodec {
 	return &TomeeSocketCodec{}
 
 }
+
+func handle(c gnet.Conn) {
+	clientdata:=c.Context().(*entity.ClientData)
+	if(clientdata.IsCrossDomain){
+		return
+
+	}
+	// 读取数据并检查是否为跨域请求
+	data, err := c.Peek(len(TEXT))
+	if err != nil {
+		log.Printf("Error reading cross-domain request: %v", err)
+		return  
+	}
+
+	if string(data) == TEXT { //判断是否是跨域请求
+		log.Printf("Received cross-domain request from %s", c.RemoteAddr())
+		// 处理跨域请求
+		c.Write([]byte(CROSS_DOMAIN))
+		c.Discard(len(TEXT))
+		
+		clientdata.IsCrossDomain=true
+		return  
+	}
+
+	return  
+}
 func (codec TomeeSocketCodec) Encode(buf []byte) ([]byte, error) {
 	bodyLen := len(buf)
 	data := make([]byte, 4+bodyLen)
-	
+
 	// 写入4字节的包长度
 	binary.BigEndian.PutUint32(data[:4], uint32(bodyLen))
 	// 写入包体
 	copy(data[4:], buf)
-	
+
 	return data, nil
 }
 
 func (codec TomeeSocketCodec) Decode(c gnet.Conn) ([]byte, error) {
+
+	handle(c)
 	// 先读取4字节的包长度
 	lenBuf, err := c.Peek(4)
 	if err != nil {
@@ -51,10 +87,10 @@ func (codec TomeeSocketCodec) Decode(c gnet.Conn) ([]byte, error) {
 		}
 		return nil, err
 	}
-	
+
 	bodyLen := binary.BigEndian.Uint32(lenBuf)
 	totalLen := 4 + int(bodyLen)
-	
+
 	// 检查整个包是否完整
 	buf, err := c.Peek(totalLen)
 	if err != nil {
@@ -63,13 +99,13 @@ func (codec TomeeSocketCodec) Decode(c gnet.Conn) ([]byte, error) {
 		}
 		return nil, err
 	}
-	
+
 	// 提取包体
 	body := make([]byte, bodyLen)
 	copy(body, buf[4:totalLen])
-	
+
 	// 从缓冲区中丢弃已读取的数据
 	_, _ = c.Discard(totalLen)
-	
+
 	return body, nil
 }