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

目录

• Collectors.toMap的用法
• 三个重载的方法
• Java8 stream特性 Collectors.toMap

Collectors.toMap的用法

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

三个重载的方法

两个参数

public static <T, K, U>
    Collector<T, ?, Map<K,U>> toMap(Function<? super T, ? extends K> keyMapper,
                                    Function<? super T, ? extends U> 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<TestVo> 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<Integer, String> map = testList.stream().collect(Collectors.toMap(TestVo::getUserId, TestVo::getName));

        for (Map.Entry<Integer, String> 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<Integer, String> 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() {
    }
}

以上为个人经验，希望能给大家一个参考，也希望大家多多支持我们。