瀏覽代碼

PostgreSQL should now work as a db prefs backend (and fix alterable field names)

simond 23 年之前
父節點
當前提交
dc26a6eba2
共有 2 個文件被更改,包括 83 次插入22 次删除
  1. 6 9
      doc/db-backend.txt
  2. 77 13
      functions/db_prefs.php

+ 6 - 9
doc/db-backend.txt

@@ -73,7 +73,8 @@ Next, edit your configuration so that the address book DSN (Data Source
 Name) is specified, this can be done using either conf.pl or via the
 Name) is specified, this can be done using either conf.pl or via the
 administration plugin. The DSN should look something like:
 administration plugin. The DSN should look something like:
 
 
- mysql://squirreluser:sqpassword@localhost/squirrelmail
+ mysql://squirreluser:sqpassword@localhost/squirrelmail or
+ pgsql://squirreluser:sqpassword@localhost/squirrelmail
 
 
 From now on all users' personal addressbooks will be stored in a
 From now on all users' personal addressbooks will be stored in a
 database.
 database.
@@ -97,25 +98,21 @@ The table structure should be similar to this (for MySQL):
 
 
 and for PostgreSQL:
 and for PostgreSQL:
 CREATE TABLE "userprefs" (
 CREATE TABLE "userprefs" (
-   "user" varchar(128) NOT NULL,
+   "username" varchar(128) NOT NULL,
    "prefkey" varchar(64) NOT NULL,
    "prefkey" varchar(64) NOT NULL,
    "prefval" text,
    "prefval" text,
-   CONSTRAINT "userprefs_pkey" PRIMARY KEY ("prefkey", "user")
+   CONSTRAINT "userprefs_pkey" PRIMARY KEY ("prefkey", "username")
 );
 );
 
 
 Next, edit your configuration so that the preferences DSN (Data Source
 Next, edit your configuration so that the preferences DSN (Data Source
 Name) is specified, this can be done using either conf.pl or via the
 Name) is specified, this can be done using either conf.pl or via the
 administration plugin. The DSN should look something like:
 administration plugin. The DSN should look something like:
 
 
- mysql://squirreluser:sqpassword@localhost/squirrelmail
+ mysql://squirreluser:sqpassword@localhost/squirrelmail or
+ pgsql://squirreluser:sqpassword@localhost/squirrelmail
 
 
 From now on all users' personal preferences will be stored in a
 From now on all users' personal preferences will be stored in a
 database.
 database.
 
 
 Default preferences can be set by altering the $default array in
 Default preferences can be set by altering the $default array in
 db_prefs.php.
 db_prefs.php.
-
-NB It seems that currently database backed preferences won't work under
-PostgreSQL at least due to using a column name of user and REPLACE which
-doesn't exist under PostgreSQL. This will be worked on and fixed as soon
-as possible.

+ 77 - 13
functions/db_prefs.php

@@ -32,6 +32,10 @@
  * $Id$
  * $Id$
  */
  */
 
 
+define('SMDB_UNKNOWN', 0);
+define('SMDB_MYSQL', 1);
+define('SMDB_PGSQL', 2);
+
 require_once('DB.php');
 require_once('DB.php');
 require_once('../config/config.php');
 require_once('../config/config.php');
 
 
@@ -75,17 +79,25 @@ class dbPrefs {
 
 
     var $dbh   = NULL;
     var $dbh   = NULL;
     var $error = NULL;
     var $error = NULL;
+    var $db_type = SMDB_UNKNOWN;
 
 
     var $default = Array('chosen_theme' => '../themes/default_theme.php',
     var $default = Array('chosen_theme' => '../themes/default_theme.php',
                          'show_html_default' => '0');
                          'show_html_default' => '0');
 
 
     function open() {
     function open() {
         global $prefs_dsn, $prefs_table;
         global $prefs_dsn, $prefs_table;
+        global $prefs_user_field, $prefs_key_field, $prefs_val_field;
 
 
         if(isset($this->dbh)) {
         if(isset($this->dbh)) {
             return true;
             return true;
         }
         }
 
 
+        if (preg_match('/^mysql/', $prefs_dsn)) {
+            $this->db_type = SMDB_MYSQL;
+        } elseif (preg_match('/^pgsql/', $prefs_dsn)) {
+            $this->db_type = SMDB_PGSQL;
+        }
+
         if (!empty($prefs_table)) {
         if (!empty($prefs_table)) {
             $this->table = $prefs_table;
             $this->table = $prefs_table;
         }
         }
@@ -168,19 +180,71 @@ class dbPrefs {
         if (!$this->open()) {
         if (!$this->open()) {
             return false;
             return false;
         }
         }
-        $query = sprintf("REPLACE INTO %s (%s, %s, %s) ".
-                         "VALUES('%s','%s','%s')",
-                         $this->table,
-                         $this->user_field,
-                         $this->key_field,
-                         $this->val_field,
-                         $this->dbh->quoteString($user),
-                         $this->dbh->quoteString($key),
-                         $this->dbh->quoteString($value));
-
-        $res = $this->dbh->simpleQuery($query);
-        if(DB::isError($res)) {
-            $this->failQuery($res);
+        if ($this->db_type == SMDB_MYSQL) {
+            $query = sprintf("REPLACE INTO %s (%s, %s, %s) ".
+                             "VALUES('%s','%s','%s')",
+                             $this->table,
+                             $this->user_field,
+                             $this->key_field,
+                             $this->val_field,
+                             $this->dbh->quoteString($user),
+                             $this->dbh->quoteString($key),
+                             $this->dbh->quoteString($value));
+
+            $res = $this->dbh->simpleQuery($query);
+            if(DB::isError($res)) {
+                $this->failQuery($res);
+            }
+        } elseif ($this->db_type == SMDB_PGSQL) {
+            $this->dbh->simpleQuery("BEGIN TRANSACTION");
+            $query = sprintf("DELETE FROM %s WHERE %s='%s' AND %s='%s'",
+                             $this->table,
+                             $this->user_field,
+                             $this->dbh->quoteString($user),
+                             $this->key_field,
+                             $this->dbh->quoteString($key));
+            $res = $this->dbh->simpleQuery($query);
+            if (DB::isError($res)) {
+                $this->dbh->simpleQuery("ROLLBACK TRANSACTION");
+                $this->failQuery($res);
+            }
+            $query = sprintf("INSERT INTO %s (%s, %s, %s) VALUES ('%s', '%s', '%s')",
+                             $this->table,
+                             $this->user_field,
+                             $this->key_field,
+                             $this->val_field,
+                             $this->dbh->quoteString($user),
+                             $this->dbh->quoteString($key),
+                             $this->dbh->quoteString($value));
+            $res = $this->dbh->simpleQuery($query);
+            if (DB::isError($res)) {
+                $this->dbh->simpleQuery("ROLLBACK TRANSACTION");
+                $this->failQuery($res);
+            }
+            $this->dbh->simpleQuery("COMMIT TRANSACTION");
+        } else {
+            $query = sprintf("DELETE FROM %s WHERE %s='%s' AND %s='%s'",
+                             $this->table,
+                             $this->user_field,
+                             $this->dbh->quoteString($user),
+                             $this->key_field,
+                             $this->dbh->quoteString($key));
+            $res = $this->dbh->simpleQuery($query);
+            if (DB::isError($res)) {
+                $this->failQuery($res);
+            }
+            $query = sprintf("INSERT INTO %s (%s, %s, %s) VALUES ('%s', '%s', '%s')",
+                             $this->table,
+                             $this->user_field,
+                             $this->key_field,
+                             $this->val_field,
+                             $this->dbh->quoteString($user),
+                             $this->dbh->quoteString($key),
+                             $this->dbh->quoteString($value));
+            $res = $this->dbh->simpleQuery($query);
+            if (DB::isError($res)) {
+                $this->failQuery($res);
+            }
         }
         }
 
 
         return true;
         return true;