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

com.hazelcast.util.SortingUtil Maven / Gradle / Ivy

There is a newer version: 4.5.4
Show newest version
/*
 * Copyright (c) 2008-2018, Hazelcast, Inc. All Rights Reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.hazelcast.util;

import com.hazelcast.internal.util.ResultSet;
import com.hazelcast.query.PagingPredicate;
import com.hazelcast.query.PagingPredicateAccessor;
import com.hazelcast.query.impl.QueryableEntry;

import java.util.Collections;
import java.util.Comparator;
import java.util.List;
import java.util.Map;

import static com.hazelcast.query.PagingPredicateAccessor.getNearestAnchorEntry;

/**
 * Utility class for generating Comparators to be used in sort methods specific to hazelcast classes.
 */
public final class SortingUtil {

    private SortingUtil() {
    }

    public static int compare(Comparator comparator, IterationType iterationType,
                              Map.Entry entry1, Map.Entry entry2) {
        if (comparator != null) {
            int result = comparator.compare(entry1, entry2);
            if (result != 0) {
                return result;
            }
            return compareIntegers(entry1.getKey().hashCode(), entry2.getKey().hashCode());
        }

        Object comparable1;
        Object comparable2;
        switch (iterationType) {
            case KEY:
                comparable1 = entry1.getKey();
                comparable2 = entry2.getKey();
                break;
            case VALUE:
                comparable1 = entry1.getValue();
                comparable2 = entry2.getValue();
                break;
            default:
                // Possibly ENTRY
                // If entries are comparable, we can compare them
                if (entry1 instanceof Comparable && entry2 instanceof Comparable) {
                    comparable1 = entry1;
                    comparable2 = entry2;
                } else {
                    // Otherwise, comparing entries directly is not meaningful.
                    // So keys can be used instead of map entries.
                    comparable1 = entry1.getKey();
                    comparable2 = entry2.getKey();
                }
                break;
        }
        checkIfComparable(comparable1);
        checkIfComparable(comparable2);

        int result = ((Comparable) comparable1).compareTo(comparable2);
        if (result != 0) {
            return result;
        }
        return compareIntegers(entry1.getKey().hashCode(), entry2.getKey().hashCode());
    }

    private static void checkIfComparable(Object comparable) {
        if (comparable instanceof Comparable) {
            return;
        }
        throw new IllegalArgumentException("Not comparable " + comparable);
    }

    /**
     * Compares two integers by considering their signs.
     * 

* Suppose that * i1 = -500.000.000 * i2 = 2.000.000.000 *

* Normally "i1 < i2", but if we use "i1 - i2" for comparison, * i1 - i2 = -500.000.000 - 2.000.000.000 and we may accept the result as "-2.500.000.000". * But the actual result is "1.794.967.296" because of overflow between * positive and negative integer bounds. *

* So, if we use "i1 - i2" for comparison, since the result is greater than 0, * "i1" is accepted as bigger that "i2". But in fact "i1" is smaller than "i2". * Therefore, "i1 - i2" is not a good method for comparison between signed integers. * * @param i1 First number to compare with second one * @param i2 Second number to compare with first one * @return +1 if i1 > i2, -1 if i2 > i1, 0 if i1 and i2 are equals */ private static int compareIntegers(int i1, int i2) { // i1 - i2 is not good way for comparison if (i1 > i2) { return +1; } else if (i2 > i1) { return -1; } else { return 0; } } public static Comparator newComparator(final Comparator comparator, final IterationType iterationType) { return new Comparator() { public int compare(Map.Entry entry1, Map.Entry entry2) { return SortingUtil.compare(comparator, iterationType, entry1, entry2); } }; } private static Comparator newComparator(final PagingPredicate pagingPredicate) { return new Comparator() { public int compare(QueryableEntry entry1, QueryableEntry entry2) { return SortingUtil.compare(pagingPredicate.getComparator(), pagingPredicate.getIterationType(), entry1, entry2); } }; } public static List getSortedSubList(List list, PagingPredicate pagingPredicate, Map.Entry nearestAnchorEntry) { if (pagingPredicate == null || list.isEmpty()) { return list; } Comparator comparator = newComparator(pagingPredicate); Collections.sort(list, comparator); int nearestPage = nearestAnchorEntry.getKey(); int pageSize = pagingPredicate.getPageSize(); int page = pagingPredicate.getPage(); long totalSize = pageSize * ((long) page - nearestPage); if (list.size() > totalSize) { // it's safe to cast totalSize back to int here since it's limited by the list size list = list.subList(0, (int) totalSize); } return list; } @SuppressWarnings("unchecked") public static ResultSet getSortedQueryResultSet(List list, PagingPredicate pagingPredicate, IterationType iterationType) { if (list.isEmpty()) { return new ResultSet(); } Comparator comparator = SortingUtil.newComparator(pagingPredicate.getComparator(), iterationType); Collections.sort(list, comparator); Map.Entry nearestAnchorEntry = getNearestAnchorEntry(pagingPredicate); int nearestPage = nearestAnchorEntry.getKey(); int page = pagingPredicate.getPage(); int pageSize = pagingPredicate.getPageSize(); long begin = pageSize * ((long) page - nearestPage - 1); int size = list.size(); if (begin > size) { return new ResultSet(); } long end = begin + pageSize; if (end > size) { end = size; } setAnchor(list, pagingPredicate, nearestPage); // it's safe to cast begin and end back to int here since they are limited by the list size List subList = list.subList((int) begin, (int) end); return new ResultSet(subList, iterationType); } public static boolean compareAnchor(PagingPredicate pagingPredicate, QueryableEntry queryEntry, Map.Entry nearestAnchorEntry) { if (pagingPredicate == null) { return true; } Map.Entry anchor = nearestAnchorEntry.getValue(); if (anchor == null) { return true; } Comparator comparator = pagingPredicate.getComparator(); IterationType iterationType = pagingPredicate.getIterationType(); return SortingUtil.compare(comparator, iterationType, anchor, queryEntry) < 0; } private static void setAnchor(List list, PagingPredicate pagingPredicate, int nearestPage) { if (list.isEmpty()) { return; } int size = list.size(); int pageSize = pagingPredicate.getPageSize(); int page = pagingPredicate.getPage(); for (int i = pageSize; i <= size && nearestPage < page; i += pageSize) { Map.Entry anchor = list.get(i - 1); nearestPage++; PagingPredicateAccessor.setAnchor(pagingPredicate, nearestPage, anchor); } } }





© 2015 - 2024 Weber Informatics LLC | Privacy Policy