Bardex\Elastic\Query

Complexity

Total Complexity

22

Size/Duplication

Total Lines	167
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	1

Importance

Changes

0

13 Methods

Rating	Name	Duplication	Size	Complexity
	getQuery()	0	1	?
	executeQuery()	0	1	?
A	__construct()	0	4	1
A	fetchAll()	0	6	1
A	fetchRaw()	0	19	2
A	createSearchResult()	0	7	1
B	extractDocuments()	0	20	6
A	extractTotal()	0	8	2
A	addListener()	0	5	1
A	removeListener()	0	9	3
A	triggerSuccess()	0	6	2
A	triggerError()	0	6	2
A	jsonSerialize()	0	4	1

<?php

namespace Bardex\Elastic;


use Elasticsearch\Client as ElasticClient;
use Psr\Log\LoggerInterface;
use Psr\Log\NullLogger;

/**
 * Class Query
 * @package Bardex\Elastic
 * @author Alexey Sumin <bardex@ya.ru>
 */
abstract class Query implements \JsonSerializable
{
    /**
     * @var ElasticClient $client
     */
    protected $elastic;

    /**
     * @var array
     */
    protected $listeners = [];

    /**
     * Получить собранный elasticsearch-запрос
     * @return array
     */
    abstract public function getQuery();


    /**
     * Отправить запрос на конкретный endpoint elasticsearch-сервера
     * @param array $query
     * @return array
     */
    abstract protected function executeQuery(array $query);


    /**
     * Конструктор
     * @param ElasticClient $elastic
     */
    public function __construct(ElasticClient $elastic)
    {
        $this->elastic = $elastic;
    }


    /**
     * Выполнить запрос к ES и вернуть результаты поиска.
     * @return SearchResult - возвращает набор документов
     */
    public function fetchAll()
    {
        $response = $this->fetchRaw();
        $result = $this->createSearchResult($response);
        return $result;
    }


    /**
     * Выполнить запрос к ES и вернуть необработанный результат (с мета-данными).
     * @return array возвращает необработанный ответ ES
     */
    public function fetchRaw()
    {
        // build query
        $query = $this->getQuery();

        // send query to elastic
        $start  = microtime(1);

        try {
            $result = $this->executeQuery($query);
            $time   = round((microtime(1) - $start) * 1000);
            $this->triggerSuccess($query, $result, $time);
            return $result;
        }
        catch (\Exception $e) {
            $this->triggerError($query, $e);
            throw $e;
        }
    }


    /**
     * Создать из ответа ES-сервера экземляр SearchResult
     * @param array $response
     * @return SearchResult
     */
    protected function createSearchResult(array $response)
    {
        $results  = $this->extractDocuments($response);
        $total    = $this->extractTotal($response);
        $searchResult = new SearchResult($results, $total);
        return $searchResult;
    }

    /**
     * Выбрать документы из ответа ES-сервера и добавить script fields.
     * @param array $response - ответ ES сервера.
     * @return array - возвращает набор документов
     */
    protected function extractDocuments(array $response)
    {
        $results = [];
        if (isset($response['hits']['hits'])) {
            foreach ($response['hits']['hits'] as $hit) {
                $row = $hit['_source'];
                if (isset($hit['fields'])) { // script fields
                    foreach ($hit['fields'] as $field => $data) {
                        if (count($data) == 1) {
                            $row[$field] = array_shift($data);
                        } else {
                            $row[$field] = $data;
                        }
                    }
                }
                $results[] = $row;
            }
        }
        return $results;
    }


    /**
     * Выбрать из ответа ES-сервера количество найденных документов.
     * @param array $response - ответ ES сервера.
     * @return integer - возвращает количество найденных документов.
     */
    protected function extractTotal(array $response)
    {
        $total = 0;
        if (isset($response['hits']['total'])) {
            $total = $response['hits']['total'];
        }
        return $total;
    }

    public function addListener(IListener $listener)
    {
        $this->listeners[] = $listener;
        return $this;
    }

    public function removeListener(IListener $listener)
    {
        foreach ($this->listeners as $i => $listItem) {
            if ($listener === $listItem) {
                unset($this->listeners[$i]);
            }
        }
        return $this;
    }

    protected function triggerSuccess(array $query, array $response, $time)
    {
        foreach ($this->listeners as $listener) {
            $listener->onSuccess($query, $response, $time);
        }
    }

    protected function triggerError(array $query, \Exception $e)
    {
        foreach ($this->listeners as $listener) {
            $listener->onError($query, $e);
        }
    }

    /**
     * Имплементация \JsonSerializable
     * @return array
     */
    public function jsonSerialize()
    {
        return $this->getQuery();
    }
}