package org.alfresco.repo.site; import java.text.Collator; import java.util.Comparator; import java.util.List; import org.alfresco.query.CannedQuerySortDetails.SortOrder; import org.alfresco.service.cmr.site.SiteInfo; import org.alfresco.service.cmr.site.SiteService; import org.alfresco.util.Pair; public class SiteMembershipComparator implements Comparator { public enum Type { SITES, MEMBERS } private List> sortPairs; private static Collator collator = Collator.getInstance(); private Type comparatorType; public SiteMembershipComparator(List> sortPairs, Type comparatorType) { if (sortPairs.size() < 1) { throw new IllegalArgumentException("Must provide at least one sort criterion"); } this.sortPairs = sortPairs; this.comparatorType = comparatorType; } private int safeCompare(Comparable o1, T o2) { int ret = 0; if (o1 == null) { if (o2 == null) { ret = 0; } else { ret = -1; } } else { if (o2 == null) { ret = 1; } else { ret = o1.compareTo(o2); } } return ret; } private int safeCompare(String s1, String s2) { int ret = 0; if (s1 == null) { if (s2 == null) { ret = 0; } else { ret = -1; } } else { if (s2 == null) { ret = 1; } else { ret = collator.compare(s1, s2); } } return ret; } private int compareMembersBody(String personId1, String personId2, String lastName1, String lastName2, String siteRole1, String siteRole2, int personId, int firstName, int lastName, int siteRole, int ret) { for (Pair pair : sortPairs) { Object name = pair.getFirst(); SortOrder sortOrder = pair.getSecond(); int multiplier = sortOrder.equals(SortOrder.ASCENDING) ? 1 : -1; if (name.equals(SiteService.SortFields.FirstName)) { ret = firstName * multiplier; } else if (name.equals(SiteService.SortFields.LastName)) { if (lastName1 == null || lastName2 == null) { continue; } ret = lastName * multiplier; } else if (name.equals(SiteService.SortFields.Role)) { if (siteRole1 == null || siteRole2 == null) { continue; } ret = siteRole * multiplier; } else if (name.equals(SiteService.SortFields.Username)) { if (personId1 == null || personId2 == null) { continue; } ret = personId * multiplier; } if (ret != 0) { break; } } return ret; } private int compareSitesBody(String shortName1, String shortName2, String siteRole1, String siteRole2, String siteTitle1, String siteTitle2, int siteShortName, int siteRole, int siteTitle, int ret) { for (Pair pair : sortPairs) { Object name = pair.getFirst(); SortOrder sortOrder = pair.getSecond(); int multiplier = sortOrder.equals(SortOrder.ASCENDING) ? 1 : -1; if (name.equals(SiteService.SortFields.SiteShortName)) { if (shortName1 == null || shortName2 == null) { continue; } ret = siteShortName * multiplier; } else if (name.equals(SiteService.SortFields.SiteTitle)) { if (siteTitle1 == null || siteTitle2 == null) { continue; } ret = siteTitle * multiplier; } else if (name.equals(SiteService.SortFields.Role)) { if (siteRole1 == null || siteRole2 == null) { continue; } ret = siteRole * multiplier; } if (ret != 0) { break; } } return ret; } @Override public int compare(SiteMembership o1, SiteMembership o2) { String personId1 = o1.getPersonId(); String personId2 = o2.getPersonId(); SiteInfo siteInfo1 = o1.getSiteInfo(); SiteInfo siteInfo2 = o2.getSiteInfo(); String shortName1 = siteInfo1.getShortName(); String shortName2 = siteInfo2.getShortName(); String firstName1 = o1.getFirstName(); String firstName2 = o2.getFirstName(); String lastName1 = o1.getLastName(); String lastName2 = o2.getLastName(); String siteRole1 = o1.getRole(); String siteRole2 = o2.getRole(); String siteTitle1 = siteInfo1.getTitle(); String siteTitle2 = siteInfo2.getTitle(); int personId = safeCompare(personId1, personId2); int firstName = safeCompare(firstName1, firstName2); int siteShortName = safeCompare(shortName1, shortName2); int lastName = safeCompare(lastName1, lastName2); int siteRole = safeCompare(siteRole1, siteRole2); int siteTitle = safeCompare(siteTitle1, siteTitle2); if (siteRole == 0 && siteShortName == 0 && personId == 0) { // equals contract return 0; } int ret = 0; switch (comparatorType) { case SITES: { ret = compareSitesBody(shortName1, shortName2, siteRole1, siteRole2, siteTitle1, siteTitle2, siteShortName, siteRole, siteTitle, ret); break; } case MEMBERS: { ret = compareMembersBody(personId1, personId2, lastName1, lastName2, siteRole1, siteRole2, personId, firstName, lastName, siteRole, ret); break; } } return ret; } }