mirror of
https://github.com/c-cube/tiny_httpd.git
synced 2025-12-05 19:00:32 -05:00
41 lines
1.7 KiB
OCaml
41 lines
1.7 KiB
OCaml
|
|
module S = Tiny_httpd
|
|
|
|
let () =
|
|
let port_ = ref 8080 in
|
|
let j = ref 32 in
|
|
Arg.parse (Arg.align [
|
|
"--port", Arg.Set_int port_, " set port";
|
|
"-p", Arg.Set_int port_, " set port";
|
|
"--debug", Arg.Unit (fun () -> S._enable_debug true), " enable debug";
|
|
"-j", Arg.Set_int j, " maximum number of connections";
|
|
]) (fun _ -> raise (Arg.Bad "")) "echo [option]*";
|
|
let server = S.create ~port:!port_ ~max_connections:!j () in
|
|
(* say hello *)
|
|
S.add_path_handler ~meth:`GET server
|
|
"/hello/%s@/" (fun name _req -> S.Response.make_string (Ok ("hello " ^name ^"!\n")));
|
|
(* echo request *)
|
|
S.add_path_handler server
|
|
"/echo" (fun req -> S.Response.make_string (Ok (Format.asprintf "echo:@ %a@." S.Request.pp req)));
|
|
S.add_path_handler ~meth:`POST server
|
|
"/debug/%B" (fun b _req -> S._enable_debug b; S.Response.make_string (Ok "ok"));
|
|
S.add_path_handler ~meth:`POST server
|
|
"/compact/" (fun _req -> Gc.compact(); S.Response.make_string (Ok "gc.compact: done"));
|
|
S.add_path_handler ~meth:`POST server
|
|
"/quit/" (fun _req -> S.stop server; S.Response.make_string (Ok "bye"));
|
|
S.add_path_handler ~meth:`PUT server
|
|
"/upload/%s" (fun path req ->
|
|
S._debug (fun k->k "start upload %S\n%!" path);
|
|
try
|
|
let oc = open_out @@ "/tmp/" ^ path in
|
|
output_string oc req.S.Request.body;
|
|
flush oc;
|
|
S.Response.make_string (Ok "uploaded file")
|
|
with e ->
|
|
S.Response.fail ~code:500 "couldn't upload file: %s" (Printexc.to_string e)
|
|
);
|
|
Printf.printf "listening on http://%s:%d\n%!" (S.addr server) (S.port server);
|
|
ignore @@ Thread.create (fun () -> Statmemprof_inuit.start 1e-4 300 2) ();
|
|
match S.run server with
|
|
| Ok () -> ()
|
|
| Error e -> raise e
|