Spring Boot - MongoDB - Page Aggregation

pageAggregation.java

public <T> Page<T> pageAggregation(
            final Pageable pageable,
            final Criteria criteria,
            final String collection,
            final Class<T> clazz) {
        long count;
        if (criteria.getCriteriaObject().isEmpty()) {
            count = Optional
                    .of(mongoTemplate.estimatedCount(clazz))
                    .orElse(0L);
        } else {
            count = Optional
                    .of(mongoTemplate.count(new org.springframework.data.mongodb.core.query.Query(criteria), clazz))
                    .orElse(0L);
        }

        final List<AggregationOperation> stagesWithPaging = new ArrayList<>();
        stagesWithPaging.add(Aggregation.match(criteria));
        stagesWithPaging.add(sort(pageable.getSort()));
        stagesWithPaging.add(skip(pageable.getOffset()));
        stagesWithPaging.add(limit(pageable.getPageSize()));
        final Aggregation resultAgg = newAggregation(stagesWithPaging);
        final List<T> result = mongoTemplate.aggregate(resultAgg, collection, clazz).getMappedResults();

        return new PageImpl<>(result, pageable, count);
    }