chillerlan\Database\Drivers\PostgreSQLDriver

Complexity

Total Complexity

16

Size/Duplication

Total Lines	162
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	3

Importance

Changes

0

10 Methods

Rating	Name	Duplication	Size	Complexity
B	connect()	0	27	3
A	disconnect()	0	3	1
A	getClientInfo()	0	5	1
A	getServerInfo()	0	5	1
A	__escape()	0	3	1
A	__getResult()	0	12	3
A	__raw()	0	3	1
A	__prepared()	0	5	1
A	__multi()	0	9	2
A	__multi_callback()	0	9	2

<?php
/**
 * Class PostgreSQLDriver
 *
 * @filesource   PostgreSQLDriver.php
 * @created      21.02.2016
 * @package      chillerlan\Database\Drivers\PostgreSQL
 * @author       Smiley <smiley@chillerlan.net>
 * @copyright    2016 Smiley
 * @license      MIT
 */

namespace chillerlan\Database\Drivers;

use chillerlan\Database\DBException;

/**
 *
 */
class PostgreSQLDriver extends DBDriverAbstract{

	/**
	 * Holds the database resource object
	 *
	 * @var resource
	 */
	protected $db;

	/**
	 * Establishes a database connection and returns the connection object
	 *
	 * @return \chillerlan\Database\Drivers\DBDriverInterface
	 * @throws \chillerlan\Database\DBException
	 */
	public function connect():DBDriverInterface{

		if(gettype($this->db) === 'resource'){
			return $this;
		}

		// i am an ugly duckling. fix me please.

		$options = [
			'--client_encoding='.$this->options->pgsql_charset,
		];

		$conn_str = [
			'host=\''.$this->options->host.'\'',
			'port=\''.(int)$this->options->port.'\'',
			'dbname=\''.$this->options->database.'\'',
			'user=\''.$this->options->username.'\'',
			'password=\''.$this->options->password.'\'',
			'options=\''.implode(' ', $options).'\'',
		];

		if(!$this->db = pg_connect(implode(' ', $conn_str))){
			throw new DBException('Could not connect to the database.');
		}

		return $this;
	}

	/**
	 * Closes a database connection
	 *
	 * @return bool
	 */
	public function disconnect():bool{
		return pg_close($this->db);
	}

	/**
	 * Returns info about the used php client
	 *
	 * @return string php's database client string
	 */
	public function getClientInfo():string{
		$ver = pg_version($this->db);

		return 'PostgreSQL '.$ver['client'].' ('.$ver['client_encoding'].')';
	}

	/**
	 * Returns info about the database server
	 *
	 * @return string database's serverinfo string
	 */
	public function getServerInfo():string{
		$ver = pg_version($this->db);

		return 'PostgreSQL '.$ver['server'].' ('.$ver['server_encoding'].', date style: '.$ver['DateStyle'].', time zone: '.$ver['TimeZone'].')';
	}

	/**
	 * @param $data
	 *
	 * @return string
	 */
	protected function __escape($data){
		return pg_escape_string($this->db, $data);
	}

	/**
	 * @param             $result
	 * @param string|null $index
	 * @param bool        $assoc
	 *
	 * @return bool|\chillerlan\Database\DBResult
	 */
	protected function __getResult($result, string $index = null, bool $assoc = true){

		if(is_bool($result)){
			return $result;
		}

		$r = $this->getResult($assoc ? 'pg_fetch_assoc' : 'pg_fetch_row', [$result], $index, $assoc);

		pg_free_result($result);

		return $r;
	}

	/**
	 * @param string      $sql
	 * @param string|null $index
	 * @param bool        $assoc
	 *
	 * @return bool|\chillerlan\Database\DBResult
	 */
	protected function __raw(string $sql, string $index = null, bool $assoc = true){
		return $this->__getResult(pg_query($sql), $index, $assoc);
	}

	/**
	 * @param string      $sql
	 * @param array       $values
	 * @param string|null $index
	 * @param bool        $assoc
	 *
	 * @return bool|\chillerlan\Database\DBResult
	 */
	protected function __prepared(string $sql, array $values = [], string $index = null, bool $assoc = true){
		pg_prepare($this->db, '', $sql);

		return $this->__getResult(pg_execute($this->db, '', $values), $index, $assoc);
	}

	/**
	 * @param string $sql
	 * @param array  $values
	 *
	 * @return bool
	 */
	protected function __multi(string $sql, array $values){
		pg_prepare($this->db, '', $sql);

		foreach($values as $row){
			pg_execute($this->db, '', $row);
		}

		return true;
	}

	/**
	 * @param string $sql
	 * @param array  $data
	 * @param        $callback
	 *
	 * @return bool
	 */
	protected function __multi_callback(string $sql, array $data, $callback){
		pg_prepare($this->db, '', $sql);

		foreach($data as $row){
			pg_execute($this->db, '', call_user_func_array($callback, [$row]));
		}

		return true;
	}

}