All Downloads are FREE. Search and download functionalities are using the official Maven repository.

com.unboundid.ldap.sdk.EntrySorter Maven / Gradle / Ivy

/*
 * Copyright 2007-2017 UnboundID Corp.
 * All Rights Reserved.
 */
/*
 * Copyright (C) 2008-2017 UnboundID Corp.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License (GPLv2 only)
 * or the terms of the GNU Lesser General Public License (LGPLv2.1 only)
 * as published by the Free Software Foundation.
 *
 * 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 .
 */
package com.unboundid.ldap.sdk;



import java.io.Serializable;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.Collections;
import java.util.Comparator;
import java.util.Iterator;
import java.util.List;
import java.util.SortedSet;
import java.util.TreeSet;

import com.unboundid.asn1.ASN1OctetString;
import com.unboundid.ldap.matchingrules.MatchingRule;
import com.unboundid.ldap.sdk.controls.SortKey;
import com.unboundid.ldap.sdk.schema.Schema;
import com.unboundid.util.ThreadSafety;
import com.unboundid.util.ThreadSafetyLevel;

import static com.unboundid.util.Debug.*;
import static com.unboundid.util.StaticUtils.*;



/**
 * This class provides a mechanism for client-side entry sorting.  Sorting may
 * be based on attributes contained in the entry, and may also be based on the
 * hierarchical location of the entry in the DIT.  The sorting may be applied
 * to any collection of entries, including the entries included in a
 * {@link SearchResult} object.
 * 

* This class provides a client-side alternative to the use of the * {@link com.unboundid.ldap.sdk.controls.ServerSideSortRequestControl}. * Client-side sorting is most appropriate for small result sets, as it requires * all entries to be held in memory at the same time. It is a good alternative * to server-side sorting when the overhead of sorting should be distributed * across client systems rather than on the server, and in cases in which the * target directory server does not support the use of the server-side sort * request control. *

* For best results, a {@link Schema} object may be used to provide an * indication as to which matching rules should be used to perform the ordering. * If no {@code Schema} object is provided, then all ordering will be performed * using case-ignore string matching. *

*

Example

* The following example may be used to obtain a sorted set of search result * entries, ordered first by sn and then by givenName, without consideration for * hierarchy: *
 * SearchResult searchResult = connection.search("dc=example,dc=com",
 *      SearchScope.SUB, Filter.createEqualityFilter("sn", "Smith"));
 *
 * EntrySorter entrySorter = new EntrySorter(false,
 *      new SortKey("sn"), new SortKey("givenName"));
 * SortedSet<Entry> sortedEntries =
 *     entrySorter.sort(searchResult.getSearchEntries());
 * 
*/ @ThreadSafety(level=ThreadSafetyLevel.COMPLETELY_THREADSAFE) public final class EntrySorter implements Comparator, Serializable { /** * The serial version UID for this serializable class. */ private static final long serialVersionUID = 7606107105238612142L; // Indicates whether entries should be sorted based on hierarchy. private final boolean sortByHierarchy; // The set of sort keys for attribute-level sorting. private final List sortKeys; // The schema to use to make the comparison, if available. private final Schema schema; /** * Creates a new entry sorter that will sort entries based only on hierarchy. * Superior entries (that is, entries closer to the root of the DIT) will be * ordered before subordinate entries. Entries below the same parent will be * sorted lexicographically based on their normalized DNs. */ public EntrySorter() { this(true, null, Collections.emptyList()); } /** * Creates a new entry sorter with the provided information. * * @param sortByHierarchy Indicates whether entries should be sorted * hierarchically, such that superior entries will * be ordered before subordinate entries. * @param sortKeys A list of sort keys that define the order in which * attributes should be compared. It may be empty * (but never {@code null}) if sorting should be done * only based on hierarchy. */ public EntrySorter(final boolean sortByHierarchy, final SortKey... sortKeys) { this(sortByHierarchy, null, Arrays.asList(sortKeys)); } /** * Creates a new entry sorter with the provided information. * * @param sortByHierarchy Indicates whether entries should be sorted * hierarchically, such that superior entries will * be ordered before subordinate entries. * @param schema The schema to use to make the determination. It * may be {@code null} if no schema is available. * @param sortKeys A list of sort keys that define the order in which * attributes should be compared. It may be empty * (but never {@code null}) if sorting should be done * only based on hierarchy. */ public EntrySorter(final boolean sortByHierarchy, final Schema schema, final SortKey... sortKeys) { this(sortByHierarchy, schema, Arrays.asList(sortKeys)); } /** * Creates a new entry sorter with the provided information. * * @param sortByHierarchy Indicates whether entries should be sorted * hierarchically, such that superior entries will * be ordered before subordinate entries. * @param sortKeys A list of sort keys that define the order in which * attributes should be compared. It may be empty or * {@code null} if sorting should be done only based * on hierarchy. */ public EntrySorter(final boolean sortByHierarchy, final List sortKeys) { this(sortByHierarchy, null, sortKeys); } /** * Creates a new entry sorter with the provided information. * * @param sortByHierarchy Indicates whether entries should be sorted * hierarchically, such that superior entries will * be ordered before subordinate entries. * @param schema The schema to use to make the determination. It * may be {@code null} if no schema is available. * @param sortKeys A list of sort keys that define the order in which * attributes should be compared. It may be empty or * {@code null} if sorting should be done only based * on hierarchy. */ public EntrySorter(final boolean sortByHierarchy, final Schema schema, final List sortKeys) { this.sortByHierarchy = sortByHierarchy; this.schema = schema; if (sortKeys == null) { this.sortKeys = Collections.emptyList(); } else { this.sortKeys = Collections.unmodifiableList(new ArrayList(sortKeys)); } } /** * Sorts the provided collection of entries according to the criteria defined * in this entry sorter. * * @param entries The collection of entries to be sorted. * * @return A sorted set, ordered in accordance with this entry sorter. */ public SortedSet sort(final Collection entries) { final TreeSet entrySet = new TreeSet(this); entrySet.addAll(entries); return entrySet; } /** * Compares the provided entries to determine the order in which they should * be placed in a sorted list. * * @param e1 The first entry to be compared. * @param e2 The second entry to be compared. * * @return A negative value if the first entry should be ordered before the * second, a positive value if the first entry should be ordered * after the second, or zero if the entries should have an equivalent * order. */ public int compare(final Entry e1, final Entry e2) { DN parsedDN1 = null; DN parsedDN2 = null; if (sortByHierarchy) { try { parsedDN1 = e1.getParsedDN(); parsedDN2 = e2.getParsedDN(); if (parsedDN1.isAncestorOf(parsedDN2, false)) { return -1; } else if (parsedDN2.isAncestorOf(parsedDN1, false)) { return 1; } } catch (LDAPException le) { debugException(le); } } for (final SortKey k : sortKeys) { final String attrName = k.getAttributeName(); final Attribute a1 = e1.getAttribute(attrName); final Attribute a2 = e2.getAttribute(attrName); if ((a1 == null) || (! a1.hasValue())) { if ((a2 == null) || (! a2.hasValue())) { // Neither entry has the attribute. Continue on with the next // attribute. continue; } else { // The first entry does not have the attribute but the second does. // The first entry should be ordered after the second. return 1; } } else { if ((a2 == null) || (! a2.hasValue())) { // The first entry has the attribute but the second does not. The // first entry should be ordered before the second. return -1; } } final MatchingRule matchingRule = MatchingRule.selectOrderingMatchingRule( attrName, k.getMatchingRuleID(), schema); if (k.reverseOrder()) { // Find the largest value for each attribute, and pick the larger of the // two. ASN1OctetString v1 = null; for (final ASN1OctetString s : a1.getRawValues()) { if (v1 == null) { v1 = s; } else { try { if (matchingRule.compareValues(s, v1) > 0) { v1 = s; } } catch (LDAPException le) { debugException(le); } } } ASN1OctetString v2 = null; for (final ASN1OctetString s : a2.getRawValues()) { if (v2 == null) { v2 = s; } else { try { if (matchingRule.compareValues(s, v2) > 0) { v2 = s; } } catch (LDAPException le) { debugException(le); } } } try { final int value = matchingRule.compareValues(v2, v1); if (value != 0) { return value; } } catch (LDAPException le) { debugException(le); } } else { // Find the smallest value for each attribute, and pick the larger of // the two. ASN1OctetString v1 = null; for (final ASN1OctetString s : a1.getRawValues()) { if (v1 == null) { v1 = s; } else { try { if (matchingRule.compareValues(s, v1) < 0) { v1 = s; } } catch (LDAPException le) { debugException(le); } } } ASN1OctetString v2 = null; for (final ASN1OctetString s : a2.getRawValues()) { if (v2 == null) { v2 = s; } else { try { if (matchingRule.compareValues(s, v2) < 0) { v2 = s; } } catch (LDAPException le) { debugException(le); } } } try { final int value = matchingRule.compareValues(v1, v2); if (value != 0) { return value; } } catch (LDAPException le) { debugException(le); } } } // If we've gotten here, then there is no difference in hierarchy or // sort attributes. Compare the DNs as a last resort. try { if (parsedDN1 == null) { parsedDN1 = e1.getParsedDN(); } if (parsedDN2 == null) { parsedDN2 = e2.getParsedDN(); } return parsedDN1.compareTo(parsedDN2); } catch (LDAPException le) { debugException(le); final String lowerDN1 = toLowerCase(e1.getDN()); final String lowerDN2 = toLowerCase(e2.getDN()); return lowerDN1.compareTo(lowerDN2); } } /** * Retrieves a hash code for this entry sorter. * * @return A hash code for this entry sorter. */ @Override() public int hashCode() { int hashCode = 0; if (sortByHierarchy) { hashCode++; } for (final SortKey k : sortKeys) { if (k.reverseOrder()) { hashCode *= -31; } else { hashCode *= 31; } hashCode += toLowerCase(k.getAttributeName()).hashCode(); } return hashCode; } /** * Indicates whether the provided object is equal to this entry sorter. * * @param o The object for which to make the determination. * * @return {@code true} if the provided object is equal to this entry sorter, * or {@code false} if not. */ @Override() public boolean equals(final Object o) { if (o == null) { return false; } if (o == this) { return true; } if (! (o instanceof EntrySorter)) { return false; } final EntrySorter s = (EntrySorter) o; if (sortByHierarchy != s.sortByHierarchy) { return false; } return sortKeys.equals(s.sortKeys); } /** * Retrieves a string representation of this entry sorter. * * @return A string representation of this entry sorter. */ @Override() public String toString() { final StringBuilder buffer = new StringBuilder(); toString(buffer); return buffer.toString(); } /** * Appends a string representation of this entry sorter to the provided * buffer. * * @param buffer The buffer to which the string representation should be * appended. */ public void toString(final StringBuilder buffer) { buffer.append("EntrySorter(sortByHierarchy="); buffer.append(sortByHierarchy); buffer.append(", sortKeys={"); final Iterator iterator = sortKeys.iterator(); while (iterator.hasNext()) { iterator.next().toString(buffer); if (iterator.hasNext()) { buffer.append(", "); } } buffer.append("})"); } }




© 2015 - 2024 Weber Informatics LLC | Privacy Policy