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

1

Browse Source
This commit is contained in:
wuaho 2021-05-25 20:07:15 +08:00
parent 285119d27e
commit d0324d5f58
3 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
single_process.py
View File

@ -35,7 +35,7 @@ def run():
ts = int(time.time()) ts = int(time.time())
if 'user' in type_: if 'user' in type_:
continue # continue
obj = getattr(handler_user, type_) obj = getattr(handler_user, type_)
handler_user.receive_data.append(UserAct(obj, msg)) handler_user.receive_data.append(UserAct(obj, msg))
if len(handler_user.receive_data) >= 1000 or last_ts + 60 < ts: if len(handler_user.receive_data) >= 1000 or last_ts + 60 < ts:

6
v2/event_attr.py
View File

@ -10,11 +10,12 @@ class EventAttr:
def get_event_attr(self, key): def get_event_attr(self, key):
attr = self.event_attr.get(key) attr = self.event_attr.get(key)
if not attr: if not attr:
self.event_attr[key] = self.rdb.get(key) or set() self.event_attr[key] = self.rdb.smembers(key) or set()
return self.event_attr[key] return self.event_attr[key]
def set_event_attr(self, key, *data): def set_event_attr(self, key, *data):
self.event_attr[key] = self.rdb.sadd(key, *data) self.rdb.sadd(key, *data)
self.event_attr[key] = data
def check_attr(self, db, data): def check_attr(self, db, data):
event_name = data.get('#event_name') event_name = data.get('#event_name')
@ -26,5 +27,6 @@ class EventAttr:
attr = self.get_event_attr(key) attr = self.get_event_attr(key)
data_attr = set(data) data_attr = set(data)
extra_attr = data_attr - attr extra_attr = data_attr - attr
if extra_attr: if extra_attr:
self.set_event_attr(key, *extra_attr) self.set_event_attr(key, *extra_attr)

4
v2/transmitter.py
View File

@ -103,6 +103,6 @@ class Transmitter:
self.check_type(self.db_name, tb, data) self.check_type(self.db_name, tb, data)
self.set_event_attr(self.db_name, tb, data) self.set_event_attr(self.db_name, tb, data)
self.__send(self.db_name, tb, [json.dumps(item) for item in data]) self.__send(self.db_name, tb, [json.dumps(item) for item in data])
except: except Exception as e:
pass self.log.eror(e)
buffer.clear() buffer.clear()