使用Java Stream API简化集合操作

大家好,我是微赚淘客系统3.0的小编,也是冬天不穿秋裤,天冷也要风度的程序猿!今天我们将探讨如何使用Java Stream API来简化集合操作。Java 8引入了Stream API,它提供了一种高效且易读的方式来处理集合数据。通过使用Stream API,我们可以避免繁琐的循环和条件判断,使代码更加简洁和优雅。

1. 什么是Stream API

Stream API是一种用于处理集合的框架,它允许我们以声明性方式(就像SQL语句)处理数据。Stream操作可以分为中间操作和终端操作,其中中间操作返回一个新的Stream,允许操作链式调用,而终端操作会产生结果或者副作用。

2. 创建Stream

Stream可以从各种数据源创建,如集合、数组或I/O通道等。下面是一些常见的创建Stream的方式:

package cn.juwatech.stream;

import java.util.Arrays;
import java.util.List;
import java.util.stream.Stream;

public class StreamCreationExample {
    public static void main(String[] args) {
        // 从集合创建
        List<String> list = Arrays.asList("a", "b", "c");
        Stream<String> streamFromList = list.stream();

        // 从数组创建
        String[] array = {"a", "b", "c"};
        Stream<String> streamFromArray = Arrays.stream(array);

        // 直接创建
        Stream<String> streamDirect = Stream.of("a", "b", "c");
    }
}

3. 常见的中间操作

中间操作返回一个新的Stream,并且是惰性执行的。常见的中间操作包括filter、map、flatMap、sorted、distinct等。

3.1 filter

filter用于对Stream中的元素进行过滤,保留满足条件的元素。

package cn.juwatech.stream;

import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;

public class StreamFilterExample {
    public static void main(String[] args) {
        List<Integer> numbers = Arrays.asList(1, 2, 3, 4, 5, 6);
        List<Integer> evenNumbers = numbers.stream()
                                           .filter(n -> n % 2 == 0)
                                           .collect(Collectors.toList());
        System.out.println(evenNumbers); // 输出: [2, 4, 6]
    }
}

3.2 map

map用于将Stream中的每个元素映射到另一个元素,通常用于元素类型的转换。

package cn.juwatech.stream;

import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;

public class StreamMapExample {
    public static void main(String[] args) {
        List<String> words = Arrays.asList("hello", "world");
        List<Integer> wordLengths = words.stream()
                                         .map(String::length)
                                         .collect(Collectors.toList());
        System.out.println(wordLengths); // 输出: [5, 5]
    }
}

3.3 flatMap

flatMap用于将每个元素转换成一个Stream,并将多个Stream合并成一个Stream。

package cn.juwatech.stream;

import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;

public class StreamFlatMapExample {
    public static void main(String[] args) {
        List<List<String>> listOfLists = Arrays.asList(
            Arrays.asList("a", "b"),
            Arrays.asList("c", "d"),
            Arrays.asList("e", "f")
        );

        List<String> flattenedList = listOfLists.stream()
                                                .flatMap(List::stream)
                                                .collect(Collectors.toList());
        System.out.println(flattenedList); // 输出: [a, b, c, d, e, f]
    }
}

3.4 sorted

sorted用于对Stream中的元素进行排序,默认是自然排序,也可以传入比较器进行自定义排序。

package cn.juwatech.stream;

import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;

public class StreamSortedExample {
    public static void main(String[] args) {
        List<String> words = Arrays.asList("banana", "apple", "cherry");
        List<String> sortedWords = words.stream()
                                        .sorted()
                                        .collect(Collectors.toList());
        System.out.println(sortedWords); // 输出: [apple, banana, cherry]
    }
}

3.5 distinct

distinct用于去重,返回由Stream中唯一元素组成的Stream。

package cn.juwatech.stream;

import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;

public class StreamDistinctExample {
    public static void main(String[] args) {
        List<Integer> numbers = Arrays.asList(1, 2, 2, 3, 3, 3, 4, 4, 4, 4);
        List<Integer> distinctNumbers = numbers.stream()
                                               .distinct()
                                               .collect(Collectors.toList());
        System.out.println(distinctNumbers); // 输出: [1, 2, 3, 4]
    }
}

4. 常见的终端操作

终端操作会触发Stream的执行,并且会产生一个结果或副作用。常见的终端操作包括forEach、collect、reduce、count、anyMatch等。

4.1 forEach

forEach用于遍历Stream中的每个元素,并对其执行指定的操作。

package cn.juwatech.stream;

import java.util.Arrays;
import java.util.List;

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

4.2 collect

collect用于将Stream中的元素收集到一个集合或其他容器中。最常见的收集器是Collectors.toList()。

package cn.juwatech.stream;

import java.util.Arrays;
import java.util.List;
import java.util.Set;
import java.util.stream.Collectors;

public class StreamCollectExample {
    public static void main(String[] args) {
        List<String> words = Arrays.asList("hello", "world", "hello");
        Set<String> uniqueWords = words.stream()
                                       .collect(Collectors.toSet());
        System.out.println(uniqueWords); // 输出: [world, hello]
    }
}

4.3 reduce

reduce用于将Stream中的元素组合起来,生成一个单一的值。常见的操作有求和、求积、求最大值等。

package cn.juwatech.stream;

import java.util.Arrays;
import java.util.List;

public class StreamReduceExample {
    public static void main(String[] args) {
        List<Integer> numbers = Arrays.asList(1, 2, 3, 4, 5);
        int sum = numbers.stream()
                         .reduce(0, Integer::sum);
        System.out.println(sum); // 输出: 15
    }
}

4.4 count

count用于计算Stream中元素的个数。

package cn.juwatech.stream;

import java.util.Arrays;
import java.util.List;

public class StreamCountExample {
    public static void main(String[] args) {
        List<String> words = Arrays.asList("hello", "world", "hello");
        long count = words.stream().count();
        System.out.println(count); // 输出: 3
    }
}

4.5 anyMatch

anyMatch用于判断Stream中是否至少有一个元素满足给定的条件。

package cn.juwatech.stream;

import java.util.Arrays;
import java.util.List;

public class StreamAnyMatchExample {
    public static void main(String[] args) {
        List<String> words = Arrays.asList("hello", "world", "hello");
        boolean hasHello = words.stream().anyMatch("hello"::equals);
        System.out.println(hasHello); // 输出: true
    }
}

通过以上示例代码,我们可以看到Java Stream API如何简化集合操作,使代码更加简洁和易读。Stream API不仅提高了代码的可维护性,也让我们能更专注于业务逻辑的实现。