feat(event): add ModuleReload event type

jxxghp · Sep 30, 2024 · 0ca4223 · 0ca4223
1 parent 835e0b4
commit 0ca4223
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/app/core/module.py b/app/core/module.py
@@ -2,8 +2,10 @@
 from typing import Generator, Optional, Tuple, Any
 
 from app.core.config import settings
+from app.core.event import eventmanager
 from app.helper.module import ModuleHelper
 from app.log import logger
+from app.schemas.types import EventType
 from app.utils.object import ObjectUtils
 from app.utils.singleton import Singleton
 
@@ -67,6 +69,7 @@ def reload(self):
         """
         self.stop()
         self.load_modules()
+        eventmanager.send_event(etype=EventType.ModuleReload, data={})
 
     def test(self, modleid: str) -> Tuple[bool, str]:
         """

diff --git a/app/schemas/types.py b/app/schemas/types.py
@@ -56,6 +56,8 @@ class EventType(Enum):
     SystemError = "system.error"
     # 刮削元数据
     MetadataScrape = "metadata.scrape"
+    # 模块需要重载
+    ModuleReload = "module.reload"
 
 
 # 同步链式事件