aQute.lib.collections.Enumerations Maven / Gradle / Ivy
package aQute.lib.collections;
import static java.util.Objects.requireNonNull;
import java.util.Enumeration;
import java.util.NoSuchElementException;
import java.util.Spliterator;
import java.util.Spliterators;
import java.util.function.Consumer;
import java.util.function.Function;
import java.util.function.Predicate;
public class Enumerations {
private Enumerations() {}
final static class EnumerationSpliterator implements Enumeration, Consumer {
private final Spliterator spliterator;
private final Function mapper;
private final Predicate filter;
private boolean hasNext = false;
private R next;
private T t;
EnumerationSpliterator(Spliterator spliterator, Function mapper,
Predicate filter) {
this.spliterator = spliterator != null ? spliterator : Spliterators.emptySpliterator();
this.mapper = requireNonNull(mapper);
this.filter = requireNonNull(filter);
}
@Override
public boolean hasMoreElements() {
if (hasNext) {
return true;
}
while (spliterator.tryAdvance(this)) {
R r = mapper.apply(t);
if (filter.test(r)) {
next = r;
return hasNext = true;
}
}
return false;
}
@Override
public R nextElement() {
if (hasMoreElements()) {
R r = next;
hasNext = false;
return r;
}
throw new NoSuchElementException();
}
@Override
public void accept(T t) {
this.t = t;
}
}
public static Enumeration enumeration(Spliterator spliterator) {
return new EnumerationSpliterator<>(spliterator, t -> t, t -> true);
}
public static Enumeration enumeration(Spliterator spliterator,
Function mapper) {
return new EnumerationSpliterator<>(spliterator, mapper, r -> true);
}
public static Enumeration enumeration(Spliterator spliterator,
Function mapper, Predicate filter) {
return new EnumerationSpliterator<>(spliterator, mapper, filter);
}
}