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))