Skip to content

HTTPS clone URL

Subversion checkout URL

You can clone with HTTPS or Subversion.

Download ZIP
tree: 7c62863b04
Fetching contributors…

Cannot retrieve contributors at this time

86 lines (73 sloc) 2.865 kb
open Threadext
type mode =
| Slave of string (* master IP *)
| Master of string (* database filename *)
let mode = ref None
let finished = ref false
let m = Mutex.create ()
let c = Condition.create ()
(** Handler for the remote database access URL *)
let remote_database_access_handler_v1 req bio =
try
Db_remote_cache_access_v1.handler req bio
with e ->
Printf.printf "Caught: %s\n" (Printexc.to_string e);
Printexc.print_backtrace stdout;
flush stdout;
raise e
(** Handler for the remote database access URL *)
let remote_database_access_handler_v2 req bio =
try
Db_remote_cache_access_v2.handler req bio
with e ->
Printf.printf "Caught: %s\n" (Printexc.to_string e);
Printexc.print_backtrace stdout;
flush stdout;
raise e
module Local_tests = Database_test.Tests(Db_cache_impl)
let schema = Test_schemas.schema
let _ =
let listen_path = ref "./database" in
let self_test = ref false in
Printexc.record_backtrace true;
Arg.parse [
"--slave-of", Arg.String (fun master -> mode := Some(Slave master)), "run as a slave of a remote db";
"--master", Arg.String (fun db -> mode := Some(Master db)), "run as a master from the given db filename";
"--listen-on", Arg.Set_string listen_path, Printf.sprintf "listen for requests on path (default %s)" !listen_path;
"--test", Arg.Set self_test, "Run unit tests in-process";
] (fun x -> Printf.fprintf stderr "Ignoring unknown parameter: %s\n%!" x)
"run a stand-alone database server";
match !mode with
| None -> failwith "Requires either --slave-of or --master arguments"
| Some mode ->
begin match mode with
| Slave _ -> failwith "unimplemented"
| Master db_filename ->
Printf.printf "Database path: %s\n%!" db_filename;
let db = Parse_db_conf.make db_filename in
Db_conn_store.initialise_db_connections [ db ];
let t = Db_backend.make () in
Db_cache_impl.make t [ db ] schema;
Db_cache_impl.sync [ db ] (Db_ref.get_database t);
Unixext.unlink_safe !listen_path;
let sockaddr = Unix.ADDR_UNIX !listen_path in
let socket = Http_svr.bind sockaddr "unix_rpc" in
let server = Http_svr.Server.empty in
Http_svr.add_handler server Http.Post "/post_remote_db_access" (Http_svr.BufIO remote_database_access_handler_v1);
Http_svr.add_handler server Http.Post "/post_remote_db_access_v2" (Http_svr.BufIO remote_database_access_handler_v2);
Http_svr.start server socket;
Printf.printf "server listening\n%!";
if !self_test then begin
Printf.printf "Running unit-tests\n%!";
Local_tests.main true;
Printf.printf "All tests passed\n%!";
end;
(* Wait for either completion *)
Mutex.execute m
(fun () ->
while not (!finished) do
Condition.wait c m
done
);
Http_svr.stop socket
end
Jump to Line
Something went wrong with that request. Please try again.