Result.php

<?php

declare(strict_types=1);

namespace Doctrine\DBAL\Driver\OCI8;

use Doctrine\DBAL\Driver\FetchUtils;
use Doctrine\DBAL\Driver\OCI8\Exception\PHPError;
use Doctrine\DBAL\Driver\Result as ResultInterface;

use function oci_cancel;
use function oci_fetch_all;
use function oci_fetch_array;
use function oci_num_fields;
use function oci_num_rows;
use function preg_match_all;
use function restore_error_handler;
use function set_error_handler;

use const E_WARNING;
use const OCI_ASSOC;
use const OCI_FETCHSTATEMENT_BY_COLUMN;
use const OCI_FETCHSTATEMENT_BY_ROW;
use const OCI_NUM;
use const OCI_RETURN_LOBS;
use const OCI_RETURN_NULLS;

final class Result implements ResultInterface
{
    /** @var resource */
    private $statement;

    /**
     * @internal The result can be only instantiated by its driver connection or statement.
     *
     * @param resource $statement
     */
    public function __construct($statement)
    {
        $this->statement = $statement;
    }

    /**
     * {@inheritDoc}
     */
    public function fetchNumeric()
    {
        return $this->fetch(OCI_NUM);
    }

    /**
     * {@inheritDoc}
     */
    public function fetchAssociative()
    {
        return $this->fetch(OCI_ASSOC);
    }

    /**
     * {@inheritDoc}
     */
    public function fetchOne()
    {
        return FetchUtils::fetchOne($this);
    }

    /**
     * {@inheritDoc}
     */
    public function fetchAllNumeric(): array
    {
        return $this->fetchAll(OCI_NUM, OCI_FETCHSTATEMENT_BY_ROW);
    }

    /**
     * {@inheritDoc}
     */
    public function fetchAllAssociative(): array
    {
        return $this->fetchAll(OCI_ASSOC, OCI_FETCHSTATEMENT_BY_ROW);
    }

    /**
     * {@inheritDoc}
     */
    public function fetchFirstColumn(): array
    {
        return $this->fetchAll(OCI_NUM, OCI_FETCHSTATEMENT_BY_COLUMN)[0];
    }

    public function rowCount(): int
    {
        $count = oci_num_rows($this->statement);

        if ($count !== false) {
            return $count;
        }

        return 0;
    }

    public function columnCount(): int
    {
        $count = oci_num_fields($this->statement);

        if ($count !== false) {
            return $count;
        }

        return 0;
    }

    public function free(): void
    {
        oci_cancel($this->statement);
    }

    /**
     * @return mixed|false
     */
    private function fetch(int $mode)
    {
        set_error_handler(
            static function (int $errno, string $errstr, string $errfile, int $errline): bool {
                /*
                 * Screen PHP error messages for any error that the OCI8 extension
                 * doesn't report via oci_error()
                 *
                 * @link http://www.dba-oracle.com/t_error_code_list.htm
                 */
                if (preg_match_all('/ORA-(\d+)/', $errstr, $matches) === 0) {
                    // If no ORA error codes are present in error message,
                    // skip this error and bubble this error up to
                    // the normal error handler
                    return false;
                }

                foreach ($matches[1] as $oraErrorCode) {
                    switch ($oraErrorCode) {
                        case '04061':
                        case '04065':
                        case '04068':
                            throw PHPError::new(
                                'There was an error before all rows could be fetched.',
                                $errno,
                                $errstr,
                                $errfile,
                                $errline
                            );
                    }
                }

                // If ORA error codes in message do not match specified blocklist
                // ORA error codes, skip this error and bubble this error up to
                // the normal error handler
                return false;
            },
            E_WARNING
        );

        try {
            return oci_fetch_array(
                $this->statement,
                $mode | OCI_RETURN_NULLS | OCI_RETURN_LOBS
            );
        } finally {
            restore_error_handler();
        }
    }

    /**
     * @return array<mixed>
     */
    private function fetchAll(int $mode, int $fetchStructure): array
    {
        oci_fetch_all(
            $this->statement,
            $result,
            0,
            -1,
            $mode | OCI_RETURN_NULLS | $fetchStructure | OCI_RETURN_LOBS
        );

        return $result;
    }
}