java根据身份层级自定义排序规则

╰+攻爆jí腚メ 2023-06-08 12:59 166阅读 0赞

## 需求： ##

根据身份的上下层级关系进行排序：  
1、没有上级的排在前。  
2、有上级的，按升序排列，即下级在前，上级在后。

## 实现： ##

建一个简单的身份类

/** * @author chenqi * @version V1.0 * @ClassName: identity * @Description: 身份类 * @Date 2019/10/18 15:02 */
    @Data
    public class identity implements Serializable { 
         private static final long serialVersionUID = 3403724365164506666L;
    
        /** 身份id */
        private Long id;
    
        /** 身份上级id */
        private Long parentId;
    }

推荐使用**lombok**，可以省略get、set方法代码

**实现代码：**

public static void main(String[] args) { 
            List<identity> list = new ArrayList<>();
            //i1的上级是i4
            identity i1 = new identity();
            i1.setId(1L);
            i1.setParentId(4L);
            //i2没有上级
            identity i2 = new identity();
            i2.setId(2L);
            //i3的上级是i1
            identity i3 = new identity();
            i3.setId(3L);
            i3.setParentId(1L);
            //i4的上级是0 则i4是最高级
            identity i4 = new identity();
            i4.setId(4L);
            i4.setParentId(0L);
            //i5的上级是i4
            identity i5 = new identity();
            i5.setId(5L);
            i5.setParentId(4L);
            list.add(i1);
            list.add(i2);
            list.add(i3);
            list.add(i4);
            list.add(i5);
            System.out.println("===排序前===");
            list.stream().forEach(item -> { 
                System.out.println(item.getId());
            });
            Collections.sort(list, new Comparator<identity>() { 
                @Override
                public int compare(identity i1, identity i2) { 
                    if(i1 == i2){ 
                        return 0;
                    }else if(CheckUtils.objCheckNull(i1.getParentId())){ 
                        return -1;//上级id为空的排在前
                    }else if(i1.getParentId().intValue() == i2.getId().intValue()) { 
                        return -1;//i1 的上级是 i2 则往前排
                    }else{ 
                        return 1;
                    }
                }
            });
            System.out.println("===排序后===");
            list.stream().forEach(item -> { 
                System.out.println(item.getId());
            });
        }

**运行结果：**  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MDYyMzczNg_size_16_color_FFFFFF_t_70]  
**思路：**  
通过重写Comparator类的compare()方法进行自定义排序规则，该方法接收两个排序对象入参，如：i1和i2，返回int值。即当返回-1时，则i1在前，i2在后，反之，当返回1时，i1在后，i2在前。

至此，完成了对有上下关系层级的数据进行自定义规则排序。  
如果该文章有帮助到您，就留言点个赞吧！您的支持与肯定是我持续更新最大的动力。

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MDYyMzczNg_size_16_color_FFFFFF_t_70]: /images/20230531/9c0fc3325c1440b09d7c845a2b0822c9.png