动态参数+reduce累加stream
通向面包的小路蜿蜒于劳动的沼泽之中，通向衣裳的小路从一块无花的土地中穿过，无论是通向面包的路还是通向衣裳的路，都是一段艰辛的历程。 ——福斯
今天有朋友问我，stream中如果要将一个User类中的username和id收集起来变成两个list怎么写，我说可以使用peek函数
@Data
@AllArgsConstructor
static class User {
private Long id;
private Long groupId;
private String username;
}就像这样：
// 用户列表
List<User> userList = Stream.iterate(1L, i -> ++i).map(id -> new User(id, 999L, Faker.instance().name().username())).limit(10).collect(Collectors.toList());
// 要收集起来的groupId
List<Long> groupIds = new ArrayList<>(userList.size() * 2);
// 要收集起来的username
List<String> usernames = new ArrayList<>(userList.size() * 2);
// 写死调用两个peek
List<Long> userIds = userList.parallelStream().peek(user -> groupIds.add(user.getGroupId())).peek(user -> usernames.add(user.getUsername())).map(User::getId).collect(Collectors.toList());
System.out.println(userIds);
System.out.println(groupIds);
System.out.println(usernames);然后运行结果：
image-20211024213325222
然后它有问，如果他不知道有多少个，有可能有一个，也有可能两个怎么办
我就给他写了另外两种实现方式：
使用动态参数+reduce实现，以及List+reduce实现
/**
* 封装peeks
*
* @param stream 流
* @param peeks 操作
* @return java.util.stream.Stream<T>
* @author <achao1441470436@gmail.com>
* @since 2021/10/24 21:12
*/
@SafeVarargs
public static <T> Stream<T> peek(Stream<T> stream, Consumer<T>... peeks) {
return Stream.of(peeks).reduce(stream, Stream::peek, Stream::concat);
}
public static <T> Stream<T> peek(Stream<T> stream, List<Consumer<T>> peeks) {
return peeks.stream().reduce(stream, Stream::peek, Stream::concat);
}这两种，使用方式如下：
// 可以写死也可以动态
userIds = peek(userList.parallelStream(), user -> groupIds.add(user.getGroupId()), user -> usernames.add(user.getUsername())).map(User::getId).collect(Collectors.toList());
// 动态
List<Consumer<User>> peeks = Arrays.asList(user -> groupIds.add(user.getGroupId()), user -> usernames.add(user.getUsername()));
Stream<User> peek = peek(userList.stream(), peeks);
// 转成array
peek(userList.stream(), peeks.stream().<Consumer<User>>toArray(Consumer[]::new));
peek(userList.stream(), peeks.<Consumer<User>>toArray(new Consumer[0]));这里原理就是用了reduce函数的累加特性，这个聚合操作：
Stream.of(peeks).reduce(stream, Stream::peek, Stream::concat);相当于：
Stream<User> stream = userList.stream();
for (Consumer<User> consumer : peeks) {
stream = stream.peek(consumer);
}关于reduce我之前的博客也写过了，就不再赘述啦~
主要就是分享这种思路，多了一种写法而已
本文参与 腾讯云自媒体分享计划，分享自作者个人站点/博客。
原始发表：2021-10-24，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读