helm/ocaml/getter/http_getter_map.ml

   1 (*
   2  * Copyright (C) 2003-2004:
   3  *    Stefano Zacchiroli <zack@cs.unibo.it>
   4  *    for the HELM Team http://helm.cs.unibo.it/
   5  *
   6  *  This file is part of HELM, an Hypertextual, Electronic
   7  *  Library of Mathematics, developed at the Computer Science
   8  *  Department, University of Bologna, Italy.
   9  *
  10  *  HELM is free software; you can redistribute it and/or
  11  *  modify it under the terms of the GNU General Public License
  12  *  as published by the Free Software Foundation; either version 2
  13  *  of the License, or (at your option) any later version.
  14  *
  15  *  HELM is distributed in the hope that it will be useful,
  16  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  17  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18  *  GNU General Public License for more details.
  19  *
  20  *  You should have received a copy of the GNU General Public License
  21  *  along with HELM; if not, write to the Free Software
  22  *  Foundation, Inc., 59 Temple Place - Suite 330, Boston,
  23  *  MA  02111-1307, USA.
  24  *
  25  *  For details, see the HELM World-Wide-Web page,
  26  *  http://helm.cs.unibo.it/
  27  *)
  28
  29 exception Key_already_in of string;;
  30 exception Key_not_found of string;;
  31
  32 class map dbname =
  33   let perm = 420 in (* permission 644 in decimal notation *)
  34   let open_dbm () = Dbm.opendbm dbname [ Dbm.Dbm_rdwr; Dbm.Dbm_create ] perm in
  35
  36   object (self)
  37
  38     inherit ThreadSafe.threadSafe
  39
  40     val mutable db = open_dbm ()
  41     val index_RE = Pcre.regexp "^theory:/.*/index.theory$"
  42
  43     (*initializer Gc.finalise Dbm.close db  (* close db on GC *)*)
  44
  45     (* Backward compatibility code to map
  46        theory:/path/t.theory into theory:/path/t/index.theory
  47        when cic:/path/t/ exists *)
  48     method private normalize_key key =
  49      if Pcre.pmatch ~rex:index_RE key &&
  50         (try ignore (Dbm.find db key); false with Not_found -> true)
  51      then
  52       (* we substitute /index.theory with .theory *)
  53       String.sub key 0 (String.length key - 13) ^ ".theory"
  54      else key
  55
  56     method add key value =
  57       self#doWriter (lazy (
  58         try
  59           Dbm.add db key value
  60         with Dbm.Dbm_error "Entry already exists" -> raise (Key_already_in key)
  61       ))
  62
  63     method replace key value =
  64       self#doWriter (lazy (
  65         Dbm.replace db key value
  66       ))
  67
  68     method remove key =
  69       self#doWriter (lazy (
  70         try
  71           Dbm.remove db key
  72         with Dbm.Dbm_error "dbm_delete" -> raise (Key_not_found key)
  73       ))
  74
  75     method resolve key =
  76       self#doReader (lazy (
  77         try
  78          Dbm.find db (self#normalize_key key)
  79         with Not_found -> raise (Key_not_found key)
  80       ))
  81
  82     method iter (f: string -> string -> unit) =
  83       self#doReader (lazy (
  84         Dbm.iter f db
  85       ))
  86
  87     method sync =
  88       self#doWriter (lazy (
  89         Dbm.close db;
  90         db <- open_dbm ()
  91       ))
  92
  93     method clear =
  94       self#doWriter (lazy (
  95         Dbm.close db;
  96         List.iter
  97           (fun ext ->
  98             let file = dbname ^ ext in
  99             if Sys.file_exists file then Sys.remove file)
 100           [".dir"; ".pag"; ".db"];
 101         db <- open_dbm ()
 102       ))
 103
 104     method close =
 105       self#doWriter (lazy (
 106         Dbm.close db
 107       ))
 108
 109   end
 110