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
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
|
/*-
* See the file LICENSE for redistribution information.
*
* Copyright (c) 2002, 2015 Oracle and/or its affiliates. All rights reserved.
*
*/
package com.sleepycat.persist.impl;
import java.util.Collection;
import java.util.Set;
import com.sleepycat.bind.tuple.TupleBase;
import com.sleepycat.compat.DbCompat;
import com.sleepycat.db.DatabaseEntry;
import com.sleepycat.db.ForeignMultiKeyNullifier;
import com.sleepycat.db.SecondaryDatabase;
import com.sleepycat.db.SecondaryKeyCreator;
import com.sleepycat.db.SecondaryMultiKeyCreator;
import com.sleepycat.persist.model.EntityMetadata;
import com.sleepycat.persist.model.Relationship;
import com.sleepycat.persist.model.SecondaryKeyMetadata;
import com.sleepycat.persist.raw.RawObject;
/**
* A persistence secondary key creator/nullifier. This class always uses
* rawAccess=true to avoid depending on the presence of the proxy class.
*
* @author Mark Hayes
*/
public class PersistKeyCreator implements SecondaryKeyCreator,
SecondaryMultiKeyCreator,
ForeignMultiKeyNullifier {
static boolean isManyType(Class cls) {
return cls.isArray() || Collection.class.isAssignableFrom(cls);
}
/* See Store.refresh for an explanation of the use of volatile fields. */
private volatile Catalog catalog;
private volatile Format priKeyFormat;
private final String keyName;
private volatile Format keyFormat;
private final boolean toMany;
/**
* Creates a key creator/nullifier for a given entity class and key name.
*/
public PersistKeyCreator(Catalog catalogParam,
EntityMetadata entityMeta,
String keyClassName,
SecondaryKeyMetadata secKeyMeta,
boolean rawAccess) {
catalog = catalogParam;
try {
getFormats(entityMeta, keyClassName, secKeyMeta, rawAccess);
} catch (RefreshException e) {
/* Must assign catalog field in constructor. */
catalog = e.refresh();
try {
getFormats(entityMeta, keyClassName, secKeyMeta, rawAccess);
} catch (RefreshException e2) {
throw DbCompat.unexpectedException(e2);
}
}
keyName = secKeyMeta.getKeyName();
Relationship rel = secKeyMeta.getRelationship();
toMany = (rel == Relationship.ONE_TO_MANY ||
rel == Relationship.MANY_TO_MANY);
}
private void getFormats(EntityMetadata entityMeta,
String keyClassName,
SecondaryKeyMetadata secKeyMeta,
boolean rawAccess)
throws RefreshException {
priKeyFormat = PersistEntityBinding.getOrCreateFormat
(catalog, entityMeta.getPrimaryKey().getClassName(), rawAccess);
keyFormat = PersistEntityBinding.getOrCreateFormat
(catalog, keyClassName, rawAccess);
if (keyFormat == null) {
throw new IllegalArgumentException
("Not a key class: " + keyClassName);
}
if (keyFormat.isPrimitive()) {
throw new IllegalArgumentException
("Use a primitive wrapper class instead of class: " +
keyFormat.getClassName());
}
}
public boolean createSecondaryKey(SecondaryDatabase secondary,
DatabaseEntry key,
DatabaseEntry data,
DatabaseEntry result) {
try {
return createSecondaryKeyInternal(secondary, key, data, result);
} catch (RefreshException e) {
e.refresh();
try {
return createSecondaryKeyInternal(secondary, key, data,
result);
} catch (RefreshException e2) {
throw DbCompat.unexpectedException(e2);
}
}
}
private boolean createSecondaryKeyInternal(SecondaryDatabase secondary,
DatabaseEntry key,
DatabaseEntry data,
DatabaseEntry result)
throws RefreshException {
if (toMany) {
throw DbCompat.unexpectedState();
}
KeyLocation loc = moveToKey(key, data);
if (loc != null) {
RecordOutput output = new RecordOutput
(catalog, true /*rawAccess*/);
loc.format.copySecKey(loc.input, output);
TupleBase.outputToEntry(output, result);
return true;
} else {
/* Key field is not present or null. */
return false;
}
}
public void createSecondaryKeys(SecondaryDatabase secondary,
DatabaseEntry key,
DatabaseEntry data,
Set results) {
try {
createSecondaryKeysInternal(secondary, key, data, results);
} catch (RefreshException e) {
e.refresh();
try {
createSecondaryKeysInternal(secondary, key, data, results);
} catch (RefreshException e2) {
throw DbCompat.unexpectedException(e2);
}
}
}
private void createSecondaryKeysInternal(SecondaryDatabase secondary,
DatabaseEntry key,
DatabaseEntry data,
Set results)
throws RefreshException {
if (!toMany) {
throw DbCompat.unexpectedState();
}
KeyLocation loc = moveToKey(key, data);
if (loc != null) {
loc.format.copySecMultiKey(loc.input, keyFormat, results);
}
/* Else key field is not present or null. */
}
public boolean nullifyForeignKey(SecondaryDatabase secondary,
DatabaseEntry key,
DatabaseEntry data,
DatabaseEntry secKey) {
try {
return nullifyForeignKeyInternal(secondary, key, data, secKey);
} catch (RefreshException e) {
e.refresh();
try {
return nullifyForeignKeyInternal(secondary, key, data, secKey);
} catch (RefreshException e2) {
throw DbCompat.unexpectedException(e2);
}
}
}
private boolean nullifyForeignKeyInternal(SecondaryDatabase secondary,
DatabaseEntry key,
DatabaseEntry data,
DatabaseEntry secKey)
throws RefreshException {
/* Deserialize the entity and get its current class format. */
RawObject entity = (RawObject) PersistEntityBinding.readEntity
(catalog, key, null, data, true /*rawAccess*/);
Format entityFormat = (Format) entity.getType();
/*
* Set the key to null. For a TO_MANY key, pass the key object to be
* removed from the array/collection.
*/
Object secKeyObject = null;
if (toMany) {
secKeyObject = PersistKeyBinding.readKey
(keyFormat, catalog, secKey.getData(), secKey.getOffset(),
secKey.getSize(), true /*rawAccess*/);
}
if (entityFormat.nullifySecKey
(catalog, entity, keyName, secKeyObject)) {
/*
* Using the current format for the entity, serialize the modified
* entity back to the data entry.
*/
PersistEntityBinding.writeEntity
(entityFormat, catalog, entity, data, true /*rawAccess*/);
return true;
} else {
/* Key field is not present or null. */
return false;
}
}
/**
* Returns the location from which the secondary key field can be copied.
*/
private KeyLocation moveToKey(DatabaseEntry priKey, DatabaseEntry data)
throws RefreshException {
RecordInput input = new RecordInput
(catalog, true /*rawAccess*/, priKey, priKeyFormat.getId(),
data.getData(), data.getOffset(), data.getSize());
int formatId = input.readPackedInt();
Format entityFormat =
catalog.getFormat(formatId, true /*expectStored*/);
input.registerEntityFormat(entityFormat);
Format fieldFormat = entityFormat.skipToSecKey(input, keyName);
if (fieldFormat != null) {
/* Returns null if key field is null. */
return input.getKeyLocation(fieldFormat);
} else {
/* Key field is not present in this class. */
return null;
}
}
/**
* See Store.refresh.
*/
void refresh(final PersistCatalog newCatalog) {
catalog = newCatalog;
keyFormat = catalog.getFormat(keyFormat.getClassName());
priKeyFormat = catalog.getFormat(priKeyFormat.getClassName());
}
}
|
