Java Stream API 操作：让你的代码更加出色-1

Java Stream

Java Stream 是一种强大的数据处理工具，可以帮助开发人员快速高效地处理和转换数据流。使用 Stream 操作可以大大简化代码，使其更具可读性和可维护性，从而提高开发效率。本文将为您介绍 Java Stream 操作的所有方面，包括 filter、map、distinct、sorted 等操作，让您的代码行云流水，更加优雅。

1. filter()：根据指定的 Predicate 保留符合条件的元素。
2. map()：根据指定的 Function 映射每个元素，生成一个新的 Stream。
3. flatMap()：将每个元素映射为一个 Stream，然后将这些 Stream 连接成一个 Stream。
4. distinct()：返回一个去重后的 Stream。
5. sorted()：对 Stream 进行排序。
6. peek()：对每个元素执行指定的操作，但并不消费元素。
7. limit()：返回一个截断后的 Stream。
8. skip()：返回一个跳过指定元素后的 Stream。
9. forEach()：对每个元素执行指定的操作。
10. toArray()：将 Stream 转换为数组。

示例

1. 使用 filter() 方法过滤奇数

public class StreamDemo01 {
    public static void main(String[] args) {
        List<Integer> numbers = Arrays.asList(1, 2, 3, 4, 5, 6, 7, 8, 9, 10);
        List<Integer> list = numbers.stream()
                .filter(n -> n % 2 == 0)
                .collect(Collectors.toList());
        System.out.println(list);
    }
}

输出

[2, 4, 6, 8, 10]

2. 使用 map() 方法将每个字符串转换为大写

public class StreamDemo02 {
    public static void main(String[] args) {
        List<String> words = Arrays.asList("hello", "world", "java", "stream");
        List<String> list = words.stream()
                .map(String::toUpperCase)
                .collect(Collectors.toList());
        System.out.println(list);
    }
}

输出

[HELLO, WORLD, JAVA, STREAM]

3. 使用 flatMap() 方法将嵌套的列表展平为一个列表

public class StreamDemo03 {
    public static void main(String[] args) {
        List<List<Integer>> numbersList = Arrays.asList(
                Arrays.asList(1, 2),
                Arrays.asList(3, 4, 5, 6),
                Arrays.asList(7, 8, 9, 10)
        );
        List<Integer> list = numbersList.stream()
                .flatMap(List::stream)
                .collect(Collectors.toList());
        System.out.println(list);
    }
}

输出

[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

4. 使用 distinct() 方法去除重复元素

public class StreamDemo04 {
    public static void main(String[] args) {
        List<Integer> numbers = Arrays.asList(1, 2, 3, 1, 2, 3, 4, 6, 9, 10, 10, 3, 6);
        List<Integer> distinctNum = numbers.stream()
                .distinct()
                .collect(Collectors.toList());
        System.out.println(distinctNum);
    }
}

输出

[1, 2, 3, 4, 6, 9, 10]

5. 使用 sorted() 方法对 Stream 进行排序

public class StreamDemo05 {
    public static void main(String[] args) {
        List<Integer> numbers = Arrays.asList(2, 1, 4, 5, 7, 9, 2, 3, 5);
        List<Integer> sortedNum = numbers.stream()
                .sorted()
                .collect(Collectors.toList());
        System.out.println(sortedNum);
    }
}

输出

[1, 2, 2, 3, 4, 5, 5, 7, 9]

6. 使用 peek() 方法打印每个元素并统计元素个数

public class StreamDemo06 {
    public static void main(String[] args) {
        List<Integer> numbers = Arrays.asList(1, 2, 3, 4, 5, 6, 7, 8, 9, 10);
        int count = numbers.stream()
                .peek(System.out::println)
                .mapToInt(Integer::intValue)
                .sum();
        System.out.println("count = " + count);
    }
}

输出

1
2
3
4
5
6
7
8
9
10
count = 55

7. 使用 limit() 方法限制结果集大小

public class StreamDemo07 {
    public static void main(String[] args) {
        List<Integer> numbers = Arrays.asList(1, 2, 3, 4, 5, 6, 7, 8, 9, 10);
        List<Integer> limitNum = numbers.stream()
                .limit(5)
                .collect(Collectors.toList());
        System.out.println(limitNum);
    }
}

输出

[1, 2, 3, 4, 5]

8. 使用 skip() 方法跳过前面的元素

public class StreamDemo08 {
    public static void main(String[] args) {
        List<Integer> numbers = Arrays.asList(1, 2, 3, 4, 5, 6, 7, 8, 9, 10);
        List<Integer> skipNum = numbers.stream()
                .skip(5)
                .collect(Collectors.toList());
        System.out.println(skipNum);
    }
}

输出

[6, 7, 8, 9, 10]

9. 使用 forEach() 方法打印每个元素

public class StreamDemo09 {
    public static void main(String[] args) {
        List<String> words = Arrays.asList("hello", "world", "java", "stream");
        words.stream()
                .forEach(System.out::println);
    }
}

输出

hello
world
java
stream

10. 使用 toArray() 方法将 Stream 转换为数组

public class StreamDemo10 {
    public static void main(String[] args) {
        List<Integer> numbers = Arrays.asList(1, 2, 3, 4, 5, 6, 7, 8, 9, 10);
        Integer[] numberArray = numbers.stream()
                .toArray(Integer[]::new);
        System.out.println(Arrays.toString(numberArray));
    }
}

输出

[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]