@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 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;AAyG7D;;GAEG;AACH,qBACa,mCACX,YAAW,sBAAsB,EAAE,qBAAqB,EAAE,eAAe,EAAE,oBAAoB;IAS7F,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;IAXhC,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;gBAG3C,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;IAiCrB,OAAO,CAAC,sBAAsB;YAmBhB,oBAAoB;YAkDpB,sBAAsB;IAWpC,OAAO,CAAC,4BAA4B;IAqBpC,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;YAsBnB,yBAAyB;YAczB,kBAAkB;YASlB,yBAAyB;IASvC,OAAO,CAAC,8BAA8B;IAoBtC,OAAO,CAAC,wBAAwB;YAUlB,uBAAuB;IAmDrC,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;YAcnB,QAAQ;YAUR,oBAAoB;IAYlC,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;CAgBzB"}
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"}
@@ -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
- await this.resolveConnectionGateways(state);
185
- await this.runConnectHandlers(state, socket);
186
- await this.finalizeConnectionBinding(state, socket);
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) {
@@ -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;AA0GvC;;GAEG;AACH,qBACa,iDACX,YAAW,sBAAsB,EAAE,qBAAqB,EAAE,eAAe,EAAE,oBAAoB;IAS7F,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;IAXhC,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;gBAG3C,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;IAqDrB,OAAO,CAAC,sBAAsB;YAmBhB,sBAAsB;IAgBpC,OAAO,CAAC,4BAA4B;IAqBpC,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;YAsBnB,yBAAyB;YAczB,kBAAkB;YAgBlB,yBAAyB;YAUzB,8BAA8B;IAqB5C,OAAO,CAAC,YAAY;YAUN,uBAAuB;IAmDrC,OAAO,CAAC,qBAAqB;IAmB7B,OAAO,CAAC,yBAAyB;IAUjC,OAAO,CAAC,8BAA8B;IAItC,OAAO,CAAC,qBAAqB;IAS7B,OAAO,CAAC,yBAAyB;IAMjC,OAAO,CAAC,sBAAsB;YAUhB,QAAQ;YAUR,oBAAoB;IAWlC,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;CAoBzB"}
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"}