添加 @kevisual/api 依赖,更新 WebSocket 消息发送逻辑,支持上下文参数

This commit is contained in:
2026-02-05 04:06:34 +08:00
parent 7bbefd8a4a
commit a51d04341e
5 changed files with 46 additions and 3 deletions

View File

@@ -2,6 +2,7 @@ import { nanoid } from 'nanoid';
import { WebSocket } from 'ws';
import { logger } from '../logger.ts';
import { EventEmitter } from 'eventemitter3';
class WsMessage {
ws: WebSocket;
user?: string;
@@ -16,7 +17,7 @@ class WsMessage {
this.emitter.emit(data.id, data?.data);
}
}
async sendData(data: any, opts?: { timeout?: number }) {
async sendData(data: any, context?: any, opts?: { timeout?: number }) {
if (this.ws.readyState !== WebSocket.OPEN) {
return { code: 500, message: 'WebSocket is not open' };
}
@@ -25,7 +26,10 @@ class WsMessage {
const message = JSON.stringify({
id,
type: 'proxy',
data,
data: {
message: data,
context: context || {},
},
});
logger.info('ws-proxy sendData', message);
this.ws.send(message);