From 328311d72368a80226c80343895566f3ff0e4ae9 Mon Sep 17 00:00:00 2001 From: liujianjiang Date: Sat, 29 Nov 2025 16:36:34 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BB=A3=E7=A0=81=E4=BC=98=E5=8C=96?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- main.py | 3 ++- public_function/config.yaml | 2 +- public_function/redis_task_manager.py | 4 ++-- 3 files changed, 5 insertions(+), 4 deletions(-) diff --git a/main.py b/main.py index 720bb64..e92a551 100644 --- a/main.py +++ b/main.py @@ -138,7 +138,8 @@ async def get_crawler_task(task_data: CrawlerItem, task_manager: Any = Depends(g data = {"task_id": result["task_id"], "status": 4} await task_manager.update_task_record(data) return {"code": 200, "message": "任务获取成功", "data": result} - raise HTTPException(status_code=404, detail="队列暂时没有任务，请等待一段时间后重新尝试") + else: + return {"code": 200, "message": "暂时没有任务", "data": None} except Exception as e: print(f"获取任务失败，失败原因: {e}") raise HTTPException(status_code=500, detail="获取任务失败；失败原因{}".format(e)) diff --git a/public_function/config.yaml b/public_function/config.yaml index 3298e82..05855bc 100644 --- a/public_function/config.yaml +++ b/public_function/config.yaml @@ -18,4 +18,4 @@ redis_config: host: 47.238.96.231 port: 6379 password: QAhL5RXKh1o21RbEqRZb - db: 0 \ No newline at end of file + db: 1 \ No newline at end of file diff --git a/public_function/redis_task_manager.py b/public_function/redis_task_manager.py index ba71f05..d30b13a 100644 --- a/public_function/redis_task_manager.py +++ b/public_function/redis_task_manager.py @@ -212,5 +212,5 @@ if __name__ == '__main__': config = read_config(config_path) redis_conn = RedisTaskManager(config) token = "opB4ztbdw45xFoJbXti20520bsEq3UDKKAtiDWHnGjjhP6v0KNFjqBM7bfzto6GLdUPviYnVdCgdCJYqe42nPoy6mvW59F3TPQZu" - # redis_conn.write_string_to_h_set("user_token", token, 1) - print(redis_conn.check_field_exists("user_token", token)) + redis_conn.write_string_to_h_set("user_token", token, 1) + # print(redis_conn.check_field_exists("user_token", token))