协议代理枚举类型修复

common/mq/nats.go

@@ -358,7 +358,8 @@ func (m *MessageQueue) gateTcpMqRecvHandleLoop(data []byte, dataBuf *[]byte) {
 	}
 	// 长度太短
 	if len(data) < 4 {
-		logger.Debug("packet len less 4 byte")
+		logger.Debug("packet len less 4 byte, data: %v", data)
+		*dataBuf = append(*dataBuf, data...)
 		return
 	}
 	// 消息的载荷部分长度

gate/client_proto/client_proto_gen_test.go

@@ -7,6 +7,7 @@ import (
 )
 
 func TestClientProtoGen(t *testing.T) {
+	// 生成根据proto类名获取对象实例的switch方法
 	dir, err := os.ReadDir("./proto")
 	if err != nil {
 		panic(err)
@@ -22,7 +23,6 @@ func TestClientProtoGen(t *testing.T) {
 		}
 		nameList = append(nameList, split[len(split)-2])
 	}
-
 	fileData := "package client_proto\n"
 	fileData += "\n"
 	fileData += "import (\n"
@@ -38,9 +38,99 @@ func TestClientProtoGen(t *testing.T) {
 	fileData += "\t\treturn nil\n"
 	fileData += "\t}\n"
 	fileData += "}\n"
-
 	err = os.WriteFile("./client_proto_gen.go", []byte(fileData), 0644)
 	if err != nil {
 		panic(err)
 	}
+	// 处理枚举
+	for _, entry := range dir {
+		rawFileData, err := os.ReadFile("./proto/" + entry.Name())
+		if err != nil {
+			panic(err)
+		}
+		rawFileStr := string(rawFileData)
+		rawFileLine := strings.Split(rawFileStr, "\n")
+		newFileStr := ""
+		for i := 0; i < len(rawFileLine); i++ {
+			line := rawFileLine[i]
+			newFileStr += line + "\n"
+			if !strings.Contains(line, "enum") {
+				continue
+			}
+			split := strings.Split(line, " ")
+			if len(split) != 3 || split[0] != "enum" || split[2] != "{" {
+				continue
+			}
+			enumName := split[1]
+			refEnum := FindEnumInDirFile("../../protocol/proto_hk4e", enumName)
+			if refEnum == nil {
+				continue
+			}
+			i++
+			x := 0
+			for {
+				nextLine := rawFileLine[i]
+				if !strings.Contains(nextLine, "}") && x < len(refEnum) {
+					newFileStr += refEnum[x] + "\n"
+					i++
+					x++
+				} else {
+					newFileStr += line + "\n"
+					break
+				}
+			}
+		}
+		err = os.WriteFile("./proto/"+entry.Name(), []byte(newFileStr), 0644)
+		if err != nil {
+			panic(err)
+		}
+	}
+}
+
+func FindEnumInDirFile(path string, name string) (lineList []string) {
+	dir, err := os.ReadDir(path)
+	if err != nil {
+		panic(err)
+	}
+	for _, entry := range dir {
+		if entry.IsDir() {
+			ret := FindEnumInDirFile(path+"/"+entry.Name(), name)
+			if ret != nil {
+				return ret
+			}
+			continue
+		}
+		fileData, err := os.ReadFile(path + "/" + entry.Name())
+		if err != nil {
+			panic(err)
+		}
+		fileStr := string(fileData)
+		fileLine := strings.Split(fileStr, "\n")
+		for i := 0; i < len(fileLine); i++ {
+			line := fileLine[i]
+			if !strings.Contains(line, "enum") {
+				continue
+			}
+			split := strings.Split(line, " ")
+			if len(split) != 3 || split[0] != "enum" || split[2] != "{" {
+				continue
+			}
+			enumName := split[1]
+			if enumName != name {
+				continue
+			}
+			i++
+			lineList := make([]string, 0)
+			for {
+				nextLine := fileLine[i]
+				if !strings.Contains(nextLine, "}") {
+					lineList = append(lineList, nextLine)
+				} else {
+					return lineList
+				}
+				i++
+			}
+		}
+	}
+	return nil
 }

pkg/object/object.go

@@ -39,7 +39,7 @@ func FastDeepCopy(dst, src any) error {
 }
 
 func CopyProtoBufSameField(dst, src pb.Message) ([]string, error) {
-	date, err := protojson.Marshal(src)
+	data, err := protojson.Marshal(src)
 	if err != nil {
 		return nil, err
 	}
@@ -50,7 +50,7 @@ func CopyProtoBufSameField(dst, src pb.Message) ([]string, error) {
 		if loopCount > 1000 {
 			return nil, errors.New("loop count limit")
 		}
-		err = protojson.Unmarshal(date, dst)
+		err = protojson.Unmarshal(data, dst)
 		if err != nil {
 			if !strings.Contains(err.Error(), "unknown field") {
 				return nil, err
@@ -61,13 +61,13 @@ func CopyProtoBufSameField(dst, src pb.Message) ([]string, error) {
 			}
 			fieldName := split[1]
 			jsonObj := make(map[string]any)
-			err = json.Unmarshal(date, &jsonObj)
+			err = json.Unmarshal(data, &jsonObj)
 			if err != nil {
 				return nil, err
 			}
 			DeleteAllKeyNameFromStringAnyMap(jsonObj, fieldName)
 			delList = append(delList, fieldName)
-			date, err = json.Marshal(jsonObj)
+			data, err = json.Marshal(jsonObj)
 			if err != nil {
 				return nil, err
 			}