Java Stream collect 收集的同时将主键加入到 Set 集合中

peek

使用Stream中的peek中间操作符，当元素流经该操作符时，进行相应的操作。

伪代码

List<User> users = List.of(user1, user2...);

Set<Integer> userIds = new HashSet<>();

Map<Integer, User> userMap = users.stream()
    .peek(user -> userIds.add(user.getId()))
    .collect(Collectors.toMap());

备注

peek还可以做一些debug的排查工作，加上打印日志等操作了解Stream执行的步骤。