-
Notifications
You must be signed in to change notification settings - Fork 8
Expand file tree
/
Copy pathInsertMultiple.php
More file actions
96 lines (77 loc) · 2.32 KB
/
InsertMultiple.php
File metadata and controls
96 lines (77 loc) · 2.32 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
<?php declare(strict_types = 1);
namespace Spameri\Elastic\Model;
class InsertMultiple
{
/**
* @var \Spameri\Elastic\Model\Insert\PrepareEntityArray
*/
private $prepareEntityArray;
/**
* @var \Spameri\Elastic\ClientProvider
*/
private $clientProvider;
/**
* @var \Spameri\ElasticQuery\Response\ResultMapper
*/
private $resultMapper;
private VersionProvider $versionProvider;
public function __construct(
\Spameri\Elastic\Model\Insert\PrepareEntityArray $prepareEntityArray,
\Spameri\Elastic\ClientProvider $clientProvider,
\Spameri\ElasticQuery\Response\ResultMapper $resultMapper,
\Spameri\Elastic\Model\VersionProvider $versionProvider
)
{
$this->prepareEntityArray = $prepareEntityArray;
$this->clientProvider = $clientProvider;
$this->resultMapper = $resultMapper;
$this->versionProvider = $versionProvider;
}
/**
* @throws \Spameri\Elastic\Exception\ElasticSearch
* @throws \Spameri\Elastic\Exception\DocumentInsertFailed
*/
public function execute(
\Spameri\Elastic\Entity\ElasticEntityCollectionInterface $entityCollection,
string $index,
?string $type = NULL
): \Spameri\ElasticQuery\Response\ResultBulk
{
if ($type === NULL) {
$type = $index;
}
if ($this->versionProvider->provide() >= \Spameri\ElasticQuery\Response\Result\Version::ELASTIC_VERSION_ID_7) {
$type = '_doc';
}
$documentsArray = [];
foreach ($entityCollection as $entity) {
$entityArray = $this->prepareEntityArray->prepare($entity);
unset($entityArray['id']);
$documentsArray[] = [
'index' => [
'_index' => $index,
'_type' => $type,
],
];
$documentsArray[] = $entityArray;
}
$document = new \Spameri\ElasticQuery\Document\Bulk($documentsArray);
try {
$response = $this->clientProvider->client()->bulk($document->toArray());
} catch (\Elasticsearch\Common\Exceptions\ElasticsearchException $exception) {
throw new \Spameri\Elastic\Exception\ElasticSearch($exception->getMessage());
}
try {
$this->clientProvider->client()->indices()->refresh(
(
new \Spameri\ElasticQuery\Document($index)
)
->toArray()
)
;
} catch (\Elasticsearch\Common\Exceptions\ElasticsearchException $exception) {
throw new \Spameri\Elastic\Exception\ElasticSearch($exception->getMessage());
}
return $this->resultMapper->mapBulkResult($response);
}
}