@replit/river 0.23.12 → 0.23.14

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 (75) hide show
  1. package/dist/{chunk-3AW3IXVD.js → chunk-4PVU7J25.js} +1 -21
  2. package/dist/chunk-4PVU7J25.js.map +1 -0
  3. package/dist/{chunk-HDBVL7EF.js → chunk-BEALFLCB.js} +2 -2
  4. package/dist/chunk-D2DHRRBN.js +476 -0
  5. package/dist/chunk-D2DHRRBN.js.map +1 -0
  6. package/dist/{chunk-7RUKEUKE.js → chunk-GCCRVSMR.js} +33 -4
  7. package/dist/chunk-GCCRVSMR.js.map +1 -0
  8. package/dist/{chunk-XZ6IOBM5.js → chunk-GN4YEXT7.js} +2 -2
  9. package/dist/chunk-GN4YEXT7.js.map +1 -0
  10. package/dist/chunk-O2AVDJCQ.js +335 -0
  11. package/dist/chunk-O2AVDJCQ.js.map +1 -0
  12. package/dist/chunk-OTVTKAN6.js +451 -0
  13. package/dist/chunk-OTVTKAN6.js.map +1 -0
  14. package/dist/chunk-WUL63FR6.js +335 -0
  15. package/dist/chunk-WUL63FR6.js.map +1 -0
  16. package/dist/{chunk-H6KTH6W6.js → chunk-YCLZWES2.js} +2 -2
  17. package/dist/client-e13979ac.d.ts +52 -0
  18. package/dist/codec/index.js +20 -2
  19. package/dist/codec/index.js.map +1 -1
  20. package/dist/{connection-8debd45f.d.ts → connection-5d0978ce.d.ts} +1 -1
  21. package/dist/{connection-581558f8.d.ts → connection-e57e98ea.d.ts} +1 -1
  22. package/dist/{transport-47af1c81.d.ts → handshake-5665ffd3.d.ts} +101 -153
  23. package/dist/{index-60f03cb7.d.ts → index-ea74cdbb.d.ts} +1 -1
  24. package/dist/logging/index.d.cts +1 -1
  25. package/dist/logging/index.d.ts +1 -1
  26. package/dist/router/index.cjs +16 -1
  27. package/dist/router/index.cjs.map +1 -1
  28. package/dist/router/index.d.cts +8 -6
  29. package/dist/router/index.d.ts +8 -6
  30. package/dist/router/index.js +2 -2
  31. package/dist/server-1cfc88d1.d.ts +24 -0
  32. package/dist/{services-ca72c9f8.d.ts → services-86c4d10d.d.ts} +3 -2
  33. package/dist/transport/impls/uds/client.cjs +303 -180
  34. package/dist/transport/impls/uds/client.cjs.map +1 -1
  35. package/dist/transport/impls/uds/client.d.cts +6 -5
  36. package/dist/transport/impls/uds/client.d.ts +6 -5
  37. package/dist/transport/impls/uds/client.js +6 -4
  38. package/dist/transport/impls/uds/client.js.map +1 -1
  39. package/dist/transport/impls/uds/server.cjs +396 -234
  40. package/dist/transport/impls/uds/server.cjs.map +1 -1
  41. package/dist/transport/impls/uds/server.d.cts +6 -5
  42. package/dist/transport/impls/uds/server.d.ts +6 -5
  43. package/dist/transport/impls/uds/server.js +8 -6
  44. package/dist/transport/impls/uds/server.js.map +1 -1
  45. package/dist/transport/impls/ws/client.cjs +305 -182
  46. package/dist/transport/impls/ws/client.cjs.map +1 -1
  47. package/dist/transport/impls/ws/client.d.cts +6 -5
  48. package/dist/transport/impls/ws/client.d.ts +6 -5
  49. package/dist/transport/impls/ws/client.js +6 -4
  50. package/dist/transport/impls/ws/client.js.map +1 -1
  51. package/dist/transport/impls/ws/server.cjs +350 -188
  52. package/dist/transport/impls/ws/server.cjs.map +1 -1
  53. package/dist/transport/impls/ws/server.d.cts +4 -3
  54. package/dist/transport/impls/ws/server.d.ts +4 -3
  55. package/dist/transport/impls/ws/server.js +8 -6
  56. package/dist/transport/impls/ws/server.js.map +1 -1
  57. package/dist/transport/index.cjs +338 -142
  58. package/dist/transport/index.cjs.map +1 -1
  59. package/dist/transport/index.d.cts +4 -2
  60. package/dist/transport/index.d.ts +4 -2
  61. package/dist/transport/index.js +14 -8
  62. package/dist/util/testHelpers.cjs +10 -6
  63. package/dist/util/testHelpers.cjs.map +1 -1
  64. package/dist/util/testHelpers.d.cts +5 -4
  65. package/dist/util/testHelpers.d.ts +5 -4
  66. package/dist/util/testHelpers.js +4 -5
  67. package/dist/util/testHelpers.js.map +1 -1
  68. package/package.json +13 -14
  69. package/dist/chunk-3AW3IXVD.js.map +0 -1
  70. package/dist/chunk-7RUKEUKE.js.map +0 -1
  71. package/dist/chunk-VRU4IKRT.js +0 -1392
  72. package/dist/chunk-VRU4IKRT.js.map +0 -1
  73. package/dist/chunk-XZ6IOBM5.js.map +0 -1
  74. /package/dist/{chunk-HDBVL7EF.js.map → chunk-BEALFLCB.js.map} +0 -0
  75. /package/dist/{chunk-H6KTH6W6.js.map → chunk-YCLZWES2.js.map} +0 -0
@@ -0,0 +1,335 @@
1
+ import {
2
+ createSessionTelemetryInfo
3
+ } from "./chunk-GCCRVSMR.js";
4
+ import {
5
+ NaiveJsonCodec
6
+ } from "./chunk-4PVU7J25.js";
7
+
8
+ // transport/session.ts
9
+ import { customAlphabet } from "nanoid";
10
+ import { SpanStatusCode } from "@opentelemetry/api";
11
+ var nanoid = customAlphabet("1234567890abcdefghijklmnopqrstuvxyz", 6);
12
+ var unsafeId = () => nanoid();
13
+ var Connection = class {
14
+ id;
15
+ telemetry;
16
+ constructor() {
17
+ this.id = `conn-${nanoid(12)}`;
18
+ }
19
+ get loggingMetadata() {
20
+ const metadata = { connId: this.id };
21
+ const spanContext = this.telemetry?.span.spanContext();
22
+ if (this.telemetry?.span.isRecording() && spanContext) {
23
+ metadata.telemetry = {
24
+ traceId: spanContext.traceId,
25
+ spanId: spanContext.spanId
26
+ };
27
+ }
28
+ return metadata;
29
+ }
30
+ };
31
+ var Session = class {
32
+ codec;
33
+ options;
34
+ telemetry;
35
+ /**
36
+ * The buffer of messages that have been sent but not yet acknowledged.
37
+ */
38
+ sendBuffer = [];
39
+ /**
40
+ * The active connection associated with this session
41
+ */
42
+ connection;
43
+ /**
44
+ * A connection that is currently undergoing handshaking. Used to distinguish between the active
45
+ * connection, but still be able to close it if needed.
46
+ */
47
+ handshakingConnection;
48
+ from;
49
+ to;
50
+ /**
51
+ * The unique ID of this session.
52
+ */
53
+ id;
54
+ /**
55
+ * What the other side advertised as their session ID
56
+ * for this session.
57
+ */
58
+ advertisedSessionId;
59
+ /**
60
+ * Number of messages we've sent along this session (excluding handshake and acks)
61
+ */
62
+ seq = 0;
63
+ /**
64
+ * Number of unique messages we've received this session (excluding handshake and acks)
65
+ */
66
+ ack = 0;
67
+ /**
68
+ * The grace period between when the inner connection is disconnected
69
+ * and when we should consider the entire session disconnected.
70
+ */
71
+ disconnectionGrace;
72
+ /**
73
+ * Number of heartbeats we've sent without a response.
74
+ */
75
+ heartbeatMisses;
76
+ /**
77
+ * The interval for sending heartbeats.
78
+ */
79
+ heartbeat;
80
+ log;
81
+ constructor(conn, from, to, options, propagationCtx) {
82
+ this.id = `session-${nanoid(12)}`;
83
+ this.options = options;
84
+ this.from = from;
85
+ this.to = to;
86
+ this.connection = conn;
87
+ this.codec = options.codec;
88
+ this.heartbeatMisses = 0;
89
+ this.heartbeat = setInterval(
90
+ () => this.sendHeartbeat(),
91
+ options.heartbeatIntervalMs
92
+ );
93
+ this.telemetry = createSessionTelemetryInfo(this, propagationCtx);
94
+ }
95
+ bindLogger(log) {
96
+ this.log = log;
97
+ }
98
+ get loggingMetadata() {
99
+ const spanContext = this.telemetry.span.spanContext();
100
+ return {
101
+ clientId: this.from,
102
+ connectedTo: this.to,
103
+ sessionId: this.id,
104
+ connId: this.connection?.id,
105
+ telemetry: {
106
+ traceId: spanContext.traceId,
107
+ spanId: spanContext.spanId
108
+ }
109
+ };
110
+ }
111
+ /**
112
+ * Sends a message over the session's connection.
113
+ * If the connection is not ready or the message fails to send, the message can be buffered for retry unless skipped.
114
+ *
115
+ * @param msg The partial message to be sent, which will be constructed into a full message.
116
+ * @param addToSendBuff Whether to add the message to the send buffer for retry.
117
+ * @returns The full transport ID of the message that was attempted to be sent.
118
+ */
119
+ send(msg) {
120
+ const fullMsg = this.constructMsg(msg);
121
+ this.log?.debug(`sending msg`, {
122
+ ...this.loggingMetadata,
123
+ transportMessage: fullMsg
124
+ });
125
+ if (this.connection) {
126
+ const ok = this.connection.send(this.codec.toBuffer(fullMsg));
127
+ if (ok)
128
+ return fullMsg.id;
129
+ this.log?.info(
130
+ `failed to send msg to ${fullMsg.to}, connection is probably dead`,
131
+ {
132
+ ...this.loggingMetadata,
133
+ transportMessage: fullMsg
134
+ }
135
+ );
136
+ } else {
137
+ this.log?.debug(
138
+ `buffering msg to ${fullMsg.to}, connection not ready yet`,
139
+ { ...this.loggingMetadata, transportMessage: fullMsg }
140
+ );
141
+ }
142
+ return fullMsg.id;
143
+ }
144
+ sendHeartbeat() {
145
+ const misses = this.heartbeatMisses;
146
+ const missDuration = misses * this.options.heartbeatIntervalMs;
147
+ if (misses > this.options.heartbeatsUntilDead) {
148
+ if (this.connection) {
149
+ this.log?.info(
150
+ `closing connection to ${this.to} due to inactivity (missed ${misses} heartbeats which is ${missDuration}ms)`,
151
+ this.loggingMetadata
152
+ );
153
+ this.telemetry.span.addEvent("closing connection due to inactivity");
154
+ this.closeStaleConnection();
155
+ }
156
+ return;
157
+ }
158
+ this.send({
159
+ streamId: "heartbeat",
160
+ controlFlags: 1 /* AckBit */,
161
+ payload: {
162
+ type: "ACK"
163
+ }
164
+ });
165
+ this.heartbeatMisses++;
166
+ }
167
+ resetBufferedMessages() {
168
+ this.sendBuffer = [];
169
+ this.seq = 0;
170
+ this.ack = 0;
171
+ }
172
+ sendBufferedMessages(conn) {
173
+ this.log?.info(`resending ${this.sendBuffer.length} buffered messages`, {
174
+ ...this.loggingMetadata,
175
+ connId: conn.id
176
+ });
177
+ for (const msg of this.sendBuffer) {
178
+ this.log?.debug(`resending msg`, {
179
+ ...this.loggingMetadata,
180
+ transportMessage: msg,
181
+ connId: conn.id
182
+ });
183
+ const ok = conn.send(this.codec.toBuffer(msg));
184
+ if (!ok) {
185
+ const errMsg = `failed to send buffered message to ${this.to} (sus, this is a fresh connection)`;
186
+ conn.telemetry?.span.setStatus({
187
+ code: SpanStatusCode.ERROR,
188
+ message: errMsg
189
+ });
190
+ this.log?.error(errMsg, {
191
+ ...this.loggingMetadata,
192
+ transportMessage: msg,
193
+ connId: conn.id,
194
+ tags: ["invariant-violation"]
195
+ });
196
+ conn.close();
197
+ return;
198
+ }
199
+ }
200
+ }
201
+ updateBookkeeping(ack, seq) {
202
+ if (seq + 1 < this.ack) {
203
+ this.log?.error(`received stale seq ${seq} + 1 < ${this.ack}`, {
204
+ ...this.loggingMetadata,
205
+ tags: ["invariant-violation"]
206
+ });
207
+ return;
208
+ }
209
+ this.sendBuffer = this.sendBuffer.filter((unacked) => unacked.seq >= ack);
210
+ this.ack = seq + 1;
211
+ }
212
+ closeStaleConnection(conn) {
213
+ if (this.connection === void 0 || this.connection === conn)
214
+ return;
215
+ this.log?.info(
216
+ `closing old inner connection from session to ${this.to}`,
217
+ this.loggingMetadata
218
+ );
219
+ this.connection.close();
220
+ this.connection = void 0;
221
+ }
222
+ replaceWithNewConnection(newConn, isTransparentReconnect) {
223
+ this.closeStaleConnection(newConn);
224
+ this.cancelGrace();
225
+ if (isTransparentReconnect) {
226
+ this.sendBufferedMessages(newConn);
227
+ }
228
+ this.connection = newConn;
229
+ this.handshakingConnection = void 0;
230
+ }
231
+ replaceWithNewHandshakingConnection(newConn) {
232
+ this.handshakingConnection = newConn;
233
+ }
234
+ beginGrace(cb) {
235
+ this.log?.info(
236
+ `starting ${this.options.sessionDisconnectGraceMs}ms grace period until session to ${this.to} is closed`,
237
+ this.loggingMetadata
238
+ );
239
+ this.cancelGrace();
240
+ this.disconnectionGrace = setTimeout(() => {
241
+ this.log?.info(
242
+ `grace period for ${this.to} elapsed`,
243
+ this.loggingMetadata
244
+ );
245
+ cb();
246
+ }, this.options.sessionDisconnectGraceMs);
247
+ }
248
+ // called on reconnect of the underlying session
249
+ cancelGrace() {
250
+ this.heartbeatMisses = 0;
251
+ clearTimeout(this.disconnectionGrace);
252
+ this.disconnectionGrace = void 0;
253
+ }
254
+ /**
255
+ * Used to close the handshaking connection, if set.
256
+ */
257
+ closeHandshakingConnection(expectedHandshakingConn) {
258
+ if (this.handshakingConnection === void 0)
259
+ return;
260
+ if (expectedHandshakingConn !== void 0 && this.handshakingConnection === expectedHandshakingConn) {
261
+ return;
262
+ }
263
+ this.handshakingConnection.close();
264
+ this.handshakingConnection = void 0;
265
+ }
266
+ // closed when we want to discard the whole session
267
+ // (i.e. shutdown or session disconnect)
268
+ close() {
269
+ this.closeStaleConnection();
270
+ this.cancelGrace();
271
+ this.resetBufferedMessages();
272
+ clearInterval(this.heartbeat);
273
+ }
274
+ get connected() {
275
+ return this.connection !== void 0;
276
+ }
277
+ get nextExpectedAck() {
278
+ return this.seq;
279
+ }
280
+ get nextExpectedSeq() {
281
+ return this.ack;
282
+ }
283
+ // This is only used in tests to make the session misbehave.
284
+ /* @internal */
285
+ advanceAckForTesting(by) {
286
+ this.ack += by;
287
+ }
288
+ constructMsg(partialMsg) {
289
+ const msg = {
290
+ ...partialMsg,
291
+ id: unsafeId(),
292
+ to: this.to,
293
+ from: this.from,
294
+ seq: this.seq,
295
+ ack: this.ack
296
+ };
297
+ this.seq++;
298
+ this.sendBuffer.push(msg);
299
+ return msg;
300
+ }
301
+ inspectSendBuffer() {
302
+ return this.sendBuffer;
303
+ }
304
+ };
305
+
306
+ // transport/options.ts
307
+ var defaultTransportOptions = {
308
+ heartbeatIntervalMs: 1e3,
309
+ heartbeatsUntilDead: 2,
310
+ sessionDisconnectGraceMs: 5e3,
311
+ codec: NaiveJsonCodec
312
+ };
313
+ var defaultConnectionRetryOptions = {
314
+ baseIntervalMs: 250,
315
+ maxJitterMs: 200,
316
+ maxBackoffMs: 32e3,
317
+ attemptBudgetCapacity: 5,
318
+ budgetRestoreIntervalMs: 200
319
+ };
320
+ var defaultClientTransportOptions = {
321
+ ...defaultTransportOptions,
322
+ ...defaultConnectionRetryOptions
323
+ };
324
+ var defaultServerTransportOptions = {
325
+ ...defaultTransportOptions
326
+ };
327
+
328
+ export {
329
+ Connection,
330
+ Session,
331
+ defaultTransportOptions,
332
+ defaultClientTransportOptions,
333
+ defaultServerTransportOptions
334
+ };
335
+ //# sourceMappingURL=chunk-O2AVDJCQ.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"sources":["../transport/session.ts","../transport/options.ts"],"sourcesContent":["import { customAlphabet } from 'nanoid';\nimport {\n ControlFlags,\n ControlMessageAckSchema,\n OpaqueTransportMessage,\n PartialTransportMessage,\n TransportClientId,\n TransportMessage,\n} from './message';\nimport { Codec } from '../codec';\nimport { Logger, MessageMetadata } from '../logging/log';\nimport { Static } from '@sinclair/typebox';\nimport {\n PropagationContext,\n TelemetryInfo,\n createSessionTelemetryInfo,\n} from '../tracing';\nimport { SpanStatusCode } from '@opentelemetry/api';\n\nconst nanoid = customAlphabet('1234567890abcdefghijklmnopqrstuvxyz', 6);\nexport const unsafeId = () => nanoid();\n\ntype SequenceNumber = number;\n\n/**\n * A connection is the actual raw underlying transport connection.\n * It’s responsible for dispatching to/from the actual connection itself\n * This should be instantiated as soon as the client/server has a connection\n * It’s tied to the lifecycle of the underlying transport connection (i.e. if the WS drops, this connection should be deleted)\n */\nexport abstract class Connection {\n id: string;\n telemetry?: TelemetryInfo;\n constructor() {\n this.id = `conn-${nanoid(12)}`; // for debugging, no collision safety needed\n }\n\n get loggingMetadata(): MessageMetadata {\n const metadata: MessageMetadata = { connId: this.id };\n const spanContext = this.telemetry?.span.spanContext();\n\n if (this.telemetry?.span.isRecording() && spanContext) {\n metadata.telemetry = {\n traceId: spanContext.traceId,\n spanId: spanContext.spanId,\n };\n }\n\n return metadata;\n }\n\n /**\n * Handle adding a callback for when a message is received.\n * @param msg The message that was received.\n */\n abstract addDataListener(cb: (msg: Uint8Array) => void): void;\n abstract removeDataListener(cb: (msg: Uint8Array) => void): void;\n\n /**\n * Handle adding a callback for when the connection is closed.\n * This should also be called if an error happens.\n * @param cb The callback to call when the connection is closed.\n */\n abstract addCloseListener(cb: () => void): void;\n\n /**\n * Handle adding a callback for when an error is received.\n * This should only be used for this.logging errors, all cleanup\n * should be delegated to addCloseListener.\n *\n * The implementer should take care such that the implemented\n * connection will call both the close and error callbacks\n * on an error.\n *\n * @param cb The callback to call when an error is received.\n */\n abstract addErrorListener(cb: (err: Error) => void): void;\n\n /**\n * Sends a message over the connection.\n * @param msg The message to send.\n * @returns true if the message was sent, false otherwise.\n */\n abstract send(msg: Uint8Array): boolean;\n\n /**\n * Closes the connection.\n */\n abstract close(): void;\n}\n\nexport interface SessionOptions {\n /**\n * Frequency at which to send heartbeat acknowledgements\n */\n heartbeatIntervalMs: number;\n /**\n * Number of elapsed heartbeats without a response message before we consider\n * the connection dead.\n */\n heartbeatsUntilDead: number;\n /**\n * Duration to wait between connection disconnect and actual session disconnect\n */\n sessionDisconnectGraceMs: number;\n /**\n * The codec to use for encoding/decoding messages over the wire\n */\n codec: Codec;\n}\n\n/**\n * A session is a higher-level abstraction that operates over the span of potentially multiple transport-level connections\n * - It’s responsible for tracking any metadata for a particular client that might need to be persisted across connections (i.e. the sendBuffer, ack, seq)\n * - This will only be considered disconnected if\n * - the server tells the client that we’ve reconnected but it doesn’t recognize us anymore (server definitely died) or\n * - we hit a grace period after a connection disconnect\n */\nexport class Session<ConnType extends Connection> {\n private codec: Codec;\n private options: SessionOptions;\n readonly telemetry: TelemetryInfo;\n\n /**\n * The buffer of messages that have been sent but not yet acknowledged.\n */\n private sendBuffer: Array<OpaqueTransportMessage> = [];\n\n /**\n * The active connection associated with this session\n */\n connection?: ConnType;\n /**\n * A connection that is currently undergoing handshaking. Used to distinguish between the active\n * connection, but still be able to close it if needed.\n */\n private handshakingConnection?: ConnType;\n readonly from: TransportClientId;\n readonly to: TransportClientId;\n\n /**\n * The unique ID of this session.\n */\n readonly id: string;\n\n /**\n * What the other side advertised as their session ID\n * for this session.\n */\n advertisedSessionId?: string;\n\n /**\n * Number of messages we've sent along this session (excluding handshake and acks)\n */\n private seq: SequenceNumber = 0;\n\n /**\n * Number of unique messages we've received this session (excluding handshake and acks)\n */\n private ack: SequenceNumber = 0;\n\n /**\n * The grace period between when the inner connection is disconnected\n * and when we should consider the entire session disconnected.\n */\n private disconnectionGrace?: ReturnType<typeof setTimeout>;\n\n /**\n * Number of heartbeats we've sent without a response.\n */\n private heartbeatMisses: number;\n\n /**\n * The interval for sending heartbeats.\n */\n private heartbeat: ReturnType<typeof setInterval>;\n private log?: Logger;\n\n constructor(\n conn: ConnType | undefined,\n from: TransportClientId,\n to: TransportClientId,\n options: SessionOptions,\n propagationCtx?: PropagationContext,\n ) {\n this.id = `session-${nanoid(12)}`;\n this.options = options;\n this.from = from;\n this.to = to;\n this.connection = conn;\n this.codec = options.codec;\n\n // setup heartbeat\n this.heartbeatMisses = 0;\n this.heartbeat = setInterval(\n () => this.sendHeartbeat(),\n options.heartbeatIntervalMs,\n );\n this.telemetry = createSessionTelemetryInfo(this, propagationCtx);\n }\n\n bindLogger(log: Logger) {\n this.log = log;\n }\n\n get loggingMetadata(): MessageMetadata {\n const spanContext = this.telemetry.span.spanContext();\n\n return {\n clientId: this.from,\n connectedTo: this.to,\n sessionId: this.id,\n connId: this.connection?.id,\n telemetry: {\n traceId: spanContext.traceId,\n spanId: spanContext.spanId,\n },\n };\n }\n\n /**\n * Sends a message over the session's connection.\n * If the connection is not ready or the message fails to send, the message can be buffered for retry unless skipped.\n *\n * @param msg The partial message to be sent, which will be constructed into a full message.\n * @param addToSendBuff Whether to add the message to the send buffer for retry.\n * @returns The full transport ID of the message that was attempted to be sent.\n */\n send(msg: PartialTransportMessage): string {\n const fullMsg: TransportMessage = this.constructMsg(msg);\n this.log?.debug(`sending msg`, {\n ...this.loggingMetadata,\n transportMessage: fullMsg,\n });\n\n if (this.connection) {\n const ok = this.connection.send(this.codec.toBuffer(fullMsg));\n if (ok) return fullMsg.id;\n this.log?.info(\n `failed to send msg to ${fullMsg.to}, connection is probably dead`,\n {\n ...this.loggingMetadata,\n transportMessage: fullMsg,\n },\n );\n } else {\n this.log?.debug(\n `buffering msg to ${fullMsg.to}, connection not ready yet`,\n { ...this.loggingMetadata, transportMessage: fullMsg },\n );\n }\n\n return fullMsg.id;\n }\n\n sendHeartbeat() {\n const misses = this.heartbeatMisses;\n const missDuration = misses * this.options.heartbeatIntervalMs;\n if (misses > this.options.heartbeatsUntilDead) {\n if (this.connection) {\n this.log?.info(\n `closing connection to ${this.to} due to inactivity (missed ${misses} heartbeats which is ${missDuration}ms)`,\n this.loggingMetadata,\n );\n this.telemetry.span.addEvent('closing connection due to inactivity');\n this.closeStaleConnection();\n }\n return;\n }\n\n this.send({\n streamId: 'heartbeat',\n controlFlags: ControlFlags.AckBit,\n payload: {\n type: 'ACK',\n } satisfies Static<typeof ControlMessageAckSchema>,\n });\n this.heartbeatMisses++;\n }\n\n resetBufferedMessages() {\n this.sendBuffer = [];\n this.seq = 0;\n this.ack = 0;\n }\n\n sendBufferedMessages(conn: ConnType) {\n this.log?.info(`resending ${this.sendBuffer.length} buffered messages`, {\n ...this.loggingMetadata,\n connId: conn.id,\n });\n for (const msg of this.sendBuffer) {\n this.log?.debug(`resending msg`, {\n ...this.loggingMetadata,\n transportMessage: msg,\n connId: conn.id,\n });\n const ok = conn.send(this.codec.toBuffer(msg));\n if (!ok) {\n // this should never happen unless the transport has an\n // incorrect implementation of `createNewOutgoingConnection`\n const errMsg = `failed to send buffered message to ${this.to} (sus, this is a fresh connection)`;\n conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: errMsg,\n });\n\n this.log?.error(errMsg, {\n ...this.loggingMetadata,\n transportMessage: msg,\n connId: conn.id,\n tags: ['invariant-violation'],\n });\n conn.close();\n return;\n }\n }\n }\n\n updateBookkeeping(ack: number, seq: number) {\n if (seq + 1 < this.ack) {\n this.log?.error(`received stale seq ${seq} + 1 < ${this.ack}`, {\n ...this.loggingMetadata,\n tags: ['invariant-violation'],\n });\n return;\n }\n\n this.sendBuffer = this.sendBuffer.filter((unacked) => unacked.seq >= ack);\n this.ack = seq + 1;\n }\n\n private closeStaleConnection(conn?: ConnType) {\n if (this.connection === undefined || this.connection === conn) return;\n this.log?.info(\n `closing old inner connection from session to ${this.to}`,\n this.loggingMetadata,\n );\n this.connection.close();\n this.connection = undefined;\n }\n\n replaceWithNewConnection(newConn: ConnType, isTransparentReconnect: boolean) {\n this.closeStaleConnection(newConn);\n this.cancelGrace();\n if (isTransparentReconnect) {\n // only send the buffered messages if this is considered a transparent reconnect. there are\n // cases where the cient reconnects but with a different session id. for those cases we should\n // not send messages from a previous session.\n\n this.sendBufferedMessages(newConn);\n }\n this.connection = newConn;\n\n // we only call replaceWithNewConnection after\n // having successfully completed a handshake so we clear\n // it here\n this.handshakingConnection = undefined;\n }\n\n replaceWithNewHandshakingConnection(newConn: ConnType) {\n this.handshakingConnection = newConn;\n }\n\n beginGrace(cb: () => void) {\n this.log?.info(\n `starting ${this.options.sessionDisconnectGraceMs}ms grace period until session to ${this.to} is closed`,\n this.loggingMetadata,\n );\n\n // Replace any old timeouts to prevent this from firing twice.\n this.cancelGrace();\n this.disconnectionGrace = setTimeout(() => {\n this.log?.info(\n `grace period for ${this.to} elapsed`,\n this.loggingMetadata,\n );\n cb();\n }, this.options.sessionDisconnectGraceMs);\n }\n\n // called on reconnect of the underlying session\n cancelGrace() {\n this.heartbeatMisses = 0;\n clearTimeout(this.disconnectionGrace);\n this.disconnectionGrace = undefined;\n }\n\n /**\n * Used to close the handshaking connection, if set.\n */\n closeHandshakingConnection(expectedHandshakingConn?: ConnType) {\n if (this.handshakingConnection === undefined) return;\n if (\n expectedHandshakingConn !== undefined &&\n this.handshakingConnection === expectedHandshakingConn\n ) {\n // If the handshaking connection is the expected one, don't close it.\n return;\n }\n this.handshakingConnection.close();\n this.handshakingConnection = undefined;\n }\n\n // closed when we want to discard the whole session\n // (i.e. shutdown or session disconnect)\n close() {\n this.closeStaleConnection();\n this.cancelGrace();\n this.resetBufferedMessages();\n clearInterval(this.heartbeat);\n }\n\n get connected() {\n return this.connection !== undefined;\n }\n\n get nextExpectedAck() {\n return this.seq;\n }\n\n get nextExpectedSeq() {\n return this.ack;\n }\n\n // This is only used in tests to make the session misbehave.\n /* @internal */ advanceAckForTesting(by: number) {\n this.ack += by;\n }\n\n constructMsg<Payload>(\n partialMsg: PartialTransportMessage<Payload>,\n ): TransportMessage<Payload> {\n const msg = {\n ...partialMsg,\n id: unsafeId(),\n to: this.to,\n from: this.from,\n seq: this.seq,\n ack: this.ack,\n };\n\n this.seq++;\n this.sendBuffer.push(msg);\n return msg;\n }\n\n inspectSendBuffer(): ReadonlyArray<OpaqueTransportMessage> {\n return this.sendBuffer;\n }\n}\n","import { NaiveJsonCodec } from '../codec/json';\nimport { ConnectionRetryOptions } from './rateLimit';\nimport { SessionOptions } from './session';\n\nexport type TransportOptions = SessionOptions;\n\nexport type ProvidedTransportOptions = Partial<TransportOptions>;\n\nexport const defaultTransportOptions: TransportOptions = {\n heartbeatIntervalMs: 1_000,\n heartbeatsUntilDead: 2,\n sessionDisconnectGraceMs: 5_000,\n codec: NaiveJsonCodec,\n};\n\nexport type ClientTransportOptions = TransportOptions & ConnectionRetryOptions;\n\nexport type ProvidedClientTransportOptions = Partial<ClientTransportOptions>;\n\nconst defaultConnectionRetryOptions: ConnectionRetryOptions = {\n baseIntervalMs: 250,\n maxJitterMs: 200,\n maxBackoffMs: 32_000,\n attemptBudgetCapacity: 5,\n budgetRestoreIntervalMs: 200,\n};\n\nexport const defaultClientTransportOptions: ClientTransportOptions = {\n ...defaultTransportOptions,\n ...defaultConnectionRetryOptions,\n};\n\nexport type ServerTransportOptions = TransportOptions;\n\nexport type ProvidedServerTransportOptions = Partial<ServerTransportOptions>;\n\nexport const defaultServerTransportOptions: ServerTransportOptions = {\n ...defaultTransportOptions,\n};\n"],"mappings":";;;;;;;;AAAA,SAAS,sBAAsB;AAiB/B,SAAS,sBAAsB;AAE/B,IAAM,SAAS,eAAe,uCAAuC,CAAC;AAC/D,IAAM,WAAW,MAAM,OAAO;AAU9B,IAAe,aAAf,MAA0B;AAAA,EAC/B;AAAA,EACA;AAAA,EACA,cAAc;AACZ,SAAK,KAAK,QAAQ,OAAO,EAAE,CAAC;AAAA,EAC9B;AAAA,EAEA,IAAI,kBAAmC;AACrC,UAAM,WAA4B,EAAE,QAAQ,KAAK,GAAG;AACpD,UAAM,cAAc,KAAK,WAAW,KAAK,YAAY;AAErD,QAAI,KAAK,WAAW,KAAK,YAAY,KAAK,aAAa;AACrD,eAAS,YAAY;AAAA,QACnB,SAAS,YAAY;AAAA,QACrB,QAAQ,YAAY;AAAA,MACtB;AAAA,IACF;AAEA,WAAO;AAAA,EACT;AAwCF;AA6BO,IAAM,UAAN,MAA2C;AAAA,EACxC;AAAA,EACA;AAAA,EACC;AAAA;AAAA;AAAA;AAAA,EAKD,aAA4C,CAAC;AAAA;AAAA;AAAA;AAAA,EAKrD;AAAA;AAAA;AAAA;AAAA;AAAA,EAKQ;AAAA,EACC;AAAA,EACA;AAAA;AAAA;AAAA;AAAA,EAKA;AAAA;AAAA;AAAA;AAAA;AAAA,EAMT;AAAA;AAAA;AAAA;AAAA,EAKQ,MAAsB;AAAA;AAAA;AAAA;AAAA,EAKtB,MAAsB;AAAA;AAAA;AAAA;AAAA;AAAA,EAMtB;AAAA;AAAA;AAAA;AAAA,EAKA;AAAA;AAAA;AAAA;AAAA,EAKA;AAAA,EACA;AAAA,EAER,YACE,MACA,MACA,IACA,SACA,gBACA;AACA,SAAK,KAAK,WAAW,OAAO,EAAE,CAAC;AAC/B,SAAK,UAAU;AACf,SAAK,OAAO;AACZ,SAAK,KAAK;AACV,SAAK,aAAa;AAClB,SAAK,QAAQ,QAAQ;AAGrB,SAAK,kBAAkB;AACvB,SAAK,YAAY;AAAA,MACf,MAAM,KAAK,cAAc;AAAA,MACzB,QAAQ;AAAA,IACV;AACA,SAAK,YAAY,2BAA2B,MAAM,cAAc;AAAA,EAClE;AAAA,EAEA,WAAW,KAAa;AACtB,SAAK,MAAM;AAAA,EACb;AAAA,EAEA,IAAI,kBAAmC;AACrC,UAAM,cAAc,KAAK,UAAU,KAAK,YAAY;AAEpD,WAAO;AAAA,MACL,UAAU,KAAK;AAAA,MACf,aAAa,KAAK;AAAA,MAClB,WAAW,KAAK;AAAA,MAChB,QAAQ,KAAK,YAAY;AAAA,MACzB,WAAW;AAAA,QACT,SAAS,YAAY;AAAA,QACrB,QAAQ,YAAY;AAAA,MACtB;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAUA,KAAK,KAAsC;AACzC,UAAM,UAA4B,KAAK,aAAa,GAAG;AACvD,SAAK,KAAK,MAAM,eAAe;AAAA,MAC7B,GAAG,KAAK;AAAA,MACR,kBAAkB;AAAA,IACpB,CAAC;AAED,QAAI,KAAK,YAAY;AACnB,YAAM,KAAK,KAAK,WAAW,KAAK,KAAK,MAAM,SAAS,OAAO,CAAC;AAC5D,UAAI;AAAI,eAAO,QAAQ;AACvB,WAAK,KAAK;AAAA,QACR,yBAAyB,QAAQ,EAAE;AAAA,QACnC;AAAA,UACE,GAAG,KAAK;AAAA,UACR,kBAAkB;AAAA,QACpB;AAAA,MACF;AAAA,IACF,OAAO;AACL,WAAK,KAAK;AAAA,QACR,oBAAoB,QAAQ,EAAE;AAAA,QAC9B,EAAE,GAAG,KAAK,iBAAiB,kBAAkB,QAAQ;AAAA,MACvD;AAAA,IACF;AAEA,WAAO,QAAQ;AAAA,EACjB;AAAA,EAEA,gBAAgB;AACd,UAAM,SAAS,KAAK;AACpB,UAAM,eAAe,SAAS,KAAK,QAAQ;AAC3C,QAAI,SAAS,KAAK,QAAQ,qBAAqB;AAC7C,UAAI,KAAK,YAAY;AACnB,aAAK,KAAK;AAAA,UACR,yBAAyB,KAAK,EAAE,8BAA8B,MAAM,wBAAwB,YAAY;AAAA,UACxG,KAAK;AAAA,QACP;AACA,aAAK,UAAU,KAAK,SAAS,sCAAsC;AACnE,aAAK,qBAAqB;AAAA,MAC5B;AACA;AAAA,IACF;AAEA,SAAK,KAAK;AAAA,MACR,UAAU;AAAA,MACV;AAAA,MACA,SAAS;AAAA,QACP,MAAM;AAAA,MACR;AAAA,IACF,CAAC;AACD,SAAK;AAAA,EACP;AAAA,EAEA,wBAAwB;AACtB,SAAK,aAAa,CAAC;AACnB,SAAK,MAAM;AACX,SAAK,MAAM;AAAA,EACb;AAAA,EAEA,qBAAqB,MAAgB;AACnC,SAAK,KAAK,KAAK,aAAa,KAAK,WAAW,MAAM,sBAAsB;AAAA,MACtE,GAAG,KAAK;AAAA,MACR,QAAQ,KAAK;AAAA,IACf,CAAC;AACD,eAAW,OAAO,KAAK,YAAY;AACjC,WAAK,KAAK,MAAM,iBAAiB;AAAA,QAC/B,GAAG,KAAK;AAAA,QACR,kBAAkB;AAAA,QAClB,QAAQ,KAAK;AAAA,MACf,CAAC;AACD,YAAM,KAAK,KAAK,KAAK,KAAK,MAAM,SAAS,GAAG,CAAC;AAC7C,UAAI,CAAC,IAAI;AAGP,cAAM,SAAS,sCAAsC,KAAK,EAAE;AAC5D,aAAK,WAAW,KAAK,UAAU;AAAA,UAC7B,MAAM,eAAe;AAAA,UACrB,SAAS;AAAA,QACX,CAAC;AAED,aAAK,KAAK,MAAM,QAAQ;AAAA,UACtB,GAAG,KAAK;AAAA,UACR,kBAAkB;AAAA,UAClB,QAAQ,KAAK;AAAA,UACb,MAAM,CAAC,qBAAqB;AAAA,QAC9B,CAAC;AACD,aAAK,MAAM;AACX;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA,EAEA,kBAAkB,KAAa,KAAa;AAC1C,QAAI,MAAM,IAAI,KAAK,KAAK;AACtB,WAAK,KAAK,MAAM,sBAAsB,GAAG,UAAU,KAAK,GAAG,IAAI;AAAA,QAC7D,GAAG,KAAK;AAAA,QACR,MAAM,CAAC,qBAAqB;AAAA,MAC9B,CAAC;AACD;AAAA,IACF;AAEA,SAAK,aAAa,KAAK,WAAW,OAAO,CAAC,YAAY,QAAQ,OAAO,GAAG;AACxE,SAAK,MAAM,MAAM;AAAA,EACnB;AAAA,EAEQ,qBAAqB,MAAiB;AAC5C,QAAI,KAAK,eAAe,UAAa,KAAK,eAAe;AAAM;AAC/D,SAAK,KAAK;AAAA,MACR,gDAAgD,KAAK,EAAE;AAAA,MACvD,KAAK;AAAA,IACP;AACA,SAAK,WAAW,MAAM;AACtB,SAAK,aAAa;AAAA,EACpB;AAAA,EAEA,yBAAyB,SAAmB,wBAAiC;AAC3E,SAAK,qBAAqB,OAAO;AACjC,SAAK,YAAY;AACjB,QAAI,wBAAwB;AAK1B,WAAK,qBAAqB,OAAO;AAAA,IACnC;AACA,SAAK,aAAa;AAKlB,SAAK,wBAAwB;AAAA,EAC/B;AAAA,EAEA,oCAAoC,SAAmB;AACrD,SAAK,wBAAwB;AAAA,EAC/B;AAAA,EAEA,WAAW,IAAgB;AACzB,SAAK,KAAK;AAAA,MACR,YAAY,KAAK,QAAQ,wBAAwB,oCAAoC,KAAK,EAAE;AAAA,MAC5F,KAAK;AAAA,IACP;AAGA,SAAK,YAAY;AACjB,SAAK,qBAAqB,WAAW,MAAM;AACzC,WAAK,KAAK;AAAA,QACR,oBAAoB,KAAK,EAAE;AAAA,QAC3B,KAAK;AAAA,MACP;AACA,SAAG;AAAA,IACL,GAAG,KAAK,QAAQ,wBAAwB;AAAA,EAC1C;AAAA;AAAA,EAGA,cAAc;AACZ,SAAK,kBAAkB;AACvB,iBAAa,KAAK,kBAAkB;AACpC,SAAK,qBAAqB;AAAA,EAC5B;AAAA;AAAA;AAAA;AAAA,EAKA,2BAA2B,yBAAoC;AAC7D,QAAI,KAAK,0BAA0B;AAAW;AAC9C,QACE,4BAA4B,UAC5B,KAAK,0BAA0B,yBAC/B;AAEA;AAAA,IACF;AACA,SAAK,sBAAsB,MAAM;AACjC,SAAK,wBAAwB;AAAA,EAC/B;AAAA;AAAA;AAAA,EAIA,QAAQ;AACN,SAAK,qBAAqB;AAC1B,SAAK,YAAY;AACjB,SAAK,sBAAsB;AAC3B,kBAAc,KAAK,SAAS;AAAA,EAC9B;AAAA,EAEA,IAAI,YAAY;AACd,WAAO,KAAK,eAAe;AAAA,EAC7B;AAAA,EAEA,IAAI,kBAAkB;AACpB,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,IAAI,kBAAkB;AACpB,WAAO,KAAK;AAAA,EACd;AAAA;AAAA;AAAA,EAGgB,qBAAqB,IAAY;AAC/C,SAAK,OAAO;AAAA,EACd;AAAA,EAEA,aACE,YAC2B;AAC3B,UAAM,MAAM;AAAA,MACV,GAAG;AAAA,MACH,IAAI,SAAS;AAAA,MACb,IAAI,KAAK;AAAA,MACT,MAAM,KAAK;AAAA,MACX,KAAK,KAAK;AAAA,MACV,KAAK,KAAK;AAAA,IACZ;AAEA,SAAK;AACL,SAAK,WAAW,KAAK,GAAG;AACxB,WAAO;AAAA,EACT;AAAA,EAEA,oBAA2D;AACzD,WAAO,KAAK;AAAA,EACd;AACF;;;AC1bO,IAAM,0BAA4C;AAAA,EACvD,qBAAqB;AAAA,EACrB,qBAAqB;AAAA,EACrB,0BAA0B;AAAA,EAC1B,OAAO;AACT;AAMA,IAAM,gCAAwD;AAAA,EAC5D,gBAAgB;AAAA,EAChB,aAAa;AAAA,EACb,cAAc;AAAA,EACd,uBAAuB;AAAA,EACvB,yBAAyB;AAC3B;AAEO,IAAM,gCAAwD;AAAA,EACnE,GAAG;AAAA,EACH,GAAG;AACL;AAMO,IAAM,gCAAwD;AAAA,EACnE,GAAG;AACL;","names":[]}