zhangqijia / Pro2dProtoBuffers

Commit ce072bf52e5ef2f267a0ba3f0c582161f21016fc

Authored by zhangqijia
1 parent ac362b38

add generate.py

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