cakephp/Cake/Database/Driver/Postgres.php

<?php
/**
 * PHP Version 5.4
 *
 * CakePHP(tm) : Rapid Development Framework (http://cakephp.org)
 * Copyright (c) Cake Software Foundation, Inc. (http://cakefoundation.org)
 *
 * Licensed under The MIT License
 * For full copyright and license information, please see the LICENSE.txt
 * Redistributions of files must retain the above copyright notice.
 *
 * @copyright     Copyright (c) Cake Software Foundation, Inc. (http://cakefoundation.org)
 * @link          http://cakephp.org CakePHP(tm) Project
 * @since         CakePHP(tm) v 3.0.0
 * @license       MIT License (http://www.opensource.org/licenses/mit-license.php)
 */
namespace Cake\Database\Driver;

use Cake\Database\Dialect\PostgresDialectTrait;
use Cake\Database\Statement;
use PDO;

class Postgres extends \Cake\Database\Driver {

	use PDODriverTrait;
	use PostgresDialectTrait;

/**
 * Base configuration settings for Postgres driver
 *
 * @var array
 */
	protected $_baseConfig = [
		'persistent' => true,
		'host' => 'localhost',
		'login' => 'root',
		'password' => '',
		'database' => 'cake',
		'schema' => 'public',
		'port' => 5432,
		'encoding' => 'utf8',
		'timezone' => 'UTC',
		'flags' => [],
		'init' => [],
		'dsn' => null
	];

/**
 * Establishes a connection to the databse server
 *
 * @return boolean true on success
 */
	public function connect() {
		if ($this->_connection) {
			return true;
		}
		$config = $this->_config;
		$config['flags'] += [
			PDO::ATTR_PERSISTENT => $config['persistent'],
			PDO::ATTR_ERRMODE => PDO::ERRMODE_EXCEPTION
		];

		if (empty($config['dsn'])) {
			$config['dsn'] = "pgsql:host={$config['host']};port={$config['port']};dbname={$config['database']}";
		}

		$this->_connect($config);
		$this->_connection = $connection = $this->connection();
		if (!empty($config['encoding'])) {
			$this->setEncoding($config['encoding']);
		}

		if (!empty($config['schema'])) {
			$this->setSchema($config['schema']);
		}

		if (!empty($config['timezone'])) {
			$config['init'][] = sprintf("SET timezone = %s", $connection->quote($config['timezone']));
		}

		foreach ($config['init'] as $command) {
			$connection->exec($command);
		}
		return true;
	}

/**
 * Returns whether php is able to use this driver for connecting to database
 *
 * @return boolean true if it is valid to use this driver
 */
	public function enabled() {
		return in_array('pgsql', PDO::getAvailableDrivers());
	}

/**
 * Sets connection encoding
 *
 * @return void
 */
	public function setEncoding($encoding) {
		$this->connect();
		$this->_connection->exec('SET NAMES ' . $this->_connection->quote($encoding));
	}

/**
 * Sets connection default schema, if any relation defined in a query is not fully qualified
 * postgres will fallback to looking the relation into defined default schema
 *
 * @return void
 */
	public function setSchema($schema) {
		$this->connect();
		$this->_connection->exec('SET search_path TO ' . $this->_connection->quote($schema));
	}

}