@replit/river 0.213.1 → 0.215.0

package/dist/router/index.d.cts

@@ -1,14 +1,14 @@
-import { A as AnyServiceSchemaMap, M as MaybeDisposable, P as ProcedureHandlerContext, a as PayloadType, I as InstantiatedServiceSchemaMap, b as AnyProcedure } from '../services-C656NVV3.cjs';
-export { J as BaseErrorSchemaType, G as CANCEL_CODE, L as Client, N as Err, T as ErrResult, F as INVALID_REQUEST_CODE, O as Ok, X as OkResult, h as ProcErrors, d as ProcHandler, e as ProcInit, f as ProcRequest, g as ProcResponse, i as ProcType, z as Procedure, B as ProcedureErrorSchemaType, r as ProcedureMap, R as RPCProcedure, y as RawReadable, v as Readable, x as ReadableBrokenError, w as ReadableResult, H as ReaderErrorSchema, _ as ResponseData, Q as Result, Z as ResultUnwrapErr, Y as ResultUnwrapOk, m as SerializedProcedureSchema, q as SerializedProcedureSchemaProtocolv1, k as SerializedServerSchema, o as SerializedServerSchemaProtocolv1, l as SerializedServiceSchema, p as SerializedServiceSchemaProtocolv1, S as Service, c as ServiceConfiguration, u as StreamProcedure, t as SubscriptionProcedure, D as UNCAUGHT_ERROR_CODE, E as UNEXPECTED_DISCONNECT_CODE, U as UploadProcedure, V as ValidProcType, W as Writable, K as createClient, j as createServiceSchema, C as flattenErrorType, s as serializeSchema, n as serializeSchemaV1Compat } from '../services-C656NVV3.cjs';
+import { A as AnyServiceSchemaMap, M as MaybeDisposable, P as ProcedureHandlerContext, a as PayloadType, I as InstantiatedServiceSchemaMap, b as AnyProcedure } from '../services-DBv2nmly.cjs';
+export { J as BaseErrorSchemaType, G as CANCEL_CODE, L as Client, N as Err, T as ErrResult, F as INVALID_REQUEST_CODE, O as Ok, X as OkResult, h as ProcErrors, d as ProcHandler, e as ProcInit, f as ProcRequest, g as ProcResponse, i as ProcType, z as Procedure, B as ProcedureErrorSchemaType, r as ProcedureMap, R as RPCProcedure, y as RawReadable, v as Readable, x as ReadableBrokenError, w as ReadableResult, H as ReaderErrorSchema, _ as ResponseData, Q as Result, Z as ResultUnwrapErr, Y as ResultUnwrapOk, m as SerializedProcedureSchema, q as SerializedProcedureSchemaProtocolv1, k as SerializedServerSchema, o as SerializedServerSchemaProtocolv1, l as SerializedServiceSchema, p as SerializedServiceSchemaProtocolv1, S as Service, c as ServiceConfiguration, u as StreamProcedure, t as SubscriptionProcedure, D as UNCAUGHT_ERROR_CODE, E as UNEXPECTED_DISCONNECT_CODE, U as UploadProcedure, V as ValidProcType, W as Writable, K as createClient, j as createServiceSchema, C as flattenErrorType, s as serializeSchema, n as serializeSchemaV1Compat } from '../services-DBv2nmly.cjs';
 import { TSchema, Static } from '@sinclair/typebox';
-import { c as TransportClientId, b as OpaqueTransportMessage } from '../message-Dlsh5WDF.cjs';
-import { C as Connection, q as ServerHandshakeOptions } from '../transport-CxT7y8Qk.cjs';
-export { r as createClientHandshakeOptions, s as createServerHandshakeOptions } from '../transport-CxT7y8Qk.cjs';
-import { S as ServerTransport } from '../server-BDSYa-CO.cjs';
+import { c as TransportClientId, b as OpaqueTransportMessage } from '../message-aABg0s5M.cjs';
+import { C as Connection, q as ServerHandshakeOptions } from '../transport-D3jzhFSi.cjs';
+export { r as createClientHandshakeOptions, s as createServerHandshakeOptions } from '../transport-D3jzhFSi.cjs';
+import { S as ServerTransport } from '../server-BR0DZaWi.cjs';
 import '@opentelemetry/api';
-import '../client-DXJRow2s.cjs';
-import '../index-D9R6UTMl.cjs';
-import '../adapter-DjiEwOYi.cjs';
+import '../client-Dk4H7qeg.cjs';
+import '../index-CHCzvZ9D.cjs';
+import '../adapter-IGrG4KLL.cjs';
 
 type StreamId = string;
 /**
@@ -74,6 +74,6 @@ declare function createServer<MetadataSchema extends TSchema, ParsedMetadata ext
     middlewares?: Array<Middleware>;
 }>): Server<Context, ParsedMetadata, Services>;
 
-var version = "0.213.1";
+var version = "0.215.0";
 
 export { MaybeDisposable, type Middleware, type MiddlewareContext, type MiddlewareParam, PayloadType, ProcedureHandlerContext, version as RIVER_VERSION, type Server, createServer };

package/dist/router/index.d.ts

@@ -1,14 +1,14 @@
-import { A as AnyServiceSchemaMap, M as MaybeDisposable, P as ProcedureHandlerContext, a as PayloadType, I as InstantiatedServiceSchemaMap, b as AnyProcedure } from '../services-ChTb4jmc.js';
-export { J as BaseErrorSchemaType, G as CANCEL_CODE, L as Client, N as Err, T as ErrResult, F as INVALID_REQUEST_CODE, O as Ok, X as OkResult, h as ProcErrors, d as ProcHandler, e as ProcInit, f as ProcRequest, g as ProcResponse, i as ProcType, z as Procedure, B as ProcedureErrorSchemaType, r as ProcedureMap, R as RPCProcedure, y as RawReadable, v as Readable, x as ReadableBrokenError, w as ReadableResult, H as ReaderErrorSchema, _ as ResponseData, Q as Result, Z as ResultUnwrapErr, Y as ResultUnwrapOk, m as SerializedProcedureSchema, q as SerializedProcedureSchemaProtocolv1, k as SerializedServerSchema, o as SerializedServerSchemaProtocolv1, l as SerializedServiceSchema, p as SerializedServiceSchemaProtocolv1, S as Service, c as ServiceConfiguration, u as StreamProcedure, t as SubscriptionProcedure, D as UNCAUGHT_ERROR_CODE, E as UNEXPECTED_DISCONNECT_CODE, U as UploadProcedure, V as ValidProcType, W as Writable, K as createClient, j as createServiceSchema, C as flattenErrorType, s as serializeSchema, n as serializeSchemaV1Compat } from '../services-ChTb4jmc.js';
+import { A as AnyServiceSchemaMap, M as MaybeDisposable, P as ProcedureHandlerContext, a as PayloadType, I as InstantiatedServiceSchemaMap, b as AnyProcedure } from '../services-B5SY771g.js';
+export { J as BaseErrorSchemaType, G as CANCEL_CODE, L as Client, N as Err, T as ErrResult, F as INVALID_REQUEST_CODE, O as Ok, X as OkResult, h as ProcErrors, d as ProcHandler, e as ProcInit, f as ProcRequest, g as ProcResponse, i as ProcType, z as Procedure, B as ProcedureErrorSchemaType, r as ProcedureMap, R as RPCProcedure, y as RawReadable, v as Readable, x as ReadableBrokenError, w as ReadableResult, H as ReaderErrorSchema, _ as ResponseData, Q as Result, Z as ResultUnwrapErr, Y as ResultUnwrapOk, m as SerializedProcedureSchema, q as SerializedProcedureSchemaProtocolv1, k as SerializedServerSchema, o as SerializedServerSchemaProtocolv1, l as SerializedServiceSchema, p as SerializedServiceSchemaProtocolv1, S as Service, c as ServiceConfiguration, u as StreamProcedure, t as SubscriptionProcedure, D as UNCAUGHT_ERROR_CODE, E as UNEXPECTED_DISCONNECT_CODE, U as UploadProcedure, V as ValidProcType, W as Writable, K as createClient, j as createServiceSchema, C as flattenErrorType, s as serializeSchema, n as serializeSchemaV1Compat } from '../services-B5SY771g.js';
 import { TSchema, Static } from '@sinclair/typebox';
-import { c as TransportClientId, b as OpaqueTransportMessage } from '../message-Dlsh5WDF.js';
-import { C as Connection, q as ServerHandshakeOptions } from '../transport-pdbkDzmJ.js';
-export { r as createClientHandshakeOptions, s as createServerHandshakeOptions } from '../transport-pdbkDzmJ.js';
-import { S as ServerTransport } from '../server-DFOzjvLh.js';
+import { c as TransportClientId, b as OpaqueTransportMessage } from '../message-aABg0s5M.js';
+import { C as Connection, q as ServerHandshakeOptions } from '../transport-BOL2p5s-.js';
+export { r as createClientHandshakeOptions, s as createServerHandshakeOptions } from '../transport-BOL2p5s-.js';
+import { S as ServerTransport } from '../server-uNzkzIRh.js';
 import '@opentelemetry/api';
-import '../client-Dw0JBBs3.js';
-import '../index-CSM8soK7.js';
-import '../adapter-Cp7_gIVA.js';
+import '../client-BhwlY0-L.js';
+import '../index-uF0cBM7z.js';
+import '../adapter-CgF7vQPu.js';
 
 type StreamId = string;
 /**
@@ -74,6 +74,6 @@ declare function createServer<MetadataSchema extends TSchema, ParsedMetadata ext
     middlewares?: Array<Middleware>;
 }>): Server<Context, ParsedMetadata, Services>;
 
-var version = "0.213.1";
+var version = "0.215.0";
 
 export { MaybeDisposable, type Middleware, type MiddlewareContext, type MiddlewareParam, PayloadType, ProcedureHandlerContext, version as RIVER_VERSION, type Server, createServer };

package/dist/router/index.js

@@ -18,7 +18,7 @@ import {
   serializeSchema,
   serializeSchemaV1Compat,
   version
-} from "../chunk-X4PP65DW.js";
+} from "../chunk-RDTQZ7HO.js";
 export {
   CANCEL_CODE,
   Err,

package/dist/{server-BDSYa-CO.d.cts → server-BR0DZaWi.d.cts}

@@ -1,8 +1,8 @@
-import { C as Connection, t as CommonSession, d as SessionState, u as CommonSessionProps, T as Transport, v as ServerTransportOptions, q as ServerHandshakeOptions, w as ServerSession, c as ProvidedServerTransportOptions, D as DeleteSessionOptions } from './transport-CxT7y8Qk.cjs';
-import { b as OpaqueTransportMessage, H as HandshakeErrorResponseCodes, P as ProtocolVersion, a as TransportMessage, c as TransportClientId } from './message-Dlsh5WDF.cjs';
+import { C as Connection, t as CommonSession, d as SessionState, u as CommonSessionProps, T as Transport, v as ServerTransportOptions, q as ServerHandshakeOptions, w as ServerSession, c as ProvidedServerTransportOptions, D as DeleteSessionOptions } from './transport-D3jzhFSi.cjs';
+import { b as OpaqueTransportMessage, H as HandshakeErrorResponseCodes, P as ProtocolVersion, a as TransportMessage, c as TransportClientId } from './message-aABg0s5M.cjs';
 import { Static, TSchema } from '@sinclair/typebox';
-import { T as Tags } from './index-D9R6UTMl.cjs';
-import { S as SendResult } from './adapter-DjiEwOYi.cjs';
+import { T as Tags } from './index-CHCzvZ9D.cjs';
+import { S as SendResult } from './adapter-IGrG4KLL.cjs';
 
 interface SessionWaitingForHandshakeListeners {
     onConnectionErrored: (err: unknown) => void;

package/dist/{server-DFOzjvLh.d.ts → server-uNzkzIRh.d.ts}

@@ -1,8 +1,8 @@
-import { C as Connection, t as CommonSession, d as SessionState, u as CommonSessionProps, T as Transport, v as ServerTransportOptions, q as ServerHandshakeOptions, w as ServerSession, c as ProvidedServerTransportOptions, D as DeleteSessionOptions } from './transport-pdbkDzmJ.js';
-import { b as OpaqueTransportMessage, H as HandshakeErrorResponseCodes, P as ProtocolVersion, a as TransportMessage, c as TransportClientId } from './message-Dlsh5WDF.js';
+import { C as Connection, t as CommonSession, d as SessionState, u as CommonSessionProps, T as Transport, v as ServerTransportOptions, q as ServerHandshakeOptions, w as ServerSession, c as ProvidedServerTransportOptions, D as DeleteSessionOptions } from './transport-BOL2p5s-.js';
+import { b as OpaqueTransportMessage, H as HandshakeErrorResponseCodes, P as ProtocolVersion, a as TransportMessage, c as TransportClientId } from './message-aABg0s5M.js';
 import { Static, TSchema } from '@sinclair/typebox';
-import { T as Tags } from './index-CSM8soK7.js';
-import { S as SendResult } from './adapter-Cp7_gIVA.js';
+import { T as Tags } from './index-uF0cBM7z.js';
+import { S as SendResult } from './adapter-CgF7vQPu.js';
 
 interface SessionWaitingForHandshakeListeners {
     onConnectionErrored: (err: unknown) => void;

package/dist/{services-C656NVV3.d.cts → services-B5SY771g.d.ts}

@@ -1,9 +1,9 @@
 import * as _sinclair_typebox from '@sinclair/typebox';
-import { TNever, TObject, TLiteral, TEnum, TString, TSchema, TUnion, Static } from '@sinclair/typebox';
+import { TObject, TLiteral, TEnum, TString, TSchema, TNever, TUnion, Static } from '@sinclair/typebox';
 import { Span } from '@opentelemetry/api';
-import { c as TransportClientId } from './message-Dlsh5WDF.cjs';
-import { C as Connection, n as ClientHandshakeOptions, z as SessionId } from './transport-CxT7y8Qk.cjs';
-import { C as ClientTransport } from './client-DXJRow2s.cjs';
+import { c as TransportClientId } from './message-aABg0s5M.js';
+import { C as Connection, n as ClientHandshakeOptions, z as SessionId } from './transport-BOL2p5s-.js';
+import { C as ClientTransport } from './client-BhwlY0-L.js';
 
 /**
  * {@link UNCAUGHT_ERROR_CODE} is the code that is used when an error is thrown
@@ -406,6 +406,22 @@ type ProcedureHandlerContext<State, Context, ParsedMetadata> = Context & {
      * the river documentation to understand the difference between the two concepts.
      */
     cancel: (message?: string) => ErrResult<Static<typeof CancelErrorSchema>>;
+    /**
+     * Register a cleanup function that will run after the procedure handler
+     * completes (whether it returns normally, throws, or is cancelled).
+     * Cleanup functions run in reverse registration order (LIFO) and each
+     * cleanup is awaited before the next one starts.
+     *
+     * Prefer this over registering async cleanup work on `signal`'s 'abort'
+     * event. Abort signal callbacks fire synchronously and do not await async
+     * work, so multiple async callbacks will interlace their execution
+     * (coroutine-like behavior) rather than running sequentially to completion.
+     * `deferCleanup` guarantees each cleanup finishes before the next begins.
+     *
+     * If a cleanup function throws, the error is recorded on the cleanup span
+     * but remaining cleanups continue to run.
+     */
+    deferCleanup: (fn: () => void | Promise<void>) => void;
     /**
      * This signal is a standard [AbortSignal](https://developer.mozilla.org/en-US/docs/Web/API/AbortSignal)
      * triggered when the procedure invocation is done. This signal tracks the invocation/request finishing

package/dist/{services-ChTb4jmc.d.ts → services-DBv2nmly.d.cts}

@@ -1,9 +1,9 @@
 import * as _sinclair_typebox from '@sinclair/typebox';
-import { TNever, TObject, TLiteral, TEnum, TString, TSchema, TUnion, Static } from '@sinclair/typebox';
+import { TObject, TLiteral, TEnum, TString, TSchema, TNever, TUnion, Static } from '@sinclair/typebox';
 import { Span } from '@opentelemetry/api';
-import { c as TransportClientId } from './message-Dlsh5WDF.js';
-import { C as Connection, n as ClientHandshakeOptions, z as SessionId } from './transport-pdbkDzmJ.js';
-import { C as ClientTransport } from './client-Dw0JBBs3.js';
+import { c as TransportClientId } from './message-aABg0s5M.cjs';
+import { C as Connection, n as ClientHandshakeOptions, z as SessionId } from './transport-D3jzhFSi.cjs';
+import { C as ClientTransport } from './client-Dk4H7qeg.cjs';
 
 /**
  * {@link UNCAUGHT_ERROR_CODE} is the code that is used when an error is thrown
@@ -406,6 +406,22 @@ type ProcedureHandlerContext<State, Context, ParsedMetadata> = Context & {
      * the river documentation to understand the difference between the two concepts.
      */
     cancel: (message?: string) => ErrResult<Static<typeof CancelErrorSchema>>;
+    /**
+     * Register a cleanup function that will run after the procedure handler
+     * completes (whether it returns normally, throws, or is cancelled).
+     * Cleanup functions run in reverse registration order (LIFO) and each
+     * cleanup is awaited before the next one starts.
+     *
+     * Prefer this over registering async cleanup work on `signal`'s 'abort'
+     * event. Abort signal callbacks fire synchronously and do not await async
+     * work, so multiple async callbacks will interlace their execution
+     * (coroutine-like behavior) rather than running sequentially to completion.
+     * `deferCleanup` guarantees each cleanup finishes before the next begins.
+     *
+     * If a cleanup function throws, the error is recorded on the cleanup span
+     * but remaining cleanups continue to run.
+     */
+    deferCleanup: (fn: () => void | Promise<void>) => void;
     /**
      * This signal is a standard [AbortSignal](https://developer.mozilla.org/en-US/docs/Web/API/AbortSignal)
      * triggered when the procedure invocation is done. This signal tracks the invocation/request finishing

package/dist/testUtil/index.cjs

@@ -360,6 +360,7 @@ var IdentifiedSession = class extends CommonSession {
   telemetry;
   to;
   protocolVersion;
+  listeners;
   /**
    * Index of the message we will send next (excluding handshake)
    */
@@ -383,7 +384,8 @@ var IdentifiedSession = class extends CommonSession {
       telemetry,
       log,
       protocolVersion,
-      seqSent: messagesSent
+      seqSent: messagesSent,
+      listeners
     } = props;
     super(props);
     this.id = id;
@@ -395,6 +397,7 @@ var IdentifiedSession = class extends CommonSession {
     this.log = log;
     this.protocolVersion = protocolVersion;
     this.seqSent = messagesSent;
+    this.listeners = listeners;
   }
   get loggingMetadata() {
     const metadata = {
@@ -411,7 +414,7 @@ var IdentifiedSession = class extends CommonSession {
     }
     return metadata;
   }
-  constructMsg(partialMsg) {
+  encodeMsg(partialMsg) {
     const msg = {
       ...partialMsg,
       id: generateId(),
@@ -420,18 +423,37 @@ var IdentifiedSession = class extends CommonSession {
       seq: this.seq,
       ack: this.ack
     };
+    const encoded = this.codec.toBuffer(msg);
+    if (!encoded.ok) {
+      this.listeners.onMessageSendFailure(
+        { ...partialMsg, seq: this.seq },
+        encoded.reason
+      );
+      return encoded;
+    }
     this.seq++;
-    return msg;
+    return {
+      ok: true,
+      value: {
+        id: msg.id,
+        seq: msg.seq,
+        msg: partialMsg,
+        data: encoded.value
+      }
+    };
   }
   nextSeq() {
     return this.sendBuffer.length > 0 ? this.sendBuffer[0].seq : this.seq;
   }
   send(msg) {
-    const constructedMsg = this.constructMsg(msg);
-    this.sendBuffer.push(constructedMsg);
+    const encodeResult = this.encodeMsg(msg);
+    if (!encodeResult.ok) {
+      return encodeResult;
+    }
+    this.sendBuffer.push(encodeResult.value);
     return {
       ok: true,
-      value: constructedMsg.id
+      value: encodeResult.value.id
     };
   }
   _handleStateExit() {
@@ -464,23 +486,6 @@ var IdentifiedSessionWithGracePeriod = class extends IdentifiedSession {
     super._handleClose();
   }
 };
-function sendMessage(conn, codec, msg) {
-  const buff = codec.toBuffer(msg);
-  if (!buff.ok) {
-    return buff;
-  }
-  const sent = conn.send(buff.value);
-  if (!sent) {
-    return {
-      ok: false,
-      reason: "failed to send message"
-    };
-  }
-  return {
-    ok: true,
-    value: msg.id
-  };
-}
 
 // transport/sessionStateMachine/SessionConnecting.ts
 var SessionConnecting = class extends IdentifiedSessionWithGracePeriod {
@@ -559,7 +564,7 @@ function coerceErrorString(err) {
 }
 
 // package.json
-var version = "0.213.1";
+var version = "0.215.0";
 
 // tracing/index.ts
 function getPropagationContext(ctx) {
@@ -642,7 +647,21 @@ var SessionWaitingForHandshake = class extends CommonSession {
     this.listeners.onHandshake(parsedMsgRes.value);
   };
   sendHandshake(msg) {
-    return sendMessage(this.conn, this.codec, msg);
+    const buff = this.codec.toBuffer(msg);
+    if (!buff.ok) {
+      return buff;
+    }
+    const sent = this.conn.send(buff.value);
+    if (!sent) {
+      return {
+        ok: false,
+        reason: "failed to send handshake"
+      };
+    }
+    return {
+      ok: true,
+      value: msg.id
+    };
   }
   _handleStateExit() {
     this.conn.removeDataListener();
@@ -691,7 +710,21 @@ var SessionHandshaking = class extends IdentifiedSessionWithGracePeriod {
     this.listeners.onHandshake(parsedMsgRes.value);
   };
   sendHandshake(msg) {
-    return sendMessage(this.conn, this.codec, msg);
+    const buff = this.codec.toBuffer(msg);
+    if (!buff.ok) {
+      return buff;
+    }
+    const sent = this.conn.send(buff.value);
+    if (!sent) {
+      return {
+        ok: false,
+        reason: "failed to send handshake"
+      };
+    }
+    return {
+      ok: true,
+      value: msg.id
+    };
   }
   _handleStateExit() {
     super._handleStateExit();
@@ -726,28 +759,35 @@ var SessionConnected = class extends IdentifiedSession {
     }
     this.startMissingHeartbeatTimeout();
   }
-  assertSendOrdering(constructedMsg) {
-    if (constructedMsg.seq > this.seqSent + 1) {
-      const msg = `invariant violation: would have sent out of order msg (seq: ${constructedMsg.seq}, expected: ${this.seqSent} + 1)`;
+  assertSendOrdering(encodedMsg) {
+    if (encodedMsg.seq > this.seqSent + 1) {
+      const msg = `invariant violation: would have sent out of order msg (seq: ${encodedMsg.seq}, expected: ${this.seqSent} + 1)`;
       this.log?.error(msg, {
         ...this.loggingMetadata,
-        transportMessage: constructedMsg,
         tags: ["invariant-violation"]
       });
       throw new Error(msg);
     }
   }
   send(msg) {
-    const constructedMsg = this.constructMsg(msg);
-    this.assertSendOrdering(constructedMsg);
-    this.sendBuffer.push(constructedMsg);
-    const res = sendMessage(this.conn, this.codec, constructedMsg);
-    if (!res.ok) {
-      this.listeners.onMessageSendFailure(constructedMsg, res.reason);
-      return res;
+    const encodeResult = this.encodeMsg(msg);
+    if (!encodeResult.ok) {
+      return encodeResult;
+    }
+    const encodedMsg = encodeResult.value;
+    this.assertSendOrdering(encodedMsg);
+    this.sendBuffer.push(encodedMsg);
+    const sent = this.conn.send(encodedMsg.data);
+    if (!sent) {
+      const reason = "failed to send message";
+      this.listeners.onMessageSendFailure(
+        { ...encodedMsg.msg, seq: encodedMsg.seq },
+        reason
+      );
+      return { ok: false, reason };
     }
-    this.seqSent = constructedMsg.seq;
-    return res;
+    this.seqSent = encodedMsg.seq;
+    return { ok: true, value: encodedMsg.id };
   }
   constructor(props) {
     super(props);
@@ -765,10 +805,14 @@ var SessionConnected = class extends IdentifiedSession {
       );
       for (const msg of this.sendBuffer) {
         this.assertSendOrdering(msg);
-        const res = sendMessage(this.conn, this.codec, msg);
-        if (!res.ok) {
-          this.listeners.onMessageSendFailure(msg, res.reason);
-          return res;
+        const sent = this.conn.send(msg.data);
+        if (!sent) {
+          const reason = "failed to send buffered message";
+          this.listeners.onMessageSendFailure(
+            { ...msg.msg, seq: msg.seq },
+            reason
+          );
+          return { ok: false, reason };
         }
         this.seqSent = msg.seq;
       }
@@ -1176,6 +1220,17 @@ var Transport = class {
     const noConnectionSession = SessionStateGraph.transition.ConnectingToNoConnection(session, {
       onSessionGracePeriodElapsed: () => {
         this.onSessionGracePeriodElapsed(noConnectionSession);
+      },
+      onMessageSendFailure: (msg, reason) => {
+        this.log?.error(`failed to send message: ${reason}`, {
+          ...noConnectionSession.loggingMetadata,
+          transportMessage: msg
+        });
+        this.protocolError({
+          type: ProtocolError.MessageSendFailure,
+          message: reason
+        });
+        this.deleteSession(noConnectionSession, { unhealthy: true });
       }
     });
     this.updateSession(noConnectionSession);
@@ -1183,18 +1238,32 @@ var Transport = class {
   }
   onConnClosed(session) {
     let noConnectionSession;
+    const listeners = {
+      onSessionGracePeriodElapsed: () => {
+        this.onSessionGracePeriodElapsed(noConnectionSession);
+      },
+      onMessageSendFailure: (msg, reason) => {
+        this.log?.error(`failed to send message: ${reason}`, {
+          ...noConnectionSession.loggingMetadata,
+          transportMessage: msg
+        });
+        this.protocolError({
+          type: ProtocolError.MessageSendFailure,
+          message: reason
+        });
+        this.deleteSession(noConnectionSession, { unhealthy: true });
+      }
+    };
     if (session.state === "Handshaking" /* Handshaking */) {
-      noConnectionSession = SessionStateGraph.transition.HandshakingToNoConnection(session, {
-        onSessionGracePeriodElapsed: () => {
-          this.onSessionGracePeriodElapsed(noConnectionSession);
-        }
-      });
+      noConnectionSession = SessionStateGraph.transition.HandshakingToNoConnection(
+        session,
+        listeners
+      );
     } else {
-      noConnectionSession = SessionStateGraph.transition.ConnectedToNoConnection(session, {
-        onSessionGracePeriodElapsed: () => {
-          this.onSessionGracePeriodElapsed(noConnectionSession);
-        }
-      });
+      noConnectionSession = SessionStateGraph.transition.ConnectedToNoConnection(
+        session,
+        listeners
+      );
     }
     this.updateSession(noConnectionSession);
     return noConnectionSession;
@@ -1355,6 +1424,17 @@ var ClientTransport = class extends Transport {
       {
         onSessionGracePeriodElapsed: () => {
           this.onSessionGracePeriodElapsed(session);
+        },
+        onMessageSendFailure: (msg, reason) => {
+          this.log?.error(`failed to send message: ${reason}`, {
+            ...session.loggingMetadata,
+            transportMessage: msg
+          });
+          this.protocolError({
+            type: ProtocolError.MessageSendFailure,
+            message: reason
+          });
+          this.deleteSession(session, { unhealthy: true });
         }
       },
       this.options,
@@ -1419,6 +1499,17 @@ var ClientTransport = class extends Transport {
         },
         onSessionGracePeriodElapsed: () => {
           this.onSessionGracePeriodElapsed(handshakingSession);
+        },
+        onMessageSendFailure: (msg, reason) => {
+          this.log?.error(`failed to send message: ${reason}`, {
+            ...handshakingSession.loggingMetadata,
+            transportMessage: msg
+          });
+          this.protocolError({
+            type: ProtocolError.MessageSendFailure,
+            message: reason
+          });
+          this.deleteSession(handshakingSession, { unhealthy: true });
         }
       }
     );
@@ -1580,6 +1671,17 @@ var ClientTransport = class extends Transport {
         },
         onSessionGracePeriodElapsed: () => {
           this.onSessionGracePeriodElapsed(backingOffSession);
+        },
+        onMessageSendFailure: (msg, reason) => {
+          this.log?.error(`failed to send message: ${reason}`, {
+            ...backingOffSession.loggingMetadata,
+            transportMessage: msg
+          });
+          this.protocolError({
+            type: ProtocolError.MessageSendFailure,
+            message: reason
+          });
+          this.deleteSession(backingOffSession, { unhealthy: true });
         }
       }
     );
@@ -1643,6 +1745,17 @@ var ClientTransport = class extends Transport {
         },
         onSessionGracePeriodElapsed: () => {
           this.onSessionGracePeriodElapsed(connectingSession);
+        },
+        onMessageSendFailure: (msg, reason) => {
+          this.log?.error(`failed to send message: ${reason}`, {
+            ...connectingSession.loggingMetadata,
+            transportMessage: msg
+          });
+          this.protocolError({
+            type: ProtocolError.MessageSendFailure,
+            message: reason
+          });
+          this.deleteSession(connectingSession, { unhealthy: true });
         }
       }
     );
@@ -1973,6 +2086,17 @@ var ServerTransport = class extends Transport {
           {
             onSessionGracePeriodElapsed: () => {
               this.onSessionGracePeriodElapsed(noConnectionSession);
+            },
+            onMessageSendFailure: (msg2, reason) => {
+              this.log?.error(`failed to send message: ${reason}`, {
+                ...noConnectionSession.loggingMetadata,
+                transportMessage: msg2
+              });
+              this.protocolError({
+                type: ProtocolError.MessageSendFailure,
+                message: reason
+              });
+              this.deleteSession(noConnectionSession, { unhealthy: true });
             }
           }
         );
@@ -2823,6 +2947,8 @@ function dummySession() {
     "server",
     {
       onSessionGracePeriodElapsed: () => {
+      },
+      onMessageSendFailure: () => {
       }
     },
     testingSessionOptions,