Maslosoft\Manganel\Adapters\Finder\ElasticSearchAdapter

Complexity

Total Complexity

11

Size/Duplication

Total Lines	77
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	3

Test Coverage

Coverage

97.06%

Importance

Changes

0

7 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	6	1
A	count()	0	5	1
A	findMany()	0	5	1
A	getQueryBuilder()	0	4	1
A	findOne()	0	10	2
A	prepare()	0	16	3
A	ensure()	0	8	2

<?php

namespace Maslosoft\Manganel\Adapters\Finder;

use Maslosoft\Mangan\Interfaces\Adapters\FinderAdapterInterface;
use Maslosoft\Mangan\Interfaces\CriteriaInterface;
use Maslosoft\Mangan\Interfaces\ModelAwareInterface;
use Maslosoft\Manganel\QueryBuilder;
use Maslosoft\Manganel\SearchCriteria;

/**
 * ElasticSearchAdapter
 *
 * @author Piotr Maselkowski <pmaselkowski at gmail.com>
 */
class ElasticSearchAdapter implements FinderAdapterInterface
{

	/**
	 *
	 * @var QueryBuilder
	 */
	private $qb = null;
	private $models = [];

	public function __construct($models)
	{
		$this->qb = new QueryBuilder();
		$this->qb->setModels($models);
		$this->models = $models;
	}

	public function count(CriteriaInterface $criteria)
	{
		$this->ensure($criteria);
		return $this->qb->setCriteria($criteria)->count();
	}

	public function findMany(CriteriaInterface $criteria, $fields = array())
	{
		$this->prepare($criteria, $fields);
		return new ElasticSearchCursor($this->qb);
	}

	public function findOne(CriteriaInterface $criteria, $fields = array())
	{
		$this->prepare($criteria, $fields);
		$data = (new ElasticSearchCursor($this->qb))->current();
		if (false === $data)
		{
			return null;
		}
		return $data;
	}

	/**
	 *
	 * @internal Used for debugging purposes, should not be used for any manipulations!
	 * @return QueryBuilder
	 */
	public function getQueryBuilder()
	{
		return $this->qb;
	}

	private function prepare(CriteriaInterface $criteria, $fields)
	{
		$this->ensure($criteria);
		assert($criteria instanceof SearchCriteria);

		$this->qb->setCriteria($criteria);
		if (!empty($fields))
		{
			$selected = array_flip($fields);
			foreach ($selected as $index => $value)
			{
				$selected[$index] = true;
			}
			$this->qb->getCriteria()->select($fields);
		}
	}

	private function ensure(&$criteria)
	{
		if (!$criteria instanceof SearchCriteria)
		{
			$criteria = new SearchCriteria($criteria);
		}
		$criteria->setModels($this->models);
	}

}