com.microsoft.azure.documentdb.internal.query.executioncomponent.TopQueryExecutionComponent Maven / Gradle / Ivy

Go to download

package com.microsoft.azure.documentdb.internal.query.executioncomponent;

import java.util.*;

import com.microsoft.azure.documentdb.*;
import com.microsoft.azure.documentdb.internal.query.QueryExecutionContext;

public final class TopQueryExecutionComponent extends QueryExecutionComponent {
    private int topCount;

    public TopQueryExecutionComponent(QueryExecutionContext source, int top) {
        super(source);
        this.topCount = top;
    }

    @Override
    public Map getResponseHeaders() {
        return super.source.getResponseHeaders();
    }

    @Override
    public boolean hasNext() {
        return super.source.hasNext() && this.topCount > 0;
    }

    @Override
    protected Document next(Document document) {
        --this.topCount;
        if (this.topCount <= 0) {
            super.source.onNotifyStop();
        }

        return document;
    }
}