From 3efbad274db77ac864d9120369a75d950a401c5f Mon Sep 17 00:00:00 2001 From: UnKownOwO <80520429@qq.com> Date: Wed, 31 May 2023 20:57:18 +0800 Subject: [PATCH] =?UTF-8?q?=E5=8D=8F=E8=AE=AE=E4=BB=A3=E7=90=86=E7=94=9F?= =?UTF-8?q?=E6=88=90=E8=A1=A5=E5=85=85=E4=BF=AE=E6=94=B9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- gate/client_proto/client_proto_gen_test.go | 24 +++++++++++----------- 1 file changed, 12 insertions(+), 12 deletions(-) diff --git a/gate/client_proto/client_proto_gen_test.go b/gate/client_proto/client_proto_gen_test.go index cb14a4d0..221f87d7 100644 --- a/gate/client_proto/client_proto_gen_test.go +++ b/gate/client_proto/client_proto_gen_test.go @@ -33,8 +33,8 @@ func TestClientProtoGen(t *testing.T) { clientCmdData := string(clientCmdFile) clientCmdLineList := strings.Split(clientCmdData, "\n") // 生成代码文件 - var fileDataBuilder strings.Builder - fileDataBuilder.WriteString(`package client_proto + var fileDataBuffer bytes.Buffer + fileDataBuffer.WriteString(`package client_proto import ( "hk4e/gate/client_proto/proto" @@ -54,32 +54,32 @@ func (c *ClientCmdProtoMap) LoadClientCmdIdAndCmdName() { } cmdName := item[0] cmdId := item[1] - _, err = fmt.Fprintf(&fileDataBuilder, ` c.clientCmdIdCmdNameMap[uint16(%s)] = "%s" + _, err = fmt.Fprintf(&fileDataBuffer, ` c.clientCmdIdCmdNameMap[uint16(%s)] = "%s" c.clientCmdNameCmdIdMap["%s"] = uint16(%s) `, cmdId, cmdName, cmdName, cmdId) if err != nil { panic(err) } } - fileDataBuilder.WriteString(`} + fileDataBuffer.WriteString(`} func (c *ClientCmdProtoMap) GetClientProtoObjByName(protoObjName string) any { switch protoObjName { `) for _, protoObjName := range protoObjNameList { - _, err = fmt.Fprintf(&fileDataBuilder, ` case "%s": + _, err = fmt.Fprintf(&fileDataBuffer, ` case "%s": return new(proto.%s) `, protoObjName, protoObjName) if err != nil { panic(err) } } - fileDataBuilder.WriteString(` default: + fileDataBuffer.WriteString(` default: return nil } } `) - err = os.WriteFile("./client_proto_gen.go", []byte(fileDataBuilder.String()), 0644) + err = os.WriteFile("./client_proto_gen.go", []byte(fileDataBuffer.String()), 0644) if err != nil { panic(err) } @@ -91,10 +91,10 @@ func (c *ClientCmdProtoMap) GetClientProtoObjByName(protoObjName string) any { } rawFileStr := string(rawFileData) rawFileLine := strings.Split(rawFileStr, "\n") - var newFileBuilder bytes.Buffer + var newFileBuffer bytes.Buffer for i := 0; i < len(rawFileLine); i++ { line := rawFileLine[i] - newFileBuilder.WriteString(line + "\n") + newFileBuffer.WriteString(line + "\n") if !strings.Contains(line, "enum") { continue } @@ -109,7 +109,7 @@ func (c *ClientCmdProtoMap) GetClientProtoObjByName(protoObjName string) any { continue } for _, ref := range refEnum { - newFileBuilder.WriteString(ref + "\n") + newFileBuffer.WriteString(ref + "\n") } i++ for { @@ -117,12 +117,12 @@ func (c *ClientCmdProtoMap) GetClientProtoObjByName(protoObjName string) any { if !strings.Contains(nextLine, "}") { i++ } else { - newFileBuilder.WriteString(nextLine + "\n") + newFileBuffer.WriteString(nextLine + "\n") break } } } - err = os.WriteFile("./proto/"+entry.Name(), newFileBuilder.Bytes(), 0644) + err = os.WriteFile("./proto/"+entry.Name(), newFileBuffer.Bytes(), 0644) if err != nil { panic(err) }