java 8 stream 将 List 转为 Map Duplicate key

目的

将 List 转为 Map,如果有多个值对应同一个key,则保留最后一个

一、准备

1⃣️、构造几个user对象,转为一个user的List,注意其中user2user0的id是相同的

/**
 * @description: 用户信息
 * @author: wx
 * @create: 2019-09-15 18:27
 */
public class User {

    private Integer id;
    private String userName;
    private String password;

    public User() {
    }
    public User(Integer id, String password) {
        this.id = id;
        this.password = password;
    }
}

public static void main(String[] args) {
    User user0 = new User(0, "000");
    User user1 = new User(1, "111");
    User user2 = new User(0, "222");
    List list = Arrays.asList(user0, user1, user2);
}

2⃣️、然后将List转为,以id为key,整个user对象为value的Map,加入如下代码

Map collect = list.stream()
                         .collect(Collectors.toMap(User::getId, listSub -> listSub));

3⃣️、会发现报如下异常

Exception in thread "main" java.lang.IllegalStateException: 
                        Duplicate key User(id=0, userName=null, password=000)

二、解决办法

修改代码List转Map代码如下:

Map collect = list.stream().collect(
        Collectors.toMap(User::getId, listSub -> listSub,
                (listSubOld, listSubNew) -> listSubNew)
);

得到的结果:

{
    0=User(id=0, userName=null, password=222), 
    1=User(id=1, userName=null, password=111)
}

三、分析

1、打开Collectors源码看,会发现toMap一共有三个重写的方法,其中根方法如下,注意看看中文注释:

/**
* @param keyMapper     a mapping function to produce keys
*                      指定key
* @param valueMapper   a mapping function to produce values
*                      指定values
* @param mergeFunction a merge function, used to resolve collisions between
*                      values associated with the same key, as supplied
*                      to {@link Map#merge(Object, Object, BiFunction)}
*                      一个合并函数,有相同的key的时候如何选取value
* @param mapSupplier   a function which returns a new, empty {@code Map} into
*                      which the results will be inserted
*                      一个函数,它返回一个新的、带有结果的空Map 
*                     指定返回的是HashMap or ConcurrentHashMap等等
*/
public static >
Collector toMap(Function keyMapper,
                            Function valueMapper,
                            BinaryOperator mergeFunction,
                            Supplier mapSupplier) {
    BiConsumer accumulator
            = (map, element) -> map.merge(keyMapper.apply(element),
                                          valueMapper.apply(element), mergeFunction);
    return new CollectorImpl<>(mapSupplier, accumulator, mapMerger(mergeFunction), CH_ID);
}

2、而我一开始用的是如下重写的toMap:

public static 
Collector> toMap(Function keyMapper,
                                Function valueMapper) {
    return toMap(keyMapper, valueMapper, throwingMerger(), HashMap::new);
}

可以看到使用了throwingMerger()这样一个合并的方法,再看看这方法的源码:

private static  BinaryOperator throwingMerger() {
    return (u,v) -> { throw new IllegalStateException(String.format("Duplicate key %s", u)); };
}

可以看到刚刚异常抛出的信息了。
总结一下:就是一旦出现一个key值有两个对应的value,就需要进行合并处理,但是默认的合并方法是throwingMerger(),只要需要合并就抛出异常。
那么就找到解决办法了,不使用默认的方法,自己写一个合并的方法。
如上就是lambda表达式(listSubOld, listSubNew) -> listSubNew,就是两个输入参数(一个相同的key对应的两个value),返回后一个值。

你可能感兴趣的:(java 8 stream 将 List 转为 Map Duplicate key)