From f3e894a0689af247abbfc40da03b134284a91e8e Mon Sep 17 00:00:00 2001 From: yu_lang <534013339@qq.com> Date: Tue, 21 Sep 2021 11:03:31 +0800 Subject: [PATCH] feat(proto):update proto auto tool --- pkg/structure/generate/tem.yaml | 5 ++- protobuf/listx.proto | 22 ++-------- protobuf/storage.proto | 25 +++-------- shell/gen_protobuf.py | 78 +++++++++++---------------------- 4 files changed, 39 insertions(+), 91 deletions(-) diff --git a/pkg/structure/generate/tem.yaml b/pkg/structure/generate/tem.yaml index 9aae05e..b81c1bf 100644 --- a/pkg/structure/generate/tem.yaml +++ b/pkg/structure/generate/tem.yaml @@ -6,4 +6,7 @@ STRING_X: - add - reduce - setbit - - getbit \ No newline at end of file + - getbit + +LIST_X: + - set \ No newline at end of file diff --git a/protobuf/listx.proto b/protobuf/listx.proto index 85a6b5c..4305a13 100644 --- a/protobuf/listx.proto +++ b/protobuf/listx.proto @@ -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 { - BaseKey key = 1; -} + +message SetRequest { + BaseKey key = 1; +} \ No newline at end of file diff --git a/protobuf/storage.proto b/protobuf/storage.proto index d13c98b..634266f 100644 --- a/protobuf/storage.proto +++ b/protobuf/storage.proto @@ -1,36 +1,23 @@ +// 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; + repeated string result = 1; } service CommServer { - 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); 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); + rpc Set (SetRequest) returns (CommendResponse); } \ No newline at end of file diff --git a/shell/gen_protobuf.py b/shell/gen_protobuf.py index 23475e4..38c6359 100644 --- a/shell/gen_protobuf.py +++ b/shell/gen_protobuf.py @@ -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 - file = open(storagePath,'w') - for l in line: - file.write(l) - file.close() + + print(f"{key}.proto", "-> success") + + +def mk_storage(cfg_camel): + storagePath = f"{protobufPath}/storage.proto" + file = open(storagePath,'w') + 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) + +