-
Notifications
You must be signed in to change notification settings - Fork 14
/
RepositoryMap.java
118 lines (97 loc) · 3.3 KB
/
RepositoryMap.java
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
/**
* Copyright (C) 2016 Czech Technical University in Prague
*
* This program is free software: you can redistribute it and/or modify it under
* the terms of the GNU General Public License as published by the Free Software
* Foundation, either version 3 of the License, or (at your option) any
* later version.
*
* This program is distributed in the hope that it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
* FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
* details. You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
package cz.cvut.kbss.jopa.sessions;
import java.net.URI;
import java.util.HashMap;
import java.util.IdentityHashMap;
import java.util.Map;
import cz.cvut.kbss.jopa.model.descriptors.Descriptor;
final class RepositoryMap {
private static final URI DEFAULT_CONTEXT = URI.create("http://defaultContext");
private final Map<URI, Map<Object, Object>> origsToClones = new HashMap<>();
private Map<Object, Descriptor> entityDescriptors;
RepositoryMap() {
}
void initDescriptors() {
this.entityDescriptors = new IdentityHashMap<>();
}
void add(Descriptor descriptor, Object original, Object clone) {
assert descriptor != null;
assert original != null;
// Null values are permitted
final Map<Object, Object> entities = getMap(descriptor);
entities.put(original, clone);
}
void remove(Descriptor descriptor, Object original) {
assert descriptor != null;
assert original != null;
final Map<Object, Object> entities = getMap(descriptor);
entities.remove(original);
}
/**
* Make sure to call {@link #initDescriptors()} before calling this.
*/
void addEntityToRepository(Object entity, Descriptor descriptor) {
assert entityDescriptors != null;
entityDescriptors.put(entity, descriptor);
}
/**
* Make sure to call {@link #initDescriptors()} before calling this.
*/
void removeEntityToRepository(Object entity) {
assert entityDescriptors != null;
entityDescriptors.remove(entity);
}
boolean contains(Descriptor descriptor, Object original) {
assert descriptor != null;
assert original != null;
final Map<Object, Object> entities = getMap(descriptor);
return entities.containsKey(original);
}
Object get(Descriptor descriptor, Object original) {
assert descriptor != null;
assert original != null;
final Map<Object, Object> entities = getMap(descriptor);
if (!entities.containsKey(original)) {
return null;
}
return entities.get(original);
}
/**
* Make sure to call {@link #initDescriptors()} before calling this.
*/
Descriptor getEntityDescriptor(Object entity) {
assert entityDescriptors != null;
assert entity != null;
return entityDescriptors.get(entity);
}
void clear() {
origsToClones.values().forEach(Map::clear);
if (entityDescriptors != null) {
initDescriptors();
}
}
private Map<Object, Object> getMap(Descriptor descriptor) {
final URI ctx = descriptor.getContext() != null ? descriptor.getContext() : DEFAULT_CONTEXT;
Map<Object, Object> entities;
if (!origsToClones.containsKey(ctx)) {
entities = new IdentityHashMap<>();
origsToClones.put(ctx, entities);
} else {
entities = origsToClones.get(ctx);
}
return entities;
}
}