shutdown server

Author: maxtori

src/server/cohttp/ezAPIServerCohttp.ml

@@ -123,6 +123,8 @@ let create_server ?catch ?allow_origin ?allow_headers ?allow_methods
     ~mode:(`TCP (`Port server_port))
     (Server.make_response_action ~callback ())
 
+let shutdown () = Lwt.return_unit
+
 let server ?catch ?allow_origin ?allow_headers ?allow_methods ?allow_credentials ?footer servers =
   Lwt.join (List.map (fun (port,kind) ->
       create_server ?catch ?allow_origin ?allow_headers ?allow_methods ?allow_credentials ?footer

src/server/httpaf/ezAPIServerHttpAf.ml

@@ -113,6 +113,8 @@ let connection_handler ?catch ?allow_origin ?allow_headers ?allow_methods
     sockaddr
     fd
 
+let shutdown = Server_common.shutdown
+
 let server ?catch ?allow_origin ?allow_headers ?allow_methods ?allow_credentials ?footer servers =
   Server_common.server ~name:"HTTPAF" ?catch ?allow_origin ?allow_headers
     ?allow_methods ?allow_credentials ?footer connection_handler servers

src/server/httpun/ezAPIServerHttpun.ml

@@ -107,5 +107,7 @@ let connection_handler ?catch ?allow_origin ?allow_headers ?allow_methods
 
   Httpun_lwt_unix.Server.create_connection_handler ~request_handler ~error_handler sockaddr fd
 
+let shutdown = Server_common.shutdown
+
 let server ?catch ?allow_origin ?allow_headers ?allow_methods ?allow_credentials ?footer servers =
   Server_common.server ?catch ?allow_origin ?allow_headers ?allow_methods ?allow_credentials ?footer connection_handler servers

src/server/server_common.ml

@@ -211,6 +211,8 @@ let mk_uri ~meth ~target ~header =
         uri
     end
 
+let server = ref None
+
 let create_server ?(name="HTTPUN") ~max_connections server_port server_kind handler =
   let open EzAPIServerUtils in
   let s = { server_port; server_kind } in
@@ -222,9 +224,14 @@ let create_server ?(name="HTTPUN") ~max_connections server_port server_kind hand
     ~nb_max_connections:max_connections
     listen_address (handler s)(* fun sockaddr fd -> *)
         (* handler ?catch ?allow_origin ?allow_headers ?allow_methods *)
-        (*   ?allow_credentials s sockaddr fd) *) >>= fun _server ->
+  (*   ?allow_credentials s sockaddr fd) *) >>= fun s ->
+  server := Some s;
   Lwt.return_unit
 
+let shutdown () = match !server with
+  | None -> Lwt.return_unit
+  | Some server -> Lazy.force server.shutdown
+
 let server ?name ?catch ?allow_origin ?allow_headers ?allow_methods ?allow_credentials ?footer handler servers =
   let max_connections =
     let n = List.length servers in

src/server/virtual/ezServer.mli

@@ -15,4 +15,6 @@ val server : ?catch:(string -> exn -> string Answer.t Lwt.t) ->
   ?allow_methods:allow_kind -> ?allow_credentials:bool -> ?footer:string ->
   (int * server_kind) list -> unit Lwt.t
 
+val shutdown : unit -> unit Lwt.t
+
 val set_debug : unit -> unit