chillerlan / php-database

src/Drivers/PostgreSQL.php

<?php
/**
 * Class PostgreSQL
 *
 * @filesource   PostgreSQL.php
 * @created      28.06.2017
 * @package      chillerlan\Database\Drivers
 * @author       Smiley <smiley@chillerlan.net>
 * @copyright    2017 Smiley
 * @license      MIT
 *
 * @noinspection PhpComposerExtensionStubsInspection
 */

namespace chillerlan\Database\Drivers;

use chillerlan\Database\{
	Dialects\Postgres, Result
};

/**
 * @property resource $db
 */
class PostgreSQL extends DriverAbstract{

	protected string $dialect = Postgres::class;

	/** @inheritdoc */
	public function connect():DriverInterface{

		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).'\'',
		];

		try{
			$this->db = pg_connect(implode(' ', $conn_str));

			return $this;
		}
		catch(\Exception $e){
			throw new DriverException('db error: [PostgreSQL]: '.$e->getMessage());
		}

	}

	/** @inheritdoc */
	public function disconnect():bool{

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

		return true;
	}

	/** @inheritdoc */
	public function getClientInfo():string{
		$ver = pg_version($this->db);

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

	/** @inheritdoc */
	public function getServerInfo():?string{
		$ver = pg_version($this->db);

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

	/** @inheritdoc */
	protected function __escape(string $data):string{
		return '\''.pg_escape_string($this->db, $data).'\''; // emulate PDO
	}

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

		if(is_bool($result)){
			return $result; // @codeCoverageIgnore
		}

		$out = new Result(null, $this->convert_encoding_src, $this->convert_encoding_dest);
		$i   = 0;

		while($row = call_user_func_array($assoc === true ? 'pg_fetch_assoc' : 'pg_fetch_row', [$result])){
			$key = $i;

			$j = 0;
			foreach($row as $k => $item){
				// https://gitter.im/arenanet/api-cdi?at=594326ba31f589c64fafe554
				$fieldType = pg_field_type($result, $j);

				if($fieldType === 'bool'){
					$row[$k] = $item === 't';
				}
				elseif(in_array($fieldType, ['int2', 'int4', 'int8'], true)){
					$row[$k] = (int)$item;

				}
				elseif(in_array($fieldType, ['float4', 'float8'], true)){
					$row[$k] = (float)$item; // @codeCoverageIgnore
				}

				$j++;
			}

			if($assoc && !empty($index)){
				$key = $row[$index];
			}

			$out[$key] = $row;
			$i++;
		}

		pg_free_result($result);

		return $i === 0 ? true : $out;
	}

	/** @inheritdoc */
	protected function raw_query(string $sql, string $index = null, bool $assoc = null){
		return $this->__getResult(pg_query($sql), $index, $assoc);

$sql of type string is incompatible with the type resource expected by parameter $connection of pg_query().

The call to pg_query() has too few arguments starting with query.

	}

	/** @inheritdoc */
	protected function prepared_query(string $sql, array $values = null, string $index = null, bool $assoc = null){
		pg_prepare($this->db, '', $this->replaceParams($sql));

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

It seems like $values can also be of type null; however, parameter $params of pg_execute() does only seem to accept array, maybe add an additional type check?

	}

	/** @inheritdoc */
	protected function multi_query(string $sql, array $values){
		pg_prepare($this->db, '', $this->replaceParams($sql));

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

		return true;
	}

	/** @inheritdoc */
	protected function multi_callback_query(string $sql, iterable $data, $callback){
		pg_prepare($this->db, '', $this->replaceParams($sql));

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

		return true;
	}

	/** @inheritdoc */
	protected function replaceParams(string $sql):string{
		$i = 0;

		return preg_replace_callback('/(\?)/', function() use (&$i){
			return '$'.++$i;
		}, $sql);
	}

}