- Patch #342950 by Josh Waihi: moving PostgreSQL stored procedures to the PostgreSQL database backend, where it belongs.

1 files changed, 58 insertions, 0 deletions

diff --git a/includes/database/pgsql/install.inc b/includes/database/pgsql/install.inc
index 222d7e734..a174bd05c 100644
--- a/includes/database/pgsql/install.inc
+++ b/includes/database/pgsql/install.inc
@@ -17,6 +17,10 @@ class DatabaseTasks_pgsql extends DatabaseTasks {
       'function' => 'checkEncoding',
       'arguments' => array(),
     );
+    $this->tasks[] = array(
+      'function' => 'initializeDatabase',
+      'arguments' => array(),
+    );
   }
 
   public function name() {
@@ -45,5 +49,59 @@ class DatabaseTasks_pgsql extends DatabaseTasks {
       $this->fail(st('Drupal could not determine the encoding of the database was set to UTF-8'));
     }
   }
+
+  /**
+   * Make PostgreSQL Drupal friendly.
+   */
+  function initializeDatabase() {
+    // We create some functions using global names instead of prefixing them
+    // like we do with table names. This is so that we don't double up if more
+    // than one instance of Drupal is running on a single database. We therefore
+    // avoid trying to create them again in that case.
+
+    try {
+      // Create functions.
+      db_query('CREATE OR REPLACE FUNCTION "greatest"(numeric, numeric) RETURNS numeric AS
+        \'SELECT CASE WHEN (($1 > $2) OR ($2 IS NULL)) THEN $1 ELSE $2 END;\'
+        LANGUAGE \'sql\''
+      );
+      db_query('CREATE OR REPLACE FUNCTION "greatest"(numeric, numeric, numeric) RETURNS numeric AS
+        \'SELECT greatest($1, greatest($2, $3));\'
+        LANGUAGE \'sql\''
+      );
+      // Don't use {} around pg_proc table.
+      if (!db_query("SELECT COUNT(*) FROM pg_proc WHERE proname = 'rand'")->fetchField()) {
+        db_query('CREATE OR REPLACE FUNCTION "rand"() RETURNS float AS
+          \'SELECT random();\'
+          LANGUAGE \'sql\''
+        );
+      }
+
+      // Don't use {} around pg_proc table.
+      if (!db_query("SELECT COUNT(*) FROM pg_proc WHERE proname = 'concat'")->fetchField()) {
+        db_query('CREATE OR REPLACE FUNCTION "concat"(text, text) RETURNS text AS
+          \'SELECT $1 || $2;\'
+          LANGUAGE \'sql\''
+        );
+      }
+      db_query('CREATE OR REPLACE FUNCTION "if"(boolean, text, text) RETURNS text AS
+        \'SELECT CASE WHEN $1 THEN $2 ELSE $3 END;\'
+        LANGUAGE \'sql\''
+      );
+      db_query('CREATE OR REPLACE FUNCTION "if"(boolean, integer, integer) RETURNS integer AS
+        \'SELECT CASE WHEN $1 THEN $2 ELSE $3 END;\'
+        LANGUAGE \'sql\''
+      );
+
+      db_query('CREATE OR REPLACE FUNCTION "substring_index"(text, text, integer) RETURNS text AS
+        \'SELECT array_to_string((string_to_array($1, $2)) [1:$3], $2);\'
+        LANGUAGE \'sql\''
+      );
+      $this->pass(st('PostgreSQL has initialized itself.'));
+    }
+    catch (Exception $e) {
+      $this->fail(st('Drupal could not be correctly setup with the existing database. Please revise any errors.'));
+    }
+  }
 }