p25734189
/
wheat-cache
forked from p93542168/wheat-cache
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix(gen-struct): add struct

This commit is contained in:
bandl 2021-09-20 16:36:47 +08:00 committed by yu_lang
parent cf4b5cd63c
commit b9060a64d5
2 changed files with 37 additions and 97 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
protobuf/listx.proto
View File

@ -1,21 +0,0 @@
syntax = "proto3";
import "base.proto";
option go_package = "pkg/proto";
message LSetRequest {
BaseKey key = 1;
}
message LGetRequest {
BaseKey key = 1;
}
message LAddRequest {
BaseKey key = 1;
}
message LQueueRequest {
BaseKey key = 1;
}
message LSetbitRequest {
BaseKey key = 1;
}
message LGetbitRequest {
BaseKey key = 1;
}

113
shell/gen_protobuf.py
View File

@ -24,92 +24,53 @@ def load_conf():
key = to_camel(key).lower()
cfg_camel[key] = [to_camel(v) for v in val]
# print(cfg_camel)
return cfg, cfg_camel
# 生成文件,并写入
def mkdir(cfg_camel):
path = "protobuf"
storagePath = f"{protobufPath}/storage.proto"
for key,value in cfg_camel.items():
'''生成、更新数据结构proto文件'''
proto_path = path + '/' + key + '.proto'
if not os.path.exists(proto_path): # 如果这个文件不存在
'''生成对应的数据结构proto文件'''
file = open(proto_path,'w')
file.write('syntax = "proto3";\nimport "base.proto";\noption go_package = "pkg/proto";\n')
for v in value:
file.write('\nmessage ' + v +'Request '+'{\n\tBaseKey key = 1;\n}')
file.close()
'''给strong.proto添加import该新的数据结构'''
import_structure = 'import "'+ key +'.proto";\n'
mess = 'message CommendResponse {\n'
with open(storagePath) as f:
line = f.readlines()
# print(line)
f.close
for i in range(len(line)):
if mess in line[i]:
break
line.insert(i,import_structure)
file = open(storagePath,'w')
for l in line:
file.write(l)
file.close()
i=0
for i in range(len(line)-1,0,-1):
if '}' in line[i]:
break
for v in value:
line.insert(i,'\trpc '+ v +' ('+ v +'Request) returns (CommendResponse);\n')
file = open(storagePath,'w')
for l in line:
file.write(l)
file.close()
else: # 如果这个文件存在
with open(proto_path) as f:
line = f.readlines()
f.close()
for v in value:
function = v + 'Request'
flag = 0
for l in line:
if function in l:
flag=1
break
if flag == 0:
file = open(proto_path,'a')
file.write('\nmessage ' + v +'Request '+'{\n\tBaseKey key = 1;\n}')
file.close()
flag = 0
'''更新storage.proto文件'''
with open(storagePath) as f:
line = f.readlines()
f.close
for v in value:
flg=0
for l in range(len(line)):
if '\trpc '+ v +' ('+ v +'Request) returns (CommendResponse);\n' in line[l]:
flg = 1
break
if flg==0:
i=0
for i in range(len(line)-1,0,-1):
if '}' in line[i]:
path = path.strip()
# 判断路径是否存在
isExists = os.path.exists(path)
# 判断结果
if not isExists:
os.makedirs(path)
print(path + ' 创建成功')
else: # 如果目录存在则不创建，并提示目录已存在
# print(path + ' 目录已存在')
for key,value in cfg_camel.items():
# print(key)
proto_path = path + '/' + key + '.proto'
if not os.path.exists(proto_path): # 如果这个文件不存在
file = open(proto_path,'w')
file.write('syntax = "proto3";\nimport "base.proto";\noption go_package = "pkg/proto";\n')
for v in value:
file.write('\nmessage ' + v +'Request '+'{\n BaseKey key = 1;\n}\n')
file.close()
else: # 如果这个文件存在
with open(proto_path) as f:
line = f.readlines()
# print(line)
f.close()
for v in value:
function = v + 'Request'
flag = 0
for l in line:
if function in l:
flag=1
break
line.insert(i,'\trpc '+ v +' ('+ v +'Request) returns (CommendResponse);\n')
i=0
file = open(storagePath,'w')
for l in line:
file.write(l)
file.close()
if flag == 0:
file = open(proto_path,'a')
file.write('\nmessage ' + v +'Request '+'{\n BaseKey key = 1;\n}\n')
file.close()
flag = 0
if __name__ == "__main__":
conf, cfg_camel = load_conf()
print(cfg_camel)
#
mkdir(cfg_camel)