routes need to explicitly ask to keep connection alive

Author: nicktrn

packages/core/src/v3/apps/http.ts

@@ -45,15 +45,15 @@ export class HttpReply {
   constructor(private response: Parameters<RequestListener>[1]) {}
 
   empty(status?: number) {
-    if (this.alreadyReplied) {
+    if (this.hasReplied) {
       return;
     }
 
     return this.response.writeHead(status ?? 200).end();
   }
 
   text(text: string, status?: number, contentType?: string) {
-    if (this.alreadyReplied) {
+    if (this.hasReplied) {
       return;
     }
 
@@ -63,7 +63,7 @@ export class HttpReply {
   }
 
   json(value: any, pretty?: boolean, status?: number) {
-    if (this.alreadyReplied) {
+    if (this.hasReplied) {
       return;
     }
 
@@ -74,7 +74,7 @@ export class HttpReply {
     );
   }
 
-  private get alreadyReplied() {
+  get hasReplied() {
     return this.response.headersSent;
   }
 }

packages/core/src/v3/serverOnly/httpServer.ts

@@ -28,6 +28,7 @@ interface RouteDefinition<
   paramsSchema?: TParams;
   querySchema?: TQuery;
   bodySchema?: TBody;
+  keepConnectionAlive?: boolean;
   handler: RouteHandler<TParams, TQuery, TBody>;
 }
 
@@ -156,7 +157,8 @@ export class HttpServer {
           return reply.empty(405);
         }
 
-        const { handler, paramsSchema, querySchema, bodySchema } = routeDefinition;
+        const { handler, paramsSchema, querySchema, bodySchema, keepConnectionAlive } =
+          routeDefinition;
 
         const params = this.parseRouteParams(route, url);
         const parsedParams = this.optionalSchema(paramsSchema, params);
@@ -202,6 +204,11 @@ export class HttpServer {
           logger.error("Route handler error", { error });
           return reply.empty(500);
         }
+
+        if (keepConnectionAlive) {
+          // Return early to keep the connection alive
+          return;
+        }
       } catch (error) {
         logger.error("Failed to handle request", { error });
         return reply.empty(500);