com.annimon.stream.operator.ObjFlatMap Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of stream-lite Show documentation

Enhancing Java 8 Streams

The newest version!

package com.annimon.stream.operator;

import com.annimon.stream.Stream;
import com.annimon.stream.function.Function;
import com.annimon.stream.iterator.LsaExtIterator;
import java.util.Iterator;

public class ObjFlatMap extends LsaExtIterator {

    private final Iterator iterator;
    private final Function> mapper;
    private Iterator inner;
    private Stream innerStream;

    public ObjFlatMap(Iterator iterator,
            Function> mapper) {
        this.iterator = iterator;
        this.mapper = mapper;
    }

    @Override
    protected void nextIteration() {
        if ((inner != null) && inner.hasNext()) {
            next = inner.next();
            hasNext = true;
            return;
        }
        while (iterator.hasNext()) {
            if (inner == null || !inner.hasNext()) {
                if (innerStream != null) {
                    innerStream.close();
                    innerStream = null;
                }
                final T arg = iterator.next();
                final Stream  result = mapper.apply(arg);
                if (result != null) {
                    inner = result.iterator();
                    innerStream = result;
                }
            }
            if ((inner != null) && inner.hasNext()) {
                next = inner.next();
                hasNext = true;
                return;
            }
        }
        hasNext = false;
        if (innerStream != null) {
            innerStream.close();
            innerStream = null;
        }
    }
}