/*-
* See the file LICENSE for redistribution information.
*
* Copyright (c) 2002, 2011 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.je.DatabaseEntry;
import com.sleepycat.je.ForeignMultiKeyNullifier;
import com.sleepycat.je.SecondaryDatabase;
import com.sleepycat.je.SecondaryKeyCreator;
import com.sleepycat.je.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());
}
}