From 789337707e366ea1fd48f9c48811c0cfc5128cbc Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=A1=9C=E8=8F=AF?= <huan0016@gmail.com>
Date: Sat, 11 Mar 2023 09:54:47 +0800
Subject: [PATCH] task model

---
 ObjectManager.py | 8 ++++----
 main.py          | 6 ++----
 2 files changed, 6 insertions(+), 8 deletions(-)

diff --git a/ObjectManager.py b/ObjectManager.py
index 54b01eb..70dcc8d 100644
--- a/ObjectManager.py
+++ b/ObjectManager.py
@@ -43,15 +43,15 @@ class TaskManager(object):
         self.tasks = {}
         # TOOD: 保持一个后台线程，定时检查任务的状态，并将任务的状态更新到数据中
 
-    def create(self, task: dict):
-        task["id"] = str(uuid.uuid4())
-        self.tasks[task["id"]] = task
+    def create(self, task: Task):
+        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):
+    def update(self, task_id, task: Task):
         self.tasks[task_id] = task
 
     def query(self, task_id: str):
diff --git a/main.py b/main.py
index c8b4f03..4f257bd 100644
--- a/main.py
+++ b/main.py
@@ -92,10 +92,8 @@ async def get_tasks():
 
 
 @app.post("/tasks", response_model=Task)
-async def create_task():
-    return task_manager.add({
-        "name": "test",
-    })
+async def create_task(task: Task):
+    return task_manager.add(task)
 
 # 维护一个任务队列, 任务队列中的任务会被分发给worker节点
 # 任务状态变化时通知对应的客户端