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
<?php
/**
* Copyright © Magento, Inc. All rights reserved.
* See COPYING.txt for license details.
*/
namespace Magento\Elasticsearch\SearchAdapter\Query\Builder;
use Magento\Framework\Search\Request\BucketInterface;
use Magento\Framework\Search\RequestInterface;
use Magento\Elasticsearch\Model\Adapter\FieldMapperInterface;
/**
* @api
* @since 100.1.0
*/
class Aggregation
{
/**
* Max number of results returned per single term bucket, i.e. limit of options for layered navigation filter.
* Default ElasticSearch limit is 10
*
* @var int
*/
private static $maxTermBacketSize = 500;
/**
* @var FieldMapperInterface
* @since 100.1.0
*/
protected $fieldMapper;
/**
* @param FieldMapperInterface $fieldMapper
*/
public function __construct(
FieldMapperInterface $fieldMapper
) {
$this->fieldMapper = $fieldMapper;
}
/**
* Build aggregation query for request
*
* @param RequestInterface $request
* @param array $searchQuery
* @return array
* @since 100.1.0
*/
public function build(
RequestInterface $request,
array $searchQuery
) {
$buckets = $request->getAggregation();
foreach ($buckets as $bucket) {
$searchQuery = $this->buildBucket($searchQuery, $bucket);
}
return $searchQuery;
}
/**
* Build aggregation query for bucket
*
* @param array $searchQuery
* @param BucketInterface $bucket
* @return array
* @since 100.1.0
*/
protected function buildBucket(
array $searchQuery,
BucketInterface $bucket
) {
$field = $this->fieldMapper->getFieldName($bucket->getField());
switch ($bucket->getType()) {
case BucketInterface::TYPE_TERM:
$searchQuery['body']['aggregations'][$bucket->getName()]= [
'terms' => [
'field' => $field,
'size' => self::$maxTermBacketSize,
],
];
break;
case BucketInterface::TYPE_DYNAMIC:
$searchQuery['body']['aggregations'][$bucket->getName()]= [
'extended_stats' => [
'field' => $field,
],
];
break;
}
return $searchQuery;
}
}