From 48e33c7fbd6969d05b2affda0321595d2d6b370a Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期一, 10 二月 2025 14:51:06 +0800
Subject: [PATCH] 增加ping機制
---
app/task/sync_account_token.py | 18 ++++++++++--------
1 files changed, 10 insertions(+), 8 deletions(-)
diff --git a/app/task/sync_account_token.py b/app/task/sync_account_token.py
index b7c2e5f..55e0082 100644
--- a/app/task/sync_account_token.py
+++ b/app/task/sync_account_token.py
@@ -5,9 +5,9 @@
from sqlalchemy.future import select
-from app.config.agent_base_url import RG_USER_LOGIN, DF_USER_LOGIN
+from app.config.agent_base_url import RG_USER_LOGIN, DF_USER_LOGIN, RG_PING, DF_PING
from app.config.config import settings
-from app.config.const import chat_server, workflow_server
+from app.config.const import chat_server, workflow_server, http_200
from app.models import UserTokenModel
from app.models.app_token_model import AppToken
from app.models.base_model import SessionLocal
@@ -29,19 +29,21 @@
:return:
"""
app_data = [{"token_id": chat_server, "url": f"{settings.fwr_base_url}{RG_USER_LOGIN}",
- "data": {"email": "", "password": ""}, "is_crypt": True},
+ "data": {"email": "", "password": ""}, "is_crypt": True, "ping_url": f"{settings.fwr_base_url}{RG_PING}",
+ "token": "{}"},
{"token_id": workflow_server, "url": f"{settings.dify_base_url}{DF_USER_LOGIN}",
- "data": {"email": "", "password": "", "remember_me": True, "language": "zh-Hans"}, "is_crypt": False}]
+ "data": {"email": "", "password": "", "remember_me": True, "language": "zh-Hans"}, "is_crypt": False
+ , "ping_url": f"{settings.dify_base_url}{DF_PING}", "token": "Bearer {}"}]
- async def sync_token_chat(token_id, url, data, is_crypt):
+ async def sync_token_chat(token_id, url, data, is_crypt, ping_url, token):
db = SessionLocal()
# pdb = PostgresqlSessionLocal()
current_time = datetime.now() - timedelta(hours=12)
try:
user_token = db.query(UserTokenModel).filter(UserTokenModel.id == token_id).first()
+ chat = ChatBaseApply()
+ if user_token and (user_token.updated_at < current_time or not user_token.access_token or await chat.chat_ping(ping_url, {}, await chat.get_chat_headers(token.format(user_token.access_token))) != http_200):
- if user_token and (user_token.updated_at < current_time or not user_token.access_token):
- chat = ChatBaseApply()
data["email"] = user_token.account
if is_crypt:
data["password"] = await chat.password_encrypt(await password_decrypted(user_token.password))
@@ -61,7 +63,7 @@
tasks = []
for app in app_data:
- tasks.append(asyncio.create_task(sync_token_chat(app["token_id"], app["url"], app["data"], app["is_crypt"])))
+ tasks.append(asyncio.create_task(sync_token_chat(app["token_id"], app["url"], app["data"], app["is_crypt"], app["ping_url"], app["token"])))
done, pending = await asyncio.wait(tasks, return_when=asyncio.ALL_COMPLETED)
--
Gitblit v1.8.0