使用 Stream API 简化集合操作

发布时间 2023-05-03 02:43:22作者: 龙凌云端

使用 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 可以大大简化集合操作的代码,使得代码更加清晰、简洁。