fix(node): notification error

Author: M03ED

app/jobs/node_checker.py

@@ -12,7 +12,7 @@
 from app.operation import OperatorType
 from app.db.crud.node import get_limited_nodes
 
-from config import JOB_CORE_HEALTH_CHECK_INTERVAL
+from config import JOB_CORE_HEALTH_CHECK_INTERVAL, JOB_CHECK_NODE_LIMITS_INTERVAL
 
 
 node_operator = NodeOperation(operator_type=OperatorType.SYSTEM)
@@ -158,9 +158,6 @@ async def node_health_check():
         check_tasks = [process_node_health_check(db_node, dict_nodes.get(db_node.id)) for db_node in db_nodes]
         await asyncio.gather(*check_tasks, return_exceptions=True)
 
-    # Check for nodes that exceeded data limits
-    await check_node_limits()
-
 
 @on_startup
 async def initialize_nodes():
@@ -175,10 +172,16 @@ async def initialize_nodes():
             await node_operator.connect_nodes_bulk(db, db_nodes)
             logger.info("All nodes' cores have been started.")
 
+    # Schedule node health check job (runs frequently)
     scheduler.add_job(
         node_health_check, "interval", seconds=JOB_CORE_HEALTH_CHECK_INTERVAL, coalesce=True, max_instances=1
     )
 
+    # Schedule node limits check job (runs less frequently)
+    scheduler.add_job(
+        check_node_limits, "interval", seconds=JOB_CHECK_NODE_LIMITS_INTERVAL, coalesce=True, max_instances=1
+    )
+
 
 @on_shutdown
 async def shutdown_nodes():

app/notification/discord/__init__.py

@@ -1,6 +1,6 @@
 from .host import create_host, modify_host, remove_host, modify_hosts
 from .user_template import create_user_template, modify_user_template, remove_user_template
-from .node import create_node, modify_node, remove_node, connect_node, error_node
+from .node import create_node, modify_node, remove_node, connect_node, error_node, limited_node
 from .group import create_group, modify_group, remove_group
 from .core import create_core, modify_core, remove_core
 from .admin import create_admin, modify_admin, remove_admin, admin_reset_usage, admin_login
@@ -27,6 +27,7 @@
     "remove_node",
     "connect_node",
     "error_node",
+    "limited_node",
     "create_group",
     "modify_group",
     "remove_group",

app/notification/telegram/__init__.py

@@ -1,6 +1,6 @@
 from .host import create_host, modify_host, remove_host, modify_hosts
 from .user_template import create_user_template, modify_user_template, remove_user_template
-from .node import create_node, modify_node, remove_node, connect_node, error_node
+from .node import create_node, modify_node, remove_node, connect_node, error_node, limited_node
 from .group import create_group, modify_group, remove_group
 from .core import create_core, modify_core, remove_core
 from .admin import create_admin, modify_admin, remove_admin, admin_reset_usage, admin_login
@@ -27,6 +27,7 @@
     "remove_node",
     "connect_node",
     "error_node",
+    "limited_node",
     "create_group",
     "modify_group",
     "remove_group",

config.py

@@ -97,4 +97,5 @@
 JOB_REMOVE_EXPIRED_USERS_INTERVAL = config("JOB_REMOVE_EXPIRED_USERS_INTERVAL", cast=int, default=3600)
 JOB_RESET_USER_DATA_USAGE_INTERVAL = config("JOB_RESET_USER_DATA_USAGE_INTERVAL", cast=int, default=600)
 JOB_RESET_NODE_USAGE_INTERVAL = config("JOB_RESET_NODE_USAGE_INTERVAL", cast=int, default=60)
+JOB_CHECK_NODE_LIMITS_INTERVAL = config("JOB_CHECK_NODE_LIMITS_INTERVAL", cast=int, default=60)
 JOB_CLEANUP_SUBSCRIPTION_UPDATES_INTERVAL = config("JOB_CLEANUP_SUBSCRIPTION_UPDATES_INTERVAL", cast=int, default=600)