java8 lambda表达式之Stream API的用法实践

来源:互联网 发布:网络综艺尺度大被叫停 编辑:程序博客网 时间:2024/06/05 02:35

1.查出所有用户,取出所有用户id的集合

 List<CircleUserVO> allUser = circleMemberInfoMapper.selectByListUserIdAndName(null, null); List<String> allUserIds = allUser.stream().map(user -> user.getUserId()).collect(Collectors.toList());

2.查出所有的用户,构造一个Map

 Map<String, CircleUserVO> userMap = allUser.stream().collect(Collectors.toMap(user -> user.getUserId(), user -> user)); 或者:Map<String, CircleUserVO> userMap2 = allUser.stream().collect(Collectors.toMap(CircleUserVO::getUserId, user -> user)); 或者:Map<String, CircleUserVO> userMap3 = allUser.stream().collect(Collectors.toMap(CircleUserVO::getUserId, Function.identity())); //java.util.function.Function

3.List转Map的方法总结
常用方式

public Map<Long, String> getIdNameMap(List<Account> accounts) {    return accounts.stream().collect(Collectors.toMap(Account::getId, Account::getUsername));}

收集成实体本身map

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));}

这里只是简单的使用后者覆盖前者来解决key重复问题。

指定具体收集的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));}

转载自:https://zacard.net/2016/03/17/java8-list-to-map/