blob: d8709baa3a1e1c46d73f0f3ceeeb5da811559a57 (
plain) (
blame)
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
|
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.document.annotation;
import com.google.common.collect.Multimap;
import com.google.common.collect.Multimaps;
import java.util.ArrayList;
import java.util.Collection;
import java.util.IdentityHashMap;
import java.util.Iterator;
import java.util.NoSuchElementException;
/**
*
* @author Einar M R Rosenvinge
*/
// TODO: Should this be removed?
public class AnnotationType2AnnotationContainer extends IteratingAnnotationContainer {
private final Multimap<AnnotationType, Annotation> annotationType2Annotation = Multimaps.newMultimap(new IdentityHashMap<>(), ArrayList::new);
@Override
void annotateAll(Collection<Annotation> annotations) {
for (Annotation a : annotations) {
annotate(a);
}
}
@Override
void annotate(Annotation annotation) {
annotationType2Annotation.put(annotation.getType(), annotation);
}
@Override
@SuppressWarnings("unchecked")
Collection<Annotation> annotations() {
return annotationType2Annotation.values();
}
@Override
Iterator<Annotation> iterator(IdentityHashMap<SpanNode, SpanNode> nodes) {
return new NonRecursiveIterator(nodes);
}
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (!(o instanceof AnnotationType2AnnotationContainer)) return false;
AnnotationType2AnnotationContainer that = (AnnotationType2AnnotationContainer) o;
if (!annotationType2Annotation.equals(that.annotationType2Annotation)) return false;
return true;
}
@Override
public int hashCode() {
return annotationType2Annotation.hashCode();
}
private class NonRecursiveIterator implements Iterator<Annotation> {
private final IdentityHashMap<SpanNode, SpanNode> nodes;
private final Iterator<Annotation> annotationIt;
private Annotation next = null;
private boolean nextCalled;
@SuppressWarnings("unchecked")
public NonRecursiveIterator(IdentityHashMap<SpanNode, SpanNode> nodes) {
this.nodes = nodes;
this.annotationIt = annotationType2Annotation.values().iterator();
}
@Override
public boolean hasNext() {
if (next != null) {
return true;
}
while (annotationIt.hasNext()) {
Annotation tmp = annotationIt.next();
if (nodes.containsKey(tmp.getSpanNodeFast())) {
next = tmp;
return true;
}
}
next = null;
return false;
}
@Override
public Annotation next() {
if (hasNext()) {
Annotation tmp = next;
next = null;
nextCalled = true;
return tmp;
}
//there is no 'next'
throw new NoSuchElementException("No next element found.");
}
@Override
public void remove() {
//only allowed to call remove immediately after next()
if (!nextCalled) {
//we have not next'ed the iterator, cannot do this:
throw new IllegalStateException("Cannot remove() before next()");
}
annotationIt.remove();
nextCalled = false;
}
}
}
|
