使用FlatMap“展平”一系列数据

在处理业务数据时，偶尔会需要从一系列逗号分隔的数据中取出各个元素并去重。本文将介绍如何通过Java 8中的FlatMap简化这个操作。

介绍需求

假设现在有这样子的一个排班信息：

日期	班组
2019-01-02	Unit-2, Unit-3
2019-01-03	Unit-2
2019-01-04	Unit-1, Unit-3
2019-01-01	Unit-2, Unit-3
2019-01-05	Unit-3
2019-01-06	Unit-1, Unit-2
2019-01-07	Unit-1
2019-01-08	Unit-2, Unit-3

而目标是，从中取出这个表中的所有班组，并去重和排序，即最终的结果是Unit-1, Unit-2, Unit-3。

准备示例数据

在代码中，我们这样来构造这些数据：

// 准备假数据
WorkSchedule workSchedule1 = new WorkSchedule("2019-01-01", "Unit-2, Unit-3");
WorkSchedule workSchedule2 = new WorkSchedule("2019-01-02", "Unit-2");
WorkSchedule workSchedule3 = new WorkSchedule("2019-01-03", "Unit-1, Unit-3");
WorkSchedule workSchedule4 = new WorkSchedule("2019-01-04", "Unit-2, Unit-3");
WorkSchedule workSchedule5 = new WorkSchedule("2019-01-05", "Unit-3");
WorkSchedule workSchedule6 = new WorkSchedule("2019-01-06", "Unit-1, Unit-2");
WorkSchedule workSchedule7 = new WorkSchedule("2019-01-07", "Unit-1");
WorkSchedule workSchedule8 = new WorkSchedule("2019-01-08", "Unit-2, Unit-3");

List<WorkSchedule> workSchedules = new ArrayList<>(8);
workSchedules.add(workSchedule1);
workSchedules.add(workSchedule2);
workSchedules.add(workSchedule3);
workSchedules.add(workSchedule4);
workSchedules.add(workSchedule5);
workSchedules.add(workSchedule6);
workSchedules.add(workSchedule7);
workSchedules.add(workSchedule8);

不使用流的做法

如果不使用流，那么我们可以通过for循环取数据，和使用TreeSet实现去重和排序。

TreeSet<String> unitNames = new TreeSet<>();

for (WorkSchedule workSchedule : workSchedules) {
    String[] arrUnitName = workSchedule.getUnits().split(",");

    for (String unitName : arrUnitName) {
        unitNames.add(unitName.trim());
    }
}

使用流的做法

如果使用流，则可以这样做：

List<String> units = workSchedules.stream()
        // 根据逗号，将班组名拆分成 String[]
        // 这一步将生成8个新的 String[] 对象
        .map(i -> i.getUnits().split(","))
        // 从上面的 String[] 生成一个新的
        .flatMap(Arrays::stream)
        // 切掉每个元素两端的空格
        .map(String::trim)
        // 去重
        .distinct()
        // 排序
        .sorted()
        // 收集到一个List中
        .collect(Collectors.toList());

如果上面代码的注释还是无法让您理解，那么，我还准备了每一步流操作的示意图。(感谢Intellij IDEA提供了这个强大的功能)

首先，在map方法中，List的每个元素都会根据逗号分隔，并生成一个String[]对象。