119 lines
3.7 KiB
PHP
119 lines
3.7 KiB
PHP
<?php
|
|
|
|
namespace App\Overrides\Matchish\ScoutElasticSearch\ElasticSearch;
|
|
|
|
use Illuminate\Database\Eloquent\Model;
|
|
use IteratorAggregate;
|
|
use Laravel\Scout\Builder;
|
|
use Laravel\Scout\Searchable;
|
|
use Traversable;
|
|
|
|
/**
|
|
* @internal
|
|
*/
|
|
final class EloquentHitsIteratorAggregate implements IteratorAggregate
|
|
{
|
|
/**
|
|
* @var array
|
|
*/
|
|
private $results;
|
|
/**
|
|
* @var callable|null
|
|
*/
|
|
private $callback;
|
|
|
|
/**
|
|
* @param array $results
|
|
* @param callable|null $callback
|
|
*/
|
|
public function __construct(array $results, callable $callback = null)
|
|
{
|
|
$this->results = $results;
|
|
$this->callback = $callback;
|
|
}
|
|
|
|
/**
|
|
* Retrieve an external iterator.
|
|
*
|
|
* @link https://php.net/manual/en/iteratoraggregate.getiterator.php
|
|
*
|
|
* @return Traversable An instance of an object implementing <b>Iterator</b> or
|
|
* <b>Traversable</b>
|
|
*
|
|
* @since 5.0.0
|
|
*/
|
|
public function getIterator(): Traversable
|
|
{
|
|
|
|
$hits = collect();
|
|
if ($this->results['hits']['total']) {
|
|
$raw = collect($this->results['hits']['hits']);
|
|
$models = $this->collectModels($raw);
|
|
$eloquentHits = $this->getEloquentHits($raw, $models);
|
|
$hits = $this->mergeHighlightsIntoModels($eloquentHits, $raw);
|
|
// $hits2 = $this->mergeCustomFieldsIntoModels($eloquentHits, $raw);
|
|
}
|
|
return new \ArrayIterator($hits);
|
|
}
|
|
|
|
private function collectModels($rawHits)
|
|
{
|
|
return collect($rawHits)->groupBy('_source.__class_name')
|
|
->map(function ($results, $class) {
|
|
$model = new $class();
|
|
$builder = new Builder($model, '');
|
|
if (! empty($this->callback)) {
|
|
$builder->query($this->callback);
|
|
}
|
|
/* @var Searchable $model */
|
|
return $models = $model->getScoutModelsByIds(
|
|
$builder,
|
|
$results->pluck('_id')->all()
|
|
);
|
|
})
|
|
->flatten()->keyBy(function ($model) {
|
|
return get_class($model).'::'.$model->getScoutKey();
|
|
});
|
|
}
|
|
|
|
private function getEloquentHits($hits, $models)
|
|
{
|
|
return collect($hits)->map(function ($hit) use ($models) {
|
|
$key = $hit['_source']['__class_name'].'::'.$hit['_id'];
|
|
return isset($models[$key]) ? $models[$key] : null;
|
|
})->filter()->all();
|
|
}
|
|
|
|
private function mergeHighlightsIntoModels($eloquentHits, $raw)
|
|
{
|
|
return collect($eloquentHits)->map(function (Model $eloquentHit) use ($raw) {
|
|
$raw = collect($raw)
|
|
->where('_source.__class_name', get_class($eloquentHit))
|
|
->where('_source.id', $eloquentHit->id)
|
|
->first();
|
|
|
|
foreach($raw['highlight'] ?? [] as $key => $highlight) {
|
|
$eloquentHit->setAttribute($key, $highlight[0]);
|
|
}
|
|
|
|
return $eloquentHit;
|
|
})->all();
|
|
}
|
|
|
|
// private function mergeCustomFieldsIntoModels($eloquentHits, $raw){
|
|
// return collect($eloquentHits)->map(function(Model $eloquentHit) use($raw){
|
|
|
|
// $raw = collect($raw)
|
|
// ->where('_source.__class_name', get_class($eloquentHit))
|
|
// ->where('_source.id', $eloquentHit->id)
|
|
// ->first()['_source'];
|
|
|
|
// $source = collect($raw);
|
|
// $eloquentHit->setAttribute('published_from', $source->get('published_from'));
|
|
// $eloquentHit->setAttribute('published_to', $source->get('published_to'));
|
|
|
|
// return $eloquentHit;
|
|
// })->all();
|
|
// }
|
|
}
|