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

feat(proto):update proto auto tool

This commit is contained in:
yu_lang 2021-09-21 11:03:31 +08:00
parent 53fe26db48
commit f3e894a068
4 changed files with 39 additions and 91 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
pkg/structure/generate/tem.yaml
View File

@ -7,3 +7,6 @@ STRING_X:
- reduce
- setbit
- getbit
LIST_X:
- set

18
protobuf/listx.proto
View File

@ -1,21 +1,7 @@
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 {
message SetRequest {
BaseKey key = 1;
}

21
protobuf/storage.proto
View File

@ -1,9 +1,13 @@
// Code generated by gen-struct. DO NOT EDIT.
// make gen-protobuf generated
syntax = "proto3";
option go_package = "pkg/proto";
import "stringx.proto";
import "listx.proto";
message CommendResponse {
repeated string result = 1;
}
@ -15,22 +19,5 @@ service CommServer {
rpc Reduce (ReduceRequest) returns (CommendResponse);
rpc Setbit (SetbitRequest) returns (CommendResponse);
rpc Getbit (GetbitRequest) returns (CommendResponse);
rpc LSet (LSetRequest) returns (CommendResponse);
rpc LGet (LGetRequest) returns (CommendResponse);
rpc LAdd (LAddRequest) returns (CommendResponse);
rpc LQueue (LQueueRequest) returns (CommendResponse);
rpc LSetbit (LSetbitRequest) returns (CommendResponse);
rpc LGetbit (LGetbitRequest) returns (CommendResponse);
rpc Set (SetRequest) returns (CommendResponse);
rpc Get (GetRequest) returns (CommendResponse);
rpc Add (AddRequest) returns (CommendResponse);
rpc Reduce (ReduceRequest) returns (CommendResponse);
rpc Setbit (SetbitRequest) returns (CommendResponse);
rpc Getbit (GetbitRequest) returns (CommendResponse);
rpc LSet (LSetRequest) returns (CommendResponse);
rpc LGet (LGetRequest) returns (CommendResponse);
rpc LAdd (LAddRequest) returns (CommendResponse);
rpc LQueue (LQueueRequest) returns (CommendResponse);
rpc LSetbit (LSetbitRequest) returns (CommendResponse);
rpc LGetbit (LGetbitRequest) returns (CommendResponse);
}

74
shell/gen_protobuf.py
View File

@ -27,12 +27,11 @@ def load_conf():
return cfg, cfg_camel
# 生成文件,并写入
def mkdir(cfg_camel):
'''生成对应的数据结构proto文件'''
def mk_structure(cfg_camel):
path = "protobuf"
storagePath = f"{protobufPath}/storage.proto"
for key,value in cfg_camel.items():
'''生成、更新数据结构proto文件'''
for key, value in cfg_camel.items():
proto_path = path + '/' + key + '.proto'
if not os.path.exists(proto_path): # 如果这个文件不存在
'''生成对应的数据结构proto文件'''
@ -41,34 +40,9 @@ def mkdir(cfg_camel):
for v in value:
file.write('\nmessage ' + v +'Request '+'{\n BaseKey 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,' rpc '+ v +' ('+ v +'Request) returns (CommendResponse);\n')
file = open(storagePath,'w')
for l in line:
file.write(l)
file.close()
else: # 如果这个文件存在
'''更新数据结构proto文件'''
with open(proto_path) as f:
line = f.readlines()
f.close()
@ -84,32 +58,30 @@ def mkdir(cfg_camel):
file.write('\nmessage ' + v +'Request '+'{\n BaseKey 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 ' rpc '+ 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]:
break
line.insert(i,' rpc '+ v +' ('+ v +'Request) returns (CommendResponse);\n')
i=0
print(f"{key}.proto", "-> success")
def mk_storage(cfg_camel):
storagePath = f"{protobufPath}/storage.proto"
file = open(storagePath,'w')
for l in line:
file.write(l)
file.write('// Code generated by gen-struct. DO NOT EDIT.\n// make gen-protobuf generated\n\nsyntax = "proto3";\n\noption go_package = "pkg/proto";\n\n')
for key,value in cfg_camel.items():
file.write('import "'+ key +'.proto";\n')
file.write('\nmessage CommendResponse {\n repeated string result = 1;\n}\n\nservice CommServer {\n')
for key,value in cfg_camel.items():
for v in value:
file.write(' rpc '+ v +' ('+ v +'Request) returns (CommendResponse);\n')
file.write('}')
file.close()
print("storage.proto", "-> success")
if __name__ == "__main__":
conf, cfg_camel = load_conf()
print(cfg_camel)
mkdir(cfg_camel)
mk_structure(cfg_camel) # 生成对应的数据结构proto文件
mk_storage(cfg_camel)