Add signal option for std server

Author: ije

server/stdserver.ts

@@ -1,4 +1,4 @@
-import { serve as stdServe, serveTLS } from 'https://deno.land/[email protected]/http/server.ts'
+import { serve as stdServe, serveTLS, Server as StdServer } from 'https://deno.land/[email protected]/http/server.ts'
 import log from '../shared/log.ts'
 import type { ServerRequest } from '../types.ts'
 import { Application } from './app.ts'
@@ -18,17 +18,15 @@ export type ServeOptions = {
   certFile?: string
   /** Server public key file. */
   keyFile?: string
+  /* The signal to close the server. */
+  signal?: AbortSignal
 }
 
 /** Create a standard Aleph server. */
-export async function serve({ app, port, hostname, certFile, keyFile }: ServeOptions) {
+export async function serve({ app, port, hostname, certFile, keyFile, signal }: ServeOptions) {
   const server = new Server(app)
   await app.ready
 
-  if (!app.isDev && app.config.compress) {
-    await compress.init()
-  }
-
   while (true) {
     try {
       let s: AsyncIterable<ServerRequest>
@@ -37,6 +35,12 @@ export async function serve({ app, port, hostname, certFile, keyFile }: ServeOpt
       } else {
         s = stdServe({ port, hostname })
       }
+      signal?.addEventListener('abort', () => {
+        (s as StdServer).close()
+      })
+      if (!app.isDev && app.config.compress) {
+        await compress.init()
+      }
       log.info(`Server ready on http://${hostname || 'localhost'}:${port}${app.config.basePath}`)
       for await (const r of s) {
         server.handle(r)