对List实体类里的属性分组排序并赋新值

深碍√TFBOYSˉ_ 2022-04-24 11:12 132阅读 0赞

对一个List<Entity>中的实体类某个或者某几个字段进行分组，然后再对实体类的另外的属性进行排序，并重新赋值

public class Test{
     
        public void test(List<Entity> entities) {
     
            /**
             * 方法一：确保要比较的字段不为null之后可以直接使用lambda
             */
            /*entities = Comparator.comparingInt(entity->fetchGroupKey(entity))
                    .thenComparingInt(entity->entity.getTwoThing());*/
     
            //-----方法二：不使用lambda进行分组排序的方法-------------------
            /**
             * OneThing、TwoThing字段都是String类型，ThreeDate是时间类型
             *
             * 1、对List<Entity>里的OneThing、ThreeDate属性相等的字段（唯一key）进行分组
             * 2、然后根据twoThing属性排序（从小到大）
             */
            entities.sort(this::compare);
     
            /**
             * 重新赋值
             * 3、对根据OneThing、ThreeDate分好组的实体类的TwoThing重新赋值
             */
            int count = 1;
            String preKey = null;
     
            for (int i = 0; i < entities.size(); i++) {
                Entity entity = entities.get(i);
                if (Objects.equals(preKey, fetchGroupKey(entity))) {
                    // count对相同key、已排序的组计数
                    count++;
                } else {
                    count = 1;
                }
     
                // 已经分组排序了，所以可以使用preKey作为比较计数的判断条件，一个preKey遍历一组
                preKey = fetchGroupKey(entity);
                /**
                 * 对TwoThing字段赋值规则：数量count不足两位（每组的实体类数量少于10），前面补0，并将其转为String类型
                 * 如果多于10，就正常赋值
                 * 值得形式如：01 02 03...10 11 12...
                 */
                entity.setTwoThing(String.format("%02d", count));
            }
        }
     
        private int compare(Entity o1, Entity o2) {
            // 以o1.getOneThing()为比较条件，进行分组
            int is = fetchGroupKey(o1).compareTo(fetchGroupKey(o2));
            // 这两个实体类不是同一组的，直接返回，继续比较下面两组
            if (is != 0) {
                return is;
            } else {
                // 这两个实体类是同一组的，进行排序
     
                // 以TwoThing字段的大小作为分组后的排序
                String o1Number = o1.getTwoThing();
                String o2Number = o2.getTwoThing();
                // 对字段进行判空，防止为null时候比较产生异常
                if (null == o1Number && null != o2Number) {
                    return -1;
                }
     
                if (null != o1Number && null == o2Number) {
                    return 1;
                }
     
                if (null == o1Number && null == o2Number) {
                    return 0;
                }
                return Integer.parseInt(o1.getTwoThing()) - Integer.parseInt(o2.getTwoThing());
            }
        }
     
        /**
         * 对该实体类的某几个字段组装唯一key，作为分组条件
         */
        private String fetchGroupKey(Entity entity) {
            // 这里用getTime()是因为效率稍微比toString()高一些
            return entity.getOneThing() + "-" + (null == entity.getDate() ? "" : entity.getThreeDate().getTime());
        }
    }