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
|
package org.zwobble.mammoth.docx;
import com.google.common.collect.ImmutableList;
import org.zwobble.mammoth.documents.DocumentElement;
import org.zwobble.mammoth.results.Result;
import org.zwobble.mammoth.results.Warning;
import java.util.List;
import java.util.function.BiFunction;
import java.util.function.Function;
import static org.zwobble.mammoth.util.MammothLists.list;
public class ReadResult {
public static final ReadResult EMPTY_SUCCESS = success(list());
public static ReadResult concat(Iterable<ReadResult> results) {
ImmutableList.Builder<DocumentElement> elements = ImmutableList.builder();
ImmutableList.Builder<DocumentElement> extra = ImmutableList.builder();
ImmutableList.Builder<Warning> warnings = ImmutableList.builder();
for (ReadResult result : results) {
elements.addAll(result.elements);
extra.addAll(result.extra);
warnings.addAll(result.warnings);
}
return new ReadResult(elements.build(), extra.build(), warnings.build());
}
public static <T> ReadResult map(
Result<T> first,
ReadResult second,
BiFunction<T, List<DocumentElement>, DocumentElement> function)
{
return new ReadResult(
list(function.apply(first.getValue(), second.elements)),
second.extra,
concat(first.getWarnings(), second.warnings));
}
public static ReadResult success(DocumentElement element) {
return success(list(element));
}
public static ReadResult success(List<DocumentElement> elements) {
return new ReadResult(elements, list(), list());
}
public static ReadResult emptyWithWarning(Warning warning) {
return new ReadResult(list(), list(), list(warning));
}
private final List<DocumentElement> elements;
private final List<DocumentElement> extra;
private final List<Warning> warnings;
public ReadResult(List<DocumentElement> elements, List<DocumentElement> extra, List<Warning> warnings) {
this.elements = elements;
this.extra = extra;
this.warnings = warnings;
}
public ReadResult map(Function<List<DocumentElement>, DocumentElement> function) {
return new ReadResult(
list(function.apply(elements)),
extra,
warnings);
}
public ReadResult toExtra() {
return new ReadResult(list(), concat(extra, elements), warnings);
}
public ReadResult appendExtra() {
return new ReadResult(concat(elements, extra), list(), warnings);
}
public Result<List<DocumentElement>> toResult() {
return new Result<>(elements, warnings);
}
private static <T> List<T> concat(List<T> first, List<T> second) {
ImmutableList.Builder<T> builder = ImmutableList.builder();
builder.addAll(first);
builder.addAll(second);
return builder.build();
}
}
|
