SqliteTable: fixups to schema handling

[org.ibex.mail.git] / src / org / ibex / mail / SqliteDB.java

diff --git a/src/org/ibex/mail/SqliteDB.java b/src/org/ibex/mail/SqliteDB.java
index e5fbc68..900e07c 100644 (file)
--- a/src/org/ibex/mail/SqliteDB.java
+++ b/src/org/ibex/mail/SqliteDB.java
@@ -22,48 +22,67 @@ public class SqliteDB {
 
     public Connection getConnection() { return conn; }
 
-    public synchronized SqliteTable getTable(String name) {
+    public synchronized SqliteTable getTable(String name, String schema) throws SQLException {
         SqliteTable ret = tables.get(name);
-        if (ret==null) ret = new SqliteTable(name);
+        if (ret==null) ret = new SqliteTable(name, schema);
         return ret;
     }
 
     // check upstream: PRAGMA encoding = "UTF-8"; 
     // create indices
-    // PRAGMA auto_vacuum=1  (can only be set before any tables are created)
-    // periodic "PRAGMA integrity_check; "?
+    // periodically run "analyze"?
 
     public void setCacheSize(int kilobytes) throws SQLException {
         conn.prepareStatement("PRAGMA cache_size="+Math.ceil(kilobytes/1.5)+";").executeUpdate();
     }
 
-    public SqliteDB(String filename, String[] tables) {
-        this(filename, tables, false);
-    }
-    public SqliteDB(String filename, String[] tables, boolean fastButDangerous) {
+    public SqliteDB(String filename) {
         this.filename = filename;
         try {
             Class.forName("org.sqlite.JDBC");
             conn = DriverManager.getConnection("jdbc:sqlite:"+filename);
-            for(String s : tables)
-                conn.prepareStatement(s).executeUpdate();
+            conn.prepareStatement("PRAGMA auto_vacuum = 1").executeUpdate();
+            conn.prepareStatement("VACUUM").executeUpdate();
+            // until we have better assurances about locking on network filesystems...
+            conn.prepareStatement("PRAGMA locking_mode = EXCLUSIVE").executeUpdate();
             conn.prepareStatement("PRAGMA temp_store = MEMORY").executeUpdate();
             conn.prepareStatement("PRAGMA page_size=4096").executeUpdate();
             conn.prepareStatement("PRAGMA cache_size=2000").executeUpdate();
-            if (fastButDangerous)
-                conn.prepareStatement("PRAGMA synchronous = OFF").executeUpdate();
+            ResultSet rs = conn.prepareStatement("PRAGMA integrity_check").executeQuery();
+            rs.next();
+            String result = rs.getString(1);
+            if (!result.equals("ok"))
+                throw new RuntimeException("PRAGMA integrity_check returned \""+result+"\"");
         }
         catch (SQLException e) { throw new RuntimeException(e); }
         catch (ClassNotFoundException e) { throw new RuntimeException(e); }
     }
 
+    public void setFastButDangerous(boolean on) throws SQLException {
+        conn.prepareStatement("PRAGMA synchronous = "+(on?"OFF":"ON")).executeUpdate();
+    }
+
+    /*
+    public void dump(OutputStream os) {
+    }
+    */
+
     public class SqliteTable {
         public final String name;
         private String reapColumn = null;
-        private SqliteTable(String name) {
+        private SqliteTable(String name, String schema) throws SQLException {
             this.name = name;
+            PreparedStatement ps = conn.prepareStatement("create table if not exists ?");
+            ps.setString(1, name+" "+schema);
+            ps.executeUpdate();
             tables.put(name, this);
         }
+        public void createIndex(String name, String column) throws SQLException {
+            PreparedStatement ps = conn.prepareStatement("create index if not exists ? on ?");
+            ps.setString(1, name);
+            ps.setString(2, column);
+            ps.executeUpdate();
+        }
         protected void reap(String reapColumn) {
             if (this.reapColumn != null) throw new RuntimeException("reapColumn already set");
             this.reapColumn = reapColumn;