使用Java Stream API简化集合操作
今天我们将探讨如何使用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不仅提高了代码的可维护性,也让我们能更专注于业务逻辑的实现。