update

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)

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()