ArangoClient\Statement\Statement

Complexity

Total Complexity

24

Size/Duplication

Total Lines	232
Duplicated Lines	0 %

Test Coverage

Coverage

96.97%

Importance

Changes

0

14 Methods

Rating	Name	Duplication	Size	Complexity
A	prepareQueryBodyContent()	0	9	2
A	getIterator()	0	3	1
A	__construct()	0	10	1
A	handleQueryResults()	0	14	4
A	execute()	0	16	1
A	parse()	0	8	1
A	getQuery()	0	3	1
A	requestOutstandingResults()	0	12	2
A	setQuery()	0	5	1
A	fetchAll()	0	3	1
A	getWritesExecuted()	0	7	4
A	profile()	0	20	3
A	explain()	0	8	1
A	getCount()	0	3	1

<?php

namespace ArangoClient\Statement;

use ArangoClient\ArangoClient;
use ArangoClient\Exceptions\ArangoException;
use ArangoClient\Manager;
use ArrayIterator;
use IteratorAggregate;
use stdClass;

/**
 * Executes queries on ArangoDB
 *
 * @see https://www.arangodb.com/docs/stable/http/aql-query-cursor.html
 *
 *
 * @template-implements \IteratorAggregate<mixed>
 */
class Statement extends Manager implements IteratorAggregate
{
    protected ArangoClient $arangoClient;

    protected string $query;

    /**
     * @var array<scalar>|null
     */
    protected ?array $bindVars;

    /**
     * @var array<mixed>
     */
    protected array $options = [];

    /**
     * @var array<mixed>
     */
    protected array $results = [];

    protected ?stdClass $stats = null;

    /**
     * @var array<mixed>
     */
    protected array $warnings = [];

    protected ?int $cursorId = null;

    protected bool $cursorHasMoreResults = false;

    protected ?int $count = null;

    protected ?stdClass $extra = null;

    /**
     * Statement constructor.
     *
     * @param  ArangoClient  $arangoClient
     * @param  string  $query
     * @param  array<scalar>  $bindVars
     * @param  array<mixed>  $options
     */
    public function __construct(
        ArangoClient $arangoClient,
        string $query,
        array $bindVars = null,
        array $options = []
    ) {
        $this->arangoClient = $arangoClient;
        $this->query = $query;
        $this->bindVars = $bindVars;
        $this->options = $options;
    }

    /**
     * A statement can be used like an array to access the results.
     *
     * @return ArrayIterator<array-key, mixed>
     */
    public function getIterator(): ArrayIterator
    {
        return new ArrayIterator($this->results);
    }

    /**
     * @throws ArangoException
     */
    public function execute(): bool
    {
        $this->results = [];

        $bodyContent = $this->prepareQueryBodyContent();

        $options = [
            'body' => $bodyContent,
        ];
        $results = $this->arangoClient->transactionAwareRequest('post', '/_api/cursor', $options);

        $this->handleQueryResults($results);

        $this->requestOutstandingResults($bodyContent);

        return true;
    }

    /**
     * @return array<mixed>
     */
    protected function prepareQueryBodyContent(): array
    {
        $bodyContent = $this->options;
        $bodyContent['query'] = $this->query;
        if (! empty($this->bindVars)) {
            $bodyContent['bindVars'] = $this->bindVars;
        }

        return $bodyContent;
    }

    protected function handleQueryResults(stdClass $results): void
    {
        $this->results = array_merge($this->results, (array) $results->result);

        if (isset($results->extra)) {
            $this->extra = (object) $results->extra;
        }

        if (isset($results->count)) {
            $this->count = (int) $results->count;
        }

        $this->cursorHasMoreResults = (bool) $results->hasMore;
        $this->cursorId = $results->hasMore ? (int) $results->id : null;
    }

    /**
     * @param  array<mixed>  $body
     *
     * @throws ArangoException
     */
    protected function requestOutstandingResults(array $body): void
    {
        while ($this->cursorHasMoreResults) {
            $uri = '/_api/cursor/'.(string) $this->cursorId;

            $options = [
                'body' => $body,
            ];

            $results = $this->arangoClient->request('put', $uri, $options);

            $this->handleQueryResults($results);
        }
    }

    /**
     * @throws ArangoException
     */
    public function explain(): stdClass
    {
        $body = $this->prepareQueryBodyContent();
        $options = [
            'body' => $body,
        ];

        return $this->arangoClient->request('post', '/_api/explain', $options);
    }

    /**
     * Parse and validate the query, will through an ArangoException if the query is invalid.
     *
     * @throws ArangoException
     */
    public function parse(): stdClass
    {
        $body = $this->prepareQueryBodyContent();
        $options = [
            'body' => $body,
        ];

        return $this->arangoClient->request('post', '/_api/query', $options);
    }

    /**
     * Execute the query and return performance information on the query.
     *
     * @see https://www.arangodb.com/docs/3.7/aql/execution-and-performance-query-profiler.html
     *
     * @throws ArangoException
     */
    public function profile(int|bool $mode = 1): stdClass|null
    {
        $bodyContent = $this->prepareQueryBodyContent();

        if (! isset($bodyContent['options']) || ! is_array($bodyContent['options'])) {
            $bodyContent['options'] = [];
        }
        $bodyContent['options']['profile'] = $mode;

        $options = [
            'body' => $bodyContent,
        ];

        $results = $this->arangoClient->request('post', '/_api/cursor', $options);

        $this->handleQueryResults($results);

        $this->requestOutstandingResults($bodyContent);

        return $this->extra;
    }

    public function setQuery(string $query): self
    {
        $this->query = $query;

        return $this;
    }

    public function getQuery(): string
    {
        return $this->query;
    }

    /**
     * Fetch all results.
     *
     * @return array<mixed>
     */
    public function fetchAll(): array
    {
        return $this->results;
    }

    /**
     * Return the total number of results. (not just the retrieved results)
     * Useful if not all results have been retrieved from the database yet.
     */
    public function getCount(): ?int
    {
        return $this->count;
    }

    public function getWritesExecuted(): int
    {
        if (! isset($this->extra->stats) || ! is_object($this->extra->stats)) {
            return 0;
        }

        return (isset($this->extra->stats->writesExecuted)) ? (int) $this->extra->stats->writesExecuted : 0;
    }
}