model

ObjectManager.py

@@ -0,0 +1,86 @@
+import uuid
+from fastapi import WebSocket
+from pydantic import BaseModel
+from datetime import datetime
+
+# 使用字典来存储websocket连接
+class ConnectionManager:
+    def __init__(self):
+        self.active_connections: map = {}
+
+    # 加入一个新的websocket连接
+    async def connect(self, websocket: WebSocket, client_id: int):
+        await websocket.accept()
+        self.active_connections[client_id] = websocket
+
+    # 删除一个websocket连接(要先检查client_id是否存在)
+    def disconnect(self, client_id: int):
+        if client_id in self.active_connections:
+            del self.active_connections[client_id]
+
+    # 向指定的websocket连接发送消息(要先检查client_id是否存在)
+    async def send_personal_message(self, message: str, client_id: int):
+        if client_id in self.active_connections:
+            await self.active_connections[client_id].send_text(message)
+
+    # 向所有的websocket连接发送消息
+    async def broadcast(self, message: str):
+        for client_id, ws in self.active_connections.items():
+            await ws.send_text(message)
+
+# Task 基本模型
+class Task(BaseModel):
+    id: int
+    name: str
+    status: str
+    created_at: datetime
+    updated_at: datetime
+
+
+# 使用字典来存储Task任务
+class TaskManager(object):
+    def __init__(self):
+        self.tasks = {}
+        # TOOD: 保持一个后台线程，定时检查任务的状态，并将任务的状态更新到数据中
+
+    def create(self, task: dict):
+        task["id"] = str(uuid.uuid4())
+        self.tasks[task["id"]] = task
+        return task
+
+    def delete(self, task_id: str):
+        del self.tasks[task_id]
+
+    def update(self, task_id, task: dict):
+        self.tasks[task_id] = task
+
+    def query(self, task_id: str):
+        return self.tasks[task_id]
+
+    def query_all(self):
+        return self.tasks
+
+# 使用字典来存储服务器池(使用腾讯云的API来管理服务器)
+class ServerManager(object):
+    def __init__(self):
+        self.servers = {}
+        # 维护一个服务器池，每个服务器都有一个状态，状态有三种：空闲，运行中，异常
+        # 如果服务器异常，需要将其从服务器池中删除
+        # 如果服务器空闲，需要通知弹性伸缩服务将其删除
+
+        # 添加一个新的服务器(如何得知增加了新服务器? 主动发现? 被动发现? 使用腾讯云的API主动增加服务器)
+        def add(self, server_name, server_info):
+            self.servers[server_name] = server_info
+    
+        def delete(self, server_name):
+            del self.servers[server_name]
+    
+        def update(self, server_name, server_info):
+            self.servers[server_name] = server_info
+    
+        def query(self, server_name):
+            return self.servers[server_name]
+    
+        def query_all(self):
+            return self.servers
+