Java集合自定义字段排序

前言

针对项目中需要对一些集合进行自定义字段排序，该方法包含object类型和map类型的排序，输入需要排序的字段名或键key，以及排序先后的sortOrder规则集合。

package cn.shencom.util;

import java.lang.reflect.Field;
import java.util.*;

public class DataSortUtil {

    // 南山一图统揽：图表的街道默认排序
    public static final List<Object> sortOrder = Arrays.asList("南山街道","南头街道","西丽街道","沙河街道","蛇口街道","招商街道","粤海街道","桃源街道");

    /**
     * 自定义字段排序
     *
     * @param list 待排序集合
     * @param fieldName 排序字段
     * @param sortOrder 排序规则
     */
    public static void sort(List<?> list, String fieldName, List<Object> sortOrder) {
        // 创建一个自定义的Comparator
        Comparator<Object> customComparator = Comparator.comparing(obj -> {
            if (obj instanceof Map) {
                Map<?, ?> map = (Map<?, ?>) obj;
                Object value = map.get(fieldName);
                return sortOrder.indexOf(value);
            } else {
                try {
                    Field field = obj.getClass().getDeclaredField(fieldName);
                    field.setAccessible(true);
                    Object value = field.get(obj);
                    return sortOrder.indexOf(value);
                } catch (NoSuchFieldException | IllegalAccessException e) {
                    e.printStackTrace();
                }
            }
            return 0;
        });

        // 对街道进行排序
        Collections.sort((List<Object>) list, customComparator);
    }

    public static void main(String[] args) {
        List<TestDto> streets = new ArrayList<>();
        streets.add(new TestDto("桃源街道"));
        streets.add(new TestDto("南头街道"));
        streets.add(new TestDto("粤海街道"));
        streets.add(new TestDto("蛇口街道"));

        // 对街道进行排序
        sort(streets, "streetName" , sortOrder);

        // 打印排序后的结果
        for (Object street : streets) {
            System.out.println(street);
        }



        List<Map<String, Object>> streetList = new ArrayList<>();
        Map<String, Object> street1 = new HashMap<>();
        street1.put("name", "Street C");
        street1.put("length", 100);
        streetList.add(street1);
        Map<String, Object> street2 = new HashMap<>();
        street2.put("name", "Street A");
        street2.put("length", 200);
        streetList.add(street2);

        List<Object> sortOrder = Arrays.asList("Street A", "Street C");
        sort(streetList, "name", sortOrder);

        System.out.println(streetList);
    }

    static class TestDto {
        private String streetName;

        public TestDto(String streetName) {
            this.streetName = streetName;
        }

        public String getStreetName() {
            return streetName;
        }

        public void setStreetName(String streetName) {
            this.streetName = streetName;
        }

        @Override
        public String toString() {
            return "Street: " + streetName;
        }
    }
}