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

发表于 更新于 分类于 Waline： 阅读次数：

在处理业务数据时，偶尔会需要从一系列逗号分隔的数据中取出各个元素并去重。本文将介绍如何通过 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。

准备示例数据

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

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
// 准备假数据
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 实现去重和排序。

1
2
3
4
5
6
7
8
9
TreeSet<String> unitNames = new TreeSet<>();

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

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

使用流的做法

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

1
2
3
4
5
6
7
8
9
10
11
12
13
14
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[] 对象。

Split

flatmap 方法会将上面 map 方法返回的各个流 “拼接” 成为一个流，即 “展平”。

FlatMap

接下来继续使用 map 方法，对流中的每个元素进行 trim 操作。

Trim

然后调用 distinct 和 sorted 方法实现去重和排序。

Distinct
Sort

最后，使用 collect 方法，将流中的数据 “收集” 到一个 List 中。

Collect