@replit/river 0.23.8 → 0.23.10

package/dist/router/index.d.cts

@@ -1,14 +1,49 @@
-import { A as AnyServiceSchemaMap, I as InstantiatedServiceSchemaMap, P as PayloadType, b as Result, R as RiverError } from '../services-fd8a9894.js';
-export { C as Client, E as Err, O as Ok, x as Output, j as ProcErrors, f as ProcHandler, g as ProcInit, h as ProcInput, i as ProcOutput, k as ProcType, a as Procedure, n as ProcedureMap, d as ProcedureResult, o as RPCProcedure, w as ResultUnwrapErr, v as ResultUnwrapOk, u as RiverErrorSchema, c as RiverUncaughtSchema, m as SerializedServerSchema, S as Service, e as ServiceConfiguration, l as ServiceSchema, q as StreamProcedure, p as SubscriptionProcedure, t as UNCAUGHT_ERROR, U as UploadProcedure, V as ValidProcType, r as createClient, s as serializeSchema } from '../services-fd8a9894.js';
+import { S as SerializedServerSchema, A as AnyServiceSchemaMap, I as InstantiatedServiceSchemaMap, P as PayloadType, b as Result, R as RiverError } from '../services-409c5545.js';
+export { C as Client, E as Err, O as Ok, z as Output, k as ProcErrors, g as ProcHandler, h as ProcInit, i as ProcInput, j as ProcOutput, l as ProcType, a as Procedure, p as ProcedureMap, d as ProcedureResult, q as RPCProcedure, y as ResultUnwrapErr, x as ResultUnwrapOk, w as RiverErrorSchema, c as RiverUncaughtSchema, o as SerializedProcedureSchema, n as SerializedServiceSchema, e as Service, f as ServiceConfiguration, m as ServiceSchema, t as StreamProcedure, r as SubscriptionProcedure, v as UNCAUGHT_ERROR, U as UploadProcedure, V as ValidProcType, u as createClient, s as serializeSchema } from '../services-409c5545.js';
 import { Static } from '@sinclair/typebox';
-import { d as ServerTransport, C as Connection, f as ServerHandshakeOptions, b as ServiceContext } from '../transport-3d34f714.js';
-export { g as ParsedMetadata, h as ServiceContextWithState, i as ServiceContextWithTransportInfo, j as createClientHandshakeOptions, k as createServerHandshakeOptions } from '../transport-3d34f714.js';
+import { d as ServerTransport, C as Connection, f as ServerHandshakeOptions, b as ServiceContext } from '../transport-cf856c41.js';
+export { g as ParsedMetadata, h as ServiceContextWithState, i as ServiceContextWithTransportInfo, j as createClientHandshakeOptions, k as createServerHandshakeOptions } from '../transport-cf856c41.js';
 import { Pushable } from 'it-pushable';
 import '../index-60f03cb7.js';
 import '@sinclair/typebox/value';
 import '@opentelemetry/api';
 import '../types-3e5768ec.js';
 
+interface ServerBreakage {
+    serviceBreakages: Record<string, ServiceBreakage>;
+}
+type ServiceBreakage = {
+    reason: 'removed';
+} | {
+    reason: 'modified';
+    procedureBreakages: Record<string, ProcedureBreakage>;
+};
+type ProcedureBreakage = {
+    reason: 'removed';
+} | {
+    reason: 'type-changed';
+    oldType: string;
+    newType: string;
+} | {
+    reason: 'modified';
+    input?: PayloadBreakage;
+    init?: PayloadBreakage;
+    output?: PayloadBreakage;
+};
+type PayloadBreakage = {
+    reason: 'type-changed';
+    oldType: string;
+    newType: string;
+} | {
+    reason: 'new-required';
+} | {
+    reason: 'removed-required';
+} | {
+    reason: 'field-breakage';
+    fieldBreakages: Record<string, PayloadBreakage>;
+};
+declare function diffServerSchema(oldServer: SerializedServerSchema, newServer: SerializedServerSchema): ServerBreakage | null;
+
 /**
  * Represents a server with a set of services. Use {@link createServer} to create it.
  * @template Services - The type of services provided by the server.
@@ -16,7 +51,6 @@ import '../types-3e5768ec.js';
 interface Server<Services extends AnyServiceSchemaMap> {
     services: InstantiatedServiceSchemaMap<Services>;
     streams: Map<string, ProcStream>;
-    close(): Promise<void>;
 }
 interface ProcStream {
     id: string;
@@ -43,6 +77,6 @@ declare function createServer<Services extends AnyServiceSchemaMap>(transport: S
     extendedContext?: Omit<ServiceContext, 'state'>;
 }>): Server<Services>;
 
-var version = "0.23.8";
+var version = "0.23.10";
 
-export { PayloadType, version as RIVER_VERSION, Result, RiverError, Server, ServiceContext, createServer };
+export { PayloadBreakage, PayloadType, ProcedureBreakage, version as RIVER_VERSION, Result, RiverError, SerializedServerSchema, Server, ServerBreakage, ServiceBreakage, ServiceContext, createServer, diffServerSchema };

package/dist/router/index.d.ts

@@ -1,14 +1,49 @@
-import { A as AnyServiceSchemaMap, I as InstantiatedServiceSchemaMap, P as PayloadType, b as Result, R as RiverError } from '../services-fd8a9894.js';
-export { C as Client, E as Err, O as Ok, x as Output, j as ProcErrors, f as ProcHandler, g as ProcInit, h as ProcInput, i as ProcOutput, k as ProcType, a as Procedure, n as ProcedureMap, d as ProcedureResult, o as RPCProcedure, w as ResultUnwrapErr, v as ResultUnwrapOk, u as RiverErrorSchema, c as RiverUncaughtSchema, m as SerializedServerSchema, S as Service, e as ServiceConfiguration, l as ServiceSchema, q as StreamProcedure, p as SubscriptionProcedure, t as UNCAUGHT_ERROR, U as UploadProcedure, V as ValidProcType, r as createClient, s as serializeSchema } from '../services-fd8a9894.js';
+import { S as SerializedServerSchema, A as AnyServiceSchemaMap, I as InstantiatedServiceSchemaMap, P as PayloadType, b as Result, R as RiverError } from '../services-409c5545.js';
+export { C as Client, E as Err, O as Ok, z as Output, k as ProcErrors, g as ProcHandler, h as ProcInit, i as ProcInput, j as ProcOutput, l as ProcType, a as Procedure, p as ProcedureMap, d as ProcedureResult, q as RPCProcedure, y as ResultUnwrapErr, x as ResultUnwrapOk, w as RiverErrorSchema, c as RiverUncaughtSchema, o as SerializedProcedureSchema, n as SerializedServiceSchema, e as Service, f as ServiceConfiguration, m as ServiceSchema, t as StreamProcedure, r as SubscriptionProcedure, v as UNCAUGHT_ERROR, U as UploadProcedure, V as ValidProcType, u as createClient, s as serializeSchema } from '../services-409c5545.js';
 import { Static } from '@sinclair/typebox';
-import { d as ServerTransport, C as Connection, f as ServerHandshakeOptions, b as ServiceContext } from '../transport-3d34f714.js';
-export { g as ParsedMetadata, h as ServiceContextWithState, i as ServiceContextWithTransportInfo, j as createClientHandshakeOptions, k as createServerHandshakeOptions } from '../transport-3d34f714.js';
+import { d as ServerTransport, C as Connection, f as ServerHandshakeOptions, b as ServiceContext } from '../transport-cf856c41.js';
+export { g as ParsedMetadata, h as ServiceContextWithState, i as ServiceContextWithTransportInfo, j as createClientHandshakeOptions, k as createServerHandshakeOptions } from '../transport-cf856c41.js';
 import { Pushable } from 'it-pushable';
 import '../index-60f03cb7.js';
 import '@sinclair/typebox/value';
 import '@opentelemetry/api';
 import '../types-3e5768ec.js';
 
+interface ServerBreakage {
+    serviceBreakages: Record<string, ServiceBreakage>;
+}
+type ServiceBreakage = {
+    reason: 'removed';
+} | {
+    reason: 'modified';
+    procedureBreakages: Record<string, ProcedureBreakage>;
+};
+type ProcedureBreakage = {
+    reason: 'removed';
+} | {
+    reason: 'type-changed';
+    oldType: string;
+    newType: string;
+} | {
+    reason: 'modified';
+    input?: PayloadBreakage;
+    init?: PayloadBreakage;
+    output?: PayloadBreakage;
+};
+type PayloadBreakage = {
+    reason: 'type-changed';
+    oldType: string;
+    newType: string;
+} | {
+    reason: 'new-required';
+} | {
+    reason: 'removed-required';
+} | {
+    reason: 'field-breakage';
+    fieldBreakages: Record<string, PayloadBreakage>;
+};
+declare function diffServerSchema(oldServer: SerializedServerSchema, newServer: SerializedServerSchema): ServerBreakage | null;
+
 /**
  * Represents a server with a set of services. Use {@link createServer} to create it.
  * @template Services - The type of services provided by the server.
@@ -16,7 +51,6 @@ import '../types-3e5768ec.js';
 interface Server<Services extends AnyServiceSchemaMap> {
     services: InstantiatedServiceSchemaMap<Services>;
     streams: Map<string, ProcStream>;
-    close(): Promise<void>;
 }
 interface ProcStream {
     id: string;
@@ -43,6 +77,6 @@ declare function createServer<Services extends AnyServiceSchemaMap>(transport: S
     extendedContext?: Omit<ServiceContext, 'state'>;
 }>): Server<Services>;
 
-var version = "0.23.8";
+var version = "0.23.10";
 
-export { PayloadType, version as RIVER_VERSION, Result, RiverError, Server, ServiceContext, createServer };
+export { PayloadBreakage, PayloadType, ProcedureBreakage, version as RIVER_VERSION, Result, RiverError, SerializedServerSchema, Server, ServerBreakage, ServiceBreakage, ServiceContext, createServer, diffServerSchema };

package/dist/router/index.js

@@ -9,11 +9,12 @@ import {
   createClientHandshakeOptions,
   createServer,
   createServerHandshakeOptions,
+  diffServerSchema,
   serializeSchema
-} from "../chunk-WQRQFAP6.js";
+} from "../chunk-B323CECK.js";
 import {
   version
-} from "../chunk-ELZRZNA5.js";
+} from "../chunk-FRICSBDW.js";
 export {
   Err,
   Ok,
@@ -26,6 +27,7 @@ export {
   createClientHandshakeOptions,
   createServer,
   createServerHandshakeOptions,
+  diffServerSchema,
   serializeSchema
 };
 //# sourceMappingURL=index.js.map

package/dist/{services-fd8a9894.d.ts → services-409c5545.d.ts}

@@ -1,5 +1,5 @@
 import { Static, TObject, TUnion, TString, TSchema, TNever, TLiteral } from '@sinclair/typebox';
-import { c as ClientTransport, C as Connection, l as ClientHandshakeOptions, i as ServiceContextWithTransportInfo } from './transport-3d34f714.js';
+import { c as ClientTransport, C as Connection, l as ClientHandshakeOptions, i as ServiceContextWithTransportInfo } from './transport-cf856c41.js';
 import { Pushable } from 'it-pushable';
 import { T as TransportClientId } from './index-60f03cb7.js';
 
@@ -477,14 +477,15 @@ interface ServiceConfiguration<State extends object> {
      */
     initializeState: () => State;
 }
+interface SerializedProcedureSchema {
+    input: PayloadType;
+    output: PayloadType;
+    errors?: RiverError;
+    type: 'rpc' | 'subscription' | 'upload' | 'stream';
+    init?: PayloadType;
+}
 interface SerializedServiceSchema {
-    procedures: Record<string, {
-        input: PayloadType;
-        output: PayloadType;
-        errors?: RiverError;
-        type: 'rpc' | 'subscription' | 'upload' | 'stream';
-        init?: PayloadType;
-    }>;
+    procedures: Record<string, SerializedProcedureSchema>;
 }
 interface SerializedServerSchema {
     handshakeSchema?: TSchema;
@@ -704,4 +705,4 @@ declare class ServiceScaffold<State extends object> {
     }>;
 }
 
-export { AnyServiceSchemaMap as A, Client as C, Err as E, InstantiatedServiceSchemaMap as I, Ok as O, PayloadType as P, RiverError as R, Service as S, UploadProcedure as U, ValidProcType as V, Procedure as a, Result as b, RiverUncaughtSchema as c, ProcedureResult as d, ServiceConfiguration as e, ProcHandler as f, ProcInit as g, ProcInput as h, ProcOutput as i, ProcErrors as j, ProcType as k, ServiceSchema as l, SerializedServerSchema as m, ProcedureMap as n, RpcProcedure as o, SubscriptionProcedure as p, StreamProcedure as q, createClient as r, serializeSchema as s, UNCAUGHT_ERROR as t, RiverErrorSchema as u, ResultUnwrapOk as v, ResultUnwrapErr as w, Output as x };
+export { AnyServiceSchemaMap as A, Client as C, Err as E, InstantiatedServiceSchemaMap as I, Ok as O, PayloadType as P, RiverError as R, SerializedServerSchema as S, UploadProcedure as U, ValidProcType as V, Procedure as a, Result as b, RiverUncaughtSchema as c, ProcedureResult as d, Service as e, ServiceConfiguration as f, ProcHandler as g, ProcInit as h, ProcInput as i, ProcOutput as j, ProcErrors as k, ProcType as l, ServiceSchema as m, SerializedServiceSchema as n, SerializedProcedureSchema as o, ProcedureMap as p, RpcProcedure as q, SubscriptionProcedure as r, serializeSchema as s, StreamProcedure as t, createClient as u, UNCAUGHT_ERROR as v, RiverErrorSchema as w, ResultUnwrapOk as x, ResultUnwrapErr as y, Output as z };

package/dist/transport/impls/uds/client.cjs

@@ -110,7 +110,7 @@ function isAck(controlFlag) {
 var import_api = require("@opentelemetry/api");
 
 // package.json
-var version = "0.23.8";
+var version = "0.23.10";
 
 // tracing/index.ts
 function getPropagationContext(ctx) {
@@ -169,7 +169,7 @@ var Connection = class {
   get loggingMetadata() {
     const metadata = { connId: this.id };
     const spanContext = this.telemetry?.span.spanContext();
-    if (spanContext) {
+    if (this.telemetry?.span.isRecording() && spanContext) {
       metadata.telemetry = {
         traceId: spanContext.traceId,
         spanId: spanContext.spanId
@@ -190,6 +190,11 @@ var Session = class {
    * The active connection associated with this session
    */
   connection;
+  /**
+   * A connection that is currently undergoing handshaking. Used to distinguish between the active
+   * connection, but still be able to close it if needed.
+   */
+  handshakingConnection;
   from;
   to;
   /**
@@ -369,23 +374,60 @@ var Session = class {
     this.cancelGrace();
     this.sendBufferedMessages(newConn);
     this.connection = newConn;
+    this.handshakingConnection = void 0;
+  }
+  replaceWithNewHandshakingConnection(newConn) {
+    this.handshakingConnection = newConn;
   }
   beginGrace(cb) {
     this.log?.info(
       `starting ${this.options.sessionDisconnectGraceMs}ms grace period until session to ${this.to} is closed`,
       this.loggingMetadata
     );
+    this.cancelGrace({ keepHeartbeatMisses: true });
     this.disconnectionGrace = setTimeout(() => {
-      this.close();
+      if (this.connection !== void 0) {
+        this.log?.warn(
+          `grace period for ${this.to} elapsed while connected. not calling callback`,
+          {
+            ...this.loggingMetadata,
+            connId: this.connection.id,
+            tags: ["invariant-violation"]
+          }
+        );
+        return;
+      }
+      this.log?.info(
+        `grace period for ${this.to} elapsed`,
+        this.loggingMetadata
+      );
       cb();
     }, this.options.sessionDisconnectGraceMs);
   }
   // called on reconnect of the underlying session
-  cancelGrace() {
-    this.heartbeatMisses = 0;
+  cancelGrace({ keepHeartbeatMisses } = {
+    keepHeartbeatMisses: false
+  }) {
+    if (!keepHeartbeatMisses) {
+      this.heartbeatMisses = 0;
+    }
+    if (this.disconnectionGrace === void 0)
+      return;
     clearTimeout(this.disconnectionGrace);
     this.disconnectionGrace = void 0;
   }
+  /**
+   * Used to close the handshaking connection, if set.
+   */
+  closeHandshakingConnection(expectedHandshakingConn) {
+    if (this.handshakingConnection === void 0)
+      return;
+    if (expectedHandshakingConn !== void 0 && this.handshakingConnection === expectedHandshakingConn) {
+      return;
+    }
+    this.handshakingConnection.close();
+    this.handshakingConnection = void 0;
+  }
   // closed when we want to discard the whole session
   // (i.e. shutdown or session disconnect)
   close() {
@@ -576,11 +618,13 @@ var createLogProxy = (log) => ({
 var ProtocolError = {
   RetriesExceeded: "conn_retry_exceeded",
   HandshakeFailed: "handshake_failed",
-  UseAfterDestroy: "use_after_destroy",
   MessageOrderingViolated: "message_ordering_violated"
 };
 var EventDispatcher = class {
   eventListeners = {};
+  removeAllListeners() {
+    this.eventListeners = {};
+  }
   numberOfListeners(eventType) {
     return this.eventListeners[eventType]?.size ?? 0;
   }
@@ -599,7 +643,8 @@ var EventDispatcher = class {
   dispatchEvent(eventType, event) {
     const handlers = this.eventListeners[eventType];
     if (handlers) {
-      for (const handler of handlers) {
+      const copy = [...handlers];
+      for (const handler of copy) {
         handler(event);
       }
     }
@@ -761,10 +806,9 @@ var defaultServerTransportOptions = {
 };
 var Transport = class {
   /**
-   * A flag indicating whether the transport has been destroyed.
-   * A destroyed transport will not attempt to reconnect and cannot be used again.
+   * The status of the transport.
    */
-  state;
+  status;
   /**
    * The {@link Codec} used to encode and decode messages.
    */
@@ -806,7 +850,7 @@ var Transport = class {
     this.sessions = /* @__PURE__ */ new Map();
     this.codec = this.options.codec;
     this.clientId = clientId;
-    this.state = "open";
+    this.status = "open";
   }
   bindLogger(fn, level) {
     if (typeof fn === "function") {
@@ -854,7 +898,13 @@ var Transport = class {
     });
     return session;
   }
-  getOrCreateSession(to, conn, sessionId, propagationCtx) {
+  getOrCreateSession({
+    to,
+    conn,
+    handshakingConn,
+    sessionId,
+    propagationCtx
+  }) {
     let session = this.sessions.get(to);
     let isReconnect = session !== void 0;
     if (session?.advertisedSessionId !== void 0 && sessionId !== void 0 && session.advertisedSessionId !== sessionId) {
@@ -862,7 +912,11 @@ var Transport = class {
         `session for ${to} already exists but has a different session id (expected: ${session.advertisedSessionId}, got: ${sessionId}), creating a new one`,
         session.loggingMetadata
       );
-      this.deleteSession(session);
+      this.deleteSession({
+        session,
+        closeHandshakingConnection: handshakingConn !== void 0,
+        handshakingConn
+      });
       isReconnect = false;
       session = void 0;
     }
@@ -876,9 +930,19 @@ var Transport = class {
     if (sessionId !== void 0) {
       session.advertisedSessionId = sessionId;
     }
+    if (handshakingConn !== void 0) {
+      session.replaceWithNewHandshakingConnection(handshakingConn);
+    }
     return { session, isReconnect };
   }
-  deleteSession(session) {
+  deleteSession({
+    session,
+    closeHandshakingConnection,
+    handshakingConn
+  }) {
+    if (closeHandshakingConnection) {
+      session.closeHandshakingConnection(handshakingConn);
+    }
     session.close();
     session.telemetry.span.end();
     this.sessions.delete(session.to);
@@ -905,7 +969,11 @@ var Transport = class {
     session.connection = void 0;
     session.beginGrace(() => {
       session.telemetry.span.addEvent("session grace period expired");
-      this.deleteSession(session);
+      this.deleteSession({
+        session,
+        closeHandshakingConnection: true,
+        handshakingConn: conn
+      });
     });
   }
   /**
@@ -944,7 +1012,7 @@ var Transport = class {
    * @param msg The received message.
    */
   handleMsg(msg, conn) {
-    if (this.state !== "open")
+    if (this.getStatus() !== "open")
       return;
     const session = this.sessions.get(msg.from);
     if (!session) {
@@ -985,7 +1053,7 @@ var Transport = class {
           code: import_api3.SpanStatusCode.ERROR,
           message: "message order violated"
         });
-        session.close();
+        this.deleteSession({ session, closeHandshakingConnection: true });
       }
       return;
     }
@@ -1023,23 +1091,16 @@ var Transport = class {
    * @returns The ID of the sent message or undefined if it wasn't sent
    */
   send(to, msg) {
-    if (this.state === "destroyed") {
-      const err = "transport is destroyed, cant send";
+    if (this.getStatus() === "closed") {
+      const err = "transport is closed, cant send";
       this.log?.error(err, {
         clientId: this.clientId,
         transportMessage: msg,
         tags: ["invariant-violation"]
       });
-      this.protocolError(ProtocolError.UseAfterDestroy, err);
-      return void 0;
-    } else if (this.state === "closed") {
-      this.log?.info(`transport closed when sending, discarding`, {
-        clientId: this.clientId,
-        transportMessage: msg
-      });
-      return void 0;
+      throw new Error(err);
     }
-    return this.getOrCreateSession(to).session.send(msg);
+    return this.getOrCreateSession({ to }).session.send(msg);
   }
   // control helpers
   sendCloseStream(to, streamId) {
@@ -1060,23 +1121,18 @@ var Transport = class {
    * Closes the transport. Any messages sent while the transport is closed will be silently discarded.
    */
   close() {
-    this.state = "closed";
+    this.status = "closed";
     for (const session of this.sessions.values()) {
-      this.deleteSession(session);
+      this.deleteSession({ session, closeHandshakingConnection: true });
     }
+    this.eventDispatcher.dispatchEvent("transportStatus", {
+      status: this.status
+    });
+    this.eventDispatcher.removeAllListeners();
     this.log?.info(`manually closed transport`, { clientId: this.clientId });
   }
-  /**
-   * Default destroy implementation for transports. You should override this in the downstream
-   * implementation if you need to do any additional cleanup and call super.destroy() at the end.
-   * Destroys the transport. Any messages sent while the transport is destroyed will throw an error.
-   */
-  destroy() {
-    this.state = "destroyed";
-    for (const session of this.sessions.values()) {
-      this.deleteSession(session);
-    }
-    this.log?.info(`manually destroyed transport`, { clientId: this.clientId });
+  getStatus() {
+    return this.status;
   }
 };
 var ClientTransport = class extends Transport {
@@ -1113,26 +1169,26 @@ var ClientTransport = class extends Transport {
     this.handshakeExtensions = options;
   }
   handleConnection(conn, to) {
-    if (this.state !== "open")
+    if (this.getStatus() !== "open")
       return;
     let session = void 0;
     const handshakeTimeout = setTimeout(() => {
-      if (!session) {
-        this.log?.warn(
-          `connection to ${to} timed out waiting for handshake, closing`,
-          { ...conn.loggingMetadata, clientId: this.clientId, connectedTo: to }
-        );
-        conn.close();
-      }
+      if (session)
+        return;
+      this.log?.warn(
+        `connection to ${to} timed out waiting for handshake, closing`,
+        { ...conn.loggingMetadata, clientId: this.clientId, connectedTo: to }
+      );
+      conn.close();
     }, this.options.sessionDisconnectGraceMs);
     const handshakeHandler = (data) => {
       const maybeSession = this.receiveHandshakeResponseMessage(data, conn);
+      clearTimeout(handshakeTimeout);
       if (!maybeSession) {
         conn.close();
         return;
       } else {
         session = maybeSession;
-        clearTimeout(handshakeTimeout);
       }
       conn.removeDataListener(handshakeHandler);
       conn.addDataListener((data2) => {
@@ -1239,11 +1295,11 @@ var ClientTransport = class extends Transport {
       connectedTo: parsed.from,
       transportMessage: parsed
     });
-    const { session, isReconnect } = this.getOrCreateSession(
-      parsed.from,
+    const { session, isReconnect } = this.getOrCreateSession({
+      to: parsed.from,
       conn,
-      parsed.payload.status.sessionId
-    );
+      sessionId: parsed.payload.status.sessionId
+    });
     this.onConnect(conn, parsed.from, session, isReconnect);
     this.retryBudget.startRestoringBudget(parsed.from);
     return session;
@@ -1253,7 +1309,7 @@ var ClientTransport = class extends Transport {
    * @param to The client ID of the node to connect to.
    */
   async connect(to) {
-    const canProceedWithConnection = () => this.state === "open";
+    const canProceedWithConnection = () => this.getStatus() === "open";
     if (!canProceedWithConnection()) {
       this.log?.info(
         `transport state is no longer open, cancelling attempt to connect to ${to}`,
@@ -1349,9 +1405,17 @@ var ClientTransport = class extends Transport {
       }
     }
   }
-  deleteSession(session) {
+  deleteSession({
+    session,
+    closeHandshakingConnection,
+    handshakingConn
+  }) {
     this.inflightConnectionPromises.delete(session.to);
-    super.deleteSession(session);
+    super.deleteSession({
+      session,
+      closeHandshakingConnection,
+      handshakingConn
+    });
   }
   async sendHandshake(to, conn) {
     let metadata = void 0;
@@ -1378,7 +1442,7 @@ var ClientTransport = class extends Transport {
         return false;
       }
     }
-    const { session } = this.getOrCreateSession(to);
+    const { session } = this.getOrCreateSession({ to, handshakingConn: conn });
     const requestMsg = handshakeRequestMessage(
       this.clientId,
       to,