fix(server): if res was piped, res should be treated as sent (#6123)

Author: yimingjfe

.changeset/tricky-countries-live.md

@@ -0,0 +1,6 @@
+---
+'@modern-js/server-core': patch
+---
+
+fix(server): if res was piped, res should be treated as sent
+fix(server): 如果响应被 pipe，响应应该被当作已经发送

packages/server/core/src/adapters/node/hono.ts

@@ -16,7 +16,9 @@ type NodeBindings = {
       __templates?: Record<string, string>;
       __serverManifest?: ServerManifest;
     };
-    res: NodeResponse;
+    res: NodeResponse & {
+      _modernBodyPiped?: boolean;
+    };
   };
 };
 
@@ -34,18 +36,26 @@ type Handler = (req: NodeRequest, res: NodeResponse) => void | Promise<void>;
 export const httpCallBack2HonoMid = (handler: Handler) => {
   return async (context: Context<ServerNodeEnv & ServerEnv>, next: Next) => {
     const { req, res } = context.env.node;
+    const onPipe = () => {
+      res._modernBodyPiped = true;
+    };
+    res.once('pipe', onPipe);
     // for bff.enableHandleWeb
     req.__honoRequest = context.req;
     req.__templates = context.get('templates') || {};
     req.__serverManifest = context.get('serverManifest') || {};
-    await handler(req, res);
-    // make sure res.headersSent is set, because when using pipe, headersSent is not set immediately
-    await new Promise(resolve => setTimeout(resolve, 0));
-    // Avoid memory leaks in node versions 18 and 20
-    delete req.__honoRequest;
-    delete req.__templates;
-    delete req.__serverManifest;
-    if (res.headersSent) {
+
+    try {
+      await handler(req, res);
+    } finally {
+      // Avoid memory leaks in node versions 18 and 20
+      delete req.__honoRequest;
+      delete req.__templates;
+      delete req.__serverManifest;
+      res.removeListener('pipe', onPipe);
+    }
+
+    if (res.headersSent || res._modernBodyPiped) {
       context.finalized = true;
     } else {
       await next();

packages/server/core/src/adapters/node/node.ts

@@ -129,7 +129,11 @@ const getRequestListener = (handler: RequestHandler) => {
        * ```
        */
 
-      if (!res.headersSent && !(response as any).res) {
+      if (
+        !res.headersSent &&
+        !(response as any).res &&
+        !(res as any)._modernBodyPiped
+      ) {
         await sendResponse(response, res);
       }
     } catch (error) {