/*
|
* CDDL HEADER START
|
*
|
* The contents of this file are subject to the terms of the
|
* Common Development and Distribution License, Version 1.0 only
|
* (the "License"). You may not use this file except in compliance
|
* with the License.
|
*
|
* You can obtain a copy of the license at
|
* trunk/opends/resource/legal-notices/OpenDS.LICENSE
|
* or https://OpenDS.dev.java.net/OpenDS.LICENSE.
|
* See the License for the specific language governing permissions
|
* and limitations under the License.
|
*
|
* When distributing Covered Code, include this CDDL HEADER in each
|
* file and include the License file at
|
* trunk/opends/resource/legal-notices/OpenDS.LICENSE. If applicable,
|
* add the following below this CDDL HEADER, with the fields enclosed
|
* by brackets "[]" replaced with your own identifying information:
|
* Portions Copyright [yyyy] [name of copyright owner]
|
*
|
* CDDL HEADER END
|
*
|
*
|
* Portions Copyright 2006-2007 Sun Microsystems, Inc.
|
*/
|
package org.opends.server.backends.jeb;
|
|
import static org.opends.server.loggers.debug.DebugLogger.debugCaught;
|
import static org.opends.server.loggers.debug.DebugLogger.debugEnabled;
|
import org.opends.server.types.DebugLogLevel;
|
|
import com.sleepycat.je.Transaction;
|
import com.sleepycat.je.DatabaseException;
|
|
import org.opends.server.protocols.asn1.ASN1OctetString;
|
import org.opends.server.types.Attribute;
|
import org.opends.server.types.AttributeType;
|
import org.opends.server.types.AttributeValue;
|
import org.opends.server.types.DirectoryException;
|
import org.opends.server.types.Entry;
|
import org.opends.server.types.Modification;
|
|
import java.util.Comparator;
|
import java.util.HashSet;
|
import java.util.List;
|
import java.util.Set;
|
import java.util.ArrayList;
|
|
/**
|
* An implementation of an Indexer for attribute equality.
|
*/
|
public class EqualityIndexer extends Indexer
|
{
|
|
|
/**
|
* The comparator for index keys generated by this class.
|
*/
|
private static final Comparator<byte[]> comparator =
|
new AttributeIndex.KeyComparator();
|
|
/**
|
* The attribute index configuration for which this instance will
|
* generate index keys.
|
*/
|
private IndexConfig indexConfig;
|
|
|
/**
|
* Create a new attribute equality indexer for the given index configuration.
|
* @param indexConfig The index configuration for which an indexer is
|
* required.
|
*/
|
public EqualityIndexer(IndexConfig indexConfig)
|
{
|
this.indexConfig = indexConfig;
|
}
|
|
/**
|
* Get a string representation of this object. The returned value is
|
* used to name an index created using this object.
|
* @return A string representation of this object.
|
*/
|
public String toString()
|
{
|
return indexConfig.getAttributeType().getNameOrOID() + ".equality";
|
}
|
|
/**
|
* Get the comparator that must be used to compare index keys
|
* generated by this class.
|
*
|
* @return A byte array comparator.
|
*/
|
public Comparator<byte[]> getComparator()
|
{
|
return comparator;
|
}
|
|
|
|
/**
|
* Generate the set of index keys for an entry.
|
*
|
* @param txn A database transaction to be used if the database need to be
|
* accessed in the course of generating the index keys.
|
* @param entry The entry.
|
* @param keys The set into which the generated keys will be inserted.
|
* @throws DatabaseException If an error occurs in the JE database.
|
*/
|
public void indexEntry(Transaction txn, Entry entry,
|
Set<ASN1OctetString> keys) throws DatabaseException
|
{
|
List<Attribute> attrList =
|
entry.getAttribute(indexConfig.getAttributeType());
|
if (attrList != null)
|
{
|
indexAttribute(attrList, keys);
|
}
|
}
|
|
|
|
/**
|
* Generate the set of index keys to be added and the set of index keys
|
* to be deleted for an entry that has been replaced.
|
*
|
* @param txn A database transaction to be used if the database need to be
|
* accessed in the course of generating the index keys.
|
* @param oldEntry The original entry contents.
|
* @param newEntry The new entry contents.
|
* @param addKeys The set into which the keys to be added will be inserted.
|
* @param delKeys The set into which the keys to be deleted will be inserted.
|
* @throws DatabaseException If an error occurs in the JE database.
|
*/
|
public void replaceEntry(Transaction txn, Entry oldEntry, Entry newEntry,
|
Set<ASN1OctetString> addKeys,
|
Set<ASN1OctetString> delKeys)
|
throws DatabaseException
|
{
|
List<Attribute> attrList;
|
|
attrList = oldEntry.getAttribute(indexConfig.getAttributeType());
|
Set<ASN1OctetString> oldSet = new HashSet<ASN1OctetString>();
|
indexAttribute(attrList, oldSet);
|
|
attrList = newEntry.getAttribute(indexConfig.getAttributeType());
|
Set<ASN1OctetString> newSet = new HashSet<ASN1OctetString>();
|
indexAttribute(attrList, newSet);
|
|
HashSet<ASN1OctetString> removeSet = new HashSet<ASN1OctetString>(oldSet);
|
removeSet.removeAll(newSet);
|
for (ASN1OctetString k : removeSet)
|
{
|
delKeys.add(k);
|
}
|
|
HashSet<ASN1OctetString> addSet = new HashSet<ASN1OctetString>(newSet);
|
addSet.removeAll(oldSet);
|
for (ASN1OctetString k : addSet)
|
{
|
addKeys.add(k);
|
}
|
}
|
|
|
|
|
/**
|
* Generate the set of index keys to be added and the set of index keys
|
* to be deleted for an entry that was modified.
|
*
|
* @param txn A database transaction to be used if the database need to be
|
* accessed in the course of generating the index keys.
|
* @param oldEntry The original entry contents.
|
* @param newEntry The new entry contents.
|
* @param mods The set of modifications that were applied to the entry.
|
* @param addKeys The set into which the keys to be added will be inserted.
|
* @param delKeys The set into which the keys to be deleted will be inserted.
|
* @throws DatabaseException If an error occurs in the JE database.
|
*/
|
public void modifyEntry(Transaction txn, Entry oldEntry, Entry newEntry,
|
List<Modification> mods,
|
Set<ASN1OctetString> addKeys,
|
Set<ASN1OctetString> delKeys)
|
throws DatabaseException
|
{
|
// Optimize for the case where there are no attribute options
|
// involved and only simple addition and deletion of individual values.
|
// An issue with attribute options is that the values can not be assumed
|
// to be unique.
|
|
List<Attribute> beforeList;
|
List<Attribute> afterList;
|
beforeList = oldEntry.getAttribute(indexConfig.getAttributeType());
|
afterList = newEntry.getAttribute(indexConfig.getAttributeType());
|
|
boolean hasOptions = false;
|
|
if (beforeList != null)
|
{
|
for (Attribute a : beforeList)
|
{
|
if (a.hasOptions())
|
{
|
hasOptions = true;
|
break;
|
}
|
}
|
}
|
|
if (afterList != null)
|
{
|
for (Attribute a : afterList)
|
{
|
if (a.hasOptions())
|
{
|
hasOptions = true;
|
break;
|
}
|
}
|
}
|
|
boolean hasOnlySimpleMods = true;
|
List<Modification> simpleMods = new ArrayList<Modification>();
|
for (Modification mod : mods)
|
{
|
Attribute modAttr = mod.getAttribute();
|
AttributeType modAttrType = modAttr.getAttributeType();
|
if (modAttrType.equals(indexConfig.getAttributeType()))
|
{
|
if (modAttr.hasOptions())
|
{
|
hasOptions = true;
|
}
|
switch (mod.getModificationType())
|
{
|
case ADD:
|
simpleMods.add(mod);
|
break;
|
|
case DELETE:
|
if (!modAttr.hasValue())
|
{
|
hasOnlySimpleMods = false;
|
}
|
else
|
{
|
simpleMods.add(mod);
|
}
|
break;
|
|
default:
|
hasOnlySimpleMods = false;
|
break;
|
}
|
}
|
}
|
|
if (hasOnlySimpleMods && !hasOptions)
|
{
|
// This is the optimized case where there are no attribute options
|
// involved and only simple addition and deletion of individual values.
|
// It should be efficient for adding and/or deleting a few values of an
|
// attribute with lots of values.
|
for (Modification mod : simpleMods)
|
{
|
Set<AttributeValue> values = mod.getAttribute().getValues();
|
Set<ASN1OctetString> keys =
|
new HashSet<ASN1OctetString>(values.size());
|
indexValues(values, keys);
|
|
switch (mod.getModificationType())
|
{
|
case ADD:
|
for (ASN1OctetString key : keys)
|
{
|
if (delKeys.contains(key))
|
{
|
delKeys.remove(key);
|
}
|
else
|
{
|
addKeys.add(key);
|
}
|
}
|
break;
|
|
case DELETE:
|
for (ASN1OctetString key : keys)
|
{
|
if (addKeys.contains(key))
|
{
|
addKeys.remove(key);
|
}
|
else
|
{
|
delKeys.add(key);
|
}
|
}
|
break;
|
}
|
}
|
}
|
else
|
{
|
// This is the non-optimized case that should be correct in all cases.
|
|
Set<ASN1OctetString> oldSet = new HashSet<ASN1OctetString>();
|
indexAttribute(beforeList, oldSet);
|
|
Set<ASN1OctetString> newSet = new HashSet<ASN1OctetString>();
|
indexAttribute(afterList, newSet);
|
|
HashSet<ASN1OctetString> removeSet = new HashSet<ASN1OctetString>(oldSet);
|
removeSet.removeAll(newSet);
|
delKeys.addAll(removeSet);
|
|
HashSet<ASN1OctetString> addSet = new HashSet<ASN1OctetString>(newSet);
|
addSet.removeAll(oldSet);
|
addKeys.addAll(addSet);
|
}
|
}
|
|
/**
|
* Generate the set of index keys for a set of attribute values.
|
* @param values The set of attribute values to be indexed.
|
* @param keys The set into which the keys will be inserted.
|
*/
|
private void indexValues(Set<AttributeValue> values,
|
Set<ASN1OctetString> keys)
|
{
|
if (values == null) return;
|
|
for (AttributeValue value : values)
|
{
|
try
|
{
|
byte[] keyBytes = value.getNormalizedValue().value();
|
|
keys.add(new ASN1OctetString(keyBytes));
|
}
|
catch (DirectoryException e)
|
{
|
if (debugEnabled())
|
{
|
debugCaught(DebugLogLevel.ERROR, e);
|
}
|
}
|
}
|
}
|
|
/**
|
* Generate the set of index keys for an attribute.
|
* @param attrList The attribute to be indexed.
|
* @param keys The set into which the keys will be inserted.
|
*/
|
private void indexAttribute(List<Attribute> attrList,
|
Set<ASN1OctetString> keys)
|
{
|
if (attrList == null) return;
|
|
for (Attribute attr : attrList)
|
{
|
indexValues(attr.getValues(), keys);
|
}
|
}
|
|
}
|
