Improve schema support for Postgre
diff --git a/system/database/drivers/pdo/subdrivers/pdo_pgsql_driver.php b/system/database/drivers/pdo/subdrivers/pdo_pgsql_driver.php
index 1300772..2eb0606 100644
--- a/system/database/drivers/pdo/subdrivers/pdo_pgsql_driver.php
+++ b/system/database/drivers/pdo/subdrivers/pdo_pgsql_driver.php
@@ -90,6 +90,26 @@
// --------------------------------------------------------------------
/**
+ * Database connection
+ *
+ * @param bool $persistent
+ * @return object
+ */
+ public function db_connect($persistent = FALSE)
+ {
+ $this->conn_id = parent::db_connect($persistent);
+
+ if (is_object($this->conn_id) && ! empty($this->schema))
+ {
+ $this->simple_query('SET search_path TO '.$this->schema.',public');
+ }
+
+ return $this->conn_id;
+ }
+
+ // --------------------------------------------------------------------
+
+ /**
* Insert ID
*
* @param string $name
diff --git a/system/database/drivers/postgre/postgre_driver.php b/system/database/drivers/postgre/postgre_driver.php
index 2a78a57..acad016 100644
--- a/system/database/drivers/postgre/postgre_driver.php
+++ b/system/database/drivers/postgre/postgre_driver.php
@@ -131,13 +131,32 @@
// --------------------------------------------------------------------
/**
- * Non-persistent database connection
+ * Database connection
*
+ * @param bool $persistent
* @return resource
*/
- public function db_connect()
+ public function db_connect($persistent = FALSE)
{
- return @pg_connect($this->dsn);
+ if ($persistent === TRUE
+ && ($conn = @pg_pconnect($this->dsn))
+ && pg_connection_status($conn) === PGSQL_CONNECTION_BAD
+ && pg_ping($conn) === FALSE
+ )
+ {
+ return FALSE;
+ }
+ else
+ {
+ $conn = @pg_connect($this->dsn);
+ }
+
+ if ($conn && ! empty($this->schema))
+ {
+ $this->simple_query('SET search_path TO '.$this->schema.',public');
+ }
+
+ return $conn;
}
// --------------------------------------------------------------------
@@ -149,15 +168,7 @@
*/
public function db_pconnect()
{
- $conn = @pg_pconnect($this->dsn);
- if ($conn && pg_connection_status($conn) === PGSQL_CONNECTION_BAD)
- {
- if (pg_ping($conn) === FALSE)
- {
- return FALSE;
- }
- }
- return $conn;
+ return $this->db_connect(TRUE);
}
// --------------------------------------------------------------------
diff --git a/user_guide_src/source/changelog.rst b/user_guide_src/source/changelog.rst
index 6570eb7..d7cfdd5 100644
--- a/user_guide_src/source/changelog.rst
+++ b/user_guide_src/source/changelog.rst
@@ -150,6 +150,7 @@
- Added ``update_batch()`` support.
- Removed ``limit()`` and ``order_by()`` support for *UPDATE* and *DELETE* queries as PostgreSQL does not support those features.
- Added a work-around for dead persistent connections to be re-created after a database restart.
+ - Changed ``db_connect()`` to include the (new) **schema** value into Postgre's **search_path** session variable.
- Improved support of the CUBRID driver, including:
- Added DSN string support.
- Added persistent connections support.