Java8提供的Stream方式进行分组GroupingBy

  有时我们需要对集合进行分组操作，这时可以使用Java8提供的Stream方式进行分组。挺好用的，此处记录下。直接贴code：

  Road实体：

@Data
@NoArgsConstructor
@AllArgsConstructor
public class Road {

    /**
     * 名称
     */
    private String name;

    /**
     * 道路长度
     */
    private double len;

    /**
     * 社区划分
     */
    private String community;
}

  Stream分组：

        Road road1 = new Road("荷花大道_东段", 100.2, "荷花社区");
        Road road2 = new Road("荷花大道西段", 88.2, "荷花社区");
        Road road3 = new Road("剑北大道_东段", 99.2, "剑北社区");
        Road road4 = new Road("剑北大道西段", 111.4, "剑北社区");
        Road road5 = new Road("剑北大道_南段", 68.9, "剑北社区")

 

        List<Road> roadList = new ArrayList<>();
        roadList.add(road1);
        roadList.add(road2);
        roadList.add(road3);
        roadList.add(road4);
        roadList.add(road5);

 

        // ①根据 社区属性 分组
        // 两种方式皆可
        Map<String, List<Road>> roadMap1 = roadList.stream().collect(Collectors.groupingBy(Road::getCommunity));
        Map<String, List<Road>> roadMap2 = roadList.stream().collect(Collectors.groupingBy(Road::getCommunity, Collectors.toList()));

 

        // ②将 道路长度大于100 的分为一组
        Map<String, List<Road>> roadMap3 = roadList.stream().collect(Collectors.groupingBy(x -> {
            if (x.getLen() > 100) {
                return "大于100";
            } else {
                return "小于等于100";
            }
        }));

 

        // ③想得到一个通过道路长度分组并有序的结果（此处道路长度都不一样，没必要分组，只是想体现有序的返回）
        TreeMap<Double, List<Road>> roadMap4 = roadList.stream().collect(Collectors.groupingBy(Road::getLen, TreeMap::new, Collectors.toList()));

 

        // ④将名字中的 大道 替换为'' 并进行分组（主要是这个，平时用的时候，有时会搞忘）
        Map<String, List<Road>> roadMap5 = roadList.stream().collect(Collectors.groupingBy(x -> x.getName().replace("_", ""), Collectors.toList()));

 

        // 按自己习惯打印输出，roadMap1...roadMap5，进行结果验证比对

  代码直接复制即可运行，可进行结果验证比对。

既然选择了远方，便只顾风雨兼程。-- 烟沙九洲