ArekX\PQL\QueryResultBuilder

Complexity

Total Complexity

17

Size/Duplication

Total Lines	165
Duplicated Lines	0 %

Test Coverage

Coverage

100%

Importance

Changes	1
Bugs	0	Features	0

14 Methods

Rating	Name	Duplication	Size	Complexity
A	pipeFilter()	0	4	1
A	pipeIndexBy()	0	5	1
A	first()	0	5	1
A	scalar()	0	5	1
A	pipeListBy()	0	12	3
A	clearPipeline()	0	4	1
A	exists()	0	5	1
A	result()	0	9	2
A	pipeSort()	0	8	1
A	all()	0	5	1
A	pipeReduce()	0	4	1
A	column()	0	5	1
A	useReader()	0	4	1
A	pipeMap()	0	4	1

<?php
/**
 * Copyright 2021 Aleksandar Panic
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

namespace ArekX\PQL;

use ArekX\PQL\Contracts\ResultBuilder;
use ArekX\PQL\Contracts\ResultReader;

/**
 * Result builder for query.
 */
class QueryResultBuilder implements ResultBuilder
{
    /**
     * Result reader which is used to read all results from.
     *
     * @var ResultReader
     */
    protected ResultReader $reader;

    /**
     * List of pipeline methods to apply when calling result()
     *
     * @see QueryResultBuilder::result()
     * @var array
     */
    protected $pipelines = [];

    /**
     * @inheritDoc
     */
    public function useReader(ResultReader $reader): ResultBuilder
    {
        $this->reader = $reader;
        return $this;
    }

    /**
     * @inheritDoc
     */
    public function first()
    {
        $result = $this->reader->getNextRow();
        $this->reader->finalize();
        return $result;
    }

    /**
     * @inheritDoc
     */
    public function scalar($index = 0)
    {
        $result = $this->reader->getNextColumn($index);
        $this->reader->finalize();
        return $result;
    }

    /**
     * @inheritDoc
     */
    public function column($index = 0): array
    {
        $result = $this->reader->getAllColumns($index);
        $this->reader->finalize();
        return $result;
    }

    /**
     * @inheritDoc
     */
    public function exists(): bool
    {
        $result = $this->reader->getNextColumn();
        $this->reader->finalize();
        return $result !== null;
    }

    /**
     * @inheritDoc
     */
    public function clearPipeline(): ResultBuilder
    {
        $this->pipelines = [];
        return $this;
    }

    /**
     * @inheritDoc
     */
    public function pipeIndexBy(string $column): ResultBuilder
    {
        return $this->pipeReduce(function ($result, $row) use ($column) {
            $result[$row[$column]] = $row;
            return $result;
        });
    }

    /**
     * @inheritDoc
     */
    public function pipeReduce(callable $reducer, $initialValue = null): ResultBuilder
    {
        $this->pipelines[] = fn($results) => array_reduce($results, $reducer, $initialValue);
        return $this;
    }

    /**
     * @inheritDoc
     */
    public function pipeSort(callable $sorter): ResultBuilder
    {
        $this->pipelines[] = function ($results) use ($sorter) {
            usort($results, $sorter);
            return $results;
        };

        return $this;
    }

    /**
     * @inheritDoc
     */
    public function pipeMap(callable $mapper): ResultBuilder
    {
        $this->pipelines[] = fn($results) => array_map($mapper, $results);
        return $this;
    }

    /**
     * @inheritDoc
     */
    public function pipeFilter(callable $filter): ResultBuilder
    {
        $this->pipelines[] = fn($results) => array_values(array_filter($results, $filter));
        return $this;
    }

    /**
     * @inheritDoc
     */
    public function result()
    {
        $result = $this->all();

        foreach ($this->pipelines as $pipelineMethod) {
            $result = $pipelineMethod($result, $this);
        }

        return $result;
    }

    /**
     * @inheritDoc
     */
    public function all(): array
    {
        $result = $this->reader->getAllRows();
        $this->reader->finalize();
        return $result;
    }

    /**
     * @inheritDoc
     */
    public function pipeListBy(string $keyColumn, string $valueColumn): ResultBuilder
    {
        return $this->pipeReduce(function ($previous, $row) use ($keyColumn, $valueColumn) {

            if (!array_key_exists($keyColumn, $row) || !array_key_exists($valueColumn, $row)) {
                throw new \Exception('$keyColumn and $valueColumn must exist in the result.');
            }

            $previous[$row[$keyColumn]] = $row[$valueColumn];

            return $previous;
        }, []);
    }
}