@faapi/faapi 6.21.1 → 6.22.0

package/dist/cli/index.js

@@ -6367,13 +6367,28 @@ function encodeSseEvent(event) {
   return out;
 }
 function createSseWriter(options = {}) {
-  const { onFirstWrite } = options;
+  const { onFirstWrite, keepOpen = false } = options;
   const encoder = new TextEncoder();
   let controller = null;
   let closed = false;
   let aborted = false;
   let firstWriteTriggered = false;
   let drainWaiters = [];
+  let closeCallbacks = [];
+  let closeCallbacksFired = false;
+  const safeRunCallback = (callback) => {
+    try {
+      callback();
+    } catch {
+    }
+  };
+  const runCloseCallbacks = () => {
+    if (closeCallbacksFired) return;
+    closeCallbacksFired = true;
+    const callbacks = closeCallbacks;
+    closeCallbacks = [];
+    for (const callback of callbacks) safeRunCallback(callback);
+  };
   const resolveDrainWaiters = () => {
     const waiters = drainWaiters;
     drainWaiters = [];
@@ -6397,6 +6412,7 @@ function createSseWriter(options = {}) {
         aborted = true;
         closed = true;
         controller = null;
+        runCloseCallbacks();
         resolveDrainWaiters();
       }
     },
@@ -6439,6 +6455,7 @@ function createSseWriter(options = {}) {
     close() {
       if (closed) return;
       closed = true;
+      runCloseCallbacks();
       resolveDrainWaiters();
       if (controller) {
         try {
@@ -6448,6 +6465,13 @@ function createSseWriter(options = {}) {
         controller = null;
       }
     },
+    onClose(callback) {
+      if (closeCallbacksFired) {
+        safeRunCallback(callback);
+        return;
+      }
+      closeCallbacks.push(callback);
+    },
     get closed() {
       return closed;
     },
@@ -6464,6 +6488,9 @@ function createSseWriter(options = {}) {
     get aborted() {
       return aborted;
     },
+    get keepOpen() {
+      return keepOpen;
+    },
     get response() {
       return response;
     }
@@ -7011,10 +7038,15 @@ function createContextFromUrl(request, url, params, config = {}, ip = "", regist
      * 接入底层连接——流式期间字节即产即达（见 createServer.md「SSE 提前接管」）。
      * 无钩子场景（app.inject / 测试直调 invokeHandler）回落为 handler 返回后交出。
      *
+     * `{ keepOpen: true }` 声明流生命周期独立于 handler 返回：handler 返回后框架
+     * 不自动 close（长连接订阅模式，清理推送源用 writer.onClose）；handler 抛错
+     * 的兜底 close 不受影响。详见 runtime/sse.md「长连接订阅模式」。
+     *
      * 与 ctx.json / ctx.html 互斥：一个 handler 只能用一种响应方式。
      */
-    sse() {
+    sse(options) {
       const writer = createSseWriter({
+        keepOpen: options?.keepOpen,
         onFirstWrite: (response) => {
           ctx.__earlyRespond?.(response);
         }
@@ -7704,7 +7736,7 @@ async function invokeHandler(handler, ctx, body, middlewares, injectors) {
   const pickSseAndAutoClose = () => {
     const sseWriter = ctx.__sseWriter;
     if (!sseWriter) return null;
-    if (!sseWriter.closed && !sseWriter.aborted) {
+    if (!sseWriter.keepOpen && !sseWriter.closed && !sseWriter.aborted) {
       sseWriter.close();
     }
     return mergeMeta(sseWriter.response, meta);