Nayjest\Querying\AbstractQuery

Complexity

Total Complexity

22

Size/Duplication

Total Lines	141
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	4

Test Coverage

Coverage

82.54%

Importance

Changes

0

14 Methods

Rating	Name	Duplication	Size	Complexity
	getHandlerFactory()	0	1	?
A	__construct()	0	7	1
A	getIterator()	0	9	2
A	initialize()	0	4	1
A	execute()	0	7	2
A	getHandlers()	0	17	4
A	get()	0	4	1
A	getArray()	0	5	2
A	getRaw()	0	9	2
A	addOperation()	0	5	1
A	addOperations()	0	7	2
A	hasOperation()	0	4	1
A	removeOperation()	0	10	2
A	getSrc()	0	4	1

<?php

namespace Nayjest\Querying;

use ArrayIterator;
use Exception;
use IteratorAggregate;
use Nayjest\Querying\Handler\AbstractFactory;
use Nayjest\Querying\Handler\HandlerInterface;
use Nayjest\Querying\Operation\HydrateOperation;
use Nayjest\Querying\Operation\OperationInterface;
use Nayjest\Querying\Row\DynamicFieldsRegistry;
use Nayjest\Querying\Row\RowInterface;

abstract class AbstractQuery implements IteratorAggregate
{
    private $dataSource;
    /**
     * @var OperationInterface[]
     */
    private $operations;

    private $dynamicFields;

    /**
     * @return AbstractFactory
     */
    abstract protected function getHandlerFactory();

    /**
     * AbstractQuery constructor.
     * @param $dataSource
     * @param string $rowClass
     */
    public function __construct($dataSource, $rowClass)
    {
        $this->dataSource = $dataSource;
        $this->dynamicFields = new DynamicFieldsRegistry();
        $this->operations = [new HydrateOperation($rowClass, $this->dynamicFields)];
        $this->initialize();
    }

    public function getIterator()
    {
        $data = $this->get();
        if (is_array($data)) {
            return new ArrayIterator($data);
        } else {
            return $data;
        }
    }

    protected function initialize()
    {

    }

    protected function execute($data)
    {
        foreach ($this->getHandlers() as $handler) {
            $data = $handler->apply($data);
        }
        return $data;
    }

    /**
     * @return HandlerInterface[]
     */
    protected function getHandlers()
    {
        $handlers = [];
        $factory = $this->getHandlerFactory();
        foreach ($this->operations as $operation) {
            $handlers[] = $factory->makeHandler($operation);
        }
        usort($handlers, function (HandlerInterface $a, HandlerInterface $b) {
            $aPos = $a->getPriority();
            $bPos = $b->getPriority();
            if ($aPos == $bPos) {
                return 0;
            }
            return ($aPos < $bPos) ? -1 : 1;
        });
        return $handlers;
    }

    /**
     * @return RowInterface[]|\Traversable
     */
    public function get()
    {
        return $this->execute($this->dataSource);
    }

    /**
     * @return RowInterface[]
     */
    public function getArray()
    {
        $data = $this->get();
        return is_array($data) ? $data : iterator_to_array($data);
    }

    /**
     * @return array
     */
    public function getRaw()
    {
        $data = $this->get();
        $res = [];
        foreach ($data as $row) {
            $res[] = $row->getSrc();
        }
        return $res;
    }

    /**
     * @param OperationInterface $operation
     * @return $this
     */
    public function addOperation(OperationInterface $operation)
    {
        $this->operations[] = $operation;
        return $this;
    }

    public function addOperations(array $operations)
    {
        foreach ($operations as $operation) {
            $this->addOperation($operation);
        }
        return $this;
    }

    public function hasOperation(OperationInterface $operation)
    {
        return in_array($operation, $this->operations);
    }

    public function removeOperation(OperationInterface $operation)
    {
        if ($this->hasOperation($operation)) {
            $key = array_search($operation, $this->operations);
            unset($this->operations[$key]);
        } else {
            throw new Exception("Trying to remove unexisting operation from query");
        }
        return $this;
    }

    public function getSrc()
    {
        return $this->dataSource;
    }
}