@orpc/standard-server-peer 1.7.10 → 1.8.0

package/README.md

@@ -30,6 +30,7 @@
 - **🔗 End-to-End Type Safety**: Ensure type-safe inputs, outputs, and errors from client to server.
 - **📘 First-Class OpenAPI**: Built-in support that fully adheres to the OpenAPI standard.
 - **📝 Contract-First Development**: Optionally define your API contract before implementation.
+- **🔍 First-Class OpenTelemetry**: Seamlessly integrate with OpenTelemetry for observability.
 - **⚙️ Framework Integrations**: Seamlessly integrate with TanStack Query (React, Vue, Solid, Svelte, Angular), Pinia Colada, and more.
 - **🚀 Server Actions**: Fully compatible with React Server Actions on Next.js, TanStack Start, and other platforms.
 - **🔠 Standard Schema Support**: Works out of the box with Zod, Valibot, ArkType, and other schema validators.
@@ -38,7 +39,6 @@
 - **📡 SSE & Streaming**: Enjoy full type-safe support for SSE and streaming.
 - **🌍 Multi-Runtime Support**: Fast and lightweight on Cloudflare, Deno, Bun, Node.js, and beyond.
 - **🔌 Extendability**: Easily extend functionality with plugins, middleware, and interceptors.
-- **🛡️ Reliability**: Well-tested, TypeScript-based, production-ready, and MIT licensed.
 
 ## Documentation
 
@@ -50,6 +50,7 @@ You can find the full documentation [here](https://orpc.unnoq.com).
 - [@orpc/server](https://www.npmjs.com/package/@orpc/server): Build your API or implement API contract.
 - [@orpc/client](https://www.npmjs.com/package/@orpc/client): Consume your API on the client with type-safety.
 - [@orpc/openapi](https://www.npmjs.com/package/@orpc/openapi): Generate OpenAPI specs and handle OpenAPI requests.
+- [@orpc/otel](https://www.npmjs.com/package/@orpc/otel): [OpenTelemetry](https://opentelemetry.io/) integration for observability.
 - [@orpc/nest](https://www.npmjs.com/package/@orpc/nest): Deeply integrate oRPC with [NestJS](https://nestjs.com/).
 - [@orpc/react](https://www.npmjs.com/package/@orpc/react): Utilities for integrating oRPC with React and React Server Actions.
 - [@orpc/tanstack-query](https://www.npmjs.com/package/@orpc/tanstack-query): [TanStack Query](https://tanstack.com/query/latest) integration.

package/dist/index.d.mts

@@ -1,4 +1,4 @@
-import { Promisable, AsyncIdQueueCloseOptions as AsyncIdQueueCloseOptions$1, AsyncIteratorClassCleanupFn, AsyncIteratorClass } from '@orpc/shared';
+import { Promisable, AsyncIdQueueCloseOptions as AsyncIdQueueCloseOptions$1, SetSpanErrorOptions, AsyncIteratorClassCleanupFn, AsyncIteratorClass } from '@orpc/shared';
 import { StandardRequest, StandardResponse, EventMeta } from '@orpc/standard-server';
 
 type EncodedMessage = string | ArrayBufferLike | Uint8Array;
@@ -16,9 +16,22 @@ interface ClientPeerCloseOptions extends AsyncIdQueueCloseOptions$1 {
 }
 declare class ClientPeer {
     private readonly idGenerator;
+    /**
+     * Queue of responses sent from server, awaiting consumption
+     */
     private readonly responseQueue;
+    /**
+     * Queue of event iterator messages sent from server, awaiting consumption
+     */
     private readonly serverEventIteratorQueue;
+    /**
+     * Controllers used to signal that the client should stop sending event iterator messages
+     */
     private readonly serverControllers;
+    /**
+     * Cleanup functions invoked when the request/response is closed
+     */
+    private readonly cleanupFns;
     private readonly send;
     constructor(send: EncodedMessageSendFn);
     get length(): number;
@@ -77,9 +90,14 @@ declare class AsyncIdQueue<T> {
     assertOpen(id: string): void;
 }
 
-declare function toEventIterator(queue: AsyncIdQueue<EventIteratorPayload>, id: string, cleanup: AsyncIteratorClassCleanupFn): AsyncIteratorClass<unknown>;
+interface ToEventIteratorOptions extends SetSpanErrorOptions {
+}
+declare function toEventIterator(queue: AsyncIdQueue<EventIteratorPayload>, id: string, cleanup: AsyncIteratorClassCleanupFn, options?: ToEventIteratorOptions): AsyncIteratorClass<unknown>;
 declare function resolveEventIterator(iterator: AsyncIterator<any>, callback: (payload: EventIteratorPayload) => Promise<'next' | 'abort'>): Promise<void>;
 
+interface ServerPeerHandleRequestFn {
+    (request: StandardRequest): Promise<StandardResponse>;
+}
 interface ServerPeerCloseOptions extends AsyncIdQueueCloseOptions$1 {
     /**
      * Should abort or not?
@@ -89,16 +107,28 @@ interface ServerPeerCloseOptions extends AsyncIdQueueCloseOptions$1 {
     abort?: boolean;
 }
 declare class ServerPeer {
+    /**
+     * Queue of event iterator messages sent from client, awaiting consumption
+     */
     private readonly clientEventIteratorQueue;
+    /**
+     * Map of active client request controllers, should be synced to request signal
+     */
     private readonly clientControllers;
     private readonly send;
     constructor(send: EncodedMessageSendFn);
     get length(): number;
     open(id: string): AbortController;
-    message(raw: EncodedMessage): Promise<[id: string, StandardRequest | undefined]>;
+    /**
+     * @todo This method will return Promise<void> in the next major version.
+     */
+    message(raw: EncodedMessage, handleRequest?: ServerPeerHandleRequestFn): Promise<[id: string, StandardRequest | undefined]>;
+    /**
+     * @deprecated Please pass the `handleRequest` (second arg) function to the `message` method instead.
+     */
     response(id: string, response: StandardResponse): Promise<void>;
     close({ abort, ...options }?: ServerPeerCloseOptions): void;
 }
 
 export { ClientPeer, MessageType, ServerPeer, decodeRequestMessage, decodeResponseMessage, encodeRequestMessage, encodeResponseMessage, resolveEventIterator, toEventIterator };
-export type { ClientPeerCloseOptions, DecodedRequestMessage, DecodedResponseMessage, EncodedMessage, EncodedMessageSendFn, EventIteratorEvent, EventIteratorPayload, RequestMessageMap, ResponseMessageMap, ServerPeerCloseOptions };
+export type { ClientPeerCloseOptions, DecodedRequestMessage, DecodedResponseMessage, EncodedMessage, EncodedMessageSendFn, EventIteratorEvent, EventIteratorPayload, RequestMessageMap, ResponseMessageMap, ServerPeerCloseOptions, ServerPeerHandleRequestFn, ToEventIteratorOptions };

package/dist/index.d.ts

@@ -1,4 +1,4 @@
-import { Promisable, AsyncIdQueueCloseOptions as AsyncIdQueueCloseOptions$1, AsyncIteratorClassCleanupFn, AsyncIteratorClass } from '@orpc/shared';
+import { Promisable, AsyncIdQueueCloseOptions as AsyncIdQueueCloseOptions$1, SetSpanErrorOptions, AsyncIteratorClassCleanupFn, AsyncIteratorClass } from '@orpc/shared';
 import { StandardRequest, StandardResponse, EventMeta } from '@orpc/standard-server';
 
 type EncodedMessage = string | ArrayBufferLike | Uint8Array;
@@ -16,9 +16,22 @@ interface ClientPeerCloseOptions extends AsyncIdQueueCloseOptions$1 {
 }
 declare class ClientPeer {
     private readonly idGenerator;
+    /**
+     * Queue of responses sent from server, awaiting consumption
+     */
     private readonly responseQueue;
+    /**
+     * Queue of event iterator messages sent from server, awaiting consumption
+     */
     private readonly serverEventIteratorQueue;
+    /**
+     * Controllers used to signal that the client should stop sending event iterator messages
+     */
     private readonly serverControllers;
+    /**
+     * Cleanup functions invoked when the request/response is closed
+     */
+    private readonly cleanupFns;
     private readonly send;
     constructor(send: EncodedMessageSendFn);
     get length(): number;
@@ -77,9 +90,14 @@ declare class AsyncIdQueue<T> {
     assertOpen(id: string): void;
 }
 
-declare function toEventIterator(queue: AsyncIdQueue<EventIteratorPayload>, id: string, cleanup: AsyncIteratorClassCleanupFn): AsyncIteratorClass<unknown>;
+interface ToEventIteratorOptions extends SetSpanErrorOptions {
+}
+declare function toEventIterator(queue: AsyncIdQueue<EventIteratorPayload>, id: string, cleanup: AsyncIteratorClassCleanupFn, options?: ToEventIteratorOptions): AsyncIteratorClass<unknown>;
 declare function resolveEventIterator(iterator: AsyncIterator<any>, callback: (payload: EventIteratorPayload) => Promise<'next' | 'abort'>): Promise<void>;
 
+interface ServerPeerHandleRequestFn {
+    (request: StandardRequest): Promise<StandardResponse>;
+}
 interface ServerPeerCloseOptions extends AsyncIdQueueCloseOptions$1 {
     /**
      * Should abort or not?
@@ -89,16 +107,28 @@ interface ServerPeerCloseOptions extends AsyncIdQueueCloseOptions$1 {
     abort?: boolean;
 }
 declare class ServerPeer {
+    /**
+     * Queue of event iterator messages sent from client, awaiting consumption
+     */
     private readonly clientEventIteratorQueue;
+    /**
+     * Map of active client request controllers, should be synced to request signal
+     */
     private readonly clientControllers;
     private readonly send;
     constructor(send: EncodedMessageSendFn);
     get length(): number;
     open(id: string): AbortController;
-    message(raw: EncodedMessage): Promise<[id: string, StandardRequest | undefined]>;
+    /**
+     * @todo This method will return Promise<void> in the next major version.
+     */
+    message(raw: EncodedMessage, handleRequest?: ServerPeerHandleRequestFn): Promise<[id: string, StandardRequest | undefined]>;
+    /**
+     * @deprecated Please pass the `handleRequest` (second arg) function to the `message` method instead.
+     */
     response(id: string, response: StandardResponse): Promise<void>;
     close({ abort, ...options }?: ServerPeerCloseOptions): void;
 }
 
 export { ClientPeer, MessageType, ServerPeer, decodeRequestMessage, decodeResponseMessage, encodeRequestMessage, encodeResponseMessage, resolveEventIterator, toEventIterator };
-export type { ClientPeerCloseOptions, DecodedRequestMessage, DecodedResponseMessage, EncodedMessage, EncodedMessageSendFn, EventIteratorEvent, EventIteratorPayload, RequestMessageMap, ResponseMessageMap, ServerPeerCloseOptions };
+export type { ClientPeerCloseOptions, DecodedRequestMessage, DecodedResponseMessage, EncodedMessage, EncodedMessageSendFn, EventIteratorEvent, EventIteratorPayload, RequestMessageMap, ResponseMessageMap, ServerPeerCloseOptions, ServerPeerHandleRequestFn, ToEventIteratorOptions };

package/dist/index.mjs

@@ -1,5 +1,5 @@
-import { isAsyncIteratorObject, stringifyJSON, readAsBuffer, AsyncIteratorClass, isTypescriptObject, SequentialIdGenerator, AsyncIdQueue } from '@orpc/shared';
-import { generateContentDisposition, flattenHeader, getFilenameFromContentDisposition, withEventMeta, ErrorEvent, getEventMeta, isEventIteratorHeaders, experimental_HibernationEventIterator } from '@orpc/standard-server';
+import { isAsyncIteratorObject, stringifyJSON, readAsBuffer, AsyncIteratorClass, startSpan, runInSpanContext, isTypescriptObject, setSpanError, runWithSpan, SequentialIdGenerator, AsyncIdQueue, getGlobalOtelConfig, clone, AbortError } from '@orpc/shared';
+import { generateContentDisposition, flattenHeader, getFilenameFromContentDisposition, withEventMeta, ErrorEvent, getEventMeta, isEventIteratorHeaders, HibernationEventIterator } from '@orpc/standard-server';
 
 var MessageType = /* @__PURE__ */ ((MessageType2) => {
   MessageType2[MessageType2["REQUEST"] = 1] = "REQUEST";
@@ -179,80 +179,138 @@ async function decodeRawMessage(raw) {
   };
 }
 
-function toEventIterator(queue, id, cleanup) {
+function toEventIterator(queue, id, cleanup, options = {}) {
+  let span;
   return new AsyncIteratorClass(async () => {
-    const item = await queue.pull(id);
-    switch (item.event) {
-      case "message": {
-        let data = item.data;
-        if (item.meta && isTypescriptObject(data)) {
-          data = withEventMeta(data, item.meta);
+    span ??= startSpan("consume_event_iterator_stream");
+    try {
+      const item = await runInSpanContext(span, () => queue.pull(id));
+      switch (item.event) {
+        case "message": {
+          let data = item.data;
+          if (item.meta && isTypescriptObject(data)) {
+            data = withEventMeta(data, item.meta);
+          }
+          span?.addEvent("message");
+          return { value: data, done: false };
         }
-        return { value: data, done: false };
-      }
-      case "error": {
-        let error = new ErrorEvent({
-          data: item.data
-        });
-        if (item.meta) {
-          error = withEventMeta(error, item.meta);
+        case "error": {
+          let error = new ErrorEvent({
+            data: item.data
+          });
+          if (item.meta) {
+            error = withEventMeta(error, item.meta);
+          }
+          span?.addEvent("error");
+          throw error;
         }
-        throw error;
-      }
-      case "done": {
-        let data = item.data;
-        if (item.meta && isTypescriptObject(data)) {
-          data = withEventMeta(data, item.meta);
+        case "done": {
+          let data = item.data;
+          if (item.meta && isTypescriptObject(data)) {
+            data = withEventMeta(data, item.meta);
+          }
+          span?.addEvent("done");
+          return { value: data, done: true };
         }
-        return { value: data, done: true };
       }
-    }
-  }, cleanup);
-}
-async function resolveEventIterator(iterator, callback) {
-  while (true) {
-    const payload = await (async () => {
-      try {
-        const { value, done } = await iterator.next();
-        if (done) {
-          return { event: "done", data: value, meta: getEventMeta(value) };
-        }
-        return { event: "message", data: value, meta: getEventMeta(value) };
-      } catch (err) {
-        return {
-          meta: getEventMeta(err),
-          event: "error",
-          data: err instanceof ErrorEvent ? err.data : void 0
-        };
+    } catch (e) {
+      if (!(e instanceof ErrorEvent)) {
+        setSpanError(span, e, options);
       }
-    })();
+      throw e;
+    }
+  }, async (reason) => {
     try {
-      const direction = await callback(payload);
-      if (payload.event === "done" || payload.event === "error") {
-        return;
+      if (reason !== "next") {
+        span?.addEvent("cancelled");
       }
-      if (direction === "abort") {
+      await runInSpanContext(span, () => cleanup(reason));
+    } catch (e) {
+      setSpanError(span, e, options);
+      throw e;
+    } finally {
+      span?.end();
+    }
+  });
+}
+function resolveEventIterator(iterator, callback) {
+  return runWithSpan(
+    { name: "stream_event_iterator" },
+    async (span) => {
+      while (true) {
+        const payload = await (async () => {
+          try {
+            const { value, done } = await iterator.next();
+            if (done) {
+              span?.addEvent("done");
+              return { event: "done", data: value, meta: getEventMeta(value) };
+            }
+            span?.addEvent("message");
+            return { event: "message", data: value, meta: getEventMeta(value) };
+          } catch (err) {
+            if (err instanceof ErrorEvent) {
+              span?.addEvent("error");
+              return {
+                event: "error",
+                data: err.data,
+                meta: getEventMeta(err)
+              };
+            } else {
+              try {
+                await callback({ event: "error", data: void 0 });
+              } catch (err2) {
+                setSpanError(span, err);
+                throw err2;
+              }
+              throw err;
+            }
+          }
+        })();
+        let isInvokeCleanupFn = false;
         try {
-          await iterator.return?.();
-        } catch {
+          const direction = await callback(payload);
+          if (payload.event === "done" || payload.event === "error") {
+            return;
+          }
+          if (direction === "abort") {
+            isInvokeCleanupFn = true;
+            await iterator.return?.();
+            return;
+          }
+        } catch (err) {
+          if (!isInvokeCleanupFn) {
+            try {
+              await iterator.return?.();
+            } catch (err2) {
+              setSpanError(span, err);
+              throw err2;
+            }
+          }
+          throw err;
         }
-        return;
-      }
-    } catch (err) {
-      try {
-        await iterator.return?.();
-      } catch {
       }
-      throw err;
     }
-  }
+  );
 }
 
 class ClientPeer {
   idGenerator = new SequentialIdGenerator();
+  /**
+   * Queue of responses sent from server, awaiting consumption
+   */
   responseQueue = new AsyncIdQueue();
+  /**
+   * Queue of event iterator messages sent from server, awaiting consumption
+   */
   serverEventIteratorQueue = new AsyncIdQueue();
+  /**
+   * Controllers used to signal that the client should stop sending event iterator messages
+   */
   serverControllers = /* @__PURE__ */ new Map();
+  /**
+   * Cleanup functions invoked when the request/response is closed
+   */
+  cleanupFns = /* @__PURE__ */ new Map();
   send;
   constructor(send) {
     this.send = async (id, ...rest) => encodeRequestMessage(id, ...rest).then(async (raw) => {
@@ -262,48 +320,85 @@ class ClientPeer {
     });
   }
   get length() {
-    return (+this.responseQueue.length + this.serverEventIteratorQueue.length + this.serverControllers.size) / 3;
+    return (+this.responseQueue.length + this.serverEventIteratorQueue.length + this.serverControllers.size + this.cleanupFns.size) / 4;
   }
   open(id) {
     this.serverEventIteratorQueue.open(id);
     this.responseQueue.open(id);
     const controller = new AbortController();
     this.serverControllers.set(id, controller);
+    this.cleanupFns.set(id, []);
     return controller;
   }
   async request(request) {
     const signal = request.signal;
-    if (signal?.aborted) {
-      throw signal.reason;
-    }
-    const id = this.idGenerator.generate();
-    const serverController = this.open(id);
-    return new Promise((resolve, reject) => {
-      this.send(id, MessageType.REQUEST, request).then(async () => {
+    return runWithSpan(
+      { name: "send_peer_request", signal },
+      async () => {
         if (signal?.aborted) {
-          await this.send(id, MessageType.ABORT_SIGNAL, void 0);
-          this.close({ id, reason: signal.reason });
-          return;
+          throw signal.reason;
         }
-        signal?.addEventListener("abort", async () => {
-          await this.send(id, MessageType.ABORT_SIGNAL, void 0);
-          this.close({ id, reason: signal.reason });
-        }, { once: true });
-        if (isAsyncIteratorObject(request.body)) {
-          await resolveEventIterator(request.body, async (payload) => {
-            if (serverController.signal.aborted) {
-              return "abort";
-            }
-            await this.send(id, MessageType.EVENT_ITERATOR, payload);
-            return "next";
+        const id = this.idGenerator.generate();
+        const serverController = this.open(id);
+        try {
+          const otelConfig = getGlobalOtelConfig();
+          if (otelConfig) {
+            const headers = clone(request.headers);
+            otelConfig.propagation.inject(otelConfig.context.active(), headers);
+            request = { ...request, headers };
+          }
+          await this.send(id, MessageType.REQUEST, request);
+          if (signal?.aborted) {
+            await this.send(id, MessageType.ABORT_SIGNAL, void 0);
+            throw signal.reason;
+          }
+          let abortListener;
+          signal?.addEventListener("abort", abortListener = async () => {
+            await this.send(id, MessageType.ABORT_SIGNAL, void 0);
+            this.close({ id, reason: signal.reason });
+          }, { once: true });
+          this.cleanupFns.get(id)?.push(() => {
+            signal?.removeEventListener("abort", abortListener);
           });
+          if (isAsyncIteratorObject(request.body)) {
+            const iterator = request.body;
+            void resolveEventIterator(iterator, async (payload) => {
+              if (serverController.signal.aborted) {
+                return "abort";
+              }
+              await this.send(id, MessageType.EVENT_ITERATOR, payload);
+              return "next";
+            });
+          }
+          const response = await this.responseQueue.pull(id);
+          if (isEventIteratorHeaders(response.headers)) {
+            const iterator = toEventIterator(
+              this.serverEventIteratorQueue,
+              id,
+              async (reason) => {
+                try {
+                  if (reason !== "next") {
+                    await this.send(id, MessageType.ABORT_SIGNAL, void 0);
+                  }
+                } finally {
+                  this.close({ id });
+                }
+              },
+              { signal }
+            );
+            return {
+              ...response,
+              body: iterator
+            };
+          }
+          this.close({ id });
+          return response;
+        } catch (err) {
+          this.close({ id, reason: err });
+          throw err;
         }
-      }).catch((err) => {
-        this.close({ id, reason: err });
-        reject(err);
-      });
-      this.responseQueue.pull(id).then(resolve).catch(reject);
-    });
+      }
+    );
   }
   async message(raw) {
     const [id, type, payload] = await decodeResponseMessage(raw);
@@ -320,31 +415,19 @@ class ClientPeer {
     if (!this.responseQueue.isOpen(id)) {
       return;
     }
-    if (isEventIteratorHeaders(payload.headers)) {
-      this.responseQueue.push(id, {
-        ...payload,
-        body: toEventIterator(this.serverEventIteratorQueue, id, async (reason) => {
-          try {
-            if (reason !== "next") {
-              await this.send(id, MessageType.ABORT_SIGNAL, void 0);
-            }
-          } finally {
-            this.close({ id });
-          }
-        })
-      });
-    } else {
-      this.responseQueue.push(id, payload);
-      this.close({ id });
-    }
+    this.responseQueue.push(id, payload);
   }
   close(options = {}) {
     if (options.id !== void 0) {
       this.serverControllers.get(options.id)?.abort(options.reason);
       this.serverControllers.delete(options.id);
+      this.cleanupFns.get(options.id)?.forEach((fn) => fn());
+      this.cleanupFns.delete(options.id);
     } else {
       this.serverControllers.forEach((c) => c.abort(options.reason));
       this.serverControllers.clear();
+      this.cleanupFns.forEach((fns) => fns.forEach((fn) => fn()));
+      this.cleanupFns.clear();
     }
     this.responseQueue.close(options);
     this.serverEventIteratorQueue.close(options);
@@ -352,7 +435,13 @@ class ClientPeer {
 }
 
 class ServerPeer {
+  /**
+   * Queue of event iterator messages sent from client, awaiting consumption
+   */
   clientEventIteratorQueue = new AsyncIdQueue();
+  /**
+   * Map of active client request controllers, should be synced to request signal
+   */
   clientControllers = /* @__PURE__ */ new Map();
   send;
   constructor(send) {
@@ -371,10 +460,13 @@ class ServerPeer {
     this.clientControllers.set(id, controller);
     return controller;
   }
-  async message(raw) {
+  /**
+   * @todo This method will return Promise<void> in the next major version.
+   */
+  async message(raw, handleRequest) {
     const [id, type, payload] = await decodeRequestMessage(raw);
     if (type === MessageType.ABORT_SIGNAL) {
-      this.close({ id });
+      this.close({ id, reason: new AbortError("Client aborted the request") });
       return [id, void 0];
     }
     if (type === MessageType.EVENT_ITERATOR) {
@@ -384,28 +476,66 @@ class ServerPeer {
       return [id, void 0];
     }
     const clientController = this.open(id);
+    const signal = clientController.signal;
     const request = {
       ...payload,
-      signal: clientController.signal,
-      body: isEventIteratorHeaders(payload.headers) ? toEventIterator(this.clientEventIteratorQueue, id, async (reason) => {
-        if (reason !== "next") {
-          await this.send(id, MessageType.ABORT_SIGNAL, void 0);
-        }
-      }) : payload.body
+      signal,
+      body: isEventIteratorHeaders(payload.headers) ? toEventIterator(
+        this.clientEventIteratorQueue,
+        id,
+        async (reason) => {
+          if (reason !== "next") {
+            await this.send(id, MessageType.ABORT_SIGNAL, void 0);
+          }
+        },
+        { signal }
+      ) : payload.body
     };
+    if (handleRequest) {
+      let context;
+      const otelConfig = getGlobalOtelConfig();
+      if (otelConfig) {
+        context = otelConfig.propagation.extract(otelConfig.context.active(), request.headers);
+      }
+      await runWithSpan(
+        { name: "receive_peer_request", context },
+        async () => {
+          const response = await runWithSpan(
+            { name: "handle_request" },
+            async () => {
+              try {
+                return await handleRequest(request);
+              } catch (reason) {
+                this.close({ id, reason, abort: false });
+                throw reason;
+              }
+            }
+          );
+          await runWithSpan(
+            { name: "send_peer_response" },
+            () => this.response(id, response)
+          );
+        }
+      );
+    }
     return [id, request];
   }
+  /**
+   * @deprecated Please pass the `handleRequest` (second arg) function to the `message` method instead.
+   */
   async response(id, response) {
     const signal = this.clientControllers.get(id)?.signal;
     if (!signal || signal.aborted) {
       return;
     }
-    await this.send(id, MessageType.RESPONSE, response).then(async () => {
+    try {
+      await this.send(id, MessageType.RESPONSE, response);
       if (!signal.aborted && isAsyncIteratorObject(response.body)) {
-        if (response.body instanceof experimental_HibernationEventIterator) {
+        if (response.body instanceof HibernationEventIterator) {
           response.body.hibernationCallback?.(id);
         } else {
-          await resolveEventIterator(response.body, async (payload) => {
+          const iterator = response.body;
+          await resolveEventIterator(iterator, async (payload) => {
             if (signal.aborted) {
               return "abort";
             }
@@ -415,20 +545,20 @@ class ServerPeer {
         }
       }
       this.close({ id, abort: false });
-    }).catch((reason) => {
+    } catch (reason) {
       this.close({ id, reason, abort: false });
       throw reason;
-    });
+    }
   }
   close({ abort = true, ...options } = {}) {
     if (options.id === void 0) {
       if (abort) {
-        this.clientControllers.forEach((c) => c.abort());
+        this.clientControllers.forEach((c) => c.abort(options.reason));
       }
       this.clientControllers.clear();
     } else {
       if (abort) {
-        this.clientControllers.get(options.id)?.abort();
+        this.clientControllers.get(options.id)?.abort(options.reason);
       }
       this.clientControllers.delete(options.id);
     }

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@orpc/standard-server-peer",
   "type": "module",
-  "version": "1.7.10",
+  "version": "1.8.0",
   "license": "MIT",
   "homepage": "https://unnoq.com",
   "repository": {
@@ -23,8 +23,8 @@
     "dist"
   ],
   "dependencies": {
-    "@orpc/shared": "1.7.10",
-    "@orpc/standard-server": "1.7.10"
+    "@orpc/shared": "1.8.0",
+    "@orpc/standard-server": "1.8.0"
   },
   "scripts": {
     "build": "unbuild",