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
121
122
123
124
125
126
127
128
129
130
131
132
133
134
|
// 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 java.util.Collections;
import java.util.HashMap;
import java.util.Map;
/**
* A registry of annotation types. This can be set up programmatically or from config.
*
* @author Einar M R Rosenvinge
*/
public class AnnotationTypeRegistry {
private final Map<Integer, AnnotationType> idMap = new HashMap<>();
private final Map<String, AnnotationType> nameMap = new HashMap<>();
/** Creates a new empty registry. */
public AnnotationTypeRegistry() {
}
/**
* Register a new annotation type.
* WARNING! Only to be used by the configuration system and in unit tests. Not to be used in production code.
*
* @param type the type to register
* @throws IllegalArgumentException if a type is already registered with this name or this id, and it is non-equal to the argument.
*/
public void register(AnnotationType type) {
if (idMap.containsKey(type.getId()) || nameMap.containsKey(type.getName())) {
AnnotationType idType = idMap.get(type.getId());
AnnotationType nameType = nameMap.get(type.getName());
if (type.equals(idType) && type.equals(nameType)) {
//it's the same one being re-registered, we're OK!
return;
}
throw new IllegalArgumentException("A type is already registered with this name or this id.");
}
idMap.put(type.getId(), type);
nameMap.put(type.getName(), type);
}
/**
* Unregisters the type given by the argument.
* WARNING! Only to be used by the configuration system and in unit tests. Not to be used in production code.
*
* @param name the name of the type to unregister
* @return true if the type was successfully unregistered, false otherwise (it was not present)
*/
public boolean unregister(String name) {
AnnotationType oldType = nameMap.remove(name);
if (oldType != null) {
idMap.remove(oldType.getId());
return true;
}
return false;
}
/**
* Unregisters the type given by the argument.
* WARNING! Only to be used by the configuration system and in unit tests. Not to be used in production code.
*
* @param id the id of the type to unregister
* @return true if the type was successfully unregistered, false otherwise (it was not present)
*/
public boolean unregister(int id) {
AnnotationType oldType = idMap.remove(id);
if (oldType != null) {
nameMap.remove(oldType.getName());
return true;
}
return false;
}
/**
* Unregisters the type given by the argument.
* WARNING! Only to be used by the configuration system and in unit tests. Not to be used in production code.
*
* @param type the AnnotationType to unregister
* @return true if the type was successfully unregistered, false otherwise (it was not present)
* @throws IllegalArgumentException if the ID and name of this annotation type are present,
* but they do not belong together.
*/
public boolean unregister(AnnotationType type) {
if (idMap.containsKey(type.getId()) && nameMap.containsKey(type.getName())) {
AnnotationType idType = idMap.get(type.getId());
AnnotationType nameType = nameMap.get(type.getName());
if (idType == nameType) {
//name and id belong together in our maps
idMap.remove(type.getId());
nameMap.remove(type.getName());
} else {
throw new IllegalArgumentException("The ID and name of this annotation type are present, " +
"but they do not belong together. Not removing type.");
}
return true;
}
// It's not there, but that's no problem
return false;
}
/**
* Returns an annotation type with the given name.
*
* @param name the name of the annotation type to return
* @return an {@link AnnotationType} with the given name, or null if it is not registered
*/
public AnnotationType getType(String name) {
return nameMap.get(name);
}
/**
* Returns an annotation type with the given id.
*
* @param id the id of the annotation type to return
* @return an {@link AnnotationType} with the given id, or null if it is not registered
*/
public AnnotationType getType(int id) {
return idMap.get(id);
}
/** Returns an unmodifiable of all types registered. */
public Map<String, AnnotationType> getTypes() {
return Collections.unmodifiableMap(nameMap);
}
/** Clears all registered annotation types. */
public void clear() {
idMap.clear();
nameMap.clear();
}
}
|
