Stream流 | Collectors.toMap 根据收集自身对象（同时防止key重复）

Stream流 | Collectors.toMap 根据收集自身对象

日常开发中我们通常会想将 List 集合根据某个成员变量为 key 值将其转成 Map 集合，如下：

GroupInfoEntity.java

@Data
public class GroupInfoEntity{
	/** 组织架构ID */
	private Long id;
	/** 组织架构名称 */
	private String name;
	/** 组织架构父ID */
	private Long parentId;
}

有一个封装上面实体的 List 集合，现在有下面两个需求： 假设 list 里面存了一些数据

List list = new ArrayList();

• 1.根据 id 和 name 将其转成 Map 集合

Map map = list.stream()
.collect(Collectors.toMap(GroupInfoEntity::getId, GroupInfoEntity::getName));

• 2.根据 id 和 对象自己 转成 Map 集合

Map map = list.stream()
.collect(Collectors.toMap(GroupInfoEntity::getId, Function.identity()));

• 3.防止key重复 一般toMap时只用到两个参数，第一个设定Key，第二个设定value。但如果Key重复的话，那就需要用到第三个参数，可以这么写（k1, k2）-> k1。 什么意思呢，就是说如果k1和k2的key都相同，那么只取k1的值去覆盖之前的值，也可以改为取k2的值，这样就解决了duplicate key的问题

Map map = list.stream()
.collect(Collectors.toMap(GroupInfoEntity::getId, GroupInfoEntity::getName, (k1, k2) -> k1));

这样就很完美的得到自己想要的数据。

注意：

这里的 Function.identity() 等价于 t -> t，就是将对象自己返回。