aws-runtime-bridge 1.1.1 → 1.1.2

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (97) hide show
  1. package/package/aws-client-agent-mcp/dist/agent-client.d.ts +166 -0
  2. package/package/aws-client-agent-mcp/dist/agent-client.d.ts.map +1 -0
  3. package/package/aws-client-agent-mcp/dist/agent-client.js +602 -0
  4. package/package/aws-client-agent-mcp/dist/agent-client.js.map +1 -0
  5. package/package/aws-client-agent-mcp/dist/agent-client.test.d.ts +2 -0
  6. package/package/aws-client-agent-mcp/dist/agent-client.test.d.ts.map +1 -0
  7. package/package/aws-client-agent-mcp/dist/agent-client.test.js +534 -0
  8. package/package/aws-client-agent-mcp/dist/agent-client.test.js.map +1 -0
  9. package/package/aws-client-agent-mcp/dist/config.d.ts +22 -0
  10. package/package/aws-client-agent-mcp/dist/config.d.ts.map +1 -0
  11. package/package/aws-client-agent-mcp/dist/config.js +106 -0
  12. package/package/aws-client-agent-mcp/dist/config.js.map +1 -0
  13. package/package/aws-client-agent-mcp/dist/config.test.d.ts +2 -0
  14. package/package/aws-client-agent-mcp/dist/config.test.d.ts.map +1 -0
  15. package/package/aws-client-agent-mcp/dist/config.test.js +139 -0
  16. package/package/aws-client-agent-mcp/dist/config.test.js.map +1 -0
  17. package/package/aws-client-agent-mcp/dist/constants.d.ts +15 -0
  18. package/package/aws-client-agent-mcp/dist/constants.d.ts.map +1 -0
  19. package/package/aws-client-agent-mcp/dist/constants.js +19 -0
  20. package/package/aws-client-agent-mcp/dist/constants.js.map +1 -0
  21. package/package/aws-client-agent-mcp/dist/http-client.d.ts +28 -0
  22. package/package/aws-client-agent-mcp/dist/http-client.d.ts.map +1 -0
  23. package/package/aws-client-agent-mcp/dist/http-client.js +96 -0
  24. package/package/aws-client-agent-mcp/dist/http-client.js.map +1 -0
  25. package/package/aws-client-agent-mcp/dist/http-client.test.d.ts +2 -0
  26. package/package/aws-client-agent-mcp/dist/http-client.test.d.ts.map +1 -0
  27. package/package/aws-client-agent-mcp/dist/http-client.test.js +228 -0
  28. package/package/aws-client-agent-mcp/dist/http-client.test.js.map +1 -0
  29. package/package/aws-client-agent-mcp/dist/index.d.ts +14 -0
  30. package/package/aws-client-agent-mcp/dist/index.d.ts.map +1 -0
  31. package/package/aws-client-agent-mcp/dist/index.js +30 -0
  32. package/package/aws-client-agent-mcp/dist/index.js.map +1 -0
  33. package/package/aws-client-agent-mcp/dist/logger.d.ts +7 -0
  34. package/package/aws-client-agent-mcp/dist/logger.d.ts.map +1 -0
  35. package/package/aws-client-agent-mcp/dist/logger.js +19 -0
  36. package/package/aws-client-agent-mcp/dist/logger.js.map +1 -0
  37. package/package/aws-client-agent-mcp/dist/mcp-server.d.ts +77 -0
  38. package/package/aws-client-agent-mcp/dist/mcp-server.d.ts.map +1 -0
  39. package/package/aws-client-agent-mcp/dist/mcp-server.js +438 -0
  40. package/package/aws-client-agent-mcp/dist/mcp-server.js.map +1 -0
  41. package/package/aws-client-agent-mcp/dist/mcp-server.test.d.ts +2 -0
  42. package/package/aws-client-agent-mcp/dist/mcp-server.test.d.ts.map +1 -0
  43. package/package/aws-client-agent-mcp/dist/mcp-server.test.js +624 -0
  44. package/package/aws-client-agent-mcp/dist/mcp-server.test.js.map +1 -0
  45. package/package/aws-client-agent-mcp/dist/mcp-tools.d.ts +78 -0
  46. package/package/aws-client-agent-mcp/dist/mcp-tools.d.ts.map +1 -0
  47. package/package/aws-client-agent-mcp/dist/mcp-tools.js +420 -0
  48. package/package/aws-client-agent-mcp/dist/mcp-tools.js.map +1 -0
  49. package/package/aws-client-agent-mcp/dist/memory-store.d.ts +61 -0
  50. package/package/aws-client-agent-mcp/dist/memory-store.d.ts.map +1 -0
  51. package/package/aws-client-agent-mcp/dist/memory-store.js +268 -0
  52. package/package/aws-client-agent-mcp/dist/memory-store.js.map +1 -0
  53. package/package/aws-client-agent-mcp/dist/memory-store.test.d.ts +2 -0
  54. package/package/aws-client-agent-mcp/dist/memory-store.test.d.ts.map +1 -0
  55. package/package/aws-client-agent-mcp/dist/memory-store.test.js +164 -0
  56. package/package/aws-client-agent-mcp/dist/memory-store.test.js.map +1 -0
  57. package/package/aws-client-agent-mcp/dist/message-buffer.d.ts +87 -0
  58. package/package/aws-client-agent-mcp/dist/message-buffer.d.ts.map +1 -0
  59. package/package/aws-client-agent-mcp/dist/message-buffer.js +185 -0
  60. package/package/aws-client-agent-mcp/dist/message-buffer.js.map +1 -0
  61. package/package/aws-client-agent-mcp/dist/message-buffer.test.d.ts +2 -0
  62. package/package/aws-client-agent-mcp/dist/message-buffer.test.d.ts.map +1 -0
  63. package/package/aws-client-agent-mcp/dist/message-buffer.test.js +44 -0
  64. package/package/aws-client-agent-mcp/dist/message-buffer.test.js.map +1 -0
  65. package/package/aws-client-agent-mcp/dist/messageContent.d.ts +53 -0
  66. package/package/aws-client-agent-mcp/dist/messageContent.d.ts.map +1 -0
  67. package/package/aws-client-agent-mcp/dist/messageContent.js +125 -0
  68. package/package/aws-client-agent-mcp/dist/messageContent.js.map +1 -0
  69. package/package/aws-client-agent-mcp/dist/runtime-launch-binding.d.ts +3 -0
  70. package/package/aws-client-agent-mcp/dist/runtime-launch-binding.d.ts.map +1 -0
  71. package/package/aws-client-agent-mcp/dist/runtime-launch-binding.js +50 -0
  72. package/package/aws-client-agent-mcp/dist/runtime-launch-binding.js.map +1 -0
  73. package/package/aws-client-agent-mcp/dist/runtime-launch-binding.test.d.ts +2 -0
  74. package/package/aws-client-agent-mcp/dist/runtime-launch-binding.test.d.ts.map +1 -0
  75. package/package/aws-client-agent-mcp/dist/runtime-launch-binding.test.js +90 -0
  76. package/package/aws-client-agent-mcp/dist/runtime-launch-binding.test.js.map +1 -0
  77. package/package/aws-client-agent-mcp/dist/status-reporter.d.ts +66 -0
  78. package/package/aws-client-agent-mcp/dist/status-reporter.d.ts.map +1 -0
  79. package/package/aws-client-agent-mcp/dist/status-reporter.js +237 -0
  80. package/package/aws-client-agent-mcp/dist/status-reporter.js.map +1 -0
  81. package/package/aws-client-agent-mcp/dist/status-reporter.test.d.ts +2 -0
  82. package/package/aws-client-agent-mcp/dist/status-reporter.test.d.ts.map +1 -0
  83. package/package/aws-client-agent-mcp/dist/status-reporter.test.js +19 -0
  84. package/package/aws-client-agent-mcp/dist/status-reporter.test.js.map +1 -0
  85. package/package/aws-client-agent-mcp/dist/types.d.ts +309 -0
  86. package/package/aws-client-agent-mcp/dist/types.d.ts.map +1 -0
  87. package/package/aws-client-agent-mcp/dist/types.js +9 -0
  88. package/package/aws-client-agent-mcp/dist/types.js.map +1 -0
  89. package/package/aws-client-agent-mcp/dist/websocket-client.d.ts +94 -0
  90. package/package/aws-client-agent-mcp/dist/websocket-client.d.ts.map +1 -0
  91. package/package/aws-client-agent-mcp/dist/websocket-client.js +316 -0
  92. package/package/aws-client-agent-mcp/dist/websocket-client.js.map +1 -0
  93. package/package/aws-client-agent-mcp/dist/websocket-client.test.d.ts +2 -0
  94. package/package/aws-client-agent-mcp/dist/websocket-client.test.d.ts.map +1 -0
  95. package/package/aws-client-agent-mcp/dist/websocket-client.test.js +191 -0
  96. package/package/aws-client-agent-mcp/dist/websocket-client.test.js.map +1 -0
  97. package/package.json +1 -1
@@ -0,0 +1,9 @@
1
+ /**
2
+ * Agent MCP Client 类型定义
3
+ */
4
+ /**
5
+ * PROFILE_INFO 消息类型
6
+ * 用于在 Agent 连接时自动推送 profile 信息
7
+ */
8
+ export const TYPE_PROFILE_INFO = "PROFILE_INFO";
9
+ //# sourceMappingURL=types.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"types.js","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA;;GAEG;AAyLH;;;GAGG;AACH,MAAM,CAAC,MAAM,iBAAiB,GAAG,cAAc,CAAC"}
@@ -0,0 +1,94 @@
1
+ /**
2
+ * WebSocket 客户端模块
3
+ * 负责与调度中心建立和维护 WebSocket 连接
4
+ */
5
+ import WebSocket from 'ws';
6
+ import { EventEmitter } from 'node:events';
7
+ import type { ServerConfig } from './types.js';
8
+ export declare class WebSocketClient extends EventEmitter {
9
+ private ws;
10
+ private config;
11
+ private agentId;
12
+ private heartbeatTimer;
13
+ private heartbeatTimeoutTimer;
14
+ private reconnectAttempts;
15
+ private reconnectTimer;
16
+ private isManualClose;
17
+ private websocketFactory;
18
+ constructor(config: ServerConfig, websocketFactory?: (url: string) => WebSocket);
19
+ /**
20
+ * 建立 WebSocket 连接
21
+ * 主流程:准备连接 -> 创建连接 -> 设置事件监听
22
+ */
23
+ connect(agentId: string): Promise<void>;
24
+ /**
25
+ * 准备连接前的状态清理
26
+ */
27
+ private prepareConnection;
28
+ /**
29
+ * 构建 WebSocket URL
30
+ */
31
+ private buildWebSocketUrl;
32
+ /**
33
+ * 创建一次性 resolve/reject 处理器
34
+ */
35
+ private createSettledHandlers;
36
+ /**
37
+ * 设置 WebSocket 事件监听器
38
+ */
39
+ private setupWebSocketEventHandlers;
40
+ /**
41
+ * 处理 WebSocket 打开事件
42
+ */
43
+ private handleWebSocketOpen;
44
+ /**
45
+ * 处理 WebSocket 消息
46
+ */
47
+ private handleWebSocketMessage;
48
+ /**
49
+ * 处理 WebSocket 关闭事件
50
+ */
51
+ private handleWebSocketClose;
52
+ /**
53
+ * 处理 WebSocket 错误事件
54
+ */
55
+ private handleWebSocketError;
56
+ /**
57
+ * 断开 WebSocket 连接
58
+ */
59
+ disconnect(): void;
60
+ /**
61
+ * 发送消息
62
+ */
63
+ send(message: unknown): boolean;
64
+ /**
65
+ * 是否已连接
66
+ */
67
+ isConnected(): boolean;
68
+ /**
69
+ * 处理接收到的消息
70
+ */
71
+ private handleMessage;
72
+ /**
73
+ * 启动心跳
74
+ */
75
+ private startHeartbeat;
76
+ /**
77
+ * 停止心跳
78
+ */
79
+ private stopHeartbeat;
80
+ private sendProtocolPing;
81
+ private markHeartbeatReceived;
82
+ private ensureHeartbeatTimeout;
83
+ private clearHeartbeatTimeout;
84
+ private resolveHeartbeatTimeout;
85
+ /**
86
+ * 安排重连
87
+ */
88
+ private scheduleReconnect;
89
+ /**
90
+ * 清除重连定时器
91
+ */
92
+ private clearReconnectTimer;
93
+ }
94
+ //# sourceMappingURL=websocket-client.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"websocket-client.d.ts","sourceRoot":"","sources":["../src/websocket-client.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,SAAS,MAAM,IAAI,CAAC;AAC3B,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAC3C,OAAO,KAAK,EAAE,YAAY,EAAoB,MAAM,YAAY,CAAC;AAKjE,qBAAa,eAAgB,SAAQ,YAAY;IAC/C,OAAO,CAAC,EAAE,CAA0B;IACpC,OAAO,CAAC,MAAM,CAAe;IAC7B,OAAO,CAAC,OAAO,CAAuB;IACtC,OAAO,CAAC,cAAc,CAA+B;IACrD,OAAO,CAAC,qBAAqB,CAA+B;IAC5D,OAAO,CAAC,iBAAiB,CAAK;IAC9B,OAAO,CAAC,cAAc,CAA+B;IACrD,OAAO,CAAC,aAAa,CAAS;IAC9B,OAAO,CAAC,gBAAgB,CAA6B;gBAEzC,MAAM,EAAE,YAAY,EAAE,gBAAgB,CAAC,EAAE,CAAC,GAAG,EAAE,MAAM,KAAK,SAAS;IAMjF;;;SAGK;IACH,OAAO,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAkBvC;;OAEG;IACH,OAAO,CAAC,iBAAiB;IAUzB;;OAEG;IACH,OAAO,CAAC,iBAAiB;IAIzB;;OAEG;IACH,OAAO,CAAC,qBAAqB;IAuB7B;;OAEG;IACH,OAAO,CAAC,2BAA2B;IAuBnC;;OAEG;IACH,OAAO,CAAC,mBAAmB;IAS3B;;OAEG;IACH,OAAO,CAAC,sBAAsB;IAS9B;;OAEG;IACH,OAAO,CAAC,oBAAoB;IAoB5B;;OAEG;IACH,OAAO,CAAC,oBAAoB;IAY5B;;OAEG;IACH,UAAU,IAAI,IAAI;IAiBlB;;OAEG;IACH,IAAI,CAAC,OAAO,EAAE,OAAO,GAAG,OAAO;IAe/B;;OAEG;IACH,WAAW,IAAI,OAAO;IAItB;;OAEG;IACH,OAAO,CAAC,aAAa;IA2BrB;;OAEG;IACH,OAAO,CAAC,cAAc;IAgBtB;;OAEG;IACH,OAAO,CAAC,aAAa;IAQrB,OAAO,CAAC,gBAAgB;IAaxB,OAAO,CAAC,qBAAqB;IAI7B,OAAO,CAAC,sBAAsB;IAgB9B,OAAO,CAAC,qBAAqB;IAO7B,OAAO,CAAC,uBAAuB;IAI/B;;OAEG;IACH,OAAO,CAAC,iBAAiB;IAmCzB;;OAEG;IACH,OAAO,CAAC,mBAAmB;CAM5B"}
@@ -0,0 +1,316 @@
1
+ /**
2
+ * WebSocket 客户端模块
3
+ * 负责与调度中心建立和维护 WebSocket 连接
4
+ */
5
+ import WebSocket from 'ws';
6
+ import { EventEmitter } from 'node:events';
7
+ import { logger } from './logger.js';
8
+ const DEFAULT_HEARTBEAT_TIMEOUT_MS = 9000;
9
+ export class WebSocketClient extends EventEmitter {
10
+ ws = null;
11
+ config;
12
+ agentId = null;
13
+ heartbeatTimer = null;
14
+ heartbeatTimeoutTimer = null;
15
+ reconnectAttempts = 0;
16
+ reconnectTimer = null;
17
+ isManualClose = false;
18
+ websocketFactory;
19
+ constructor(config, websocketFactory) {
20
+ super();
21
+ this.config = config;
22
+ this.websocketFactory = websocketFactory ?? ((url) => new WebSocket(url));
23
+ }
24
+ /**
25
+ * 建立 WebSocket 连接
26
+ * 主流程:准备连接 -> 创建连接 -> 设置事件监听
27
+ */
28
+ connect(agentId) {
29
+ return new Promise((resolve, reject) => {
30
+ this.prepareConnection(agentId);
31
+ const wsUrl = this.buildWebSocketUrl(agentId);
32
+ logger.info(`[WebSocket] 正在连接: ${wsUrl}`);
33
+ const { rejectOnce, resolveOnce } = this.createSettledHandlers(resolve, reject);
34
+ try {
35
+ this.ws = this.websocketFactory(wsUrl);
36
+ this.setupWebSocketEventHandlers(this.ws, rejectOnce, resolveOnce);
37
+ }
38
+ catch (error) {
39
+ rejectOnce(error instanceof Error ? error : new Error(String(error)));
40
+ }
41
+ });
42
+ }
43
+ /**
44
+ * 准备连接前的状态清理
45
+ */
46
+ prepareConnection(agentId) {
47
+ this.agentId = agentId;
48
+ this.isManualClose = false;
49
+ if (this.ws && this.ws.readyState !== WebSocket.CLOSED) {
50
+ this.ws.removeAllListeners();
51
+ this.ws.terminate();
52
+ }
53
+ }
54
+ /**
55
+ * 构建 WebSocket URL
56
+ */
57
+ buildWebSocketUrl(agentId) {
58
+ return `${this.config.serverUrl}?agentId=${agentId}`;
59
+ }
60
+ /**
61
+ * 创建一次性 resolve/reject 处理器
62
+ */
63
+ createSettledHandlers(resolve, reject) {
64
+ let settled = false;
65
+ const rejectOnce = (error) => {
66
+ if (!settled) {
67
+ settled = true;
68
+ reject(error);
69
+ }
70
+ };
71
+ const resolveOnce = () => {
72
+ if (!settled) {
73
+ settled = true;
74
+ resolve();
75
+ }
76
+ };
77
+ return { rejectOnce, resolveOnce };
78
+ }
79
+ /**
80
+ * 设置 WebSocket 事件监听器
81
+ */
82
+ setupWebSocketEventHandlers(ws, rejectOnce, resolveOnce) {
83
+ let opened = false;
84
+ ws.on('open', () => {
85
+ opened = true;
86
+ this.handleWebSocketOpen(resolveOnce);
87
+ });
88
+ ws.on('message', (data) => this.handleWebSocketMessage(data));
89
+ ws.on('pong', () => this.markHeartbeatReceived());
90
+ ws.on('close', (code, reason) => this.handleWebSocketClose(code, reason, opened, rejectOnce));
91
+ ws.on('error', (error) => this.handleWebSocketError(error, opened, rejectOnce));
92
+ }
93
+ /**
94
+ * 处理 WebSocket 打开事件
95
+ */
96
+ handleWebSocketOpen(resolveOnce) {
97
+ logger.info('[WebSocket] 连接已建立');
98
+ this.reconnectAttempts = 0;
99
+ this.clearReconnectTimer();
100
+ this.startHeartbeat();
101
+ this.emit('connected');
102
+ resolveOnce();
103
+ }
104
+ /**
105
+ * 处理 WebSocket 消息
106
+ */
107
+ handleWebSocketMessage(data) {
108
+ try {
109
+ const message = JSON.parse(data.toString());
110
+ this.handleMessage(message);
111
+ }
112
+ catch (error) {
113
+ logger.warn('[WebSocket] 消息解析失败:', error);
114
+ }
115
+ }
116
+ /**
117
+ * 处理 WebSocket 关闭事件
118
+ */
119
+ handleWebSocketClose(code, reason, opened, rejectOnce) {
120
+ logger.warn(`[WebSocket] 连接关闭: code=${code}, reason=${reason.toString()}`);
121
+ this.stopHeartbeat();
122
+ this.clearHeartbeatTimeout();
123
+ this.emit('disconnected', code, reason);
124
+ if (!opened) {
125
+ rejectOnce(new Error(`WebSocket 连接在建立前关闭: code=${code}, reason=${reason.toString()}`));
126
+ }
127
+ if (!this.isManualClose) {
128
+ this.scheduleReconnect();
129
+ }
130
+ }
131
+ /**
132
+ * 处理 WebSocket 错误事件
133
+ */
134
+ handleWebSocketError(error, opened, rejectOnce) {
135
+ logger.error('[WebSocket] 连接错误:', error.message);
136
+ this.emit('error', error);
137
+ if (!opened) {
138
+ rejectOnce(error);
139
+ }
140
+ }
141
+ /**
142
+ * 断开 WebSocket 连接
143
+ */
144
+ disconnect() {
145
+ this.isManualClose = true;
146
+ this.stopHeartbeat();
147
+ this.clearHeartbeatTimeout();
148
+ this.clearReconnectTimer();
149
+ if (this.ws) {
150
+ if (this.ws.readyState === WebSocket.OPEN) {
151
+ this.ws.close(1000, 'Agent logout');
152
+ }
153
+ this.ws = null;
154
+ }
155
+ this.agentId = null;
156
+ logger.info('[WebSocket] 已断开连接');
157
+ }
158
+ /**
159
+ * 发送消息
160
+ */
161
+ send(message) {
162
+ if (!this.ws || this.ws.readyState !== WebSocket.OPEN) {
163
+ logger.warn('[WebSocket] 连接未建立,无法发送消息');
164
+ return false;
165
+ }
166
+ try {
167
+ this.ws.send(JSON.stringify(message));
168
+ return true;
169
+ }
170
+ catch (error) {
171
+ logger.error('[WebSocket] 发送消息失败:', error);
172
+ return false;
173
+ }
174
+ }
175
+ /**
176
+ * 是否已连接
177
+ */
178
+ isConnected() {
179
+ return this.ws !== null && this.ws.readyState === WebSocket.OPEN;
180
+ }
181
+ /**
182
+ * 处理接收到的消息
183
+ */
184
+ handleMessage(message) {
185
+ switch (message.type) {
186
+ case 'GROUP_MSG':
187
+ this.emit('groupMessage', message.data);
188
+ break;
189
+ case 'DM_MSG':
190
+ this.emit('directMessage', message.data);
191
+ break;
192
+ case 'ONLINE_CHANGE':
193
+ this.emit('onlineChange', message.data);
194
+ break;
195
+ case 'HEARTBEAT':
196
+ this.markHeartbeatReceived();
197
+ break;
198
+ case 'PROFILE_INFO':
199
+ logger.info('[WebSocket] 收到 PROFILE_INFO 消息');
200
+ this.emit('profileInfo', message.data);
201
+ break;
202
+ case 'AGENT_STATUS':
203
+ logger.info('[WebSocket] 收到 AGENT_STATUS 消息');
204
+ this.emit('agentStatus', message.data);
205
+ break;
206
+ default:
207
+ logger.warn('[WebSocket] 未知消息类型:', message.type);
208
+ }
209
+ }
210
+ /**
211
+ * 启动心跳
212
+ */
213
+ startHeartbeat() {
214
+ this.stopHeartbeat();
215
+ this.heartbeatTimer = setInterval(() => {
216
+ if (this.isConnected()) {
217
+ this.ensureHeartbeatTimeout();
218
+ this.sendProtocolPing();
219
+ this.send({
220
+ type: 'HEARTBEAT',
221
+ data: 'ping',
222
+ timestamp: new Date().toISOString()
223
+ });
224
+ }
225
+ }, this.config.heartbeatInterval);
226
+ }
227
+ /**
228
+ * 停止心跳
229
+ */
230
+ stopHeartbeat() {
231
+ if (this.heartbeatTimer) {
232
+ clearInterval(this.heartbeatTimer);
233
+ this.heartbeatTimer = null;
234
+ }
235
+ this.clearHeartbeatTimeout();
236
+ }
237
+ sendProtocolPing() {
238
+ if (!this.ws || this.ws.readyState !== WebSocket.OPEN) {
239
+ return;
240
+ }
241
+ try {
242
+ this.ws.ping();
243
+ }
244
+ catch (error) {
245
+ logger.warn('[WebSocket] 协议层 ping 发送失败,终止连接并重连:', error instanceof Error ? error.message : String(error));
246
+ this.ws.terminate();
247
+ }
248
+ }
249
+ markHeartbeatReceived() {
250
+ this.clearHeartbeatTimeout();
251
+ }
252
+ ensureHeartbeatTimeout() {
253
+ if (this.heartbeatTimeoutTimer) {
254
+ return;
255
+ }
256
+ this.heartbeatTimeoutTimer = setTimeout(() => {
257
+ this.heartbeatTimeoutTimer = null;
258
+ if (!this.ws || this.ws.readyState !== WebSocket.OPEN) {
259
+ return;
260
+ }
261
+ logger.warn(`[WebSocket] ${this.resolveHeartbeatTimeout()}ms 未收到心跳响应,终止连接并触发重连`);
262
+ this.ws.terminate();
263
+ }, this.resolveHeartbeatTimeout());
264
+ }
265
+ clearHeartbeatTimeout() {
266
+ if (this.heartbeatTimeoutTimer) {
267
+ clearTimeout(this.heartbeatTimeoutTimer);
268
+ this.heartbeatTimeoutTimer = null;
269
+ }
270
+ }
271
+ resolveHeartbeatTimeout() {
272
+ return this.config.heartbeatTimeout ?? DEFAULT_HEARTBEAT_TIMEOUT_MS;
273
+ }
274
+ /**
275
+ * 安排重连
276
+ */
277
+ scheduleReconnect() {
278
+ const hasRetryLimit = this.config.maxReconnectAttempts !== -1;
279
+ if (hasRetryLimit && this.reconnectAttempts >= this.config.maxReconnectAttempts) {
280
+ logger.error('[WebSocket] 重连次数已达上限,放弃重连');
281
+ this.emit('reconnectFailed');
282
+ return;
283
+ }
284
+ if (this.reconnectTimer) {
285
+ return;
286
+ }
287
+ this.reconnectAttempts++;
288
+ // 首次重连也走配置的重连间隔,给服务端和前端一个稳定窗口来感知离线状态。
289
+ // 否则 0ms 立即重连会让 agent 在 DB 中几乎瞬间从 offline 又变回 online,
290
+ // 页面上看起来就像“从未离线”。
291
+ const delay = this.config.reconnectInterval;
292
+ const retryInfo = hasRetryLimit
293
+ ? `第 ${this.reconnectAttempts}/${this.config.maxReconnectAttempts} 次`
294
+ : `第 ${this.reconnectAttempts} 次(无限重试模式)`;
295
+ logger.warn(`[WebSocket] ${delay}ms 后尝试${retryInfo}重连...`);
296
+ this.reconnectTimer = setTimeout(() => {
297
+ this.reconnectTimer = null;
298
+ if (this.agentId) {
299
+ this.connect(this.agentId).catch((error) => {
300
+ logger.warn('[WebSocket] 重连失败,继续重试:', error instanceof Error ? error.message : String(error));
301
+ this.scheduleReconnect();
302
+ });
303
+ }
304
+ }, delay);
305
+ }
306
+ /**
307
+ * 清除重连定时器
308
+ */
309
+ clearReconnectTimer() {
310
+ if (this.reconnectTimer) {
311
+ clearTimeout(this.reconnectTimer);
312
+ this.reconnectTimer = null;
313
+ }
314
+ }
315
+ }
316
+ //# sourceMappingURL=websocket-client.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"websocket-client.js","sourceRoot":"","sources":["../src/websocket-client.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,SAAS,MAAM,IAAI,CAAC;AAC3B,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAE3C,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAErC,MAAM,4BAA4B,GAAG,IAAI,CAAC;AAE1C,MAAM,OAAO,eAAgB,SAAQ,YAAY;IACvC,EAAE,GAAqB,IAAI,CAAC;IAC5B,MAAM,CAAe;IACrB,OAAO,GAAkB,IAAI,CAAC;IAC9B,cAAc,GAA0B,IAAI,CAAC;IAC7C,qBAAqB,GAA0B,IAAI,CAAC;IACpD,iBAAiB,GAAG,CAAC,CAAC;IACtB,cAAc,GAA0B,IAAI,CAAC;IAC7C,aAAa,GAAG,KAAK,CAAC;IACtB,gBAAgB,CAA6B;IAErD,YAAY,MAAoB,EAAE,gBAA6C;QAC7E,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,IAAI,CAAC,CAAC,GAAW,EAAE,EAAE,CAAC,IAAI,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC;IACpF,CAAC;IAEH;;;SAGK;IACH,OAAO,CAAC,OAAe;QACrB,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;YAEhC,MAAM,KAAK,GAAG,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;YAC9C,MAAM,CAAC,IAAI,CAAC,qBAAqB,KAAK,EAAE,CAAC,CAAC;YAE1C,MAAM,EAAE,UAAU,EAAE,WAAW,EAAE,GAAG,IAAI,CAAC,qBAAqB,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;YAEhF,IAAI,CAAC;gBACH,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC;gBACvC,IAAI,CAAC,2BAA2B,CAAC,IAAI,CAAC,EAAE,EAAE,UAAU,EAAE,WAAW,CAAC,CAAC;YACrE,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,UAAU,CAAC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;YACxE,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACK,iBAAiB,CAAC,OAAe;QACvC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC;QAE3B,IAAI,IAAI,CAAC,EAAE,IAAI,IAAI,CAAC,EAAE,CAAC,UAAU,KAAK,SAAS,CAAC,MAAM,EAAE,CAAC;YACvD,IAAI,CAAC,EAAE,CAAC,kBAAkB,EAAE,CAAC;YAC7B,IAAI,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC;QACtB,CAAC;IACH,CAAC;IAED;;OAEG;IACK,iBAAiB,CAAC,OAAe;QACvC,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,YAAY,OAAO,EAAE,CAAC;IACvD,CAAC;IAED;;OAEG;IACK,qBAAqB,CAC3B,OAAmB,EACnB,MAA8B;QAE9B,IAAI,OAAO,GAAG,KAAK,CAAC;QAEpB,MAAM,UAAU,GAAG,CAAC,KAAY,EAAQ,EAAE;YACxC,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,OAAO,GAAG,IAAI,CAAC;gBACf,MAAM,CAAC,KAAK,CAAC,CAAC;YAChB,CAAC;QACH,CAAC,CAAC;QAEF,MAAM,WAAW,GAAG,GAAS,EAAE;YAC7B,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,OAAO,GAAG,IAAI,CAAC;gBACf,OAAO,EAAE,CAAC;YACZ,CAAC;QACH,CAAC,CAAC;QAEF,OAAO,EAAE,UAAU,EAAE,WAAW,EAAE,CAAC;IACrC,CAAC;IAED;;OAEG;IACK,2BAA2B,CACjC,EAAa,EACb,UAAkC,EAClC,WAAuB;QAEvB,IAAI,MAAM,GAAG,KAAK,CAAC;QAEnB,EAAE,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE;YACjB,MAAM,GAAG,IAAI,CAAC;YACd,IAAI,CAAC,mBAAmB,CAAC,WAAW,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,IAAuB,EAAE,EAAE,CAAC,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC,CAAC,CAAC;QAEjF,EAAE,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,qBAAqB,EAAE,CAAC,CAAC;QAElD,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,IAAY,EAAE,MAAc,EAAE,EAAE,CAC9C,IAAI,CAAC,oBAAoB,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC,CAAC;QAE/D,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAY,EAAE,EAAE,CAC9B,IAAI,CAAC,oBAAoB,CAAC,KAAK,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC,CAAC;IAC1D,CAAC;IAED;;OAEG;IACK,mBAAmB,CAAC,WAAuB;QACjD,MAAM,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QACjC,IAAI,CAAC,iBAAiB,GAAG,CAAC,CAAC;QAC3B,IAAI,CAAC,mBAAmB,EAAE,CAAC;QAC3B,IAAI,CAAC,cAAc,EAAE,CAAC;QACtB,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACvB,WAAW,EAAE,CAAC;IAChB,CAAC;IAED;;OAEG;IACK,sBAAsB,CAAC,IAAuB;QACpD,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE,CAA8B,CAAC;YACzE,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;QAC9B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,IAAI,CAAC,qBAAqB,EAAE,KAAK,CAAC,CAAC;QAC5C,CAAC;IACH,CAAC;IAED;;OAEG;IACK,oBAAoB,CAC1B,IAAY,EACZ,MAAc,EACd,MAAe,EACf,UAAkC;QAElC,MAAM,CAAC,IAAI,CAAC,0BAA0B,IAAI,YAAY,MAAM,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;QAC3E,IAAI,CAAC,aAAa,EAAE,CAAC;QACrB,IAAI,CAAC,qBAAqB,EAAE,CAAC;QAC7B,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC;QAExC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,UAAU,CAAC,IAAI,KAAK,CAAC,4BAA4B,IAAI,YAAY,MAAM,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,CAAC;QACzF,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,CAAC;YACxB,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAC3B,CAAC;IACH,CAAC;IAED;;OAEG;IACK,oBAAoB,CAC1B,KAAY,EACZ,MAAe,EACf,UAAkC;QAElC,MAAM,CAAC,KAAK,CAAC,mBAAmB,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;QACjD,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;QAC1B,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,UAAU,CAAC,KAAK,CAAC,CAAC;QACpB,CAAC;IACH,CAAC;IAED;;OAEG;IACH,UAAU;QACR,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC;QAC1B,IAAI,CAAC,aAAa,EAAE,CAAC;QACrB,IAAI,CAAC,qBAAqB,EAAE,CAAC;QAC7B,IAAI,CAAC,mBAAmB,EAAE,CAAC;QAE3B,IAAI,IAAI,CAAC,EAAE,EAAE,CAAC;YACZ,IAAI,IAAI,CAAC,EAAE,CAAC,UAAU,KAAK,SAAS,CAAC,IAAI,EAAE,CAAC;gBAC1C,IAAI,CAAC,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,cAAc,CAAC,CAAC;YACtC,CAAC;YACD,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC;QACjB,CAAC;QAED,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;QACpB,MAAM,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;IACnC,CAAC;IAED;;OAEG;IACH,IAAI,CAAC,OAAgB;QACnB,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,IAAI,CAAC,EAAE,CAAC,UAAU,KAAK,SAAS,CAAC,IAAI,EAAE,CAAC;YACtD,MAAM,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;YACxC,OAAO,KAAK,CAAC;QACf,CAAC;QAED,IAAI,CAAC;YACH,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC;YACtC,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,qBAAqB,EAAE,KAAK,CAAC,CAAC;YAC3C,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED;;OAEG;IACH,WAAW;QACT,OAAO,IAAI,CAAC,EAAE,KAAK,IAAI,IAAI,IAAI,CAAC,EAAE,CAAC,UAAU,KAAK,SAAS,CAAC,IAAI,CAAC;IACnE,CAAC;IAED;;OAEG;IACK,aAAa,CAAC,OAAkC;QACtD,QAAQ,OAAO,CAAC,IAAI,EAAE,CAAC;YACrB,KAAK,WAAW;gBACd,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;gBACxC,MAAM;YACR,KAAK,QAAQ;gBACX,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;gBACzC,MAAM;YACR,KAAK,eAAe;gBAClB,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;gBACxC,MAAM;YACR,KAAK,WAAW;gBACd,IAAI,CAAC,qBAAqB,EAAE,CAAC;gBAC7B,MAAM;YACR,KAAK,cAAc;gBACjB,MAAM,CAAC,IAAI,CAAC,gCAAgC,CAAC,CAAC;gBAC9C,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;gBACvC,MAAM;YACR,KAAK,cAAc;gBACjB,MAAM,CAAC,IAAI,CAAC,gCAAgC,CAAC,CAAC;gBAC9C,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;gBACvC,MAAM;YACR;gBACE,MAAM,CAAC,IAAI,CAAC,qBAAqB,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;QACrD,CAAC;IACH,CAAC;IAED;;OAEG;IACK,cAAc;QACpB,IAAI,CAAC,aAAa,EAAE,CAAC;QAErB,IAAI,CAAC,cAAc,GAAG,WAAW,CAAC,GAAG,EAAE;YACrC,IAAI,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC;gBACvB,IAAI,CAAC,sBAAsB,EAAE,CAAC;gBAC9B,IAAI,CAAC,gBAAgB,EAAE,CAAC;gBACxB,IAAI,CAAC,IAAI,CAAC;oBACR,IAAI,EAAE,WAAW;oBACjB,IAAI,EAAE,MAAM;oBACZ,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;iBACpC,CAAC,CAAC;YACL,CAAC;QACH,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC;IACpC,CAAC;IAED;;OAEG;IACK,aAAa;QACnB,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,aAAa,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YACnC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC7B,CAAC;QACD,IAAI,CAAC,qBAAqB,EAAE,CAAC;IAC/B,CAAC;IAEO,gBAAgB;QACtB,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,IAAI,CAAC,EAAE,CAAC,UAAU,KAAK,SAAS,CAAC,IAAI,EAAE,CAAC;YACtD,OAAO;QACT,CAAC;QAED,IAAI,CAAC;YACH,IAAI,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;QACjB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,IAAI,CAAC,oCAAoC,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;YAC1G,IAAI,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC;QACtB,CAAC;IACH,CAAC;IAEO,qBAAqB;QAC3B,IAAI,CAAC,qBAAqB,EAAE,CAAC;IAC/B,CAAC;IAEO,sBAAsB;QAC5B,IAAI,IAAI,CAAC,qBAAqB,EAAE,CAAC;YAC/B,OAAO;QACT,CAAC;QAED,IAAI,CAAC,qBAAqB,GAAG,UAAU,CAAC,GAAG,EAAE;YAC3C,IAAI,CAAC,qBAAqB,GAAG,IAAI,CAAC;YAClC,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,IAAI,CAAC,EAAE,CAAC,UAAU,KAAK,SAAS,CAAC,IAAI,EAAE,CAAC;gBACtD,OAAO;YACT,CAAC;YAED,MAAM,CAAC,IAAI,CAAC,eAAe,IAAI,CAAC,uBAAuB,EAAE,sBAAsB,CAAC,CAAC;YACjF,IAAI,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC;QACtB,CAAC,EAAE,IAAI,CAAC,uBAAuB,EAAE,CAAC,CAAC;IACrC,CAAC;IAEO,qBAAqB;QAC3B,IAAI,IAAI,CAAC,qBAAqB,EAAE,CAAC;YAC/B,YAAY,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;YACzC,IAAI,CAAC,qBAAqB,GAAG,IAAI,CAAC;QACpC,CAAC;IACH,CAAC;IAEO,uBAAuB;QAC7B,OAAO,IAAI,CAAC,MAAM,CAAC,gBAAgB,IAAI,4BAA4B,CAAC;IACtE,CAAC;IAED;;OAEG;IACK,iBAAiB;QACvB,MAAM,aAAa,GAAG,IAAI,CAAC,MAAM,CAAC,oBAAoB,KAAK,CAAC,CAAC,CAAC;QAC9D,IAAI,aAAa,IAAI,IAAI,CAAC,iBAAiB,IAAI,IAAI,CAAC,MAAM,CAAC,oBAAoB,EAAE,CAAC;YAChF,MAAM,CAAC,KAAK,CAAC,2BAA2B,CAAC,CAAC;YAC1C,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;YAC7B,OAAO;QACT,CAAC;QAED,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,OAAO;QACT,CAAC;QAED,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACzB,sCAAsC;QACtC,sDAAsD;QACtD,kBAAkB;QAClB,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC;QAC5C,MAAM,SAAS,GAAG,aAAa;YAC7B,CAAC,CAAC,KAAK,IAAI,CAAC,iBAAiB,IAAI,IAAI,CAAC,MAAM,CAAC,oBAAoB,IAAI;YACrE,CAAC,CAAC,KAAK,IAAI,CAAC,iBAAiB,YAAY,CAAC;QAE5C,MAAM,CAAC,IAAI,CAAC,eAAe,KAAK,SAAS,SAAS,OAAO,CAAC,CAAC;QAE3D,IAAI,CAAC,cAAc,GAAG,UAAU,CAAC,GAAG,EAAE;YACpC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;YAC3B,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;gBACjB,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;oBACzC,MAAM,CAAC,IAAI,CAAC,wBAAwB,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;oBAC9F,IAAI,CAAC,iBAAiB,EAAE,CAAC;gBAC3B,CAAC,CAAC,CAAC;YACL,CAAC;QACH,CAAC,EAAE,KAAK,CAAC,CAAC;IACZ,CAAC;IAGD;;OAEG;IACK,mBAAmB;QACzB,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,YAAY,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAClC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC7B,CAAC;IACH,CAAC;CACF"}
@@ -0,0 +1,2 @@
1
+ export {};
2
+ //# sourceMappingURL=websocket-client.test.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"websocket-client.test.d.ts","sourceRoot":"","sources":["../src/websocket-client.test.ts"],"names":[],"mappings":""}
@@ -0,0 +1,191 @@
1
+ import { describe, it } from 'node:test';
2
+ import assert from 'node:assert/strict';
3
+ import { EventEmitter } from 'node:events';
4
+ import { performance } from 'node:perf_hooks';
5
+ import WebSocket from 'ws';
6
+ import { WebSocketClient } from './websocket-client.js';
7
+ class FakeSocket extends EventEmitter {
8
+ readyState;
9
+ constructor() {
10
+ super();
11
+ this.readyState = WebSocket.CONNECTING;
12
+ }
13
+ close() {
14
+ this.readyState = WebSocket.CLOSED;
15
+ this.emit('close', 1000, Buffer.from('manual close'));
16
+ }
17
+ terminate() {
18
+ this.readyState = WebSocket.CLOSED;
19
+ }
20
+ send(_payload) {
21
+ // test stub
22
+ }
23
+ ping() {
24
+ setImmediate(() => this.emit('pong'));
25
+ }
26
+ }
27
+ class SilentPingSocket extends FakeSocket {
28
+ ping() {
29
+ // intentionally no pong
30
+ }
31
+ terminate() {
32
+ this.readyState = WebSocket.CLOSED;
33
+ this.emit('close', 1006, Buffer.from('heartbeat timeout'));
34
+ }
35
+ }
36
+ describe('WebSocketClient reconnect behavior', () => {
37
+ it('should keep retrying at interval after scheduler restart until connected', async () => {
38
+ const createdSockets = [];
39
+ let attempt = 0;
40
+ const client = new WebSocketClient({
41
+ serverUrl: 'ws://localhost:8080/ws/agent',
42
+ heartbeatInterval: 1000,
43
+ heartbeatTimeout: 3000,
44
+ reconnectInterval: 20,
45
+ maxReconnectAttempts: -1
46
+ }, (_url) => {
47
+ const socket = new FakeSocket();
48
+ createdSockets.push(socket);
49
+ attempt += 1;
50
+ if (attempt === 1 || attempt >= 4) {
51
+ setImmediate(() => {
52
+ socket.readyState = WebSocket.OPEN;
53
+ socket.emit('open');
54
+ });
55
+ }
56
+ else {
57
+ setImmediate(() => {
58
+ socket.emit('error', new Error('ECONNREFUSED'));
59
+ socket.emit('close', 1006, Buffer.from('connect fail'));
60
+ });
61
+ }
62
+ return socket;
63
+ });
64
+ try {
65
+ client.on('error', () => {
66
+ // suppress unhandled error event in test environment
67
+ });
68
+ await client.connect('agent-test');
69
+ const firstSocket = createdSockets[0];
70
+ assert.ok(firstSocket, 'first socket should exist');
71
+ firstSocket.readyState = WebSocket.CLOSED;
72
+ firstSocket.emit('close', 1006, Buffer.from('server restart'));
73
+ await new Promise((resolve, reject) => {
74
+ const timer = setTimeout(() => {
75
+ reject(new Error('expected reconnect to eventually succeed'));
76
+ }, 500);
77
+ client.on('connected', () => {
78
+ if (attempt >= 4) {
79
+ clearTimeout(timer);
80
+ resolve();
81
+ }
82
+ });
83
+ });
84
+ assert.equal(attempt, 4, 'should attempt initial + retries until success');
85
+ assert.equal(client.isConnected(), true);
86
+ }
87
+ finally {
88
+ client.disconnect();
89
+ }
90
+ });
91
+ it('should wait reconnect interval before first reconnect attempt', async () => {
92
+ const reconnectInterval = 80;
93
+ const attemptTimes = [];
94
+ let attempt = 0;
95
+ const client = new WebSocketClient({
96
+ serverUrl: 'ws://localhost:8080/ws/agent',
97
+ heartbeatInterval: 1000,
98
+ heartbeatTimeout: 3000,
99
+ reconnectInterval,
100
+ maxReconnectAttempts: 2,
101
+ }, (_url) => {
102
+ const socket = new FakeSocket();
103
+ attempt += 1;
104
+ attemptTimes.push(performance.now());
105
+ if (attempt === 1) {
106
+ setImmediate(() => {
107
+ socket.readyState = WebSocket.OPEN;
108
+ socket.emit('open');
109
+ });
110
+ }
111
+ else {
112
+ setImmediate(() => {
113
+ socket.readyState = WebSocket.OPEN;
114
+ socket.emit('open');
115
+ });
116
+ }
117
+ return socket;
118
+ });
119
+ try {
120
+ client.on('error', () => {
121
+ // suppress unhandled error event in test environment
122
+ });
123
+ await client.connect('agent-test');
124
+ const firstSocketTime = attemptTimes[0];
125
+ const firstSocket = client.ws;
126
+ assert.ok(firstSocket, 'first socket should exist');
127
+ firstSocket.readyState = WebSocket.CLOSED;
128
+ firstSocket.emit('close', 1006, Buffer.from('server restart'));
129
+ await new Promise((resolve, reject) => {
130
+ const timer = setTimeout(() => {
131
+ reject(new Error('expected reconnect attempt'));
132
+ }, 500);
133
+ const check = setInterval(() => {
134
+ if (attempt >= 2) {
135
+ clearInterval(check);
136
+ clearTimeout(timer);
137
+ resolve();
138
+ }
139
+ }, 5);
140
+ });
141
+ assert.ok(attemptTimes[1] - firstSocketTime >= reconnectInterval - 15, `first reconnect should wait about ${reconnectInterval}ms, actual=${attemptTimes[1] - firstSocketTime}ms`);
142
+ }
143
+ finally {
144
+ client.disconnect();
145
+ }
146
+ });
147
+ it('should terminate and reconnect when heartbeat response times out', async () => {
148
+ const createdSockets = [];
149
+ let attempt = 0;
150
+ const client = new WebSocketClient({
151
+ serverUrl: 'ws://localhost:8080/ws/agent',
152
+ heartbeatInterval: 10,
153
+ heartbeatTimeout: 20,
154
+ reconnectInterval: 10,
155
+ maxReconnectAttempts: 2,
156
+ }, (_url) => {
157
+ attempt += 1;
158
+ const socket = attempt === 1 ? new SilentPingSocket() : new FakeSocket();
159
+ createdSockets.push(socket);
160
+ setImmediate(() => {
161
+ socket.readyState = WebSocket.OPEN;
162
+ socket.emit('open');
163
+ });
164
+ return socket;
165
+ });
166
+ try {
167
+ client.on('error', () => {
168
+ // suppress unhandled error event in test environment
169
+ });
170
+ await client.connect('agent-test');
171
+ await new Promise((resolve, reject) => {
172
+ const timer = setTimeout(() => {
173
+ reject(new Error('expected reconnect after heartbeat timeout'));
174
+ }, 500);
175
+ client.on('connected', () => {
176
+ if (attempt >= 2) {
177
+ clearTimeout(timer);
178
+ resolve();
179
+ }
180
+ });
181
+ });
182
+ assert.equal(createdSockets[0]?.readyState, WebSocket.CLOSED);
183
+ assert.equal(attempt, 2);
184
+ assert.equal(client.isConnected(), true);
185
+ }
186
+ finally {
187
+ client.disconnect();
188
+ }
189
+ });
190
+ });
191
+ //# sourceMappingURL=websocket-client.test.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"websocket-client.test.js","sourceRoot":"","sources":["../src/websocket-client.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,WAAW,CAAC;AACzC,OAAO,MAAM,MAAM,oBAAoB,CAAC;AACxC,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAC3C,OAAO,EAAE,WAAW,EAAE,MAAM,iBAAiB,CAAC;AAC9C,OAAO,SAAS,MAAM,IAAI,CAAC;AAC3B,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AAExD,MAAM,UAAW,SAAQ,YAAY;IAC5B,UAAU,CAAS;IAE1B;QACE,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC,UAAU,CAAC;IACzC,CAAC;IAED,KAAK;QACH,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC,MAAM,CAAC;QACnC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC;IACxD,CAAC;IAED,SAAS;QACP,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC,MAAM,CAAC;IACrC,CAAC;IAED,IAAI,CAAC,QAAgB;QACnB,YAAY;IACd,CAAC;IAED,IAAI;QACF,YAAY,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;IACxC,CAAC;CACF;AAED,MAAM,gBAAiB,SAAQ,UAAU;IAC9B,IAAI;QACX,wBAAwB;IAC1B,CAAC;IAEQ,SAAS;QAChB,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC,MAAM,CAAC;QACnC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC,CAAC;IAC7D,CAAC;CACF;AAED,QAAQ,CAAC,oCAAoC,EAAE,GAAG,EAAE;IAClD,EAAE,CAAC,0EAA0E,EAAE,KAAK,IAAI,EAAE;QACxF,MAAM,cAAc,GAAiB,EAAE,CAAC;QACxC,IAAI,OAAO,GAAG,CAAC,CAAC;QAEhB,MAAM,MAAM,GAAG,IAAI,eAAe,CAChC;YACE,SAAS,EAAE,8BAA8B;YACzC,iBAAiB,EAAE,IAAI;YACvB,gBAAgB,EAAE,IAAI;YACtB,iBAAiB,EAAE,EAAE;YACrB,oBAAoB,EAAE,CAAC,CAAC;SACzB,EACD,CAAC,IAAY,EAAE,EAAE;YACf,MAAM,MAAM,GAAG,IAAI,UAAU,EAAE,CAAC;YAChC,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC5B,OAAO,IAAI,CAAC,CAAC;YAEb,IAAI,OAAO,KAAK,CAAC,IAAI,OAAO,IAAI,CAAC,EAAE,CAAC;gBAClC,YAAY,CAAC,GAAG,EAAE;oBAChB,MAAM,CAAC,UAAU,GAAG,SAAS,CAAC,IAAI,CAAC;oBACnC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBACtB,CAAC,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,YAAY,CAAC,GAAG,EAAE;oBAChB,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC;oBAChD,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC;gBAC1D,CAAC,CAAC,CAAC;YACL,CAAC;YAED,OAAO,MAA8B,CAAC;QACxC,CAAC,CACF,CAAC;QAEF,IAAI,CAAC;YACH,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;gBACtB,qDAAqD;YACvD,CAAC,CAAC,CAAC;YAEH,MAAM,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;YAEnC,MAAM,WAAW,GAAG,cAAc,CAAC,CAAC,CAAC,CAAC;YACtC,MAAM,CAAC,EAAE,CAAC,WAAW,EAAE,2BAA2B,CAAC,CAAC;YAEpD,WAAW,CAAC,UAAU,GAAG,SAAS,CAAC,MAAM,CAAC;YAC1C,WAAW,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC;YAE/D,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;gBAC1C,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE;oBAC5B,MAAM,CAAC,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAC,CAAC;gBAChE,CAAC,EAAE,GAAG,CAAC,CAAC;gBAER,MAAM,CAAC,EAAE,CAAC,WAAW,EAAE,GAAG,EAAE;oBAC1B,IAAI,OAAO,IAAI,CAAC,EAAE,CAAC;wBACjB,YAAY,CAAC,KAAK,CAAC,CAAC;wBACpB,OAAO,EAAE,CAAC;oBACZ,CAAC;gBACH,CAAC,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC,EAAE,gDAAgD,CAAC,CAAC;YAC3E,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,WAAW,EAAE,EAAE,IAAI,CAAC,CAAC;QAC3C,CAAC;gBAAS,CAAC;YACT,MAAM,CAAC,UAAU,EAAE,CAAC;QACtB,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC7E,MAAM,iBAAiB,GAAG,EAAE,CAAC;QAC7B,MAAM,YAAY,GAAa,EAAE,CAAC;QAClC,IAAI,OAAO,GAAG,CAAC,CAAC;QAEhB,MAAM,MAAM,GAAG,IAAI,eAAe,CAChC;YACE,SAAS,EAAE,8BAA8B;YACzC,iBAAiB,EAAE,IAAI;YACvB,gBAAgB,EAAE,IAAI;YACtB,iBAAiB;YACjB,oBAAoB,EAAE,CAAC;SACxB,EACD,CAAC,IAAY,EAAE,EAAE;YACf,MAAM,MAAM,GAAG,IAAI,UAAU,EAAE,CAAC;YAChC,OAAO,IAAI,CAAC,CAAC;YACb,YAAY,CAAC,IAAI,CAAC,WAAW,CAAC,GAAG,EAAE,CAAC,CAAC;YAErC,IAAI,OAAO,KAAK,CAAC,EAAE,CAAC;gBAClB,YAAY,CAAC,GAAG,EAAE;oBAChB,MAAM,CAAC,UAAU,GAAG,SAAS,CAAC,IAAI,CAAC;oBACnC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBACtB,CAAC,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,YAAY,CAAC,GAAG,EAAE;oBAChB,MAAM,CAAC,UAAU,GAAG,SAAS,CAAC,IAAI,CAAC;oBACnC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBACtB,CAAC,CAAC,CAAC;YACL,CAAC;YAED,OAAO,MAA8B,CAAC;QACxC,CAAC,CACF,CAAC;QAEF,IAAI,CAAC;YACH,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;gBACtB,qDAAqD;YACvD,CAAC,CAAC,CAAC;YAEH,MAAM,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;YACnC,MAAM,eAAe,GAAG,YAAY,CAAC,CAAC,CAAC,CAAC;YACxC,MAAM,WAAW,GAAI,MAA+C,CAAC,EAAE,CAAC;YACxE,MAAM,CAAC,EAAE,CAAC,WAAW,EAAE,2BAA2B,CAAC,CAAC;YAEpD,WAAW,CAAC,UAAU,GAAG,SAAS,CAAC,MAAM,CAAC;YAC1C,WAAW,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC;YAE/D,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;gBAC1C,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE;oBAC5B,MAAM,CAAC,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC,CAAC;gBAClD,CAAC,EAAE,GAAG,CAAC,CAAC;gBAER,MAAM,KAAK,GAAG,WAAW,CAAC,GAAG,EAAE;oBAC7B,IAAI,OAAO,IAAI,CAAC,EAAE,CAAC;wBACjB,aAAa,CAAC,KAAK,CAAC,CAAC;wBACrB,YAAY,CAAC,KAAK,CAAC,CAAC;wBACpB,OAAO,EAAE,CAAC;oBACZ,CAAC;gBACH,CAAC,EAAE,CAAC,CAAC,CAAC;YACR,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,EAAE,CAAC,YAAY,CAAC,CAAC,CAAC,GAAG,eAAe,IAAI,iBAAiB,GAAG,EAAE,EACnE,qCAAqC,iBAAiB,cAAc,YAAY,CAAC,CAAC,CAAC,GAAG,eAAe,IAAI,CAAC,CAAC;QAC/G,CAAC;gBAAS,CAAC;YACT,MAAM,CAAC,UAAU,EAAE,CAAC;QACtB,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;QAChF,MAAM,cAAc,GAAiB,EAAE,CAAC;QACxC,IAAI,OAAO,GAAG,CAAC,CAAC;QAEhB,MAAM,MAAM,GAAG,IAAI,eAAe,CAChC;YACE,SAAS,EAAE,8BAA8B;YACzC,iBAAiB,EAAE,EAAE;YACrB,gBAAgB,EAAE,EAAE;YACpB,iBAAiB,EAAE,EAAE;YACrB,oBAAoB,EAAE,CAAC;SACxB,EACD,CAAC,IAAY,EAAE,EAAE;YACf,OAAO,IAAI,CAAC,CAAC;YACb,MAAM,MAAM,GAAG,OAAO,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,gBAAgB,EAAE,CAAC,CAAC,CAAC,IAAI,UAAU,EAAE,CAAC;YACzE,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC5B,YAAY,CAAC,GAAG,EAAE;gBAChB,MAAM,CAAC,UAAU,GAAG,SAAS,CAAC,IAAI,CAAC;gBACnC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACtB,CAAC,CAAC,CAAC;YACH,OAAO,MAA8B,CAAC;QACxC,CAAC,CACF,CAAC;QAEF,IAAI,CAAC;YACH,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;gBACtB,qDAAqD;YACvD,CAAC,CAAC,CAAC;YAEH,MAAM,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;YAEnC,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;gBAC1C,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE;oBAC5B,MAAM,CAAC,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC,CAAC;gBAClE,CAAC,EAAE,GAAG,CAAC,CAAC;gBAER,MAAM,CAAC,EAAE,CAAC,WAAW,EAAE,GAAG,EAAE;oBAC1B,IAAI,OAAO,IAAI,CAAC,EAAE,CAAC;wBACjB,YAAY,CAAC,KAAK,CAAC,CAAC;wBACpB,OAAO,EAAE,CAAC;oBACZ,CAAC;gBACH,CAAC,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,SAAS,CAAC,MAAM,CAAC,CAAC;YAC9D,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;YACzB,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,WAAW,EAAE,EAAE,IAAI,CAAC,CAAC;QAC3C,CAAC;gBAAS,CAAC;YACT,MAAM,CAAC,UAAU,EAAE,CAAC;QACtB,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "aws-runtime-bridge",
3
- "version": "1.1.1",
3
+ "version": "1.1.2",
4
4
  "description": "AgentsWorkStudio runtime bridge service for machine-level agent runtime integration",
5
5
  "type": "module",
6
6
  "main": "dist/index.js",