1 (* Copyright (C) 2005, HELM Team.
3 * This file is part of HELM, an Hypertextual, Electronic
4 * Library of Mathematics, developed at the Computer Science
5 * Department, University of Bologna, Italy.
7 * HELM is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU General Public License
9 * as published by the Free Software Foundation; either version 2
10 * of the License, or (at your option) any later version.
12 * HELM is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 * GNU General Public License for more details.
17 * You should have received a copy of the GNU General Public License
18 * along with HELM; if not, write to the Free Software
19 * Foundation, Inc., 59 Temple Place - Suite 330, Boston,
22 * For details, see the HELM World-Wide-Web page,
23 * http://cs.unibo.it/helm/.
26 (* $Id: hMysql.ml 5769 2006-01-08 18:00:22Z sacerdot $ *)
29 type result = Mysql.result option
32 type result = Sqlite3.row list
33 type dbd = Sqlite3.db option
42 | GENERIC_ERROR of string
44 exception Error of string
46 let prerr_endline s = () (*HLog.debug s*);;
48 let profiler = HExtlib.profile "Sqlite3"
52 ?(host = Helm_registry.get "matita.basedir")
53 ?(database = "sqlite") ?port ?password ?user ()
55 let username = Helm_registry.get "user.name" in
56 let host_mangled = Pcre.replace ~pat:"[/ .]" ~templ:"_" host in
58 "/dev/shm/matita.db." ^ username ^ "." ^ host_mangled ^ "." ^
59 string_of_int (Unix.getpid ())
61 let db_name = host ^ "/" ^ database in
63 if HExtlib.is_dir "/dev/shm/" && HExtlib.writable_dir "/dev/shm/"
66 (let cp_to_ram_cmd = "cp " ^ db_name ^ " " ^ tmp_db_name in
67 ignore (Sys.command cp_to_ram_cmd);
68 let mv_to_disk_cmd _ =
69 ignore (Sys.command ("mv " ^ tmp_db_name ^ " " ^ db_name))
71 at_exit mv_to_disk_cmd;
76 let db = Sqlite3.db_open db_to_open in
77 (* attach the REGEX function *)
78 Sqlite3.create_fun2 db "REGEXP"
82 | Sqlite3.Data.TEXT rex, Sqlite3.Data.BLOB s
83 | Sqlite3.Data.TEXT rex, Sqlite3.Data.TEXT s ->
84 let r = Str.regexp rex in
85 if Str.string_match r s 0 then
89 | _ -> raise (Error "wrong types to 'REGEXP'")
90 with exn -> HLog.error (Printexc.to_string exn); raise exn);
98 let b = Sqlite3.db_close db in
99 if b=false then prerr_endline "No Closed DataBase"
102 (* XXX hack, sqlite has a print "%q" that should be used, but is not bound *)
104 let s_escaped = Pcre.replace ~pat:"'" ~templ:"''" s in
105 (*let s_escaped = Pcre.replace ~pat:"([^'])'([^'])" ~templ:"$1''$2" s in*)
107 prerr_endline s_escaped;
111 let string_of_rc = function
112 |Sqlite3.Rc.OK -> "Sqlite3.Rc.OK"
113 |Sqlite3.Rc.ERROR -> "Sqlite3.Rc.ERROR"
114 |Sqlite3.Rc.INTERNAL -> "Sqlite3.Rc.INTERNAL"
115 |Sqlite3.Rc.PERM -> "Sqlite3.Rc.PERM"
116 |Sqlite3.Rc.ABORT -> "Sqlite3.Rc.ABORT"
117 |Sqlite3.Rc.BUSY -> "Sqlite3.Rc.BUSY"
118 |Sqlite3.Rc.LOCKED -> "Sqlite3.Rc.LOCKED"
119 |Sqlite3.Rc.NOMEM -> "Sqlite3.Rc.NOMEM"
120 |Sqlite3.Rc.READONLY -> "Sqlite3.Rc.READONLY"
121 |Sqlite3.Rc.INTERRUPT -> "Sqlite3.Rc.INTERRUPT"
122 |Sqlite3.Rc.IOERR -> "Sqlite3.Rc.IOERR"
123 |Sqlite3.Rc.CORRUPT -> "Sqlite3.Rc.CORRUPT"
124 |Sqlite3.Rc.NOTFOUND -> "Sqlite3.Rc.NOTFOUND"
125 |Sqlite3.Rc.FULL -> "Sqlite3.Rc.FULL"
126 |Sqlite3.Rc.CANTOPEN -> "Sqlite3.Rc.CANTOPEN"
127 |Sqlite3.Rc.PROTOCOL -> "Sqlite3.Rc.PROTOCOL"
128 |Sqlite3.Rc.EMPTY -> "Sqlite3.Rc.EMPTY"
129 |Sqlite3.Rc.SCHEMA -> "Sqlite3.Rc.SCHEMA"
130 |Sqlite3.Rc.TOOBIG -> "Sqlite3.Rc.TOOBIG"
131 |Sqlite3.Rc.CONSTRAINT -> "Sqlite3.Rc.CONSTRAINT"
132 |Sqlite3.Rc.MISMATCH -> "Sqlite3.Rc.MISMATCH"
133 |Sqlite3.Rc.MISUSE -> "Sqlite3.Rc.MISUSE"
134 |Sqlite3.Rc.NOFLS -> "Sqlite3.Rc.NOFLS"
135 |Sqlite3.Rc.AUTH -> "Sqlite3.Rc.AUTH"
136 |Sqlite3.Rc.FORMAT -> "Sqlite3.Rc.FORMAT"
137 |Sqlite3.Rc.RANGE -> "Sqlite3.Rc.RANGE"
138 |Sqlite3.Rc.NOTADB -> "Sqlite3.Rc.NOTADB"
139 |Sqlite3.Rc.ROW -> "Sqlite3.Rc.ROW"
140 |Sqlite3.Rc.DONE -> "Sqlite3.Rc.DONE"
141 |Sqlite3.Rc.UNKNOWN n ->
142 "Sqlite3.Rc.UNKNOWN " ^ string_of_int (Sqlite3.Rc.int_of_unknown n)
145 let pp_rc rc = prerr_endline (string_of_rc rc);;
149 let stored_result = ref [] in
151 stored_result := row :: !stored_result
157 profiler.HExtlib.profile (Sqlite3.exec_no_headers db ~cb:store) s
160 | Sqlite3.Rc.OK -> !stored_result
161 | _ -> raise (Error (string_of_rc rc ^ ": " ^ Sqlite3.errmsg db ))
165 let map f = List.map f res in
166 profiler.HExtlib.profile map f
170 let iter f = List.iter f res in
171 profiler.HExtlib.profile iter f
177 match Sqlite3.errcode db with
180 let errmsg = (Sqlite3.errmsg db) in
181 if Pcre.pmatch errmsg ~pat:"^table .* already exists" then
184 if Pcre.pmatch errmsg ~pat:"^index .* already exists" then Dup_keyname
185 else if Pcre.pmatch errmsg ~pat:"^no such table: .*" then No_such_table
186 else if Pcre.pmatch errmsg ~pat:"^no such index: .*" then No_such_index
187 else GENERIC_ERROR errmsg
188 |Sqlite3.Rc.INTERNAL |Sqlite3.Rc.PERM |Sqlite3.Rc.ABORT
189 |Sqlite3.Rc.BUSY |Sqlite3.Rc.LOCKED |Sqlite3.Rc.NOMEM
190 |Sqlite3.Rc.READONLY |Sqlite3.Rc.INTERRUPT |Sqlite3.Rc.IOERR
191 |Sqlite3.Rc.CORRUPT |Sqlite3.Rc.NOTFOUND |Sqlite3.Rc.FULL
192 |Sqlite3.Rc.CANTOPEN |Sqlite3.Rc.PROTOCOL |Sqlite3.Rc.EMPTY
193 |Sqlite3.Rc.SCHEMA |Sqlite3.Rc.TOOBIG |Sqlite3.Rc.CONSTRAINT
194 |Sqlite3.Rc.MISMATCH |Sqlite3.Rc.MISUSE |Sqlite3.Rc.NOFLS
195 |Sqlite3.Rc.AUTH |Sqlite3.Rc.FORMAT |Sqlite3.Rc.RANGE
196 |Sqlite3.Rc.NOTADB |Sqlite3.Rc.ROW |Sqlite3.Rc.DONE
197 |Sqlite3.Rc.UNKNOWN _ -> GENERIC_ERROR "Sqlite3_generic_error"
202 let escape_string_for_like = ("ESCAPE \"\\\"" : ('a,'b,'c,'a) format4);;