hanneskod\yaysondb\Filter\LimitFilter

Complexity

Total Complexity

5

Size/Duplication

Total Lines	43
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	0

Importance

Changes

0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	5	1
A	filter()	0	18	4

<?php

declare(strict_types = 1);

namespace hanneskod\yaysondb\Filter;

/**
 * Limit the number of returned documents
 */
class LimitFilter implements FilterInterface
{
    /**
     * @var int The number of documents to return
     */
    private $count;

    /**
     * @var int The offset of the first document to return
     */
    private $offset;

    /**
     * Set limit clause
     *
     * @param int $count  The number of documents to return
     * @param int $offset The offset of the first document to return
     */
    public function __construct(int $count, int $offset = 0)
    {
        $this->count = $count;
        $this->offset = $offset;
    }

    public function filter(\Traversable $documents): \Generator
    {
        $count = 0;

        foreach ($documents as $id => $doc) {
            $count++;

            if ($count <= $this->offset) {
                continue;
            }

            if ($count > $this->count + $this->offset) {
                break;
            }

            yield $id => $doc;
        }
    }
}