Megghy/vtsuru.live
1
0
Fork 0
mirror of https://github.com/Megghy/vtsuru.live.git synced 2025-12-06 18:36:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

并发控制

Browse Source
This commit is contained in:
Megghy
2025-04-26 06:19:21 +08:00
parent 1276cc0381
commit 1bf2a40516
1 changed files with 82 additions and 79 deletions
Download Patch File Download Diff File Expand all files Collapse all files

161
src/store/useWebFetcher.ts
View File

@@ -211,91 +211,94 @@ export const useWebFetcher = defineStore('WebFetcher', () => {
/** /**
* 连接 SignalR 服务器 * 连接 SignalR 服务器
*/ */
async function connectSignalR() { async function connectSignalR(): Promise<boolean> {
if (signalRClient.value && signalRClient.value.state !== signalR.HubConnectionState.Disconnected) { return await navigator.locks.request('webFetcherStartLock', async () => {
console.log(prefix.value + "SignalR 已连接或正在连接"); if (signalRClient.value && signalRClient.value.state !== signalR.HubConnectionState.Disconnected) {
return true; console.log(prefix.value + "SignalR 已连接或正在连接");
} return true;
signalRClient.value?.stop(); }
signalRClient.value = undefined; signalRClient.value?.stop();
signalRConnectionId.value = undefined; signalRClient.value = undefined;
console.log(prefix.value + '正在连接到 vtsuru 服务器...'); signalRConnectionId.value = undefined;
const connection = new signalR.HubConnectionBuilder() console.log(prefix.value + '正在连接到 vtsuru 服务器...');
.withUrl(BASE_HUB_URL + 'web-fetcher?token=' + (route.query.token ?? account.value.token), { // 使用 account.token const connection = new signalR.HubConnectionBuilder()
headers: { Authorization: `Bearer ${cookie.value?.cookie}` }, .withUrl(BASE_HUB_URL + 'web-fetcher?token=' + (route.query.token ?? account.value.token), { // 使用 account.token
skipNegotiation: true, headers: { Authorization: `Bearer ${cookie.value?.cookie}` },
transport: signalR.HttpTransportType.WebSockets skipNegotiation: true,
}) transport: signalR.HttpTransportType.WebSockets
.withAutomaticReconnect({ })
nextRetryDelayInMilliseconds: retryContext => { .withAutomaticReconnect({
return retryContext.elapsedMilliseconds < 60 * 1000 ? 10 * 1000 : 30 * 1000; nextRetryDelayInMilliseconds: retryContext => {
} return retryContext.elapsedMilliseconds < 60 * 1000 ? 10 * 1000 : 30 * 1000;
}) // 自动重连策略 }
.withHubProtocol(new msgpack.MessagePackHubProtocol()) // 使用 MessagePack 协议 }) // 自动重连策略
.build(); .withHubProtocol(new msgpack.MessagePackHubProtocol()) // 使用 MessagePack 协议
.build();
// --- SignalR 事件监听 --- connection.on('Disconnect', (reason: unknown) => {
connection.onreconnecting(error => { console.log(prefix.value + '被服务器断开连接: ' + reason);
console.log(prefix.value + `与服务器断开，正在尝试重连... ${error?.message || ''}`); disconnectedByServer = true; // 标记是服务器主动断开
state.value = 'connecting'; // 更新状态为连接中 window.$message.error(`被服务器要求断开连接: ${reason}`);
signalRConnectionId.value = undefined; // 连接断开，ID失效 });
}); // --- 尝试启动连接 ---
try {
await connection.start();
signalRConnectionId.value = connection.connectionId ?? undefined; // 保存连接ID
signalRId.value = await sendSelfInfo(connection); // 发送客户端信息
await connection.send('Finished'); // 通知服务器已准备好
signalRClient.value = connection; // 保存实例
connection.onreconnected(async connectionId => { // --- SignalR 事件监听 ---
console.log(prefix.value + `与服务器重新连接成功! ConnectionId: ${connectionId}`); connection.onreconnecting(error => {
signalRConnectionId.value = connectionId ?? undefined; console.log(prefix.value + `与服务器断开，正在尝试重连... ${error?.message || ''}`);
state.value = 'connected'; // 更新状态为已连接 state.value = 'connecting'; // 更新状态为连接中
signalRId.value = connectionId ?? await sendSelfInfo(connection); // 更新连接ID signalRConnectionId.value = undefined; // 连接断开，ID失效
connection.send('Reconnected').catch(err => console.error(prefix.value + "Send Reconnected failed: " + err)); });
});
connection.onclose(async (error) => { connection.onreconnected(async connectionId => {
// 只有在不是由 Stop() 或服务器明确要求断开时才记录错误并尝试独立重连（虽然 withAutomaticReconnect 应该处理） console.log(prefix.value + `与服务器重新连接成功! ConnectionId: ${connectionId}`);
if (state.value !== 'disconnected' && !disconnectedByServer) { signalRConnectionId.value = connectionId ?? undefined;
console.error(prefix.value + `与服务器连接关闭: ${error?.message || '未知原因'}. 30秒后将自动重启`); state.value = 'connected'; // 更新状态为已连接
//state.value = 'connecting'; // 标记为连接中，等待自动重连 signalRId.value = connectionId ?? await sendSelfInfo(connection); // 更新连接ID
//signalRConnectionId.value = undefined; connection.send('Reconnected').catch(err => console.error(prefix.value + "Send Reconnected failed: " + err));
//await connection.start(); });
// 停止 SignalR 连接
signalRClient.value?.stop(); connection.onclose(async (error) => {
signalRClient.value = undefined; // 只有在不是由 Stop() 或服务器明确要求断开时才记录错误并尝试独立重连（虽然 withAutomaticReconnect 应该处理）
if (state.value !== 'disconnected' && !disconnectedByServer) {
console.error(prefix.value + `与服务器连接关闭: ${error?.message || '未知原因'}. 30秒后将自动重启`);
//state.value = 'connecting'; // 标记为连接中，等待自动重连
//signalRConnectionId.value = undefined;
//await connection.start();
// 停止 SignalR 连接
signalRClient.value?.stop();
signalRClient.value = undefined;
signalRConnectionId.value = undefined;
setTimeout(() => {
console.log(prefix.value + '尝试重启...');
connectSignalR(); // 30秒后尝试重启
}, 30 * 1000); // 30秒后自动重启
} else if (disconnectedByServer) {
console.log(prefix.value + `连接已被服务器关闭.`);
//Stop(); // 服务器要求断开，则彻底停止
} else {
console.log(prefix.value + `连接已手动关闭.`);
}
});
connection.on('Request', async (url: string, method: string, body: string, useCookie: boolean) => onRequest(url, method, body, useCookie));
connection.on('Notification', (type: string, data: any) => { onReceivedNotification(type, data); });
console.log(prefix.value + '已连接到 vtsuru 服务器, ConnectionId: ' + signalRId.value); // 调试输出连接状态
// state.value = 'connected'; // 状态将在 Start 函数末尾统一设置
return true;
} catch (e) {
console.error(prefix.value + '无法连接到 vtsuru 服务器: ' + e);
signalRConnectionId.value = undefined; signalRConnectionId.value = undefined;
setTimeout(() => { signalRClient.value = undefined;
console.log(prefix.value + '尝试重启...'); // state.value = 'disconnected'; // 保持 connecting 或由 Start 控制
connectSignalR(); // 30秒后尝试重启 return false;
}, 30 * 1000); // 30秒后自动重启
} else if (disconnectedByServer) {
console.log(prefix.value + `连接已被服务器关闭.`);
//Stop(); // 服务器要求断开，则彻底停止
} else {
console.log(prefix.value + `连接已手动关闭.`);
} }
}); });
connection.on('Disconnect', (reason: unknown) => {
console.log(prefix.value + '被服务器断开连接: ' + reason);
disconnectedByServer = true; // 标记是服务器主动断开
window.$message.error(`被服务器要求断开连接: ${reason}`);
});
connection.on('Request', async (url: string, method: string, body: string, useCookie: boolean) => onRequest(url, method, body, useCookie));
connection.on('Notification', (type: string, data: any) => { onReceivedNotification(type, data); });
// --- 尝试启动连接 ---
try {
await connection.start();
signalRConnectionId.value = connection.connectionId ?? undefined; // 保存连接ID
signalRId.value = await sendSelfInfo(connection); // 发送客户端信息
await connection.send('Finished'); // 通知服务器已准备好
signalRClient.value = connection; // 保存实例
console.log(prefix.value + '已连接到 vtsuru 服务器, ConnectionId: ' + signalRId.value); // 调试输出连接状态
// state.value = 'connected'; // 状态将在 Start 函数末尾统一设置
return true;
} catch (e) {
console.error(prefix.value + '无法连接到 vtsuru 服务器: ' + e);
signalRConnectionId.value = undefined;
signalRClient.value = undefined;
// state.value = 'disconnected'; // 保持 connecting 或由 Start 控制
return false;
}
} }
async function sendSelfInfo(client: signalR.HubConnection) { async function sendSelfInfo(client: signalR.HubConnection) {
return client.invoke('SetSelfInfo', return client.invoke('SetSelfInfo',