result.php

<?php
namespace Bitrix\Main\DB;
 
abstract class Result implements \IteratorAggregate
{
    protected $connection;
    protected $resource;
    protected $trackerQuery = null;
 
    protected $converters = array();
    protected $serializedFields = array();
    protected $replacedAliases = array();
    protected $fetchDataModifiers = array();
 
    protected $count;
 
    public function __construct($result, Connection $dbConnection = null, \Bitrix\Main\Diag\SqlTrackerQuery $trackerQuery = null)
    {
        $this->resource = $result;
        $this->connection = $dbConnection;
        $this->trackerQuery = $trackerQuery;
        $resultFields = $this->getFields();
        if ($resultFields && $this->connection)
        {
            $helper = $this->connection->getSqlHelper();
            foreach ($resultFields as $key => $type)
            {
                $converter = $helper->getConverter($resultFields[$key]);
                if (is_callable($converter))
                {
                    $this->converters[$key] = $converter;
                }
            }
        }
    }
 
    public function getResource()
    {
        return $this->resource;
    }
 
    public function setReplacedAliases(array $replacedAliases)
    {
        $this->replacedAliases = $replacedAliases;
    }
 
    public function addReplacedAliases(array $replacedAliases)
    {
        $this->replacedAliases = array_merge($this->replacedAliases, $replacedAliases);
    }
 
    public function setSerializedFields(array $serializedFields)
    {
        $this->serializedFields = $serializedFields;
    }
 
    public function addFetchDataModifier($fetchDataModifier)
    {
        if (!is_callable($fetchDataModifier))
        {
            throw new \Bitrix\Main\ArgumentException('Data Modifier should be a callback');
        }
 
        $this->fetchDataModifiers[] = $fetchDataModifier;
    }
 
    public function fetchRaw()
    {
        if ($this->trackerQuery != null)
        {
            $this->trackerQuery->restartQuery();
        }
 
        $data = $this->fetchRowInternal();
 
        if ($this->trackerQuery != null)
        {
            $this->trackerQuery->refinishQuery();
        }
 
        if (!$data)
        {
            return false;
        }
 
        return $data;
    }
 
    public function fetch(\Bitrix\Main\Text\Converter $converter = null)
    {
        $data = $this->fetchRaw();
 
        if (!$data)
        {
            return false;
        }
 
        if ($this->converters)
        {
            foreach ($this->converters as $field => $convertDataModifier)
            {
                $data[$field] = call_user_func_array($convertDataModifier, array($data[$field]));
            }
        }
 
        if ($this->serializedFields)
        {
            foreach ($this->serializedFields as $field)
            {
                if (isset($data[$field]))
                    $data[$field] = unserialize($data[$field]);
            }
        }
 
        if ($this->replacedAliases)
        {
            foreach ($this->replacedAliases as $tech => $human)
            {
                $data[$human] = $data[$tech];
                unset($data[$tech]);
            }
        }
 
        if ($this->fetchDataModifiers)
        {
            foreach ($this->fetchDataModifiers as $fetchDataModifier)
            {
                $result = call_user_func_array($fetchDataModifier, array(&$data));
 
                if (is_array($result))
                {
                    $data = $result;
                }
            }
        }
 
        if ($converter !== null)
        {
            foreach ($data as $key => $val)
            {
                $data[$key] = $converter->encode(
                    $val,
                    ($data[$key."_TYPE"] ?? \Bitrix\Main\Text\Converter::TEXT)
                );
            }
        }
 
        return $data;
    }
 
    public function fetchAll(\Bitrix\Main\Text\Converter $converter = null)
    {
        $res = array();
        while ($ar = $this->fetch($converter))
        {
            $res[] = $ar;
        }
        return $res;
    }
 
    abstract public function getFields();
 
    abstract public function getSelectedRowsCount();
 
    abstract protected function fetchRowInternal();
 
    public function getTrackerQuery()
    {
        return $this->trackerQuery;
    }
 
    public function getConverters()
    {
        return $this->converters;
    }
 
    public function setConverters($converters)
    {
        $this->converters = $converters;
    }
 
    public function setCount($n)
    {
        $this->count = (int)$n;
    }
 
    public function getCount()
    {
        if($this->count !== null)
        {
            return $this->count;
        }
        throw new \Bitrix\Main\ObjectPropertyException("count");
    }
 
    public function getIterator(): \Traversable
    {
        return new ResultIterator($this);
    }
}