1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
package io.dico.dicore.task;
import java.util.*;
import java.util.function.BiConsumer;
import java.util.function.BiPredicate;
import java.util.function.Consumer;
import java.util.function.Predicate;
public abstract class IteratorTask<T> extends BaseTask<T> {
private Iterator<? extends T> iterator;
public IteratorTask() {
}
@SuppressWarnings("unchecked")
public IteratorTask(Iterable<? extends T> iterable, boolean clone) {
refresh(iterable, clone);
}
public IteratorTask(Iterable<? extends T> iterable) {
this(iterable, false);
}
public IteratorTask(Iterator<? extends T> iterator) {
refresh(iterator);
}
@Override
protected void doStartChecks() {
super.doStartChecks();
if (iterator == null) {
throw new IllegalStateException("An iterator must be supplied first");
}
}
protected final void refresh(Iterable<? extends T> iterable, boolean clone) {
if (clone) {
Collection<T> collection;
if (!(iterable instanceof Collection)) {
collection = new LinkedList<>();
for (T next : iterable) {
collection.add(next);
}
} else {
collection = new ArrayList((Collection) iterable);
}
iterator = collection.iterator();
} else {
iterator = iterable.iterator();
}
}
protected final void refresh(Iterator<? extends T> iterator) {
Objects.requireNonNull(iterator);
this.iterator = iterator;
}
@Override
protected T supply() {
return iterator.next();
}
protected void remove() {
iterator.remove();
}
// One argument: The processed object
public static <T> IteratorTask<T> create(Iterable<? extends T> iterable, Consumer<T> processor) {
return create(iterable, false, processor);
}
public static <T> IteratorTask<T> create(Iterable<? extends T> iterable, boolean clone, Consumer<T> processor) {
return create(iterable, clone, object -> {
processor.accept(object);
return true;
});
}
public static <T> IteratorTask<T> create(Iterable<? extends T> iterable, Predicate<T> processor) {
return create(iterable, false, processor);
}
public static <T> IteratorTask<T> create(Iterable<? extends T> iterable, boolean clone, Predicate<T> processor) {
return new IteratorTask<T>(iterable, clone) {
@Override
protected boolean process(T object) {
return processor.test(object);
}
};
}
// Two arguments: the processed object, and a runnable to remove it from the iterator.
public static <T> IteratorTask<T> create(Iterable<? extends T> iterable, BiConsumer<T, Runnable> processor) {
return create(iterable, false, processor);
}
public static <T> IteratorTask<T> create(Iterable<? extends T> iterable, boolean clone, BiConsumer<T, Runnable> processor) {
return create(iterable, clone, (object, runnable) -> {
processor.accept(object, runnable);
return true;
});
}
public static <T> IteratorTask<T> create(Iterable<? extends T> iterable, BiPredicate<T, Runnable> processor) {
return create(iterable, false, processor);
}
public static <T> IteratorTask<T> create(Iterable<? extends T> iterable, boolean clone, BiPredicate<T, Runnable> processor) {
return new IteratorTask<T>(iterable, clone) {
@Override
protected boolean process(T object) {
return processor.test(object, this::remove);
}
};
}
}
|