type result = Mysql.result option
*)
+
+let debug = false
+let debug_print =
+ if debug then prerr_endline else (fun _ ->())
+;;
+
type result = Sqlite3.row list
type dbd = Sqlite3.db option
exception Error of string
-let prerr_endline s = ()(*HLog.debug s;;*)
-
let profiler = HExtlib.profile "Sqlite3"
let quick_connect
=
let username = Helm_registry.get "user.name" in
let host_mangled = Pcre.replace ~pat:"[/ .]" ~templ:"_" host in
- let tmp_db_name =
- "/dev/shm/matita.db." ^ username ^ "." ^ host_mangled ^ "." ^
- string_of_int (Unix.getpid ())
+ let find_db _ =
+ let base = "matita.db." ^ username ^ "." ^ host_mangled ^ "." in
+ let root = "/dev/shm/" in
+ let files = HExtlib.find ~test:(Pcre.pmatch ~pat:(base^"[0-9]+")) root in
+ let rec aux = function
+ | [] ->
+ debug_print ("HSqlite3: no valid db files found in memory");
+ let name = root ^ base ^ string_of_int (Unix.getpid ()) in
+ debug_print ("HSqlite3: memory db file name: "^name);
+ name, true
+ | x::tl ->
+ debug_print ("HSqlite3: found a .db in memory: " ^ x);
+ match Array.to_list (Pcre.extract ~pat:"\\.([0-9]+)$" x) with
+ | [] | _::_::_::_ -> assert false
+ | [_;p] when HExtlib.is_dir ("/proc/" ^ p) ->
+ debug_print ("HSqlite3: found valid db file: " ^ x);
+ x, false
+ | _ ->
+ HLog.warn ("HSqlite3: dead process db file found: " ^ x);
+ HLog.warn ("HSqlite3: removing: " ^ x);
+ ignore (Sys.command ("rm " ^ x));
+ aux tl
+ in
+ aux files
in
let db_name = host ^ "/" ^ database in
let db_to_open =
if HExtlib.is_dir "/dev/shm/" && HExtlib.writable_dir "/dev/shm/"
&& (not is_library)
- then
- (let cp_to_ram_cmd = "cp " ^ db_name ^ " " ^ tmp_db_name in
- ignore (Sys.command cp_to_ram_cmd);
+ then (
+ let tmp_db_name, first_time = find_db () in
+ let cp_to_ram_cmd = "cp " ^ db_name ^ " " ^ tmp_db_name in
+ if first_time then
+ if HExtlib.is_regular db_name then ignore (Sys.command cp_to_ram_cmd)
+ else debug_print ("HSqlite3: no initial db: " ^ db_name)
+ else debug_print "HSqlite3: not copying the db, already in memory";
let mv_to_disk_cmd _ =
- ignore (Sys.command ("mv " ^ tmp_db_name ^ " " ^ db_name))
+ if first_time then ignore (Sys.command ("mv "^tmp_db_name^" "^db_name))
+ else debug_print "HSqlite3: not copying back the db"
in
at_exit mv_to_disk_cmd;
tmp_db_name)
(* attach the REGEX function *)
Sqlite3.create_fun2 db "REGEXP"
(fun s rex ->
+ try
match rex, s with
| Sqlite3.Data.TEXT rex, Sqlite3.Data.BLOB s
| Sqlite3.Data.TEXT rex, Sqlite3.Data.TEXT s ->
Sqlite3.Data.INT 1L
else
Sqlite3.Data.INT 0L
- | _ -> raise (Error "wrong types to 'REGEXP'"));
+ | _ -> raise (Error "wrong types to 'REGEXP'")
+ with exn -> HLog.error (Printexc.to_string exn); raise exn);
Some db
;;
| None -> ()
| Some db ->
let b = Sqlite3.db_close db in
- if b=false then prerr_endline "No Closed DataBase"
+ if b=false then debug_print "No Closed DataBase"
;;
(* XXX hack, sqlite has a print "%q" that should be used, but is not bound *)
-let escape s = Pcre.replace ~pat:"([^'])'([^'])" ~templ:"$1''$2" s
+let escape s =
+ let s_escaped = Pcre.replace ~pat:"'" ~templ:"''" s in
+ (*let s_escaped = Pcre.replace ~pat:"([^'])'([^'])" ~templ:"$1''$2" s in*)
+ debug_print s;
+ debug_print s_escaped;
+ s_escaped
+;;
let string_of_rc = function
|Sqlite3.Rc.OK -> "Sqlite3.Rc.OK"
"Sqlite3.Rc.UNKNOWN " ^ string_of_int (Sqlite3.Rc.int_of_unknown n)
;;
-let pp_rc rc = prerr_endline (string_of_rc rc);;
+let pp_rc rc = debug_print (string_of_rc rc);;
let exec s db =
- prerr_endline s;
+ debug_print s;
let stored_result = ref [] in
let store row =
stored_result := row :: !stored_result
in
match rc with
| Sqlite3.Rc.OK -> !stored_result
- | _ -> raise (Error (string_of_rc rc))
+ | _ -> raise (Error (string_of_rc rc ^ ": " ^ Sqlite3.errmsg db ))
;;
let rec map res ~f =