You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
88 lines
2.5 KiB
88 lines
2.5 KiB
<?php |
|
|
|
/** |
|
* @link http://www.yiiframework.com/ |
|
* @copyright Copyright (c) 2008 Yii Software LLC |
|
* @license http://www.yiiframework.com/license/ |
|
*/ |
|
|
|
namespace yii\db\pgsql; |
|
|
|
/** |
|
* This is an extension of the default PDO class for PostgreSQL drivers. |
|
* It provides additional low level functionality for setting database |
|
* configuration parameters. |
|
* |
|
* @author Gevik babakhani <gevikb@gmail.com> |
|
* @since 2.0 |
|
*/ |
|
class PDO extends \PDO { |
|
|
|
const OPT_SEARCH_PATH = 'search_path'; |
|
const OPT_DEFAULT_SCHEMA = 'default_schema'; |
|
const DEFAULT_SCHEMA = 'public'; |
|
|
|
private $_currentDatabase = null; |
|
|
|
/** |
|
* Here we override the default PDO constructor in order to |
|
* find and set the default schema search path. |
|
*/ |
|
public function __construct($dsn, $username, $passwd, $options) { |
|
$searchPath = null; |
|
if (is_array($options)) { |
|
if (isset($options[self::OPT_SEARCH_PATH])) { |
|
$matches = null; |
|
if (preg_match("/(\s?)+(\w)+((\s+)?,(\s+)?\w+)*/", $options[self::OPT_SEARCH_PATH], $matches) === 1) { |
|
$searchPath = $matches[0]; |
|
} |
|
} |
|
if (isset($options[self::OPT_DEFAULT_SCHEMA])) { |
|
$schema = trim($options[self::OPT_DEFAULT_SCHEMA]); |
|
if ($schema !== '') { |
|
Schema::$DEFAULT_SCHEMA = $schema; |
|
} |
|
} |
|
if (Schema::$DEFAULT_SCHEMA === null || Schema::$DEFAULT_SCHEMA === '') { |
|
Schema::$DEFAULT_SCHEMA = self::DEFAULT_SCHEMA; |
|
} |
|
} |
|
parent::__construct($dsn, $username, $passwd, $options); |
|
if (!is_null($searchPath)) { |
|
$this->setSchemaSearchPath($searchPath); |
|
} |
|
} |
|
|
|
/** |
|
* Returns the name of the current (connected) database |
|
* @return string |
|
*/ |
|
public function getCurrentDatabase() { |
|
if (is_null($this->_currentDatabase)) { |
|
return $this->query('select current_database()')->fetchColumn(); |
|
} |
|
} |
|
|
|
/** |
|
* Sets the schema search path of the current users session. |
|
* The syntax of the path is a comma separated string with |
|
* your custom search path at the beginning and the "public" |
|
* schema at the end. |
|
* |
|
* This method automatically adds the "public" schema at the |
|
* end of the search path if it is not provied. |
|
* @param string custom schema search path. defaults to public |
|
*/ |
|
public function setSchemaSearchPath($searchPath = 'public') { |
|
$schemas = explode(',', str_replace(' ', '', $searchPath)); |
|
if (end($schemas) !== 'public') { |
|
$schemas[] = 'public'; |
|
} |
|
foreach ($schemas as $k => $item) { |
|
$schemas[$k] = '"' . str_replace(array('"', "'", ';'), '', $item) . '"'; |
|
} |
|
$path = implode(', ', $schemas); |
|
$this->exec('SET search_path TO ' . $path); |
|
} |
|
|
|
}
|
|
|