TNTSearchEngine::modelCollection() - Code Metrics - Inspection of "Fixes the results total count." - teamtnt/laravel-scout-tntsearch-driver - Measure and Improve Code Quality continuously with Scrutinizer

Completed

Pull Request — master (#128)

unknown

created 2017-12-12 11:54 UTC

TNTSearchEngine::modelCollection() B

↳ Parent: TNTSearchEngine

Complexity

Conditions	5
Paths	2

Size

Total Lines	22
Code Lines	15

Duplication

Lines	0
Ratio	0 %

Importance

Changes

Metric	Value
dl	0
loc	22
rs	8.6737
c	0
b	0
f	0
cc	5
eloc	15
nc	2
nop	1

<?php

namespace TeamTNT\Scout\Engines;

use Illuminate\Database\Eloquent\Collection;
use Laravel\Scout\Builder;
use Laravel\Scout\Engines\Engine;
use TeamTNT\TNTSearch\TNTSearch;
use TeamTNT\TNTSearch\Exceptions\IndexNotFoundException;

class TNTSearchEngine extends Engine
{
    const NO_RESULTS_COUNT = 0;

    /**
     * @var TNTSearch
     */
    protected $tnt;

    /**
     * @var Builder
     */
    protected $builder;

    protected $modelCollection;

    /**
     * Create a new engine instance.
     *
     * @param TNTSearch $tnt
     */
    public function __construct(TNTSearch $tnt)
    {
        $this->tnt = $tnt;
    }

    /**
     * Update the given model in the index.
     *
     * @param Collection $models
     *
     * @return void
     */
    public function update($models)
    {
        $this->initIndex($models->first());
        $this->tnt->selectIndex("{$models->first()->searchableAs()}.index");
        $index = $this->tnt->getIndex();
        $index->setPrimaryKey($models->first()->getKeyName());

        $index->indexBeginTransaction();
        $models->each(function ($model) use ($index) {
            $array = $model->toSearchableArray();

            if (empty($array)) {
                return;
            }

            if ($model->getKey()) {
                $index->update($model->getKey(), $array);
            } else {
                $index->insert($array);
            }
        });
        $index->indexEndTransaction();
    }

    /**
     * Remove the given model from the index.
     *
     * @param Collection $models
     *
     * @return void
     */
    public function delete($models)
    {
        $this->initIndex($models->first());
        $models->each(function ($model) {
            $this->tnt->selectIndex("{$model->searchableAs()}.index");
            $index = $this->tnt->getIndex();
            $index->setPrimaryKey($model->getKeyName());
            $index->delete($model->getKey());
        });
    }

    /**
     * Perform the given search on the engine.
     *
     * @param Builder $builder
     *
     * @return mixed
     */
    public function search(Builder $builder)
    {
        try {
            return $this->performSearch($builder);
        } catch (IndexNotFoundException $e) {
            $this->initIndex($builder->model);
        }
    }

    /**
     * Perform the given search on the engine.
     *
     * @param Builder $builder
     * @param int     $perPage
     * @param int     $page
     *
     * @return mixed
     */
    public function paginate(Builder $builder, $perPage, $page)
    {
        $results = $this->performSearch($builder);

        if ($builder->limit) {
            $results['hits'] = $builder->limit;
        }

        $chunks = array_chunk($results['ids'], $perPage);

        if (!empty($chunks)) {
            if (array_key_exists($page - 1, $chunks)) {
                $results['ids'] = $chunks[$page - 1];
            } else {
                $results['ids'] = end($chunks);
            }
        }

        return $results;
    }

    /**
     * Perform the given search on the engine.
     *
     * @param Builder $builder
     *
     * @return mixed
     */
    protected function performSearch(Builder $builder, array $options = [])
    {
        $index = $builder->index ?: $builder->model->searchableAs();
        $limit = $builder->limit ?: 10000;
        $this->tnt->selectIndex("{$index}.index");

        $this->builder = $builder;
        $this->tnt->asYouType = $builder->model->asYouType ?: false;

        if ($builder->callback) {
            return call_user_func(
                $builder->callback,
                $this->tnt,
                $builder->query,
                $options
            );
        }
        if (isset($this->tnt->config['searchBoolean']) ? $this->tnt->config['searchBoolean'] : false) {
            return $this->tnt->searchBoolean($builder->query, $limit);
        } else {
            return $this->tnt->search($builder->query, $limit);
        }
    }

    /**
     * Get the filter array for the query.
     *
     * @param Builder $builder
     *
     * @return array
     */
    protected function filters(Builder $builder)
    {
        return collect($builder->wheres)->map(function ($value, $key) {
            return $key.'='.$value;
        })->values()->all();
    }

    /**
     * Map the given results to instances of the given model.
     *
     * @param mixed                               $results
     * @param \Illuminate\Database\Eloquent\Model $model
     *
     * @return Collection
     */
    public function map($results, $model)
    {
        if (count($results['ids']) === 0) {
            return Collection::make();
        }

        return $this->modelCollection($results);
    }

    /**
     * Pluck and return the primary keys of the given results.
     *
     * @param mixed $results
     * @return \Illuminate\Support\Collection
     */
    public function mapIds($results)
    {
        return collect($results['ids'])->values();
    }

    /**
     * Get the total count from a raw result returned by the engine.
     *
     * @param mixed $results
     *
     * @return int
     */
    public function getTotalCount($results)
    {
        if (count($results['ids']) === 0) {
            return self::NO_RESULTS_COUNT;
        }

        return $this->modelCollection($results)->count();
    }

    public function initIndex($model)
    {
        $indexName = $model->searchableAs();

        if (!file_exists($this->tnt->config['storage']."/{$indexName}.index")) {
            $indexer = $this->tnt->createIndex("$indexName.index");
            $indexer->setDatabaseHandle($model->getConnection()->getPdo());
            $indexer->setPrimaryKey($model->getKeyName());
        }
    }

    protected function modelCollection($results)
    {
        if (! is_null($this->modelCollection)) {
            return $this->modelCollection;
        }

        $keys   = collect($results['ids'])->values()->all();
        $fieldsWheres = array_keys($this->builder->wheres);
        $models = $this->builder->model->whereIn(
            $this->builder->model->getQualifiedKeyName(), $keys
        )->get()->keyBy($this->builder->model->getKeyName());

        $this->modelCollection = collect($results['ids'])->map(function ($hit) use ($models) {
            return $models->has($hit) ? $models[$hit] : null;
        })->filter(function ($model) use ($fieldsWheres) {
            return !is_null($model) && array_reduce($fieldsWheres, function ($carry, $item) use($model) {
                    return $carry && $model[$item] == $this->builder->wheres[$item];
                }, true);
        });

        return $this->modelCollection;
    }
}


1			<?php
2
3			namespace TeamTNT\Scout\Engines;
4
5			use Illuminate\Database\Eloquent\Collection;
6			use Laravel\Scout\Builder;
7			use Laravel\Scout\Engines\Engine;
8			use TeamTNT\TNTSearch\TNTSearch;
9			use TeamTNT\TNTSearch\Exceptions\IndexNotFoundException;
10
11			class TNTSearchEngine extends Engine
12			{
13			const NO_RESULTS_COUNT = 0;
14
15			/**
16			* @var TNTSearch
17			*/
18			protected $tnt;
19
20			/**
21			* @var Builder
22			*/
23			protected $builder;
24
25			protected $modelCollection;
26
27			/**
28			* Create a new engine instance.
29			*
30			* @param TNTSearch $tnt
31			*/
32			public function __construct(TNTSearch $tnt)
33			{
34			$this->tnt = $tnt;
35			}
36
37			/**
38			* Update the given model in the index.
39			*
40			* @param Collection $models
41			*
42			* @return void
43			*/
44			public function update($models)
45			{
46			$this->initIndex($models->first());
47			$this->tnt->selectIndex("{$models->first()->searchableAs()}.index");
48			$index = $this->tnt->getIndex();
49			$index->setPrimaryKey($models->first()->getKeyName());
50
51			$index->indexBeginTransaction();
52			$models->each(function ($model) use ($index) {
53			$array = $model->toSearchableArray();
54
55			if (empty($array)) {
56			return;
57			}
58
59			if ($model->getKey()) {
60			$index->update($model->getKey(), $array);
61			} else {
62			$index->insert($array);
63			}
64			});
65			$index->indexEndTransaction();
66			}
67
68			/**
69			* Remove the given model from the index.
70			*
71			* @param Collection $models
72			*
73			* @return void
74			*/
75			public function delete($models)
76			{
77			$this->initIndex($models->first());
78			$models->each(function ($model) {
79			$this->tnt->selectIndex("{$model->searchableAs()}.index");
80			$index = $this->tnt->getIndex();
81			$index->setPrimaryKey($model->getKeyName());
82			$index->delete($model->getKey());
83			});
84			}
85
86			/**
87			* Perform the given search on the engine.
88			*
89			* @param Builder $builder
90			*
91			* @return mixed
92			*/
93			public function search(Builder $builder)
94			{
95			try {
96			return $this->performSearch($builder);
97			} catch (IndexNotFoundException $e) {
98			$this->initIndex($builder->model);
99			}
100			}
101
102			/**
103			* Perform the given search on the engine.
104			*
105			* @param Builder $builder
106			* @param int $perPage
107			* @param int $page
108			*
109			* @return mixed
110			*/
111			public function paginate(Builder $builder, $perPage, $page)
112			{
113			$results = $this->performSearch($builder);
114
115			if ($builder->limit) {
116			$results['hits'] = $builder->limit;
117			}
118
119			$chunks = array_chunk($results['ids'], $perPage);
120
121			if (!empty($chunks)) {
122			if (array_key_exists($page - 1, $chunks)) {
123			$results['ids'] = $chunks[$page - 1];
124			} else {
125			$results['ids'] = end($chunks);
126			}
127			}
128
129			return $results;
130			}
131
132			/**
133			* Perform the given search on the engine.
134			*
135			* @param Builder $builder
136			*
137			* @return mixed
138			*/
139			protected function performSearch(Builder $builder, array $options = [])
140			{
141			$index = $builder->index ?: $builder->model->searchableAs();
142			$limit = $builder->limit ?: 10000;
143			$this->tnt->selectIndex("{$index}.index");
144
145			$this->builder = $builder;
146			$this->tnt->asYouType = $builder->model->asYouType ?: false;
147
148			if ($builder->callback) {
149			return call_user_func(
150			$builder->callback,
151			$this->tnt,
152			$builder->query,
153			$options
154			);
155			}
156			if (isset($this->tnt->config['searchBoolean']) ? $this->tnt->config['searchBoolean'] : false) {
157			return $this->tnt->searchBoolean($builder->query, $limit);
158			} else {
159			return $this->tnt->search($builder->query, $limit);
160			}
161			}
162
163			/**
164			* Get the filter array for the query.
165			*
166			* @param Builder $builder
167			*
168			* @return array
169			*/
170			protected function filters(Builder $builder)
171			{
172			return collect($builder->wheres)->map(function ($value, $key) {
173			return $key.'='.$value;
174			})->values()->all();
175			}
176
177			/**
178			* Map the given results to instances of the given model.
179			*
180			* @param mixed $results
181			* @param \Illuminate\Database\Eloquent\Model $model
182			*
183			* @return Collection
184			*/
185			public function map($results, $model)
186			{
187			if (count($results['ids']) === 0) {
188			return Collection::make();
189			}
190
191			return $this->modelCollection($results);
192			}
193
194			/**
195			* Pluck and return the primary keys of the given results.
196			*
197			* @param mixed $results
198			* @return \Illuminate\Support\Collection
199			*/
200			public function mapIds($results)
201			{
202			return collect($results['ids'])->values();
203			}
204
205			/**
206			* Get the total count from a raw result returned by the engine.
207			*
208			* @param mixed $results
209			*
210			* @return int
211			*/
212			public function getTotalCount($results)
213			{
214			if (count($results['ids']) === 0) {
215			return self::NO_RESULTS_COUNT;
216			}
217
218			return $this->modelCollection($results)->count();
219			}
220
221			public function initIndex($model)
222			{
223			$indexName = $model->searchableAs();
224
225			if (!file_exists($this->tnt->config['storage']."/{$indexName}.index")) {
226			$indexer = $this->tnt->createIndex("$indexName.index");
227			$indexer->setDatabaseHandle($model->getConnection()->getPdo());
228			$indexer->setPrimaryKey($model->getKeyName());
229			}
230			}
231
232			protected function modelCollection($results)
233			{
234			if (! is_null($this->modelCollection)) {
235			return $this->modelCollection;
236			}
237
238			$keys = collect($results['ids'])->values()->all();
239			$fieldsWheres = array_keys($this->builder->wheres);
240			$models = $this->builder->model->whereIn(
241			$this->builder->model->getQualifiedKeyName(), $keys
242			)->get()->keyBy($this->builder->model->getKeyName());
243
244			$this->modelCollection = collect($results['ids'])->map(function ($hit) use ($models) {
245			return $models->has($hit) ? $models[$hit] : null;
246			})->filter(function ($model) use ($fieldsWheres) {
247			return !is_null($model) && array_reduce($fieldsWheres, function ($carry, $item) use($model) {
248			return $carry && $model[$item] == $this->builder->wheres[$item];
249			}, true);
250			});
251
252			return $this->modelCollection;
253			}
254			}
255

teamtnt / laravel-scout-tntsearch-driver

Pull Request — master (#128)

TNTSearchEngine::modelCollection() B

Complexity

Size

Duplication

Importance

Duplication Side-by-Side

Filter issues like