fix(node): disconnect function

Author: M03ED

app/db/crud/node.py

@@ -90,7 +90,7 @@ async def get_nodes(
             query = query.where(Node.status == status)
 
     if enabled:
-        query = query.where(Node.status != NodeStatus.disabled)
+        query = query.where(Node.status.not_in([NodeStatus.disabled, NodeStatus.limited]))
 
     if core_id:
         query = query.where(Node.core_config_id == core_id)

app/jobs/node_checker.py

@@ -133,6 +133,9 @@ async def check_node_limits():
         limited_nodes = await get_limited_nodes(db)
 
         for db_node in limited_nodes:
+            # Disconnect the node first (stop it from running)
+            await node_operator.disconnect_single_node(db_node.id)
+
             # Update status to limited
             await NodeOperation._update_single_node_status(
                 db, db_node.id, NodeStatus.limited, message="Data limit exceeded", send_notification=False

app/operation/node.py

@@ -207,8 +207,8 @@ async def modify_node(
         except IntegrityError:
             await self.raise_error(message=f'Node "{db_node.name}" already exists', code=409, db=db)
 
-        if db_node.status is NodeStatus.disabled:
-            await node_manager.remove_node(db_node.id)
+        if db_node.status in (NodeStatus.disabled, NodeStatus.limited):
+            await self.disconnect_single_node(db_node.id)
         else:
             try:
                 await node_manager.update_node(db_node)
@@ -410,6 +410,18 @@ async def connect_single_node(self, db: AsyncSession, node_id: int) -> None:
             )
             asyncio.create_task(notification.error_node(node_notif))
 
+    async def disconnect_single_node(self, node_id: int) -> None:
+        """
+        Disconnect a single node from the node manager (stop it from running).
+
+        Used when a node needs to be stopped (e.g., when limited or disabled).
+
+        Args:
+            node_id (int): ID of the node to disconnect.
+        """
+        await node_manager.remove_node(node_id)
+        logger.info(f'Node "{node_id}" disconnected')
+
     async def restart_node(self, db: AsyncSession, node_id: Node, admin: AdminDetails) -> None:
         await self.connect_single_node(db, node_id)
         logger.info(f'Node "{node_id}" restarted by admin "{admin.username}"')