在工作中常常会遇到将List转化为Map的场景,下面总结了经常使用到的几种方式;并简单的做了简单测试,供大家学习使用。
准备工作:
List<UserEntity> entityList = new ArrayList<>();
UserEntity userEntity = new UserEntity();
userEntity.setUserId("0001");
userEntity.setUserName("0001Name");
userEntity.setUserMobile("0001Phone");
entityList.add(userEntity);
UserEntity userEntity2 = new UserEntity();
userEntity2.setUserId("0002");
userEntity2.setUserName("0002Name");
userEntity2.setUserMobile("0002Phone");
entityList.add(userEntity2);
方式一:key和value都是对象中的某个属性值
key为userId、value为userName
Map<String, String> userMap = entityList.stream().collect(Collectors.toMap(UserEntity::getUserId, UserEntity::getUserName));
注:当userId出现重复的情况,会报 Duplicate key
的错误。
方式二:key是对象中的某个属性值,value是对象本身。
key为userId、value为UserEntity对象
Map<String, UserEntity> userMap = entityList.stream().collect(Collectors.toMap(UserEntity::getUserId, user -> user));
另外一种写法:
Map<String, UserEntity> userMap = entityList.stream().collect(Collectors.toMap(UserEntity::getUserId, Function.identity()));
注:以上两种写法,当userId出现重复的情况,同样会报 Duplicate key
的错误。
如果存在key重复的情况,使用第二个key去覆盖第一个key的值,写法如下:
Map<String, UserEntity> userMap = entityList.stream().collect(Collectors.toMap(UserEntity::getUserId, Function.identity(), (oldValue, newValue) -> newValue));
方式三:List根据key进行分组
根据userId进行分组
Map<String, List<UserEntity>> userIdGroupByList = entityList.stream().collect(Collectors.groupingBy(UserEntity::getUserId));