35 lines
838 B
Python
35 lines
838 B
Python
from settings import settings
|
|
from v2 import *
|
|
|
|
db_client = CK(**settings.CK_CONFIG)
|
|
sketch = Sketch(db_client)
|
|
handler_event = HandlerEvent(db_client)
|
|
handler_user = HandlerUser(db_client)
|
|
transmitter = Transmitter(db_client, sketch)
|
|
|
|
|
|
def run():
|
|
transmitter.add_source(handler_event, 1000, 60)
|
|
transmitter.add_source(handler_user, 100, 60)
|
|
for topic, msg in consumer():
|
|
# print(msg)
|
|
type_ = msg['#type']
|
|
db = settings.APPID_TO_CKDB.get(msg['#app_id'])
|
|
if 'user' in type_:
|
|
# continue
|
|
obj = getattr(handler_user, type_)
|
|
elif 'track' in type_:
|
|
# continue
|
|
obj = getattr(handler_event, type_)
|
|
else:
|
|
continue
|
|
|
|
del msg['#type']
|
|
obj(db, msg)
|
|
|
|
transmitter.run()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
run()
|