Lamdar List转Map

298 阅读1分钟

前言

时不时总是忘记怎么写,网上baidu下来也都是乱七八糟复制粘贴的内容,今天自己来汇总下。

List转Map

Map<key,List<value>>

Map<Long, List<ActivityUserMissionDO>> map = activityUserMissionDos.stream().
		collect(Collectors.groupingBy(ActivityUserMissionDO::getParentModuleId));

Map<key,value>:

指定key和value:

toMap方法value为null时会报java.lang.NullPointException,所以最好过滤一下

Map<String, Integer> lastDayPushNumMap = 
	lastDayPushNumList.stream().filter(t -> t.getLastDayPushNum() != null)
    		.collect(Collectors.toMap(key -> key.getUserId() + Const.SPILIT + key.getHostId(), 
            		value -> value.getLastDayPushNum()));

指定key,model作为value:

public Map<Long, Account> getIdAccountMap(List<Account> accounts) {
	return accounts.stream().collect(Collectors.toMap(Account::getId, account -> account));
}

account -> account是一个返回本身的lambda表达式,其实还可以使用Function接口中的一个默认方法代替,使整个方法更简洁优雅:

public Map<Long, Account> getIdAccountMap(List<Account> accounts) {
	return accounts.stream().collect(Collectors.toMap(Account::getId, Function.identity()));
}

重复key的情况

public Map<String, Account> getNameAccountMap(List<Account> accounts) {
	return accounts.stream().collect(Collectors.toMap(Account::getUsername, Function.identity()));
}

这个方法可能报错java.lang.IllegalStateException: Duplicate key,因为name是有可能重复的。
toMap有个重载方法,可以传入一个合并的函数来解决key冲突问题:

public Map<String, Account> getNameAccountMap(List<Account> accounts) {
	return accounts.stream().collect(Collectors.toMap(Account::getUsername, Function.identity(), (key1, key2) -> key2));
}

指定具体收集的map

toMap还有另一个重载方法,可以指定一个Map的具体实现,来收集数据:

public Map<String, Account> getNameAccountMap(List<Account> accounts) {
	return accounts.stream().collect(Collectors.toMap(Account::getUsername, 
    		Function.identity(), 
            (key1, key2) -> key2, LinkedHashMap::new));
}