leguadmin/to_ck
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

ck 断开重连

Browse Source
This commit is contained in:
wuaho 2021-04-21 18:23:23 +08:00
parent 3df9a6fb5c
commit f0501b0615
2 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
ck/ck.py
View File

@ -35,6 +35,9 @@ class CK:
self.bulk_max = bulk_max self.bulk_max = bulk_max
self.__send_ts = int(time.time()) self.__send_ts = int(time.time())
def set_connect(self, client):
self.client = client
def up_tb_struct(self, db, tb, data): def up_tb_struct(self, db, tb, data):
struct_dict = self.struct_dict.get(f'{db}_{tb}', {}) struct_dict = self.struct_dict.get(f'{db}_{tb}', {})

5
main.py
View File

@ -1,3 +1,6 @@
import clickhouse_driver
from clickhouse_driver import Client
from ck.ck import CK from ck.ck import CK
from common import * from common import *
@ -21,6 +24,8 @@ def run():
del msg['type'] del msg['type']
ck_client.send(db, table, msg) ck_client.send(db, table, msg)
except clickhouse_driver.errors.NetworkError:
ck_client.set_connect(Client(**settings.CK_CONFIG))
except Exception as e: except Exception as e:
print(traceback.print_exc()) print(traceback.print_exc())