时间:2022-10-09 15:33:16 | 栏目:JAVA代码 | 点击:次
在我们实际开发过程中经常使用到将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); }
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() { } }