postgres/lib/PqBufferedResultSet.php

<?php

namespace Amp\Postgres;

use Amp\Promise;
use Amp\Success;
use pq;

class PqBufferedResultSet implements ResultSet {
    /** @var \pq\Result */
    private $result;

    /** @var int */
    private $position = 0;

    /** @var mixed Last row emitted. */
    private $currentRow;

    /** @var int Next row fetch type. */
    private $type = self::FETCH_ASSOC;

    /**
     * @param pq\Result $result PostgreSQL result object.
     */
    public function __construct(pq\Result $result) {
        $this->result = $result;
    }

    /**
     * {@inheritdoc}
     */
    public function advance(int $type = self::FETCH_ASSOC): Promise {
        $this->currentRow = null;
        $this->type = $type;

        if (++$this->position > $this->result->numRows) {
            return new Success(false);
        }

        return new Success(true);
    }

    /**
     * {@inheritdoc}
     */
    public function getCurrent() {
        if ($this->currentRow !== null) {
            return $this->currentRow;
        }

        if ($this->position > $this->result->numRows) {
            throw new \Error("No more rows remain in the result set");
        }

        switch ($this->type) {
            case self::FETCH_ASSOC:
                return $this->currentRow = $this->result->fetchRow(pq\Result::FETCH_ASSOC);
            case self::FETCH_ARRAY:
                return $this->currentRow = $this->result->fetchRow(pq\Result::FETCH_ARRAY);
            case self::FETCH_OBJECT:
                return $this->currentRow = $this->result->fetchRow(pq\Result::FETCH_OBJECT);
            default:
                throw new \Error("Invalid result fetch type");
        }
    }

    public function numRows(): int {
        return $this->result->numRows;
    }

    public function numFields(): int {
        return $this->result->numCols;
    }
}