aozhiwei 9d39452286 1
2019-03-04 19:11:50 +08:00

72 lines
2.1 KiB
Python

#coding utf8
#!/usr/bin/python
import os
import sys
import time
import threading
import subprocess
g_is_terminated = False
def is_terminated():
return g_is_terminated
def printp_stdout(p):
try:
while not is_terminated():
line = p.stdout.readline()
if len(line) > 0:
print(line, end = '')
except Exception as e:
print('build_pb stdout error:' + str(e))
def printp_stderr(p):
try:
while is_terminated():
line = p.stderr.readline()
if len(line) > 0:
print(line, end = '')
except Exception as e:
print('build_pb stderr error:' + str(e))
def need_rebuild():
for proto_name in ('ss_proto', 'ss_msgid'):
if not os.path.isfile(proto_name + '.pb.cc'):
return True
s1 = os.stat(proto_name + '.pb.cc')
s2 = os.stat('../tools/protobuild/' + proto_name + '.proto')
if s1.st_mtime < s2.st_mtime:
return True
return False
def rebuild():
global g_is_terminated
try:
p = subprocess.Popen(
'protoc --proto_path=../tools/protobuild --cpp_out=../dbproxy ../tools/protobuild/ss_proto.proto && ' +
'protoc --proto_path=../tools/protobuild --cpp_out=../dbproxy ../tools/protobuild/ss_msgid.proto ',
stdin = subprocess.PIPE,
stdout = subprocess.PIPE,
stderr = None,
shell = True)
t1 = threading.Thread(target = printp_stdout, args=(p, ))
t2 = threading.Thread(target = printp_stderr, args=(p, ))
t1.start()
t2.start()
p.wait()
g_is_terminated = True
t1.join()
t2.join()
sys.exit(p.returncode)
except Exception as e:
print('build_protocol rebuild error:' + str(e))
def repair_githooks():
os.system('/bin/bash ../tools/scripts/githooks/install.sh')
repair_githooks()
if need_rebuild():
rebuild()
else:
print('pb files already is the latest')