pgsqlresult.php

<?php
namespace Bitrix\Main\DB;
 
class PgsqlResult extends Result
{
    protected $resource;
 
    private $resultFields = null;
 
    public function getSelectedRowsCount()
    {
        return pg_num_rows($this->resource);
    }
 
    public function getFields()
    {
        if ($this->resultFields == null)
        {
            $this->resultFields = array();
 
            if (
                $this->connection
                && (is_resource($this->resource) || is_object($this->resource))
            )
            {
                $fields = pg_num_fields($this->resource);
                if ($fields)
                {
                    $helper = $this->connection->getSqlHelper();
                    for ($i = 0; $i < $fields; $i++)
                    {
                        $fieldName = mb_strtoupper(pg_field_name($this->resource, $i));
                        $fieldType = pg_field_type($this->resource, $i);
                        $this->resultFields[$fieldName] = $helper->getFieldByColumnType($fieldName, $fieldType);
                    }
                }
            }
        }
 
        return $this->resultFields;
    }
 
    protected $byteaFields = false;
 
    protected function fetchRowInternal()
    {
        $result = pg_fetch_assoc($this->resource);
        if ($result)
        {
            if ($this->byteaFields === false)
            {
                $this->byteaFields = [];
                $fieldNum = 0;
                foreach ($result as $fieldName => $_)
                {
                    $fieldType = pg_field_type($this->resource, $fieldNum);
                    if ($fieldType === 'bytea')
                    {
                        $this->byteaFields[$fieldName] = $fieldType;
                    }
                    $fieldNum++;
                }
            }
 
            if ($this->byteaFields)
            {
                foreach ($this->byteaFields as $fieldName => $fieldType)
                {
                    $result[$fieldName] = pg_unescape_bytea($result[$fieldName]);
                }
            }
 
            return array_change_key_case($result, \CASE_UPPER);
        }
        else
        {
            return $result;
        }
    }
}