代码如下:
addEventListener('fetch', event => { event.respondWith(handleRequest(event.request)) }) async function handleRequest(request) { if (request.headers.get('Upgrade') !== 'websocket') { return new Response('Expected Upgrade: websocket', { status: 426 }) } const targetUrl = 'wss://openai websocket 的api网关地址' const [client, server] = Object.values(new WebSocketPair()) await server.accept() const targetWebSocket = new WebSocket(targetUrl) targetWebSocket.addEventListener('open', () => { console.log('Connected to target WebSocket server') }) server.addEventListener('message', event => { console.log('Message from client:', event.data) if (targetWebSocket.readyState === WebSocket.OPEN) { targetWebSocket.send(event.data) } }) targetWebSocket.addEventListener('message', event => { console.log('Message from target:', event.data) if (server.readyState === WebSocket.OPEN) { server.send(event.data) } }) server.addEventListener('close', () => { console.log('Client connection closed') targetWebSocket.close() }) targetWebSocket.addEventListener('close', () => { console.log('Target connection closed') server.close() }) // 错误处理 server.addEventListener('error', error => { console.error('Server WebSocket error:', error) }) targetWebSocket.addEventListener('error', error => { console.error('Target WebSocket error:', error) }) // 保持连接活跃 const keepAlive = setInterval(() => { if (server.readyState === WebSocket.OPEN) { server.send('ping') } }, 25000) server.addEventListener('close', () => { clearInterval(keepAlive) }) return new Response(null, { status: 101, webSocket: client, }) }
网友回复