generator.py
4.33 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
import os
ProtoCodeGO = './cmd/gameserver/action/protocode.go'
ProtoCodeGOTest = './cmd/test/action/protocode.go'
ProtoCodeFile = './protos/protocode.proto'
ProtoFileDir = "./protos"
ProtoCodeStr = "syntax = \"proto3\";\noption go_package = \"../pb;pb\";\n\npackage protocode;\n\nenum ProtoCode\n{{\n " \
"UNKNOWN = 0;\n {}\n}} "
ProtoCodeLineReq = "\t{}Rpc = {};\n"
ProtoCodeLineRsp = "\t{}Rsp = {};\n"
ProtoCodeLineNty = "\t{}Nty = {};\n"
GoProtoCodeStr = "package action\n\nimport (\n\t\"pro2d/common/logger\"\n\t\"pro2d/pb\"\n)\n\nfunc GetActionMap() " \
"map[interface{{}}]interface{{}} {{\n\tlogger.Debug(\"init protocode...\")\n\tam := make(map[interface{{" \
"}}]interface{{}})\n{}\n\treturn am\n}}"
GoProtoCodeLine = "\tam[uint32(pb.ProtoCode_{}Rpc)] = {}Rpc\n"
GoProtoCodeTestStr = "package action\n\nimport (\n\t\"pro2d/pb\"\n)\n\nfunc GetTestActionMap() " \
"map[interface{{}}]interface{{}} {{\n\tam := make(map[interface{{" \
"}}]interface{{}})\n{}\n\treturn am\n}}"
GoProtoCodeTestReqLine = "\tam[uint32(pb.ProtoCode_{}Rpc)] = {}Rsp\n"
GoProtoCodeTestRspLine = "\tam[uint32(pb.ProtoCode_{}Rsp)] = {}Rsp\n"
GoProtoCodeTestNtyLine = "\tam[uint32(pb.ProtoCode_{}Nty)] = {}Nty\n"
def generatorProto(path):
RpcStartCode = 500
NtyStartCode = 1000
files = os.listdir(path)
ProtoCodeData = ""
GoCodeData = ""
GoCodeTestData = ""
for file in files:
if file.find("account.proto") != -1:
continue
if os.path.isdir(file):
continue
with open(path + "/" + file, 'r', encoding='utf-8', errors='ignore') as f:
lines = f.readlines() # 读取所有行
firstline = lines[0]
if firstline.find("proto3") == -1:
continue
# req + rar
for line in lines:
if line.find("message") == -1:
continue
sline = line.split(' ')
if len(sline) < 2:
continue
messageStr = sline[1].replace('\n', '').replace('{', "")
n1 = messageStr.find('Req')
n2 = messageStr.find('Rar')
loginReq = messageStr.find('LoginReq')
if n1 != -1:
RpcStartCode = RpcStartCode + 1
ProtoCodeData += ProtoCodeLineReq.format(messageStr[:n1], RpcStartCode)
if loginReq != -1:
continue
GoCodeData += GoProtoCodeLine.format(messageStr[:n1], messageStr[:n1])
GoCodeTestData += GoProtoCodeTestReqLine.format(messageStr[:n1], messageStr[:n1])
elif n2 != -1:
RpcStartCode = RpcStartCode + 1
ProtoCodeData += ProtoCodeLineReq.format(messageStr[:n2], RpcStartCode)
if loginReq != -1:
continue
GoCodeData += GoProtoCodeLine.format(messageStr[:n2], messageStr[:n2])
GoCodeTestData += GoProtoCodeTestReqLine.format(messageStr[:n2], messageStr[:n2])
ProtoCodeData += '\n'
# nty
for line in lines:
if line.find("message") == -1:
continue
sline = line.split(' ')
if len(sline) < 2:
continue
messageStr = sline[1].replace('\n', '').replace('{', "")
n3 = messageStr.find('Nty')
loginReq = messageStr.find('LoginReq')
if n3 != -1:
NtyStartCode = NtyStartCode + 1
ProtoCodeData += ProtoCodeLineNty.format(messageStr[:n3], NtyStartCode)
if loginReq != -1:
continue
GoCodeTestData += GoProtoCodeTestNtyLine.format(messageStr[:n3], messageStr[:n3])
# protocode.go
gostr = GoProtoCodeStr.format(GoCodeData)
fo = open(ProtoCodeGO, "w")
fo.write(gostr)
fo.close()
# protocode.go
gostr = GoProtoCodeTestStr.format(GoCodeTestData)
fo = open(ProtoCodeGOTest, "w")
fo.write(gostr)
fo.close()
#protocode.proto
protostr = ProtoCodeStr.format(ProtoCodeData)
fo = open(ProtoCodeFile, "w")
fo.write(protostr)
fo.close()
if __name__ == "__main__":
generatorProto(ProtoFileDir)