使用 Stream API 简化集合操作

使用 Stream API 简化集合操作

---

 

在 Java 8 中，引入了 Stream API，使得我们可以通过链式调用来进行集合操作，避免了传统的循环迭代方式。本文将介绍如何使用 Stream API 简化集合操作，并通过代码示例进行演示。

 

一、常用的 Stream API 操作

在 Stream API 中，常用的操作包括：

• map：对集合中的元素进行映射操作；
• filter：对集合中的元素进行过滤操作；
• distinct：去重操作；
• sorted：排序操作；
• limit：限制返回结果的个数；
• collect：将结果转换为集合。

 

二、示例代码

下面通过一个示例代码来演示如何使用 Stream API 简化集合操作。假设我们有一个包含多个字符串的集合，我们需要对其中的字符串进行如下操作：

1. 将字符串全部转换为大写；
2. 过滤出包含字母 "A" 的字符串；
3. 去重；
4. 按字符串长度进行排序；
5. 返回前 3 个字符串。

使用传统的循环迭代方式实现代码如下：

List<String> strings = Arrays.asList("apple", "banana", "orange", "pear", "grape", "lemon");

List<String> result = new ArrayList<>();
for (String string : strings) {
    if (string.contains("A")) {
        result.add(string.toUpperCase());
    }
}

Set<String> set = new HashSet<>();
for (String string : result) {
    set.add(string);
}

List<String> sortedList = new ArrayList<>(set);
Collections.sort(sortedList, new Comparator<String>() {
    @Override
    public int compare(String o1, String o2) {
        return o1.length() - o2.length();
    }
});

List<String> finalList = new ArrayList<>();
for (int i = 0; i < Math.min(3, sortedList.size()); i++) {
    finalList.add(sortedList.get(i));
}

System.out.println(finalList);

 

使用 Stream API 实现代码如下：

List<String> strings = Arrays.asList("apple", "banana", "orange", "pear", "grape", "lemon");

List<String> finalList = strings.stream()
        .map(String::toUpperCase)
        .filter(s -> s.contains("A"))
        .distinct()
        .sorted(Comparator.comparing(String::length))
        .limit(3)
        .collect(Collectors.toList());

System.out.println(finalList);

 

从上面的代码可以看出，使用 Stream API 可以大大简化集合操作的代码，使得代码更加清晰、简洁。