日常记录-java 自定义排序

import org.springframework.util.CollectionUtils;

import java.util.List;
import java.util.function.Function;

public class SortUtil {

    /**
     * 对列表进行排序。如果列表中的元素在指定的顺序列表中不存在，则将它们排序到列表的末尾。
     *
     * @param resultList   要排序的列表
     * @param <T>          列表元素的类型
     * @param <K>          排序键的类型
     * @param keyExtractor 从列表元素中提取排序键的函数式接口
     * @param orderList    指定排序顺序的列表，其中包含排序键
     */
    public static <T, K extends Comparable<K>> void setListOrder(List<T> resultList, Function<T, K> keyExtractor, List<K> orderList) {
        if (CollectionUtils.isEmpty(resultList) || CollectionUtils.isEmpty(orderList)) {
            return;
        }
        resultList.sort((o1, o2) -> {
            K key1 = keyExtractor.apply(o1);
            K key2 = keyExtractor.apply(o2);
            int i1 = orderList.indexOf(key1);
            int a = i1 < 0 ? Integer.MAX_VALUE : i1;
            int i2 = orderList.indexOf(key2);
            int b = i2 < 0 ? Integer.MAX_VALUE : i2;
            return a - b;
        });
    }
}