diff --git a/handler/__init__.py b/handler/__init__.py index 2cae881..af3758a 100644 --- a/handler/__init__.py +++ b/handler/__init__.py @@ -1,3 +1,5 @@ +from loguru import logger + from .handle_event import HandlerEvent from .handle_paylist import HandlerPaylist from .handle_user import HandlerUser @@ -11,5 +13,8 @@ HANDLER_DICT = { def handle_factory(data): type_ = data['type'] - obj = HANDLER_DICT[type_](data) - return obj + obj = HANDLER_DICT.get(type_) + if not obj: + logger.warning(f'未知类型{type_}') + return + return obj(data) diff --git a/main.py b/main.py index debcf75..545ace6 100644 --- a/main.py +++ b/main.py @@ -36,6 +36,8 @@ def main(): st = time.time() * 1000 data = msg.value obj = handle_factory(data) + if not obj: + continue obj.run() logger.debug(time.time() * 1000 - st)