@fluojs/websockets 1.0.0-beta.2 → 1.0.0-beta.4
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.
- package/README.ko.md +3 -1
- package/README.md +3 -1
- package/dist/bun/bun-service.d.ts +11 -0
- package/dist/bun/bun-service.d.ts.map +1 -1
- package/dist/bun/bun-service.js +185 -5
- package/dist/bun/bun-types.d.ts +27 -0
- package/dist/bun/bun-types.d.ts.map +1 -1
- package/dist/cloudflare-workers/cloudflare-workers-service.d.ts +11 -0
- package/dist/cloudflare-workers/cloudflare-workers-service.d.ts.map +1 -1
- package/dist/cloudflare-workers/cloudflare-workers-service.js +219 -50
- package/dist/cloudflare-workers/cloudflare-workers-types.d.ts +30 -0
- package/dist/cloudflare-workers/cloudflare-workers-types.d.ts.map +1 -1
- package/dist/deno/deno-service.d.ts +11 -0
- package/dist/deno/deno-service.d.ts.map +1 -1
- package/dist/deno/deno-service.js +218 -49
- package/dist/deno/deno-types.d.ts +27 -0
- package/dist/deno/deno-types.d.ts.map +1 -1
- package/dist/internal/shared.d.ts +81 -0
- package/dist/internal/shared.d.ts.map +1 -1
- package/dist/internal/shared.js +93 -0
- package/dist/metadata.d.ts +38 -0
- package/dist/metadata.d.ts.map +1 -1
- package/dist/metadata.js +44 -0
- package/package.json +7 -7
package/README.ko.md
CHANGED
|
@@ -101,7 +101,7 @@ WebSocketModule.forRoot({
|
|
|
101
101
|
});
|
|
102
102
|
```
|
|
103
103
|
|
|
104
|
-
옵션을 생략하면 `@fluojs/websockets`는 동시 연결 수와 인바운드 페이로드 크기에 대해 기본 제한값을 적용합니다. 또한 server-backed Node 리스너는 `heartbeat.enabled`를 명시적으로 `false`로 두지 않는 한 heartbeat 타이머를 활성화합니다.
|
|
104
|
+
옵션을 생략하면 `@fluojs/websockets`는 동시 연결 수와 인바운드 페이로드 크기에 대해 기본 제한값을 적용합니다. 또한 server-backed Node 리스너는 `heartbeat.enabled`를 명시적으로 `false`로 두지 않는 한 heartbeat 타이머를 활성화합니다. 공식 fetch-style 런타임 모듈(`@fluojs/websockets/bun`, `@fluojs/websockets/deno`, `@fluojs/websockets/cloudflare-workers`)은 애플리케이션 shutdown 시 추적 중인 websocket 클라이언트를 닫고 `shutdown.timeoutMs` 범위 안에서 `@OnDisconnect()` cleanup이 마무리될 수 있도록 bounded 기회를 제공합니다.
|
|
105
105
|
|
|
106
106
|
## 공개 API 개요
|
|
107
107
|
|
|
@@ -131,3 +131,5 @@ WebSocketModule.forRoot({
|
|
|
131
131
|
- `packages/websockets/src/public-surface.test.ts`
|
|
132
132
|
- `packages/websockets/src/node/node.test.ts`
|
|
133
133
|
- `packages/websockets/src/bun/bun.test.ts`
|
|
134
|
+
- `packages/websockets/src/deno/deno.test.ts`
|
|
135
|
+
- `packages/websockets/src/cloudflare-workers/cloudflare-workers.test.ts`
|
package/README.md
CHANGED
|
@@ -101,7 +101,7 @@ WebSocketModule.forRoot({
|
|
|
101
101
|
});
|
|
102
102
|
```
|
|
103
103
|
|
|
104
|
-
When omitted, `@fluojs/websockets` now applies bounded defaults for concurrent connections and inbound payload size. Server-backed Node listeners also enable heartbeat timers unless you explicitly set `heartbeat.enabled` to `false`.
|
|
104
|
+
When omitted, `@fluojs/websockets` now applies bounded defaults for concurrent connections and inbound payload size. Server-backed Node listeners also enable heartbeat timers unless you explicitly set `heartbeat.enabled` to `false`. The official fetch-style runtime modules (`@fluojs/websockets/bun`, `@fluojs/websockets/deno`, and `@fluojs/websockets/cloudflare-workers`) close tracked websocket clients during application shutdown and give `@OnDisconnect()` cleanup a bounded chance to finish within `shutdown.timeoutMs`.
|
|
105
105
|
|
|
106
106
|
## Public API Overview
|
|
107
107
|
|
|
@@ -131,3 +131,5 @@ Use the runtime subpaths when you want an explicit runtime binding instead of th
|
|
|
131
131
|
- `packages/websockets/src/public-surface.test.ts`
|
|
132
132
|
- `packages/websockets/src/node/node.test.ts`
|
|
133
133
|
- `packages/websockets/src/bun/bun.test.ts`
|
|
134
|
+
- `packages/websockets/src/deno/deno.test.ts`
|
|
135
|
+
- `packages/websockets/src/cloudflare-workers/cloudflare-workers.test.ts`
|
|
@@ -12,11 +12,14 @@ export declare class BunWebSocketGatewayLifecycleService implements OnApplicatio
|
|
|
12
12
|
private readonly logger;
|
|
13
13
|
private readonly adapter;
|
|
14
14
|
private readonly moduleOptions;
|
|
15
|
+
private isShuttingDown;
|
|
16
|
+
private readonly pendingUpgradeOperations;
|
|
15
17
|
private pendingUpgradeReservations;
|
|
16
18
|
private readonly roomSockets;
|
|
17
19
|
private shutdownPromise;
|
|
18
20
|
private readonly socketRegistry;
|
|
19
21
|
private readonly socketRooms;
|
|
22
|
+
private readonly socketStates;
|
|
20
23
|
constructor(runtimeContainer: Container, compiledModules: readonly CompiledModule[], logger: ApplicationLogger, adapter: HttpApplicationAdapter, moduleOptions: WebSocketModuleOptions);
|
|
21
24
|
onApplicationBootstrap(): Promise<void>;
|
|
22
25
|
private assertNoServerBackedGatewayOptIn;
|
|
@@ -27,6 +30,9 @@ export declare class BunWebSocketGatewayLifecycleService implements OnApplicatio
|
|
|
27
30
|
private handleUpgradeRequest;
|
|
28
31
|
private bindConnectionHandlers;
|
|
29
32
|
private createConnectionHandlerState;
|
|
33
|
+
private settleOpenRegistration;
|
|
34
|
+
private settleConnectLifecycle;
|
|
35
|
+
private settleDisconnectLifecycle;
|
|
30
36
|
private getBufferedMessageCount;
|
|
31
37
|
private getQueuedMessageCount;
|
|
32
38
|
private maybeCompactBufferedMessages;
|
|
@@ -52,8 +58,13 @@ export declare class BunWebSocketGatewayLifecycleService implements OnApplicatio
|
|
|
52
58
|
private releaseUpgradeReservation;
|
|
53
59
|
private resolveMaxPayloadBytes;
|
|
54
60
|
private resolveIdleTimeoutSeconds;
|
|
61
|
+
private resolveShutdownTimeoutMs;
|
|
55
62
|
private shutdown;
|
|
56
63
|
private runShutdownLifecycle;
|
|
64
|
+
private trackPendingUpgradeOperation;
|
|
65
|
+
private awaitPendingUpgradeOperations;
|
|
66
|
+
private closeActiveSockets;
|
|
67
|
+
private awaitHandlerQueueDrain;
|
|
57
68
|
joinRoom(socketId: string, room: string): void;
|
|
58
69
|
leaveRoom(socketId: string, room: string): void;
|
|
59
70
|
broadcastToRoom(room: string, event: string, data: unknown): void;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"bun-service.d.ts","sourceRoot":"","sources":["../../src/bun/bun-service.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AAC5C,OAAO,KAAK,EAAE,iBAAiB,EAAE,cAAc,EAAE,sBAAsB,EAAE,qBAAqB,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC;AASzI,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,cAAc,CAAC;AAa3D,OAAO,KAAK,EAGV,oBAAoB,EACrB,MAAM,aAAa,CAAC;AACrB,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,gBAAgB,CAAC;
|
|
1
|
+
{"version":3,"file":"bun-service.d.ts","sourceRoot":"","sources":["../../src/bun/bun-service.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AAC5C,OAAO,KAAK,EAAE,iBAAiB,EAAE,cAAc,EAAE,sBAAsB,EAAE,qBAAqB,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC;AASzI,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,cAAc,CAAC;AAa3D,OAAO,KAAK,EAGV,oBAAoB,EACrB,MAAM,aAAa,CAAC;AACrB,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,gBAAgB,CAAC;AA4H7D;;GAEG;AACH,qBACa,mCACX,YAAW,sBAAsB,EAAE,qBAAqB,EAAE,eAAe,EAAE,oBAAoB;IAY7F,OAAO,CAAC,QAAQ,CAAC,gBAAgB;IACjC,OAAO,CAAC,QAAQ,CAAC,eAAe;IAChC,OAAO,CAAC,QAAQ,CAAC,MAAM;IACvB,OAAO,CAAC,QAAQ,CAAC,OAAO;IACxB,OAAO,CAAC,QAAQ,CAAC,aAAa;IAdhC,OAAO,CAAC,cAAc,CAAS;IAC/B,OAAO,CAAC,QAAQ,CAAC,wBAAwB,CAA4B;IACrE,OAAO,CAAC,0BAA0B,CAAK;IACvC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAkC;IAC9D,OAAO,CAAC,eAAe,CAA4B;IACnD,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAwD;IACvF,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAkC;IAC9D,OAAO,CAAC,QAAQ,CAAC,YAAY,CAA6C;gBAGvD,gBAAgB,EAAE,SAAS,EAC3B,eAAe,EAAE,SAAS,cAAc,EAAE,EAC1C,MAAM,EAAE,iBAAiB,EACzB,OAAO,EAAE,sBAAsB,EAC/B,aAAa,EAAE,sBAAsB;IAGlD,sBAAsB,IAAI,OAAO,CAAC,IAAI,CAAC;IAqB7C,OAAO,CAAC,gCAAgC;IAclC,qBAAqB,IAAI,OAAO,CAAC,IAAI,CAAC;IAItC,eAAe,IAAI,OAAO,CAAC,IAAI,CAAC;IAItC,OAAO,CAAC,aAAa;IAmCrB,OAAO,CAAC,sBAAsB;YAmBhB,oBAAoB;YA0DpB,sBAAsB;IA0BpC,OAAO,CAAC,4BAA4B;IAkCpC,OAAO,CAAC,sBAAsB;IAS9B,OAAO,CAAC,sBAAsB;IAS9B,OAAO,CAAC,yBAAyB;IASjC,OAAO,CAAC,uBAAuB;IAI/B,OAAO,CAAC,qBAAqB;IAI7B,OAAO,CAAC,4BAA4B;IASpC,OAAO,CAAC,qBAAqB;IAK7B,OAAO,CAAC,0BAA0B;IASlC,OAAO,CAAC,mBAAmB;IAM3B,OAAO,CAAC,mBAAmB;IAe3B,OAAO,CAAC,iBAAiB;IAYzB,OAAO,CAAC,qBAAqB;IA0C7B,OAAO,CAAC,sBAAsB;YAwDhB,iBAAiB;IAmB/B,OAAO,CAAC,yBAAyB;YAyBnB,yBAAyB;YAczB,kBAAkB;YASlB,yBAAyB;IASvC,OAAO,CAAC,8BAA8B;IAoBtC,OAAO,CAAC,wBAAwB;YAUlB,uBAAuB;IA0DrC,OAAO,CAAC,qBAAqB;IAmB7B,OAAO,CAAC,yBAAyB;IAUjC,OAAO,CAAC,8BAA8B;IAItC,OAAO,CAAC,qBAAqB;IAS7B,OAAO,CAAC,yBAAyB;IAMjC,OAAO,CAAC,sBAAsB;IAU9B,OAAO,CAAC,yBAAyB;IAcjC,OAAO,CAAC,wBAAwB;YAUlB,QAAQ;YAUR,oBAAoB;IAoBlC,OAAO,CAAC,4BAA4B;YAetB,6BAA6B;YA4C7B,kBAAkB;YAoBlB,sBAAsB;IAkDpC,QAAQ,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,IAAI;IAmB9C,SAAS,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,IAAI;IAc/C,eAAe,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,GAAG,IAAI;IA4BjE,QAAQ,CAAC,QAAQ,EAAE,MAAM,GAAG,WAAW,CAAC,MAAM,CAAC;IAU/C,OAAO,CAAC,gBAAgB;CAiBzB"}
|
package/dist/bun/bun-service.js
CHANGED
|
@@ -11,6 +11,7 @@ import { WEBSOCKET_OPTIONS_INTERNAL } from '../options-token.internal.js';
|
|
|
11
11
|
const DEFAULT_MAX_PENDING_MESSAGES_PER_SOCKET = 256;
|
|
12
12
|
const DEFAULT_MAX_WEBSOCKET_CONNECTIONS = 1_000;
|
|
13
13
|
const DEFAULT_MAX_WEBSOCKET_PAYLOAD_BYTES = 1_048_576;
|
|
14
|
+
const DEFAULT_WEBSOCKET_SHUTDOWN_TIMEOUT_MS = 5_000;
|
|
14
15
|
const LIFECYCLE_LOG_CONTEXT = 'WebSocketGatewayLifecycleService';
|
|
15
16
|
function hasBunWebSocketBindingHost(adapter) {
|
|
16
17
|
return 'configureWebSocketBinding' in adapter && typeof adapter.configureWebSocketBinding === 'function';
|
|
@@ -40,6 +41,16 @@ function resolveMessageByteLength(message) {
|
|
|
40
41
|
function isWebSocketUpgradeRequest(request) {
|
|
41
42
|
return request.headers.get('upgrade')?.toLowerCase() === 'websocket';
|
|
42
43
|
}
|
|
44
|
+
function createCompletionSignal() {
|
|
45
|
+
let resolve;
|
|
46
|
+
const promise = new Promise(res => {
|
|
47
|
+
resolve = res;
|
|
48
|
+
});
|
|
49
|
+
return {
|
|
50
|
+
promise,
|
|
51
|
+
resolve
|
|
52
|
+
};
|
|
53
|
+
}
|
|
43
54
|
|
|
44
55
|
/**
|
|
45
56
|
* Boots Bun-backed websocket gateways and manages their room lifecycle state.
|
|
@@ -49,11 +60,14 @@ class BunWebSocketGatewayLifecycleService {
|
|
|
49
60
|
static {
|
|
50
61
|
[_BunWebSocketGatewayL, _initClass] = _applyDecs(this, [Inject(RUNTIME_CONTAINER, COMPILED_MODULES, APPLICATION_LOGGER, HTTP_APPLICATION_ADAPTER, WEBSOCKET_OPTIONS_INTERNAL)], []).c;
|
|
51
62
|
}
|
|
63
|
+
isShuttingDown = false;
|
|
64
|
+
pendingUpgradeOperations = new Set();
|
|
52
65
|
pendingUpgradeReservations = 0;
|
|
53
66
|
roomSockets = new Map();
|
|
54
67
|
shutdownPromise;
|
|
55
68
|
socketRegistry = new Map();
|
|
56
69
|
socketRooms = new Map();
|
|
70
|
+
socketStates = new Map();
|
|
57
71
|
constructor(runtimeContainer, compiledModules, logger, adapter, moduleOptions) {
|
|
58
72
|
this.runtimeContainer = runtimeContainer;
|
|
59
73
|
this.compiledModules = compiledModules;
|
|
@@ -90,7 +104,7 @@ class BunWebSocketGatewayLifecycleService {
|
|
|
90
104
|
createBinding(descriptors) {
|
|
91
105
|
const descriptorsByPath = this.groupDescriptorsByPath(descriptors);
|
|
92
106
|
return {
|
|
93
|
-
fetch: (request, server) => this.handleUpgradeRequest(request, server, descriptorsByPath),
|
|
107
|
+
fetch: (request, server) => this.trackPendingUpgradeOperation(this.handleUpgradeRequest(request, server, descriptorsByPath)),
|
|
94
108
|
websocket: {
|
|
95
109
|
backpressureLimit: this.resolveBackpressureLimit(),
|
|
96
110
|
close: (socket, code, reason) => {
|
|
@@ -108,7 +122,7 @@ class BunWebSocketGatewayLifecycleService {
|
|
|
108
122
|
this.handleSocketMessage(socket, message);
|
|
109
123
|
},
|
|
110
124
|
open: socket => {
|
|
111
|
-
void this.bindConnectionHandlers(socket).catch(error => {
|
|
125
|
+
void this.trackPendingUpgradeOperation(this.bindConnectionHandlers(socket)).catch(error => {
|
|
112
126
|
this.unregisterSocket(socket.data.state.socketId);
|
|
113
127
|
this.logger.error('WebSocket gateway open lifecycle failed.', error, LIFECYCLE_LOG_CONTEXT);
|
|
114
128
|
socket.close(1011, 'Internal server error');
|
|
@@ -155,7 +169,14 @@ class BunWebSocketGatewayLifecycleService {
|
|
|
155
169
|
status: rejection.status
|
|
156
170
|
});
|
|
157
171
|
}
|
|
172
|
+
if (this.isShuttingDown) {
|
|
173
|
+
this.releaseUpgradeReservation();
|
|
174
|
+
return new Response('WebSocket server is shutting down.', {
|
|
175
|
+
status: 503
|
|
176
|
+
});
|
|
177
|
+
}
|
|
158
178
|
const state = this.createConnectionHandlerState(request, [...descriptors]);
|
|
179
|
+
void this.trackPendingUpgradeOperation(state.openRegistrationPromise);
|
|
159
180
|
let upgraded = false;
|
|
160
181
|
try {
|
|
161
182
|
upgraded = server.upgrade(request, {
|
|
@@ -164,10 +185,12 @@ class BunWebSocketGatewayLifecycleService {
|
|
|
164
185
|
}
|
|
165
186
|
});
|
|
166
187
|
} catch (error) {
|
|
188
|
+
this.settleOpenRegistration(state);
|
|
167
189
|
this.releaseUpgradeReservation();
|
|
168
190
|
throw error;
|
|
169
191
|
}
|
|
170
192
|
if (!upgraded) {
|
|
193
|
+
this.settleOpenRegistration(state);
|
|
171
194
|
this.releaseUpgradeReservation();
|
|
172
195
|
return new Response(null, {
|
|
173
196
|
status: 400
|
|
@@ -181,27 +204,73 @@ class BunWebSocketGatewayLifecycleService {
|
|
|
181
204
|
} = socket.data;
|
|
182
205
|
this.releaseUpgradeReservation();
|
|
183
206
|
this.socketRegistry.set(state.socketId, socket);
|
|
184
|
-
|
|
185
|
-
|
|
186
|
-
|
|
207
|
+
this.socketStates.set(state.socketId, state);
|
|
208
|
+
this.settleOpenRegistration(state);
|
|
209
|
+
try {
|
|
210
|
+
await this.resolveConnectionGateways(state);
|
|
211
|
+
await this.runConnectHandlers(state, socket);
|
|
212
|
+
await this.finalizeConnectionBinding(state, socket);
|
|
213
|
+
if (this.isShuttingDown && socket.readyState === 1) {
|
|
214
|
+
socket.close(1001, 'Server shutting down');
|
|
215
|
+
await state.disconnectLifecyclePromise;
|
|
216
|
+
}
|
|
217
|
+
} finally {
|
|
218
|
+
if (!state.handlersReady && state.bufferedDisconnect) {
|
|
219
|
+
this.settleDisconnectLifecycle(state);
|
|
220
|
+
}
|
|
221
|
+
this.settleConnectLifecycle(state);
|
|
222
|
+
}
|
|
187
223
|
}
|
|
188
224
|
createConnectionHandlerState(request, descriptors) {
|
|
225
|
+
const connectLifecycle = createCompletionSignal();
|
|
226
|
+
const disconnectLifecycle = createCompletionSignal();
|
|
227
|
+
const openRegistration = createCompletionSignal();
|
|
189
228
|
return {
|
|
190
229
|
bufferedDisconnect: undefined,
|
|
191
230
|
bufferedMessages: [],
|
|
192
231
|
bufferedMessagesStartIndex: 0,
|
|
232
|
+
connectLifecycleSettled: false,
|
|
233
|
+
connectLifecyclePromise: connectLifecycle.promise,
|
|
193
234
|
descriptors,
|
|
235
|
+
disconnectLifecycleSettled: false,
|
|
236
|
+
disconnectLifecyclePromise: disconnectLifecycle.promise,
|
|
194
237
|
enqueuedMessageCount: 0,
|
|
195
238
|
handlerQueue: Promise.resolve(),
|
|
196
239
|
handlersReady: false,
|
|
240
|
+
openRegistrationSettled: false,
|
|
241
|
+
openRegistrationPromise: openRegistration.promise,
|
|
197
242
|
processingMessageQueue: false,
|
|
198
243
|
queuedMessages: [],
|
|
199
244
|
queuedMessagesStartIndex: 0,
|
|
200
245
|
request,
|
|
246
|
+
resolveOpenRegistration: openRegistration.resolve,
|
|
247
|
+
resolveConnectLifecycle: connectLifecycle.resolve,
|
|
248
|
+
resolveDisconnectLifecycle: disconnectLifecycle.resolve,
|
|
201
249
|
resolved: [],
|
|
202
250
|
socketId: crypto.randomUUID()
|
|
203
251
|
};
|
|
204
252
|
}
|
|
253
|
+
settleOpenRegistration(state) {
|
|
254
|
+
if (state.openRegistrationSettled) {
|
|
255
|
+
return;
|
|
256
|
+
}
|
|
257
|
+
state.openRegistrationSettled = true;
|
|
258
|
+
state.resolveOpenRegistration();
|
|
259
|
+
}
|
|
260
|
+
settleConnectLifecycle(state) {
|
|
261
|
+
if (state.connectLifecycleSettled) {
|
|
262
|
+
return;
|
|
263
|
+
}
|
|
264
|
+
state.connectLifecycleSettled = true;
|
|
265
|
+
state.resolveConnectLifecycle();
|
|
266
|
+
}
|
|
267
|
+
settleDisconnectLifecycle(state) {
|
|
268
|
+
if (state.disconnectLifecycleSettled) {
|
|
269
|
+
return;
|
|
270
|
+
}
|
|
271
|
+
state.disconnectLifecycleSettled = true;
|
|
272
|
+
state.resolveDisconnectLifecycle();
|
|
273
|
+
}
|
|
205
274
|
getBufferedMessageCount(state) {
|
|
206
275
|
return state.bufferedMessages.length - state.bufferedMessagesStartIndex;
|
|
207
276
|
}
|
|
@@ -330,6 +399,8 @@ class BunWebSocketGatewayLifecycleService {
|
|
|
330
399
|
await dispatchGatewayDisconnect(state.resolved, socket, disconnectEvent.code, disconnectEvent.reason, state.socketId, this.logger, LIFECYCLE_LOG_CONTEXT);
|
|
331
400
|
}).catch(error => {
|
|
332
401
|
this.logger.error('WebSocket gateway disconnect dispatch failed.', error, LIFECYCLE_LOG_CONTEXT);
|
|
402
|
+
}).finally(() => {
|
|
403
|
+
this.settleDisconnectLifecycle(state);
|
|
333
404
|
});
|
|
334
405
|
}
|
|
335
406
|
async resolveConnectionGateways(state) {
|
|
@@ -379,6 +450,12 @@ class BunWebSocketGatewayLifecycleService {
|
|
|
379
450
|
return configured;
|
|
380
451
|
}
|
|
381
452
|
async resolveUpgradeRejection(request, path) {
|
|
453
|
+
if (this.isShuttingDown) {
|
|
454
|
+
return {
|
|
455
|
+
body: 'WebSocket server is shutting down.',
|
|
456
|
+
status: 503
|
|
457
|
+
};
|
|
458
|
+
}
|
|
382
459
|
if (!this.tryReserveUpgradeSlot()) {
|
|
383
460
|
return {
|
|
384
461
|
body: 'WebSocket connection limit exceeded.',
|
|
@@ -465,6 +542,13 @@ class BunWebSocketGatewayLifecycleService {
|
|
|
465
542
|
}
|
|
466
543
|
return Math.max(1, Math.ceil(configured / 1000));
|
|
467
544
|
}
|
|
545
|
+
resolveShutdownTimeoutMs() {
|
|
546
|
+
const configured = this.moduleOptions.shutdown?.timeoutMs;
|
|
547
|
+
if (typeof configured !== 'number' || !Number.isFinite(configured) || configured <= 0) {
|
|
548
|
+
return DEFAULT_WEBSOCKET_SHUTDOWN_TIMEOUT_MS;
|
|
549
|
+
}
|
|
550
|
+
return Math.floor(configured);
|
|
551
|
+
}
|
|
468
552
|
async shutdown() {
|
|
469
553
|
if (this.shutdownPromise) {
|
|
470
554
|
await this.shutdownPromise;
|
|
@@ -474,15 +558,110 @@ class BunWebSocketGatewayLifecycleService {
|
|
|
474
558
|
await this.shutdownPromise;
|
|
475
559
|
}
|
|
476
560
|
async runShutdownLifecycle() {
|
|
561
|
+
this.isShuttingDown = true;
|
|
477
562
|
if (hasBunWebSocketBindingHost(this.adapter)) {
|
|
478
563
|
const bunAdapter = this.adapter;
|
|
479
564
|
bunAdapter.configureWebSocketBinding(undefined);
|
|
480
565
|
}
|
|
566
|
+
const shutdownTimeoutMs = this.resolveShutdownTimeoutMs();
|
|
567
|
+
await this.awaitPendingUpgradeOperations(shutdownTimeoutMs);
|
|
568
|
+
await this.closeActiveSockets(shutdownTimeoutMs);
|
|
481
569
|
this.pendingUpgradeReservations = 0;
|
|
482
570
|
this.socketRegistry.clear();
|
|
483
571
|
this.socketRooms.clear();
|
|
572
|
+
this.socketStates.clear();
|
|
484
573
|
this.roomSockets.clear();
|
|
485
574
|
}
|
|
575
|
+
trackPendingUpgradeOperation(operation) {
|
|
576
|
+
let trackedOperation;
|
|
577
|
+
trackedOperation = operation.then(() => undefined, () => undefined).finally(() => {
|
|
578
|
+
if (trackedOperation) {
|
|
579
|
+
this.pendingUpgradeOperations.delete(trackedOperation);
|
|
580
|
+
}
|
|
581
|
+
});
|
|
582
|
+
this.pendingUpgradeOperations.add(trackedOperation);
|
|
583
|
+
return operation;
|
|
584
|
+
}
|
|
585
|
+
async awaitPendingUpgradeOperations(timeoutMs) {
|
|
586
|
+
if (this.pendingUpgradeOperations.size === 0) {
|
|
587
|
+
return;
|
|
588
|
+
}
|
|
589
|
+
await new Promise((resolve, reject) => {
|
|
590
|
+
let settled = false;
|
|
591
|
+
const timeout = setTimeout(() => {
|
|
592
|
+
if (settled) {
|
|
593
|
+
return;
|
|
594
|
+
}
|
|
595
|
+
settled = true;
|
|
596
|
+
reject(new Error(`Timed out while waiting for in-flight Bun websocket upgrades after ${String(timeoutMs)}ms.`));
|
|
597
|
+
}, timeoutMs);
|
|
598
|
+
Promise.all([...this.pendingUpgradeOperations]).then(() => {
|
|
599
|
+
if (settled) {
|
|
600
|
+
return;
|
|
601
|
+
}
|
|
602
|
+
settled = true;
|
|
603
|
+
clearTimeout(timeout);
|
|
604
|
+
resolve();
|
|
605
|
+
}).catch(error => {
|
|
606
|
+
if (settled) {
|
|
607
|
+
return;
|
|
608
|
+
}
|
|
609
|
+
settled = true;
|
|
610
|
+
clearTimeout(timeout);
|
|
611
|
+
reject(error);
|
|
612
|
+
});
|
|
613
|
+
}).catch(error => {
|
|
614
|
+
this.logger.error(`Failed to wait for in-flight Bun websocket upgrades within ${String(timeoutMs)}ms.`, error, LIFECYCLE_LOG_CONTEXT);
|
|
615
|
+
});
|
|
616
|
+
}
|
|
617
|
+
async closeActiveSockets(timeoutMs) {
|
|
618
|
+
const activeSockets = [...this.socketRegistry.entries()];
|
|
619
|
+
if (activeSockets.length === 0) {
|
|
620
|
+
return;
|
|
621
|
+
}
|
|
622
|
+
const activeStates = activeSockets.map(([socketId]) => this.socketStates.get(socketId)).filter(state => state !== undefined);
|
|
623
|
+
for (const [, socket] of activeSockets) {
|
|
624
|
+
if (socket.readyState === 1) {
|
|
625
|
+
socket.close(1001, 'Server shutting down');
|
|
626
|
+
}
|
|
627
|
+
}
|
|
628
|
+
await this.awaitHandlerQueueDrain(activeStates, timeoutMs);
|
|
629
|
+
}
|
|
630
|
+
async awaitHandlerQueueDrain(states, timeoutMs) {
|
|
631
|
+
if (states.length === 0) {
|
|
632
|
+
return;
|
|
633
|
+
}
|
|
634
|
+
await new Promise((resolve, reject) => {
|
|
635
|
+
let settled = false;
|
|
636
|
+
const timeout = setTimeout(() => {
|
|
637
|
+
if (settled) {
|
|
638
|
+
return;
|
|
639
|
+
}
|
|
640
|
+
settled = true;
|
|
641
|
+
reject(new Error(`Timed out while closing Bun websocket connections after ${String(timeoutMs)}ms.`));
|
|
642
|
+
}, timeoutMs);
|
|
643
|
+
Promise.all(states.map(async state => {
|
|
644
|
+
await state.connectLifecyclePromise;
|
|
645
|
+
await state.disconnectLifecyclePromise;
|
|
646
|
+
})).then(() => {
|
|
647
|
+
if (settled) {
|
|
648
|
+
return;
|
|
649
|
+
}
|
|
650
|
+
settled = true;
|
|
651
|
+
clearTimeout(timeout);
|
|
652
|
+
resolve();
|
|
653
|
+
}).catch(error => {
|
|
654
|
+
if (settled) {
|
|
655
|
+
return;
|
|
656
|
+
}
|
|
657
|
+
settled = true;
|
|
658
|
+
clearTimeout(timeout);
|
|
659
|
+
reject(error);
|
|
660
|
+
});
|
|
661
|
+
}).catch(error => {
|
|
662
|
+
this.logger.error(`Failed to close Bun websocket connections within ${String(timeoutMs)}ms.`, error, LIFECYCLE_LOG_CONTEXT);
|
|
663
|
+
});
|
|
664
|
+
}
|
|
486
665
|
joinRoom(socketId, room) {
|
|
487
666
|
let rooms = this.socketRooms.get(socketId);
|
|
488
667
|
if (!rooms) {
|
|
@@ -539,6 +718,7 @@ class BunWebSocketGatewayLifecycleService {
|
|
|
539
718
|
}
|
|
540
719
|
unregisterSocket(socketId) {
|
|
541
720
|
this.socketRegistry.delete(socketId);
|
|
721
|
+
this.socketStates.delete(socketId);
|
|
542
722
|
const rooms = this.socketRooms.get(socketId);
|
|
543
723
|
if (rooms) {
|
|
544
724
|
for (const room of rooms) {
|
package/dist/bun/bun-types.d.ts
CHANGED
|
@@ -1,5 +1,11 @@
|
|
|
1
1
|
import type { WebSocketModuleOptions as SharedWebSocketModuleOptions } from '../types.js';
|
|
2
|
+
/**
|
|
3
|
+
* Defines the bun web socket message type.
|
|
4
|
+
*/
|
|
2
5
|
export type BunWebSocketMessage = string | ArrayBuffer | Uint8Array;
|
|
6
|
+
/**
|
|
7
|
+
* Describes the bun server web socket contract.
|
|
8
|
+
*/
|
|
3
9
|
export interface BunServerWebSocket<TData = unknown> {
|
|
4
10
|
readonly data: TData;
|
|
5
11
|
readonly readyState: number;
|
|
@@ -13,6 +19,9 @@ export interface BunServerWebSocket<TData = unknown> {
|
|
|
13
19
|
subscribe(topic: string): void;
|
|
14
20
|
unsubscribe(topic: string): void;
|
|
15
21
|
}
|
|
22
|
+
/**
|
|
23
|
+
* Describes the bun server like contract.
|
|
24
|
+
*/
|
|
16
25
|
export interface BunServerLike {
|
|
17
26
|
fetch?(request: Request): Response | Promise<Response> | undefined | Promise<Response | undefined>;
|
|
18
27
|
hostname?: string;
|
|
@@ -24,6 +33,9 @@ export interface BunServerLike {
|
|
|
24
33
|
}): boolean;
|
|
25
34
|
url?: URL;
|
|
26
35
|
}
|
|
36
|
+
/**
|
|
37
|
+
* Describes the bun web socket handler contract.
|
|
38
|
+
*/
|
|
27
39
|
export interface BunWebSocketHandler<TData = unknown> {
|
|
28
40
|
backpressureLimit?: number;
|
|
29
41
|
close?(socket: BunServerWebSocket<TData>, code: number, reason: string): void | Promise<void>;
|
|
@@ -42,17 +54,32 @@ export interface BunWebSocketHandler<TData = unknown> {
|
|
|
42
54
|
publishToSelf?: boolean;
|
|
43
55
|
sendPings?: boolean;
|
|
44
56
|
}
|
|
57
|
+
/**
|
|
58
|
+
* Describes the bun web socket binding contract.
|
|
59
|
+
*/
|
|
45
60
|
export interface BunWebSocketBinding<TData = unknown> {
|
|
46
61
|
fetch(request: Request, server: BunServerLike): Response | Promise<Response> | undefined | Promise<Response | undefined>;
|
|
47
62
|
websocket: BunWebSocketHandler<TData>;
|
|
48
63
|
}
|
|
64
|
+
/**
|
|
65
|
+
* Describes the bun web socket binding host contract.
|
|
66
|
+
*/
|
|
49
67
|
export interface BunWebSocketBindingHost {
|
|
50
68
|
configureWebSocketBinding<TData>(binding: BunWebSocketBinding<TData> | undefined): void;
|
|
51
69
|
}
|
|
70
|
+
/**
|
|
71
|
+
* Defines the typed on message handler type.
|
|
72
|
+
*/
|
|
52
73
|
export type TypedOnMessageHandler<TEvents extends Record<string, unknown>, K extends keyof TEvents> = (payload: TEvents[K], socket: BunServerWebSocket, request: Request) => void | Promise<void>;
|
|
74
|
+
/**
|
|
75
|
+
* Describes the web socket gateway context contract.
|
|
76
|
+
*/
|
|
53
77
|
export interface WebSocketGatewayContext {
|
|
54
78
|
request: Request;
|
|
55
79
|
socket: BunServerWebSocket;
|
|
56
80
|
}
|
|
81
|
+
/**
|
|
82
|
+
* Defines the web socket module options type.
|
|
83
|
+
*/
|
|
57
84
|
export type WebSocketModuleOptions = SharedWebSocketModuleOptions;
|
|
58
85
|
//# sourceMappingURL=bun-types.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"bun-types.d.ts","sourceRoot":"","sources":["../../src/bun/bun-types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,sBAAsB,IAAI,4BAA4B,EAAE,MAAM,aAAa,CAAC;AAE1F,MAAM,MAAM,mBAAmB,GAAG,MAAM,GAAG,WAAW,GAAG,UAAU,CAAC;AAEpE,MAAM,WAAW,kBAAkB,CAAC,KAAK,GAAG,OAAO;IACjD,QAAQ,CAAC,IAAI,EAAE,KAAK,CAAC;IACrB,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC;IAC5B,QAAQ,CAAC,aAAa,EAAE,MAAM,CAAC;IAC/B,QAAQ,CAAC,aAAa,EAAE,MAAM,EAAE,CAAC;IACjC,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5C,IAAI,CAAC,QAAQ,EAAE,CAAC,MAAM,EAAE,kBAAkB,CAAC,KAAK,CAAC,KAAK,IAAI,GAAG,IAAI,CAAC;IAClE,YAAY,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC;IACrC,OAAO,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,mBAAmB,GAAG,IAAI,CAAC;IAC3D,IAAI,CAAC,OAAO,EAAE,mBAAmB,EAAE,QAAQ,CAAC,EAAE,OAAO,GAAG,MAAM,CAAC;IAC/D,SAAS,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;IAC/B,WAAW,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;CAClC;AAED,MAAM,WAAW,aAAa;IAC5B,KAAK,CAAC,CAAC,OAAO,EAAE,OAAO,GAAG,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC,GAAG,SAAS,GAAG,OAAO,CAAC,QAAQ,GAAG,SAAS,CAAC,CAAC;IACnG,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,sBAAsB,CAAC,EAAE,OAAO,GAAG,IAAI,CAAC;IAC7C,OAAO,CAAC,KAAK,GAAG,OAAO,EACrB,OAAO,EAAE,OAAO,EAChB,OAAO,CAAC,EAAE;QACR,IAAI,CAAC,EAAE,KAAK,CAAC;QACb,OAAO,CAAC,EAAE,WAAW,CAAC;KACvB,GACA,OAAO,CAAC;IACX,GAAG,CAAC,EAAE,GAAG,CAAC;CACX;AAED,MAAM,WAAW,mBAAmB,CAAC,KAAK,GAAG,OAAO;IAClD,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,KAAK,CAAC,CAAC,MAAM,EAAE,kBAAkB,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC9F,wBAAwB,CAAC,EAAE,OAAO,CAAC;IACnC,IAAI,CAAC,EAAE,KAAK,CAAC;IACb,KAAK,CAAC,CAAC,MAAM,EAAE,kBAAkB,CAAC,KAAK,CAAC,GAAG,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAChE,KAAK,CAAC,CAAC,MAAM,EAAE,kBAAkB,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,KAAK,GAAG,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC9E,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,OAAO,CAAC,CAAC,MAAM,EAAE,kBAAkB,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,mBAAmB,GAAG,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAChG,IAAI,CAAC,CAAC,MAAM,EAAE,kBAAkB,CAAC,KAAK,CAAC,GAAG,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC/D,iBAAiB,CAAC,EACd,OAAO,GACP;QACE,QAAQ,CAAC,EAAE,OAAO,GAAG,OAAO,GAAG,MAAM,GAAG,OAAO,GAAG,MAAM,GAAG,KAAK,GAAG,KAAK,GAAG,MAAM,GAAG,KAAK,GAAG,WAAW,GAAG,SAAS,GAAG,QAAQ,CAAC;QAC/H,UAAU,CAAC,EAAE,OAAO,GAAG,OAAO,GAAG,MAAM,GAAG,OAAO,GAAG,MAAM,GAAG,KAAK,GAAG,KAAK,GAAG,MAAM,GAAG,KAAK,GAAG,WAAW,GAAG,SAAS,GAAG,QAAQ,CAAC;KAClI,CAAC;IACN,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,SAAS,CAAC,EAAE,OAAO,CAAC;CACrB;AAED,MAAM,WAAW,mBAAmB,CAAC,KAAK,GAAG,OAAO;IAClD,KAAK,CAAC,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,aAAa,GAAG,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC,GAAG,SAAS,GAAG,OAAO,CAAC,QAAQ,GAAG,SAAS,CAAC,CAAC;IACzH,SAAS,EAAE,mBAAmB,CAAC,KAAK,CAAC,CAAC;CACvC;AAED,MAAM,WAAW,uBAAuB;IACtC,yBAAyB,CAAC,KAAK,EAAE,OAAO,EAAE,mBAAmB,CAAC,KAAK,CAAC,GAAG,SAAS,GAAG,IAAI,CAAC;CACzF;AAED,MAAM,MAAM,qBAAqB,CAAC,OAAO,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,CAAC,SAAS,MAAM,OAAO,IAAI,CACpG,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC,EACnB,MAAM,EAAE,kBAAkB,EAC1B,OAAO,EAAE,OAAO,KACb,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;AAE1B,MAAM,WAAW,uBAAuB;IACtC,OAAO,EAAE,OAAO,CAAC;IACjB,MAAM,EAAE,kBAAkB,CAAC;CAC5B;AAED,MAAM,MAAM,sBAAsB,GAAG,4BAA4B,CAAC"}
|
|
1
|
+
{"version":3,"file":"bun-types.d.ts","sourceRoot":"","sources":["../../src/bun/bun-types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,sBAAsB,IAAI,4BAA4B,EAAE,MAAM,aAAa,CAAC;AAE1F;;GAEG;AACH,MAAM,MAAM,mBAAmB,GAAG,MAAM,GAAG,WAAW,GAAG,UAAU,CAAC;AAEpE;;GAEG;AACH,MAAM,WAAW,kBAAkB,CAAC,KAAK,GAAG,OAAO;IACjD,QAAQ,CAAC,IAAI,EAAE,KAAK,CAAC;IACrB,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC;IAC5B,QAAQ,CAAC,aAAa,EAAE,MAAM,CAAC;IAC/B,QAAQ,CAAC,aAAa,EAAE,MAAM,EAAE,CAAC;IACjC,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5C,IAAI,CAAC,QAAQ,EAAE,CAAC,MAAM,EAAE,kBAAkB,CAAC,KAAK,CAAC,KAAK,IAAI,GAAG,IAAI,CAAC;IAClE,YAAY,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC;IACrC,OAAO,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,mBAAmB,GAAG,IAAI,CAAC;IAC3D,IAAI,CAAC,OAAO,EAAE,mBAAmB,EAAE,QAAQ,CAAC,EAAE,OAAO,GAAG,MAAM,CAAC;IAC/D,SAAS,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;IAC/B,WAAW,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;CAClC;AAED;;GAEG;AACH,MAAM,WAAW,aAAa;IAC5B,KAAK,CAAC,CAAC,OAAO,EAAE,OAAO,GAAG,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC,GAAG,SAAS,GAAG,OAAO,CAAC,QAAQ,GAAG,SAAS,CAAC,CAAC;IACnG,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,sBAAsB,CAAC,EAAE,OAAO,GAAG,IAAI,CAAC;IAC7C,OAAO,CAAC,KAAK,GAAG,OAAO,EACrB,OAAO,EAAE,OAAO,EAChB,OAAO,CAAC,EAAE;QACR,IAAI,CAAC,EAAE,KAAK,CAAC;QACb,OAAO,CAAC,EAAE,WAAW,CAAC;KACvB,GACA,OAAO,CAAC;IACX,GAAG,CAAC,EAAE,GAAG,CAAC;CACX;AAED;;GAEG;AACH,MAAM,WAAW,mBAAmB,CAAC,KAAK,GAAG,OAAO;IAClD,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,KAAK,CAAC,CAAC,MAAM,EAAE,kBAAkB,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC9F,wBAAwB,CAAC,EAAE,OAAO,CAAC;IACnC,IAAI,CAAC,EAAE,KAAK,CAAC;IACb,KAAK,CAAC,CAAC,MAAM,EAAE,kBAAkB,CAAC,KAAK,CAAC,GAAG,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAChE,KAAK,CAAC,CAAC,MAAM,EAAE,kBAAkB,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,KAAK,GAAG,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC9E,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,OAAO,CAAC,CAAC,MAAM,EAAE,kBAAkB,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,mBAAmB,GAAG,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAChG,IAAI,CAAC,CAAC,MAAM,EAAE,kBAAkB,CAAC,KAAK,CAAC,GAAG,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC/D,iBAAiB,CAAC,EACd,OAAO,GACP;QACE,QAAQ,CAAC,EAAE,OAAO,GAAG,OAAO,GAAG,MAAM,GAAG,OAAO,GAAG,MAAM,GAAG,KAAK,GAAG,KAAK,GAAG,MAAM,GAAG,KAAK,GAAG,WAAW,GAAG,SAAS,GAAG,QAAQ,CAAC;QAC/H,UAAU,CAAC,EAAE,OAAO,GAAG,OAAO,GAAG,MAAM,GAAG,OAAO,GAAG,MAAM,GAAG,KAAK,GAAG,KAAK,GAAG,MAAM,GAAG,KAAK,GAAG,WAAW,GAAG,SAAS,GAAG,QAAQ,CAAC;KAClI,CAAC;IACN,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,SAAS,CAAC,EAAE,OAAO,CAAC;CACrB;AAED;;GAEG;AACH,MAAM,WAAW,mBAAmB,CAAC,KAAK,GAAG,OAAO;IAClD,KAAK,CAAC,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,aAAa,GAAG,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC,GAAG,SAAS,GAAG,OAAO,CAAC,QAAQ,GAAG,SAAS,CAAC,CAAC;IACzH,SAAS,EAAE,mBAAmB,CAAC,KAAK,CAAC,CAAC;CACvC;AAED;;GAEG;AACH,MAAM,WAAW,uBAAuB;IACtC,yBAAyB,CAAC,KAAK,EAAE,OAAO,EAAE,mBAAmB,CAAC,KAAK,CAAC,GAAG,SAAS,GAAG,IAAI,CAAC;CACzF;AAED;;GAEG;AACH,MAAM,MAAM,qBAAqB,CAAC,OAAO,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,CAAC,SAAS,MAAM,OAAO,IAAI,CACpG,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC,EACnB,MAAM,EAAE,kBAAkB,EAC1B,OAAO,EAAE,OAAO,KACb,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;AAE1B;;GAEG;AACH,MAAM,WAAW,uBAAuB;IACtC,OAAO,EAAE,OAAO,CAAC;IACjB,MAAM,EAAE,kBAAkB,CAAC;CAC5B;AAED;;GAEG;AACH,MAAM,MAAM,sBAAsB,GAAG,4BAA4B,CAAC"}
|
|
@@ -12,20 +12,26 @@ export declare class CloudflareWorkersWebSocketGatewayLifecycleService implement
|
|
|
12
12
|
private readonly logger;
|
|
13
13
|
private readonly adapter;
|
|
14
14
|
private readonly moduleOptions;
|
|
15
|
+
private isShuttingDown;
|
|
16
|
+
private readonly pendingUpgradeOperations;
|
|
15
17
|
private pendingUpgradeReservations;
|
|
16
18
|
private readonly roomSockets;
|
|
17
19
|
private shutdownPromise;
|
|
18
20
|
private readonly socketRegistry;
|
|
19
21
|
private readonly socketRooms;
|
|
22
|
+
private readonly socketStates;
|
|
20
23
|
constructor(runtimeContainer: Container, compiledModules: readonly CompiledModule[], logger: ApplicationLogger, adapter: HttpApplicationAdapter, moduleOptions: WebSocketModuleOptions);
|
|
21
24
|
onApplicationBootstrap(): Promise<void>;
|
|
22
25
|
private assertNoServerBackedGatewayOptIn;
|
|
23
26
|
onApplicationShutdown(): Promise<void>;
|
|
24
27
|
onModuleDestroy(): Promise<void>;
|
|
25
28
|
private createBinding;
|
|
29
|
+
private handleUpgradeRequest;
|
|
26
30
|
private groupDescriptorsByPath;
|
|
27
31
|
private bindConnectionHandlers;
|
|
28
32
|
private createConnectionHandlerState;
|
|
33
|
+
private settleConnectLifecycle;
|
|
34
|
+
private settleDisconnectLifecycle;
|
|
29
35
|
private attachConnectionListeners;
|
|
30
36
|
private getBufferedMessageCount;
|
|
31
37
|
private getQueuedMessageCount;
|
|
@@ -50,8 +56,13 @@ export declare class CloudflareWorkersWebSocketGatewayLifecycleService implement
|
|
|
50
56
|
private tryReserveUpgradeSlot;
|
|
51
57
|
private releaseUpgradeReservation;
|
|
52
58
|
private resolveMaxPayloadBytes;
|
|
59
|
+
private resolveShutdownTimeoutMs;
|
|
53
60
|
private shutdown;
|
|
54
61
|
private runShutdownLifecycle;
|
|
62
|
+
private trackPendingUpgradeOperation;
|
|
63
|
+
private awaitPendingUpgradeOperations;
|
|
64
|
+
private closeActiveSockets;
|
|
65
|
+
private awaitHandlerQueueDrain;
|
|
55
66
|
joinRoom(socketId: string, room: string): void;
|
|
56
67
|
leaveRoom(socketId: string, room: string): void;
|
|
57
68
|
broadcastToRoom(room: string, event: string, data: unknown): void;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"cloudflare-workers-service.d.ts","sourceRoot":"","sources":["../../src/cloudflare-workers/cloudflare-workers-service.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AAC5C,OAAO,KAAK,EAAE,iBAAiB,EAAE,cAAc,EAAE,sBAAsB,EAAE,qBAAqB,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC;AAEzI,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,cAAc,CAAC;AAa3D,OAAO,KAAK,EAA8B,oBAAoB,EAA6B,MAAM,aAAa,CAAC;AAC/G,OAAO,KAAK,EAKV,sBAAsB,EACvB,MAAM,+BAA+B,CAAC;
|
|
1
|
+
{"version":3,"file":"cloudflare-workers-service.d.ts","sourceRoot":"","sources":["../../src/cloudflare-workers/cloudflare-workers-service.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AAC5C,OAAO,KAAK,EAAE,iBAAiB,EAAE,cAAc,EAAE,sBAAsB,EAAE,qBAAqB,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC;AAEzI,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,cAAc,CAAC;AAa3D,OAAO,KAAK,EAA8B,oBAAoB,EAA6B,MAAM,aAAa,CAAC;AAC/G,OAAO,KAAK,EAKV,sBAAsB,EACvB,MAAM,+BAA+B,CAAC;AA0HvC;;GAEG;AACH,qBACa,iDACX,YAAW,sBAAsB,EAAE,qBAAqB,EAAE,eAAe,EAAE,oBAAoB;IAY7F,OAAO,CAAC,QAAQ,CAAC,gBAAgB;IACjC,OAAO,CAAC,QAAQ,CAAC,eAAe;IAChC,OAAO,CAAC,QAAQ,CAAC,MAAM;IACvB,OAAO,CAAC,QAAQ,CAAC,OAAO;IACxB,OAAO,CAAC,QAAQ,CAAC,aAAa;IAdhC,OAAO,CAAC,cAAc,CAAS;IAC/B,OAAO,CAAC,QAAQ,CAAC,wBAAwB,CAA4B;IACrE,OAAO,CAAC,0BAA0B,CAAK;IACvC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAkC;IAC9D,OAAO,CAAC,eAAe,CAA4B;IACnD,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAgD;IAC/E,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAkC;IAC9D,OAAO,CAAC,QAAQ,CAAC,YAAY,CAA6C;gBAGvD,gBAAgB,EAAE,SAAS,EAC3B,eAAe,EAAE,SAAS,cAAc,EAAE,EAC1C,MAAM,EAAE,iBAAiB,EACzB,OAAO,EAAE,sBAAsB,EAC/B,aAAa,EAAE,sBAAsB;IAGlD,sBAAsB,IAAI,OAAO,CAAC,IAAI,CAAC;IAoB7C,OAAO,CAAC,gCAAgC;IAclC,qBAAqB,IAAI,OAAO,CAAC,IAAI,CAAC;IAItC,eAAe,IAAI,OAAO,CAAC,IAAI,CAAC;IAItC,OAAO,CAAC,aAAa;YAUP,oBAAoB;IAwDlC,OAAO,CAAC,sBAAsB;YAmBhB,sBAAsB;IA8BpC,OAAO,CAAC,4BAA4B;IA8BpC,OAAO,CAAC,sBAAsB;IAS9B,OAAO,CAAC,yBAAyB;IASjC,OAAO,CAAC,yBAAyB;IAyCjC,OAAO,CAAC,uBAAuB;IAI/B,OAAO,CAAC,qBAAqB;IAI7B,OAAO,CAAC,4BAA4B;IASpC,OAAO,CAAC,qBAAqB;IAK7B,OAAO,CAAC,0BAA0B;IASlC,OAAO,CAAC,mBAAmB;IAM3B,OAAO,CAAC,qBAAqB;IA0C7B,OAAO,CAAC,sBAAsB;YAyDhB,iBAAiB;YAqBjB,gBAAgB;IAU9B,OAAO,CAAC,yBAAyB;YAyBnB,yBAAyB;YAczB,kBAAkB;YAgBlB,yBAAyB;YAUzB,8BAA8B;IAqB5C,OAAO,CAAC,YAAY;YAUN,uBAAuB;IA0DrC,OAAO,CAAC,qBAAqB;IAmB7B,OAAO,CAAC,yBAAyB;IAUjC,OAAO,CAAC,8BAA8B;IAItC,OAAO,CAAC,qBAAqB;IAS7B,OAAO,CAAC,yBAAyB;IAMjC,OAAO,CAAC,sBAAsB;IAU9B,OAAO,CAAC,wBAAwB;YAUlB,QAAQ;YAUR,oBAAoB;IAmBlC,OAAO,CAAC,4BAA4B;YAyBtB,6BAA6B;YA4C7B,kBAAkB;YAoBlB,sBAAsB;IAkDpC,QAAQ,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,IAAI;IAmB9C,SAAS,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,IAAI;IAc/C,eAAe,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,GAAG,IAAI;IA4BjE,QAAQ,CAAC,QAAQ,EAAE,MAAM,GAAG,WAAW,CAAC,MAAM,CAAC;IAU/C,OAAO,CAAC,gBAAgB;CAqBzB"}
|