Java8 Stream collect(Collectors.toMap())的使用

Collectors.toMap 用法

在我们实际开发过程中经常使用到将List 转为Map的过程,在Java8 中Stream提供简便开发的方式

三个重载的方法

两个参数

public static 
    Collector> toMap(Function keyMapper,
                                    Function valueMapper) {
        return new CollectorImpl<>(HashMap::new,
                                   uniqKeysMapAccumulator(keyMapper, valueMapper),
                                   uniqKeysMapMerger(),
                                   CH_ID);
    }

Java8 stream特性 Collectors.toMap

package stream; 
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.UUID;
import java.util.function.Function;
import java.util.stream.Collectors;
 
/**
 * @author zzl
 * @Date 2022/1/15
 * @description Java stream特性:Collectors.toMap
 */
public class CollectorsToMapTest {
    public static void main(String[] args) {
        List testList = new ArrayList<>();
        // 初始化list
        for (int i = 0; i < 3; i++) {
            TestVo vo = new TestVo();
            vo.setUserId(i);
            vo.setName(UUID.randomUUID().toString().replaceAll("-", ""));
            testList.add(vo);
        }
        // toMap(p1,p2),p1参数是map的key值,p2参数是map的value值,当value为对象时,可以用Function.identity()表示value值
        Map map = testList.stream().collect(Collectors.toMap(TestVo::getUserId, TestVo::getName));
 
        for (Map.Entry entry : map.entrySet()) {
            System.out.println("key=" + entry.getKey() + ",value=" + entry.getValue());
        }
        System.out.println("==================================================");
 
        testList = new ArrayList<>();
        testList.add(new TestVo(1, "a"));
        testList.add(new TestVo(2, "b"));
        testList.add(new TestVo(1, "c"));
        // toMap(p1,p2,p3),其中p3参数是为了解决key值冲突时,决定value取值的
        Map map2 = testList.stream()
                .collect(Collectors.toMap(TestVo::getUserId, TestVo::getName, (oldValue, newValue) -> oldValue));
 
        System.out.println("(oldValue, newValue) -> oldValue的方式:key值冲突时,value取值为旧的key对应的value值");
        map2.forEach((k, v) -> System.out.println("key=" + k + ",value=" + v));
 
        System.out.println("(oldValue, newValue) -> newValue的方式:key值冲突时,value取值为新的key对应的value值");
        map2 = testList.stream()
                .collect(Collectors.toMap(TestVo::getUserId, TestVo::getName, (oldValue, newValue) -> newValue));
        map2.forEach((k, v) -> System.out.println("key=" + k + ",value=" + v));
    }
}

执行结果 :

key=0,value=acd45a638a2b43a4b7ccab7781290916
key=1,value=6fa7e201faaf4de0b4d6645214966285
key=2,value=468721a42ff14dc38a0b4efd2bf288eb
==================================================
(oldValue, newValue) -> oldValue的方式:key值冲突时,value取值为旧的key对应的value值
key=1,value=a
key=2,value=b
(oldValue, newValue) -> newValue的方式:key值冲突时,value取值为新的key对应的value值
key=1,value=c
key=2,value=b

package stream;  
/**
 * @author zzl
 * @Date 2022/1/15
 * @description
 */
public class TestVo {
    private Integer userId;
    private String name;
 
    public Integer getUserId() {
        return userId;
    }
 
    public void setUserId(Integer userId) {
        this.userId = userId;
    }
 
    public String getName() {
        return name;
    }
 
    public void setName(String name) {
        this.name = name;
    }
 
    public TestVo(Integer userId, String name) {
        this.userId = userId;
        this.name = name;
    }
 
    public TestVo() { 
    }
}

以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。

你可能感兴趣的