Simple adaptation of the Doctrine ORM Paginator (Doctrine\ORM\Tools\Pagination\Paginator) for Doctrine ODM MongoDB

Paginator.php

<?php

namespace AppBundle\Doctrine\ODM\Pagination;

use Doctrine\ODM\MongoDB\Query\Builder;
use Doctrine\ODM\MongoDB\Query\Query;

/**
 * Simple adaptation of Doctrine\ORM\Tools\Pagination\Paginator
 * for Doctrine ODM MongoDB
 */
class Paginator implements \Countable, \IteratorAggregate
{
    /**
     * @var Doctrine\ODM\MongoDB\Query\Query
     */
    protected $query;

    /**
     * Constructor.
     *
     * @param Query|Builder $query
     */
    public function __construct($query)
    {
        $this->query = ($query instanceof Builder) ? $query->getQuery() : $query;
    }

    /**
     * Returns the query.
     *
     * @return Query
     */
    public function getQuery()
    {
        return $this->query;
    }

    /**
     * @return integer
     */
    public function count()
    {
        return $this->query->count();
    }

    /**
     * @return ArrayIterator
     */
    public function getIterator()
    {
        return new \ArrayIterator($this->query->execute()->toArray(false));
    }
}