@orpc/standard-server-peer 0.0.0-next.fc23c8d → 0.0.0-next.fc2dc8f

package/dist/index.mjs

@@ -1,6 +1,8 @@
-import { isAsyncIteratorObject, stringifyJSON, AsyncIteratorClass, isTypescriptObject, SequentialIdGenerator, AsyncIdQueue } from '@orpc/shared';
-import { flattenHeader, getFilenameFromContentDisposition, generateContentDisposition, withEventMeta, ErrorEvent, getEventMeta, 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';
 
+const SHORTABLE_ORIGIN = "http://orpc";
+const SHORTABLE_ORIGIN_MATCHER = /^http:\/\/orpc\//;
 var MessageType = /* @__PURE__ */ ((MessageType2) => {
   MessageType2[MessageType2["REQUEST"] = 1] = "REQUEST";
   MessageType2[MessageType2["RESPONSE"] = 2] = "RESPONSE";
@@ -8,7 +10,7 @@ var MessageType = /* @__PURE__ */ ((MessageType2) => {
   MessageType2[MessageType2["ABORT_SIGNAL"] = 4] = "ABORT_SIGNAL";
   return MessageType2;
 })(MessageType || {});
-async function encodeRequestMessage(id, type, payload) {
+function serializeRequestMessage(id, type, payload) {
   if (type === 3 /* EVENT_ITERATOR */) {
     const eventPayload = payload;
     const serializedPayload2 = {
@@ -16,35 +18,26 @@ async function encodeRequestMessage(id, type, payload) {
       d: eventPayload.data,
       m: eventPayload.meta
     };
-    return encodeRawMessage({ i: id, t: type, p: serializedPayload2 });
+    return { i: id, t: type, p: serializedPayload2 };
   }
   if (type === 4 /* ABORT_SIGNAL */) {
-    return encodeRawMessage({ i: id, t: type, p: payload });
+    return { i: id, t: type, p: payload };
   }
   const request = payload;
-  const { body: processedBody, headers: processedHeaders } = await prepareBodyAndHeadersForSerialization(
-    request.body,
-    request.headers
-  );
   const serializedPayload = {
-    u: request.url.toString().replace(/^orpc:\//, "/"),
-    b: processedBody instanceof Blob ? void 0 : processedBody,
-    h: Object.keys(processedHeaders).length > 0 ? processedHeaders : void 0,
+    u: request.url.toString().replace(SHORTABLE_ORIGIN_MATCHER, "/"),
+    b: request.body,
+    h: Object.keys(request.headers).length > 0 ? request.headers : void 0,
     m: request.method === "POST" ? void 0 : request.method
   };
-  const baseMessage = {
+  return {
     i: id,
     p: serializedPayload
   };
-  if (processedBody instanceof Blob) {
-    return encodeRawMessage(baseMessage, processedBody);
-  }
-  return encodeRawMessage(baseMessage);
 }
-async function decodeRequestMessage(raw) {
-  const { json: message, blobData } = await decodeRawMessage(raw);
+function deserializeRequestMessage(message) {
   const id = message.i;
-  const type = message.t;
+  const type = message.t ?? 1 /* REQUEST */;
   if (type === 3 /* EVENT_ITERATOR */) {
     const payload2 = message.p;
     return [id, type, { event: payload2.e, data: payload2.d, meta: payload2.m }];
@@ -53,24 +46,14 @@ async function decodeRequestMessage(raw) {
     return [id, type, message.p];
   }
   const payload = message.p;
-  const headers = payload.h ?? {};
-  let body = payload.b;
-  const contentType = flattenHeader(headers["content-type"]);
-  if (blobData) {
-    const contentDisposition = flattenHeader(headers["content-disposition"]);
-    if (contentDisposition === void 0 && contentType?.startsWith("multipart/form-data")) {
-      const tempRes = new Response(blobData, { headers: { "content-type": contentType } });
-      body = await tempRes.formData();
-    } else {
-      const filename = getFilenameFromContentDisposition(contentDisposition) ?? "blob";
-      body = new File([blobData], filename, { type: contentType });
-    }
-  } else if (contentType?.startsWith("application/x-www-form-urlencoded") && typeof body === "string") {
-    body = new URLSearchParams(body);
-  }
-  return [id, 1 /* REQUEST */, { url: new URL(payload.u, "orpc:/"), headers, method: payload.m ?? "POST", body }];
+  return [id, 1 /* REQUEST */, {
+    url: payload.u.startsWith("/") ? new URL(`${SHORTABLE_ORIGIN}${payload.u}`) : new URL(payload.u),
+    headers: payload.h ?? {},
+    method: payload.m ?? "POST",
+    body: payload.b
+  }];
 }
-async function encodeResponseMessage(id, type, payload) {
+function serializeResponseMessage(id, type, payload) {
   if (type === 3 /* EVENT_ITERATOR */) {
     const eventPayload = payload;
     const serializedPayload2 = {
@@ -78,32 +61,23 @@ async function encodeResponseMessage(id, type, payload) {
       d: eventPayload.data,
       m: eventPayload.meta
     };
-    return encodeRawMessage({ i: id, t: type, p: serializedPayload2 });
+    return { i: id, t: type, p: serializedPayload2 };
   }
   if (type === 4 /* ABORT_SIGNAL */) {
-    return encodeRawMessage({ i: id, t: type, p: void 0 });
+    return { i: id, t: type, p: void 0 };
   }
   const response = payload;
-  const { body: processedBody, headers: processedHeaders } = await prepareBodyAndHeadersForSerialization(
-    response.body,
-    response.headers
-  );
   const serializedPayload = {
     s: response.status === 200 ? void 0 : response.status,
-    h: Object.keys(processedHeaders).length > 0 ? processedHeaders : void 0,
-    b: processedBody instanceof Blob ? void 0 : processedBody
+    h: Object.keys(response.headers).length > 0 ? response.headers : void 0,
+    b: response.body
   };
-  const baseMessage = {
+  return {
     i: id,
     p: serializedPayload
   };
-  if (processedBody instanceof Blob) {
-    return encodeRawMessage(baseMessage, processedBody);
-  }
-  return encodeRawMessage(baseMessage);
 }
-async function decodeResponseMessage(raw) {
-  const { json: message, blobData } = await decodeRawMessage(raw);
+function deserializeResponseMessage(message) {
   const id = message.i;
   const type = message.t;
   if (type === 3 /* EVENT_ITERATOR */) {
@@ -114,24 +88,73 @@ async function decodeResponseMessage(raw) {
     return [id, type, message.p];
   }
   const payload = message.p;
-  const headers = payload.h ?? {};
-  let body = payload.b;
-  const contentType = flattenHeader(headers["content-type"]);
-  if (blobData) {
-    const contentDisposition = flattenHeader(headers["content-disposition"]);
-    if (contentDisposition === void 0 && contentType?.startsWith("multipart/form-data")) {
-      const tempRes = new Response(blobData, { headers: { "content-type": contentType } });
-      body = await tempRes.formData();
-    } else {
-      const filename = getFilenameFromContentDisposition(contentDisposition) ?? "blob";
-      body = new File([blobData], filename, { type: contentType });
-    }
-  } else if (contentType?.startsWith("application/x-www-form-urlencoded") && typeof body === "string") {
-    body = new URLSearchParams(body);
+  return [id, 2 /* RESPONSE */, {
+    status: payload.s ?? 200,
+    headers: payload.h ?? {},
+    body: payload.b
+  }];
+}
+async function encodeRequestMessage(id, type, payload) {
+  if (type === 3 /* EVENT_ITERATOR */ || type === 4 /* ABORT_SIGNAL */) {
+    return encodeRawMessage(serializeRequestMessage(id, type, payload));
   }
-  return [id, 2 /* RESPONSE */, { status: payload.s ?? 200, headers, body }];
+  const request = payload;
+  const { body: processedBody, headers: processedHeaders } = await serializeBodyAndHeaders(
+    request.body,
+    request.headers
+  );
+  const modifiedRequest = {
+    ...request,
+    body: processedBody instanceof Blob ? void 0 : processedBody,
+    headers: processedHeaders
+  };
+  const baseMessage = serializeRequestMessage(id, 1 /* REQUEST */, modifiedRequest);
+  if (processedBody instanceof Blob) {
+    return encodeRawMessage(baseMessage, processedBody);
+  }
+  return encodeRawMessage(baseMessage);
+}
+async function decodeRequestMessage(raw) {
+  const { json: message, buffer } = await decodeRawMessage(raw);
+  const [id, type, payload] = deserializeRequestMessage(message);
+  if (type === 3 /* EVENT_ITERATOR */ || type === 4 /* ABORT_SIGNAL */) {
+    return [id, type, payload];
+  }
+  const request = payload;
+  const body = await deserializeBody(request.headers, request.body, buffer);
+  return [id, type, { ...request, body }];
+}
+async function encodeResponseMessage(id, type, payload) {
+  if (type === 3 /* EVENT_ITERATOR */ || type === 4 /* ABORT_SIGNAL */) {
+    return encodeRawMessage(serializeResponseMessage(id, type, payload));
+  }
+  const response = payload;
+  const { body: processedBody, headers: processedHeaders } = await serializeBodyAndHeaders(
+    response.body,
+    response.headers
+  );
+  const modifiedResponse = {
+    ...response,
+    body: processedBody instanceof Blob ? void 0 : processedBody,
+    headers: processedHeaders
+  };
+  const baseMessage = serializeResponseMessage(id, 2 /* RESPONSE */, modifiedResponse);
+  if (processedBody instanceof Blob) {
+    return encodeRawMessage(baseMessage, processedBody);
+  }
+  return encodeRawMessage(baseMessage);
 }
-async function prepareBodyAndHeadersForSerialization(body, originalHeaders) {
+async function decodeResponseMessage(raw) {
+  const { json: message, buffer } = await decodeRawMessage(raw);
+  const [id, type, payload] = deserializeResponseMessage(message);
+  if (type === 3 /* EVENT_ITERATOR */ || type === 4 /* ABORT_SIGNAL */) {
+    return [id, type, payload];
+  }
+  const response = payload;
+  const body = await deserializeBody(response.headers, response.body, buffer);
+  return [id, type, { ...response, body }];
+}
+async function serializeBodyAndHeaders(body, originalHeaders) {
   const headers = { ...originalHeaders };
   const originalContentDisposition = headers["content-disposition"];
   delete headers["content-type"];
@@ -159,167 +182,298 @@ async function prepareBodyAndHeadersForSerialization(body, originalHeaders) {
   }
   return { body, headers };
 }
-function isEventIteratorHeaders(headers) {
-  return Boolean(flattenHeader(headers["content-type"])?.startsWith("text/event-stream") && headers["content-disposition"] === void 0);
+async function deserializeBody(headers, body, buffer) {
+  const contentType = flattenHeader(headers["content-type"]);
+  const contentDisposition = flattenHeader(headers["content-disposition"]);
+  if (typeof contentDisposition === "string") {
+    const filename = getFilenameFromContentDisposition(contentDisposition) ?? "blob";
+    return new File(buffer === void 0 ? [] : [buffer], filename, { type: contentType });
+  }
+  if (contentType?.startsWith("multipart/form-data")) {
+    const tempRes = new Response(buffer, { headers: { "content-type": contentType } });
+    return tempRes.formData();
+  }
+  if (contentType?.startsWith("application/x-www-form-urlencoded") && typeof body === "string") {
+    return new URLSearchParams(body);
+  }
+  return body;
 }
 const JSON_AND_BINARY_DELIMITER = 255;
-async function encodeRawMessage(data, blobData) {
+async function encodeRawMessage(data, blob) {
   const json = stringifyJSON(data);
-  if (blobData === void 0) {
+  if (blob === void 0 || blob.size === 0) {
     return json;
   }
-  return new Blob([
+  return readAsBuffer(new Blob([
     new TextEncoder().encode(json),
     new Uint8Array([JSON_AND_BINARY_DELIMITER]),
-    blobData
-  ]).arrayBuffer();
+    blob
+  ]));
 }
 async function decodeRawMessage(raw) {
   if (typeof raw === "string") {
     return { json: JSON.parse(raw) };
   }
-  const buffer = new Uint8Array(raw instanceof Blob ? await raw.arrayBuffer() : raw);
+  const buffer = raw instanceof Uint8Array ? raw : new Uint8Array(raw);
   const delimiterIndex = buffer.indexOf(JSON_AND_BINARY_DELIMITER);
   if (delimiterIndex === -1) {
     const jsonPart2 = new TextDecoder().decode(buffer);
     return { json: JSON.parse(jsonPart2) };
   }
-  const jsonPart = new TextDecoder().decode(buffer.slice(0, delimiterIndex));
-  const blobData = buffer.slice(delimiterIndex + 1);
+  const jsonPart = new TextDecoder().decode(buffer.subarray(0, delimiterIndex));
+  const bufferPart = buffer.subarray(delimiterIndex + 1);
   return {
     json: JSON.parse(jsonPart),
-    blobData: blobData.buffer
+    buffer: bufferPart
   };
 }
 
-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 {
+  peer;
+  constructor(send) {
+    this.peer = new experimental_ClientPeerWithoutCodec(async ([id, type, payload]) => {
+      await send(await encodeRequestMessage(id, type, payload));
+    });
+  }
+  get length() {
+    return this.peer.length;
+  }
+  open(id) {
+    return this.peer.open(id);
+  }
+  async request(request) {
+    return this.peer.request(request);
+  }
+  async message(raw) {
+    return this.peer.message(await decodeResponseMessage(raw));
+  }
+  close(options = {}) {
+    return this.peer.close(options);
+  }
+}
+class experimental_ClientPeerWithoutCodec {
   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) => {
+    this.send = async (message) => {
+      const id = message[0];
       if (this.serverControllers.has(id)) {
-        await send(raw);
+        await send(message);
       }
-    });
+    };
   }
   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);
+  async message([id, type, payload]) {
     if (type === MessageType.ABORT_SIGNAL) {
       this.serverControllers.get(id)?.abort();
       return;
@@ -333,31 +487,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);
@@ -365,15 +507,51 @@ class ClientPeer {
 }
 
 class ServerPeer {
+  peer;
+  constructor(send) {
+    this.peer = new experimental_ServerPeerWithoutCodec(async ([id, type, payload]) => {
+      await send(await encodeResponseMessage(id, type, payload));
+    });
+  }
+  get length() {
+    return this.peer.length;
+  }
+  open(id) {
+    return this.peer.open(id);
+  }
+  /**
+   * @todo This method will return Promise<void> in the next major version.
+   */
+  async message(raw, handleRequest) {
+    return this.peer.message(await decodeRequestMessage(raw), handleRequest);
+  }
+  /**
+   * @deprecated Please pass the `handleRequest` (second arg) function to the `message` method instead.
+   */
+  async response(id, response) {
+    return this.peer.response(id, response);
+  }
+  close({ abort = true, ...options } = {}) {
+    return this.peer.close({ ...options, abort });
+  }
+}
+class experimental_ServerPeerWithoutCodec {
+  /**
+   * 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) {
-    this.send = (id, ...rest) => encodeResponseMessage(id, ...rest).then(async (raw) => {
+    this.send = async (message) => {
+      const id = message[0];
       if (this.clientControllers.has(id)) {
-        await send(raw);
+        await send(message);
       }
-    });
+    };
   }
   get length() {
     return (this.clientEventIteratorQueue.length + this.clientControllers.size) / 2;
@@ -384,10 +562,12 @@ class ServerPeer {
     this.clientControllers.set(id, controller);
     return controller;
   }
-  async message(raw) {
-    const [id, type, payload] = await decodeRequestMessage(raw);
+  /**
+   * @todo This method will return Promise<void> in the next major version.
+   */
+  async message([id, type, payload], handleRequest) {
     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) {
@@ -397,51 +577,89 @@ 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";
             }
-            await this.send(id, MessageType.EVENT_ITERATOR, payload);
+            await this.send([id, MessageType.EVENT_ITERATOR, payload]);
             return "next";
           });
         }
       }
       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);
     }
@@ -449,4 +667,4 @@ class ServerPeer {
   }
 }
 
-export { ClientPeer, MessageType, ServerPeer, decodeRequestMessage, decodeResponseMessage, encodeRequestMessage, encodeResponseMessage, isEventIteratorHeaders, resolveEventIterator, toEventIterator };
+export { ClientPeer, MessageType, ServerPeer, decodeRequestMessage, decodeResponseMessage, deserializeRequestMessage, deserializeResponseMessage, encodeRequestMessage, encodeResponseMessage, experimental_ClientPeerWithoutCodec, experimental_ServerPeerWithoutCodec, resolveEventIterator, serializeRequestMessage, serializeResponseMessage, toEventIterator };