Skip to content

Commit 70d8a45

Browse files
committed
chore: 增加 Token 刷新失败后,重新获取
1 parent 2af48a5 commit 70d8a45

File tree

2 files changed

+22
-17
lines changed

2 files changed

+22
-17
lines changed

core/cluster.py

Lines changed: 21 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -98,24 +98,28 @@ async def _schedule_refresh_token(self, ttl: float):
9898
logger.ttraceback("cluster.refresh_token", id=self._id, name=self.display_name)
9999

100100
async def refresh_token(self):
101-
async with aiohttp.ClientSession(
102-
base_url=cfg.base_url,
103-
headers={
104-
"Authorization": f"Bearer {self._token}",
105-
"User-Agent": USER_AGENT,
106-
}
107-
) as session:
108-
async with session.post(
109-
"/openbmclapi-agent/token",
110-
json={
111-
"clusterId": self._id,
112-
"token": self._token
101+
try:
102+
async with aiohttp.ClientSession(
103+
base_url=cfg.base_url,
104+
headers={
105+
"Authorization": f"Bearer {self._token}",
106+
"User-Agent": USER_AGENT,
113107
}
114-
) as resp:
115-
data = await resp.json()
116-
self._token = data['token']
117-
ttl = data['ttl'] / 1000.0
118-
self.schedule_refresh_token(ttl)
108+
) as session:
109+
async with session.post(
110+
"/openbmclapi-agent/token",
111+
json={
112+
"clusterId": self._id,
113+
"token": self._token
114+
}
115+
) as resp:
116+
data = await resp.json()
117+
self._token = data['token']
118+
ttl = data['ttl'] / 1000.0
119+
self.schedule_refresh_token(ttl)
120+
except:
121+
logger.ttraceback("cluster.refresh_token", id=self._id, name=self.display_name)
122+
await self.fetch_token()
119123

120124
async def get_socketio_token(self):
121125
return {

locale/zh_cn.json

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
"debug.web.uvicorn.port": "Web 服务器监听端口 [${port}]",
66
"error.cluster.enable": "节点 [${name} (${id})] 上线失败 [${err}]",
77
"traceback.cluster.enable.timeout": "节点 [${name} (${id})] 注册超时",
8+
"traceback.cluster.refresh_token": "节点 [${name} (${id})] Token 刷新失败",
89
"error.cluster.exception": "节点 [${name} (${id})] 异常:[${msg}]",
910
"error.cluster.kicked": "节点 [${name} (${id})] 被主控踢出下线",
1011
"error.cluster.refresh_token": "刷新节点 [${name} (${id})] Token 失败",

0 commit comments

Comments
 (0)