Ytake\PrestoClient\ResultsSession

Complexity

Total Complexity

7

Size/Duplication

Total Lines	68
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	1

Importance

Changes

0

5 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	6	1
A	execute()	0	6	1
A	yieldResults()	0	7	2
A	getResults()	0	9	2
A	addResults()	0	4	1

<?php
declare(strict_types=1);

namespace Ytake\PrestoClient;

/**
 * Class ResultsSession
 */
class ResultsSession
{
    /** @var QueryResult[] */
    private $results = [];

    /** @var StatementClient */
    private $prestoClient;

    /** @var int */
    private $timeout = 500000;

    /** @var bool */
    private $debug = false;

    /**
     * @param StatementClient $prestoClient
     * @param int             $timeout
     * @param bool            $debug
     */
    public function __construct(StatementClient $prestoClient, int $timeout = 500000, bool $debug = false)
    {
        $this->prestoClient = $prestoClient;
        $this->timeout = $timeout;
        $this->debug = $debug;
    }

    /**
     * @return ResultsSession
     */
    public function execute(): ResultsSession
    {
        $this->prestoClient->execute($this->timeout, $this->debug);

        return $this;
    }

    /**
     * @return \Generator
     */
    public function yieldResults(): \Generator
    {
        while ($this->prestoClient->isValid()) {
            yield $this->prestoClient->current();
            $this->prestoClient->advance();
        }
    }

    /**
     * @return array
     */
    public function getResults(): array
    {
        while ($this->prestoClient->isValid()) {
            $this->addResults($this->prestoClient->current());
            $this->prestoClient->advance();
        }

        return $this->results;
    }

    /**
     * @param QueryResult $queryResult
     */
    private function addResults(QueryResult $queryResult)
    {
        $this->results[] = $queryResult;
    }
}