TBolier\RethinkQL\Response\Cursor

Complexity

Total Complexity

19

Size/Duplication

Total Lines	170
Duplicated Lines	0 %

Test Coverage

Coverage

41.51%

Importance

Changes

0

11 Methods

Rating	Name	Duplication	Size	Complexity
A	next()	0	5	1
A	__construct()	0	10	1
A	key()	0	3	1
A	current()	0	13	3
A	rewind()	0	5	1
A	valid()	0	3	2
A	addResponse()	0	6	2
A	request()	0	10	2
A	seek()	0	8	3
A	close()	0	10	2
A	count()	0	3	1

<?php
declare(strict_types = 1);

namespace TBolier\RethinkQL\Response;

use TBolier\RethinkQL\Connection\ConnectionCursorInterface;
use TBolier\RethinkQL\Connection\ConnectionException;
use TBolier\RethinkQL\Message\MessageInterface;
use TBolier\RethinkQL\Types\Response\ResponseType;

class Cursor implements \Iterator, \Countable
{
    /**
     * @var ConnectionCursorInterface
     */
    private $connection;
    /**
     * @var int
     */
    private $index;
    /**
     * @var bool
     */
    private $isComplete;
    /**
     * @var MessageInterface
     */
    private $message;
    /**
     * @var ResponseInterface
     */
    private $response;
    /**
     * @var int
     */
    private $size;
    /**
     * @var int
     */
    private $token;

    /**
     * @param ConnectionCursorInterface $connection
     * @param int $token
     * @param ResponseInterface $response
     * @param MessageInterface $message
     */
    public function __construct(
        ConnectionCursorInterface $connection,
        int $token,
        ResponseInterface $response,
        MessageInterface $message
    ) {
        $this->connection = $connection;
        $this->token = $token;
        $this->addResponse($response);
        $this->message = $message;
    }

    /**
     * @inheritdoc
     * @throws \Exception
     */
    public function current()
    {
        $this->seek();

        if (!$this->valid()) {
            return;
        }

        if ($this->response->isAtomic()) {
            return $this->response->getData();
        }

        return $this->response->getData()[$this->index];
    }

    /**
     * @inheritdoc
     * @throws \Exception
     */
    public function next(): void
    {
        $this->index++;

        $this->seek();
    }

    /**
     * @inheritdoc
     */
    public function key(): int
    {
        return $this->index;
    }

    /**
     * @inheritdoc
     */
    public function valid(): bool
    {
        return (!$this->isComplete || ($this->index < $this->size));
    }

    /**
     * @inheritdoc
     * @throws ConnectionException
     */
    public function rewind(): void
    {
        $this->close();

        $this->addResponse($this->connection->rewindFromCursor($this->message));
    }

    /**
     * @inheritdoc
     */
    public function count()
    {
        return $this->size;
    }

    /**
     * @param ResponseInterface $response
     */
    private function addResponse(ResponseInterface $response)
    {
        $this->index = 0;
        $this->isComplete = $response->getType() === ResponseType::SUCCESS_SEQUENCE;
        $this->size = $response->isAtomic() ? 1 : \count($response->getData());
        $this->response = $response;
    }

    /**
     * @return void
     * @throws \Exception
     */
    private function seek(): void
    {
        while ($this->index === $this->size) {
            if ($this->isComplete) {
                return;
            }

            $this->request();
        }
    }

    /**
     * @return void
     * @throws \Exception
     */
    private function request(): void
    {
        try {
            $response = $this->connection->continueQuery($this->token);
            $this->addResponse($response);
        } catch (\Exception $e) {
            $this->isComplete = true;
            $this->close();

            throw $e;
        }
    }

    /**
     * @return void
     */
    private function close(): void
    {
        if (!$this->isComplete) {
            $this->connection->stopQuery($this->token);
            $this->isComplete = true;
        }

        $this->index = 0;
        $this->size = 0;
        $this->response = null;
    }
}