Isswp101\Persimmon\DAL\ElasticsearchDAL

Complexity

Total Complexity

17

Size/Duplication

Total Lines	116
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	6

Test Coverage

Coverage

31.87%

Importance

Changes	8
Bugs	1	Features	0

7 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	6	1
A	getModel()	0	4	1
A	delete()	0	4	1
A	getParams()	0	10	2
A	get()	0	16	3
B	put()	0	24	4
B	search()	0	39	5

<?php

namespace Isswp101\Persimmon\DAL;

use Elasticsearch\Client;
use Illuminate\Support\Arr;
use Isswp101\Persimmon\Collection\ElasticsearchCollection;
use Isswp101\Persimmon\ElasticsearchModel;
use Psr\Log\LoggerInterface;

class ElasticsearchDAL implements IDAL
{
    protected $model;
    protected $client;
    protected $logger;

    public function __construct(ElasticsearchModel $model, Client $client, LoggerInterface $logger = null)
    {
        $this->model = $model;
        $this->client = $client;
        $this->logger = $logger;
    }

    public function getModel()
    {
        return $this->model;
    }

    public function get($id, array $options = [])
    {
        $params = $this->model->getPath()->toArray();

        if (!empty($options['columns'])) {
            $params['_source'] = $options['columns'];
        }

        if (!empty($options['parent'])) {
            $params['parent'] = $options['parent'];
        }

        $response = $this->client->get($params);

        return $this->model->fillByResponse($response);
    }

    public function put(array $columns = ['*'])
    {
        $params = $this->getParams();

        if (!$this->model->_exist || $columns == ['*']) {
            if (!$params['id']) {
                unset($params['id']);
            }

            $params['body'] = $this->model->toArray();

            $response = $this->client->index($params);
        } else {
            $params['body'] = [
                'doc' => array_only($this->model->toArray(), $columns)
            ];

            $response = $this->client->update($params);
        }

        $this->model->setId($response['_id']);

        return $this->model->getId();
    }

    public function delete()
    {
        return $this->client->delete($this->getParams());
    }

    protected function getParams()
    {
        $params = $this->model->getPath()->toArray();

        if ($this->model->getParentId()) {
            $params['parent'] = $this->model->getParentId();
        }

        return $params;
    }

    public function search(array $query)
    {
        if (empty($query['body']['query']) && empty($query['body']['filter'])) {
            $query['body']['query'] = [
                'match_all' => []
            ];
        }

        $params = [
            'index' => $this->model->getIndex(),
            'type' => $this->model->getType(),
            'from' => Arr::get($query, 'from', 0),
            'size' => Arr::get($query, 'size', 50),
            'body' => $query['body']
        ];

        if ($this->logger) {
            $this->logger->debug('Query', $params);
        }

        $collection = new ElasticsearchCollection();

        $response = $this->client->search($params);

        $collection->response($response);

        $from = (int)$params['from'];
        foreach ($response['hits']['hits'] as $hit) {
            $model = $this->model->createInstance();
            $model->_score = $hit['_score'];
            $model->_position = $from++;
            $model->_exist = true;
            $model->fillByResponse($hit);
            $model->fillByInnerHits($hit);
            $collection->put($model->getId(), $model);
        }

        return $collection;
    }
}