-
Notifications
You must be signed in to change notification settings - Fork 8
Expand file tree
/
Copy pathScroll.php
More file actions
114 lines (98 loc) · 2.48 KB
/
Scroll.php
File metadata and controls
114 lines (98 loc) · 2.48 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
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
<?php declare(strict_types = 1);
namespace Spameri\Elastic\Model;
readonly class Scroll
{
public function __construct(
private \Spameri\Elastic\ClientProvider $clientProvider,
private \Spameri\ElasticQuery\Response\ResultMapper $resultMapper,
private VersionProvider $versionProvider,
)
{
}
/**
* @throws \Spameri\Elastic\Exception\ElasticSearch
*/
public function execute(
\Spameri\ElasticQuery\ElasticQuery $elasticQuery,
string $index,
string|null $type = NULL,
): \Spameri\ElasticQuery\Response\ResultSearch
{
if ($type === NULL) {
$type = $index;
}
if ($this->versionProvider->provide() >= \Spameri\ElasticQuery\Response\Result\Version::ELASTIC_VERSION_ID_7) {
$type = NULL;
}
try {
if ($elasticQuery->options()->scrollId() === NULL) {
$result = $this->clientProvider->client()->search(
(
new \Spameri\ElasticQuery\Document(
$index,
new \Spameri\ElasticQuery\Document\Body\Plain($elasticQuery->toArray()),
$type,
NULL,
[
'scroll' => $elasticQuery->options()->scroll(),
],
)
)
->toArray(),
)->asArray()
;
if (isset($result['_scroll_id'])) {
$elasticQuery->options()->scrollInitialized($result['_scroll_id']);
} else {
throw new \Spameri\Elastic\Exception\ScrollNotInitialized(
'ElasticSearch did not return scroll id.',
);
}
} else {
$result = $this->clientProvider->client()->scroll(
(
new \Spameri\ElasticQuery\Document(
NULL,
NULL,
NULL,
NULL,
[
'scroll' => $elasticQuery->options()->scroll(),
'scroll_id' => $elasticQuery->options()->scrollId(),
],
)
)
->toArray(),
)->asArray()
;
}
} catch (\Elastic\Elasticsearch\Exception\ElasticsearchException $exception) {
throw new \Spameri\Elastic\Exception\ElasticSearch($exception->getMessage());
}
return $this->resultMapper->mapSearchResults($result);
}
public function closeScroll(
\Spameri\ElasticQuery\ElasticQuery $elasticQuery,
): void
{
try {
$this->clientProvider->client()->clearScroll(
(
new \Spameri\ElasticQuery\Document(
NULL,
NULL,
NULL,
NULL,
[
'scroll_id' => $elasticQuery->options()->scrollId(),
],
)
)
->toArray(),
)
;
} catch (\Elastic\Elasticsearch\Exception\ElasticsearchException $exception) {
throw new \Spameri\Elastic\Exception\ElasticSearch($exception->getMessage());
}
}
}