From 9340bf8e58e58afac387796818e95905311f2bf7 Mon Sep 17 00:00:00 2001 From: wuaho Date: Sat, 14 Aug 2021 09:25:36 +0800 Subject: [PATCH] 1 --- v2/transmitter.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/v2/transmitter.py b/v2/transmitter.py index 35cb0e1..d38a9be 100644 --- a/v2/transmitter.py +++ b/v2/transmitter.py @@ -19,22 +19,23 @@ class Transmitter: self.lock = lock self.event_attr = event_attr self.p = p + self.ping_ts = 0 def add_source(self, handler, bulk_max=1000, time_out=60): self.slots[handler] = {'bulk_max': bulk_max, 'time_out': time_out, "ts": int(time.time())} def check_send(self): + ts = int(time.time()) for h, p in self.slots.items(): - ts = int(time.time()) tb, buffer = h.buffer_pool buffer_size = len(buffer) if (p['ts'] + p['time_out'] <= ts or buffer_size >= p['bulk_max']) and buffer_size > 0: p['ts'] = ts yield tb, buffer - elif p['ts'] + p['time_out'] <= ts: - # 保持连接 - print('保持连接') - self.db_client.execute('select 1') + if self.ping_ts + 60 < ts: + # 保持连接 + print('保持连接') + self.db_client.execute('select 1') @staticmethod def flat_data(data: dict):