122 lines
3.5 KiB
TypeScript
122 lines
3.5 KiB
TypeScript
import { createSkill, tool } from '@kevisual/router';
|
||
import { app, cnb } from '../../app.ts';
|
||
|
||
import { createKeepAlive } from '../../../src/keep.ts';
|
||
|
||
type AliveInfo = {
|
||
startTime: number;
|
||
updatedTime?: number;
|
||
KeepAlive: ReturnType<typeof createKeepAlive>;
|
||
}
|
||
|
||
const keepAliveMap = new Map<string, AliveInfo>();
|
||
|
||
// 保持工作空间存活技能
|
||
app.route({
|
||
path: 'cnb',
|
||
key: 'keep-workspace-alive',
|
||
description: '保持工作空间存活技能,参数wsUrl:工作空间访问URL，cookie:访问工作空间所需的cookie',
|
||
middleware: ['auth'],
|
||
metadata: {
|
||
tags: [],
|
||
...({
|
||
args: {
|
||
wsUrl: tool.schema.string().describe('工作空间的访问URL'),
|
||
cookie: tool.schema.string().describe('访问工作空间所需的cookie')
|
||
}
|
||
})
|
||
}
|
||
}).define(async (ctx) => {
|
||
const wsUrl = ctx.query?.wsUrl as string;
|
||
const cookie = ctx.query?.cookie as string;
|
||
if (!wsUrl) {
|
||
ctx.throw(400, '缺少工作空间访问URL参数');
|
||
}
|
||
if (!cookie) {
|
||
ctx.throw(400, '缺少访问工作空间所需的cookie参数');
|
||
}
|
||
|
||
// 检测是否已在运行
|
||
const existing = keepAliveMap.get(wsUrl);
|
||
if (existing) {
|
||
ctx.body = { message: `工作空间 ${wsUrl} 的保持存活任务已在运行中` };
|
||
return;
|
||
}
|
||
|
||
console.log(`启动保持工作空间 ${wsUrl} 存活的任务`);
|
||
const keep = createKeepAlive({
|
||
wsUrl,
|
||
cookie,
|
||
onConnect: () => {
|
||
console.log(`工作空间 ${wsUrl} 保持存活任务已连接`);
|
||
},
|
||
onMessage: (data) => {
|
||
// 可选：处理收到的消息
|
||
// console.log(`工作空间 ${wsUrl} 收到消息: ${data}`);
|
||
const aliveInfo = keepAliveMap.get(wsUrl);
|
||
if (aliveInfo) {
|
||
aliveInfo.updatedTime = Date.now();
|
||
}
|
||
},
|
||
debug: true,
|
||
onExit: (code) => {
|
||
console.log(`工作空间 ${wsUrl} 保持存活任务已退出，退出码: ${code}`);
|
||
keepAliveMap.delete(wsUrl);
|
||
}
|
||
});
|
||
|
||
keepAliveMap.set(wsUrl, { startTime: Date.now(), updatedTime: Date.now(), KeepAlive: keep });
|
||
|
||
ctx.body = { message: `已启动保持工作空间 ${wsUrl} 存活的任务` };
|
||
}).addTo(app);
|
||
|
||
// 获取保持工作空间存活任务列表技能
|
||
app.route({
|
||
path: 'cnb',
|
||
key: 'list-keep-alive-tasks',
|
||
description: '获取保持工作空间存活任务列表技能',
|
||
middleware: ['auth'],
|
||
metadata: {
|
||
tags: [],
|
||
}
|
||
}).define(async (ctx) => {
|
||
const list = Array.from(keepAliveMap.entries()).map(([wsUrl, info]) => ({
|
||
wsUrl,
|
||
startTime: info.startTime,
|
||
updatedTime: info.updatedTime
|
||
}));
|
||
ctx.body = { list };
|
||
}).addTo(app);
|
||
|
||
// 停止保持工作空间存活技能
|
||
app.route({
|
||
path: 'cnb',
|
||
key: 'stop-keep-workspace-alive',
|
||
description: '停止保持工作空间存活技能, 参数wsUrl:工作空间访问URL',
|
||
middleware: ['auth'],
|
||
metadata: {
|
||
tags: [],
|
||
...({
|
||
args: {
|
||
wsUrl: tool.schema.string().describe('工作空间的访问URL'),
|
||
}
|
||
})
|
||
}
|
||
}).define(async (ctx) => {
|
||
const wsUrl = ctx.query?.wsUrl as string;
|
||
if (!wsUrl) {
|
||
ctx.throw(400, '缺少工作空间访问URL参数');
|
||
}
|
||
|
||
const keepAlive = keepAliveMap.get(wsUrl);
|
||
if (keepAlive) {
|
||
const endTime = Date.now();
|
||
const duration = endTime - keepAlive.startTime;
|
||
keepAlive?.KeepAlive?.disconnect();
|
||
keepAliveMap.delete(wsUrl);
|
||
ctx.body = { message: `已停止保持工作空间 ${wsUrl} 存活的任务，持续时间: ${duration}ms` };
|
||
} else {
|
||
ctx.body = { message: `没有找到工作空间 ${wsUrl} 的保持存活任务` };
|
||
}
|
||
}).addTo(app);
|