com.redhat.lightblue.assoc.ep.StepResultDocumentStream Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of lightblue-core-crud Show documentation

There is a newer version: 2.18.0

package com.redhat.lightblue.assoc.ep;

import java.util.Iterator;
import java.util.ArrayList;
import java.util.function.Consumer;
import java.util.stream.Stream;

import com.redhat.lightblue.crud.DocCtx;
import com.redhat.lightblue.crud.DocumentStream;

/**
 * Interface between DocumentStream used in CRUDOperationContext, and StepResult
 */
public class StepResultDocumentStream implements DocumentStream {

    private final Iterator itr;
    private final ArrayList> listeners=new ArrayList<>();

    public StepResultDocumentStream(StepResult result) {
        this.itr=result.stream().iterator();
    }

    @Override
    public boolean hasNext() {
        return itr.hasNext();
    }

    @Override
    public DocCtx next() {
        DocCtx doc=itr.next();
        for(Consumer c:listeners)
            c.accept(doc);
        return doc;
    }

    @Override
    public void close() {}

    @Override
    public void addListener(Consumer listener) {
        listeners.add(listener);
    }
}