1
|
|
|
<?php |
2
|
|
|
|
3
|
|
|
namespace Isswp101\Persimmon\Repository; |
4
|
|
|
|
5
|
|
|
use Elasticsearch\Client; |
6
|
|
|
use Elasticsearch\Common\Exceptions\Missing404Exception; |
7
|
|
|
use Isswp101\Persimmon\Collection\ElasticsearchCollection; |
8
|
|
|
use Isswp101\Persimmon\Collection\IElasticsearchCollection; |
9
|
|
|
use Isswp101\Persimmon\CollectionParser\ElasticsearchCollectionParser; |
10
|
|
|
use Isswp101\Persimmon\Contracts\Storable; |
11
|
|
|
use Isswp101\Persimmon\Exceptions\ClassTypeErrorException; |
12
|
|
|
use Isswp101\Persimmon\Exceptions\ModelNotFoundException; |
13
|
|
|
use Isswp101\Persimmon\Model\IElasticsearchModel; |
14
|
|
|
use Isswp101\Persimmon\QueryBuilder\IQueryBuilder; |
15
|
|
|
use Isswp101\Persimmon\Relationship\RelationshipKey; |
16
|
|
|
use Isswp101\Persimmon\Response\ElasticsearchCollectionResponse; |
17
|
|
|
use Isswp101\Persimmon\Response\ElasticsearchItemResponse; |
18
|
|
|
|
19
|
|
|
class ElasticsearchRepository implements IRepository |
20
|
|
|
{ |
21
|
|
|
private $client; |
22
|
|
|
|
23
|
|
|
public function __construct(Client $client) |
24
|
|
|
{ |
25
|
|
|
$this->client = $client; |
26
|
|
|
} |
27
|
|
|
|
28
|
|
View Code Duplication |
public function instantiate(string $class): Storable |
|
|
|
|
29
|
|
|
{ |
30
|
|
|
$instance = new $class; |
31
|
|
|
if (!$instance instanceof IElasticsearchModel) { |
32
|
|
|
throw new ClassTypeErrorException(IElasticsearchModel::class); |
33
|
|
|
} |
34
|
|
|
return $instance; |
35
|
|
|
} |
36
|
|
|
|
37
|
|
|
protected function fill(Storable $model, ElasticsearchItemResponse $response) |
38
|
|
|
{ |
39
|
|
|
$model->fill($response->source()); |
40
|
|
|
$relationshipKey = new RelationshipKey($response->id(), $response->parent()); |
41
|
|
|
$model->setPrimaryKey($relationshipKey->build()); |
42
|
|
|
} |
43
|
|
|
|
44
|
|
|
public function find($id, string $class, array $columns = []): Storable |
45
|
|
|
{ |
46
|
|
|
$model = $this->instantiate($class); |
47
|
|
|
$relationshipKey = RelationshipKey::parse($id); |
48
|
|
|
$collection = new ElasticsearchCollectionParser($model->getCollection()); |
49
|
|
|
$params = [ |
50
|
|
|
'index' => $collection->getIndex(), |
51
|
|
|
'type' => $collection->getType(), |
52
|
|
|
'id' => $relationshipKey->getId(), |
53
|
|
|
'parent' => $relationshipKey->getParentId(), |
54
|
|
|
'_source' => $columns |
55
|
|
|
]; |
56
|
|
|
try { |
57
|
|
|
$response = new ElasticsearchItemResponse($this->client->get($params)); |
|
|
|
|
58
|
|
|
} catch (Missing404Exception $e) { |
59
|
|
|
throw new ModelNotFoundException($class, $id); |
60
|
|
|
} |
61
|
|
|
$this->fill($model, $response); |
62
|
|
|
return $model; |
63
|
|
|
} |
64
|
|
|
|
65
|
|
|
public function all(IQueryBuilder $query, string $class, callable $callback = null): IElasticsearchCollection |
66
|
|
|
{ |
67
|
|
|
$model = $this->instantiate($class); |
68
|
|
|
$collection = new ElasticsearchCollectionParser($model->getCollection()); |
69
|
|
|
$params = [ |
70
|
|
|
'index' => $collection->getIndex(), |
71
|
|
|
'type' => $collection->getType(), |
72
|
|
|
'body' => $query->build() |
73
|
|
|
]; |
74
|
|
|
$response = new ElasticsearchCollectionResponse($this->client->search($params)); |
|
|
|
|
75
|
|
|
$models = new ElasticsearchCollection([], $response); |
76
|
|
|
foreach ($response->hits() as $hit) { |
77
|
|
|
$model = $this->instantiate($class); |
78
|
|
|
$this->fill($model, new ElasticsearchItemResponse($hit)); |
79
|
|
|
if ($callback != null) { |
80
|
|
|
$callback($model); |
81
|
|
|
} |
82
|
|
|
$models->put($model->getPrimaryKey(), $model); |
83
|
|
|
} |
84
|
|
|
return $models; |
85
|
|
|
} |
86
|
|
|
|
87
|
|
View Code Duplication |
public function insert(Storable $model) |
|
|
|
|
88
|
|
|
{ |
89
|
|
|
$relationshipKey = RelationshipKey::parse($model->getPrimaryKey()); |
90
|
|
|
$collection = new ElasticsearchCollectionParser($model->getCollection()); |
91
|
|
|
$params = [ |
92
|
|
|
'index' => $collection->getIndex(), |
93
|
|
|
'type' => $collection->getType(), |
94
|
|
|
'id' => $relationshipKey->getId(), |
95
|
|
|
'parent' => $relationshipKey->getParentId(), |
96
|
|
|
'body' => $model->toArray() |
97
|
|
|
]; |
98
|
|
|
$this->client->index($params); |
99
|
|
|
} |
100
|
|
|
|
101
|
|
View Code Duplication |
public function update(Storable $model) |
|
|
|
|
102
|
|
|
{ |
103
|
|
|
$relationshipKey = RelationshipKey::parse($model->getPrimaryKey()); |
104
|
|
|
$collection = new ElasticsearchCollectionParser($model->getCollection()); |
105
|
|
|
$params = [ |
106
|
|
|
'index' => $collection->getIndex(), |
107
|
|
|
'type' => $collection->getType(), |
108
|
|
|
'id' => $relationshipKey->getId(), |
109
|
|
|
'parent' => $relationshipKey->getParentId(), |
110
|
|
|
'body' => $model->toArray() |
111
|
|
|
]; |
112
|
|
|
$this->client->index($params); |
113
|
|
|
} |
114
|
|
|
|
115
|
|
View Code Duplication |
public function delete(Storable $model) |
|
|
|
|
116
|
|
|
{ |
117
|
|
|
$relationshipKey = RelationshipKey::parse($model->getPrimaryKey()); |
118
|
|
|
$collection = new ElasticsearchCollectionParser($model->getCollection()); |
119
|
|
|
$params = [ |
120
|
|
|
'index' => $collection->getIndex(), |
121
|
|
|
'type' => $collection->getType(), |
122
|
|
|
'id' => $relationshipKey->getId(), |
123
|
|
|
'parent' => $relationshipKey->getParentId() |
124
|
|
|
]; |
125
|
|
|
$this->client->delete($params); |
126
|
|
|
} |
127
|
|
|
} |
Duplicated code is one of the most pungent code smells. If you need to duplicate the same code in three or more different places, we strongly encourage you to look into extracting the code into a single class or operation.
You can also find more detailed suggestions in the “Code” section of your repository.