diff --git a/handler/__init__.py b/handler/__init__.py index 953bfd8..8cb2c0a 100644 --- a/handler/__init__.py +++ b/handler/__init__.py @@ -5,16 +5,15 @@ from .handle_paylist import HandlerPaylist from .handle_user import HandlerUser HANDLER_DICT = { - 'user': HandlerUser, - 'paylist': HandlerPaylist, - 'event': HandlerEvent, + b'user': HandlerUser, + b'paylist': HandlerPaylist, + b'event': HandlerEvent, } -def handle_factory(data): - type_ = data['type'] - obj = HANDLER_DICT.get(type_) +def handle_factory(key, data): + obj = HANDLER_DICT.get(key) if not obj: - logger.warning(f'未知类型{type_}') + logger.warning(f'未知类型{key}') return return obj(data) diff --git a/main.py b/main.py index 20200b4..4343d05 100644 --- a/main.py +++ b/main.py @@ -34,8 +34,9 @@ def main(): consumer.subscribe((conf['topic_name'],)) for msg in consumer: st = time.time() * 1000 + key = msg.key data = msg.value - obj = handle_factory(data) + obj = handle_factory(key, data) if not obj: continue obj.run()