严浩
90 lines
2.9 KiB
TypeScript
90 lines
2.9 KiB
TypeScript
export default {
|
|
async fetch(request, env) {
|
|
const url = new URL(request.url);
|
|
|
|
// 本地开发环境延迟处理
|
|
if (url.hostname === 'localhost' || url.hostname === '127.0.0.1') {
|
|
await new Promise((r) => setTimeout(r, 250));
|
|
}
|
|
|
|
// API 路由处理
|
|
if (url.pathname.startsWith('/api/')) {
|
|
await env.KV.put(
|
|
'events:api:last-call',
|
|
`${new Date().toISOString()} ${request.method} ${url.pathname}`,
|
|
);
|
|
|
|
// 获取所有可用的键名
|
|
const availableKeys = [
|
|
'events:api:last-call',
|
|
'events:ws:connection',
|
|
'events:ws:message',
|
|
'events:ws:disconnection',
|
|
];
|
|
|
|
return Response.json({
|
|
timestamp: Date.now(),
|
|
lastApiCall: await env.KV.get('events:api:last-call'),
|
|
availableKeys: availableKeys,
|
|
storedData: {
|
|
apiLastCall: await env.KV.get('events:api:last-call'),
|
|
wsConnection: await env.KV.get('events:ws:connection'),
|
|
wsMessage: await env.KV.get('events:ws:message'),
|
|
wsDisconnection: await env.KV.get('events:ws:disconnection'),
|
|
},
|
|
});
|
|
}
|
|
|
|
// WebSocket 连接处理
|
|
if (url.pathname === '/ws') {
|
|
const upgradeHeader = request.headers.get('Upgrade');
|
|
if (upgradeHeader !== 'websocket') {
|
|
return new Response('Expected websocket', { status: 400 });
|
|
}
|
|
|
|
const [client, server] = Object.values(new WebSocketPair());
|
|
|
|
// 处理服务器端WebSocket消息
|
|
server.accept();
|
|
env.KV.put('events:ws:connection', `${new Date().toISOString()} ${url.pathname}`);
|
|
|
|
// accept 后立即发送欢迎消息
|
|
if (url.hostname === 'localhost' || url.hostname === '127.0.0.1') {
|
|
await new Promise((r) => setTimeout(r, 250));
|
|
}
|
|
|
|
server.send(
|
|
`欢迎连接到WebSocket服务器！连接时间: ${new Date().toLocaleString('zh-CN', { timeZone: 'Asia/Shanghai' })}`,
|
|
);
|
|
|
|
server.addEventListener('message', async (event) => {
|
|
console.log('收到客户端消息:', event.data);
|
|
await env.KV.put('events:ws:message', `${new Date().toISOString()} ${event.data}`);
|
|
|
|
// 回复消息给客户端
|
|
if (url.hostname === 'localhost' || url.hostname === '127.0.0.1') {
|
|
await new Promise((r) => setTimeout(r, 250));
|
|
}
|
|
|
|
server.send(
|
|
`服务器收到: ${event.data} (时间: ${new Date().toLocaleString('zh-CN', { timeZone: 'Asia/Shanghai' })})`,
|
|
);
|
|
});
|
|
|
|
server.addEventListener('close', (event) => {
|
|
console.log('WebSocket连接关闭');
|
|
console.log('[close] event :>> ', event);
|
|
env.KV.put('events:ws:disconnection', `${new Date().toISOString()} ${url.pathname}`);
|
|
server.close(event.code, `连接关闭: ${event.reason}`);
|
|
});
|
|
|
|
return new Response(null, {
|
|
status: 101,
|
|
webSocket: client,
|
|
});
|
|
}
|
|
|
|
return new Response(null, { status: 404 });
|
|
},
|
|
} satisfies ExportedHandler<Env>;
|