ONGR\ElasticsearchDSL\Aggregation\Bucketing\CompositeAggregation

Complexity

Total Complexity

12

Size/Duplication

Total Lines	130
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	2

Importance

Changes

0

8 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	8	2
A	addSource()	0	12	2
A	getArray()	0	16	3
A	getType()	0	4	1
A	setSize()	0	6	1
A	getSize()	0	4	1
A	setAfter()	0	6	1
A	getAfter()	0	4	1

<?php

/*
 * This file is part of the ONGR package.
 *
 * (c) NFQ Technologies UAB <info@nfq.com>
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

namespace ONGR\ElasticsearchDSL\Aggregation\Bucketing;

use ONGR\ElasticsearchDSL\Aggregation\AbstractAggregation;
use ONGR\ElasticsearchDSL\Aggregation\Type\BucketingTrait;
use ONGR\ElasticsearchDSL\BuilderInterface;

/**
 * Class representing composite aggregation.
 *
 * @link https://www.elastic.co/guide/en/elasticsearch/reference/current/search-aggregations-bucket-composite-aggregation.html
 */
class CompositeAggregation extends AbstractAggregation
{
    use BucketingTrait;

    /**
     * @var BuilderInterface[]
     */
    private $sources = [];

    /**
     * @var int
     */
    private $size;

    /**
     * @var array
     */
    private $after;

    /**
     * Inner aggregations container init.
     *
     * @param string             $name
     * @param AbstractAggregation[] $sources
     */
    public function __construct($name, $sources = [])
    {
        parent::__construct($name);

        foreach ($sources as $agg) {
            $this->addSource($agg);
        }
    }

    /**
     * @param AbstractAggregation $agg
     *
     * @throws \LogicException
     *
     * @return self
     */
    public function addSource(AbstractAggregation $agg)
    {
        $array = $agg->getArray();

        $array = is_array($array) ? array_merge($array, $agg->getParameters()) : $array;

        $this->sources[] = [
            $agg->getName() => [ $agg->getType() => $array ]
        ];

        return $this;
    }

    /**
     * {@inheritdoc}
     */
    public function getArray()
    {
        $array = [
            'sources' => $this->sources,
        ];

        if ($this->size !== null) {
            $array['size'] = $this->size;
        }

        if (!empty($this->after)) {
            $array['after'] = $this->after;
        }

        return $array;
    }

    /**
     * {@inheritdoc}
     */
    public function getType()
    {
        return 'composite';
    }

    /**
     * Sets size
     *
     * @param int $size Size
     *
     * @return $this
     */
    public function setSize($size)
    {
        $this->size = $size;

        return $this;
    }

    /**
     * Returns size
     *
     * @return int
     */
    public function getSize()
    {
        return $this->size;
    }

    /**
     * Sets after
     *
     * @param array $after After
     *
     * @return $this
     */
    public function setAfter(array $after)
    {
        $this->after = $after;

        return $this;
    }

    /**
     * Returns after
     *
     * @return array
     */
    public function getAfter()
    {
        return $this->after;
    }
}