用于将流映射到具有标识键的映射的快捷方式

Question

考虑以下示例代码，通过将每个字符串映射到其长度并打印每个映射条目来处理字符串列表

package com.dbenergie.ndm.bnb.business;

import java.util.Arrays;
import java.util.List;
import java.util.Map;
import java.util.function.Function;
import java.util.stream.Collectors;

public class Application {

    public static void main(String[] args) {
        final List<String> strings = Arrays.asList("Hello", "World", "!");

        // 1
        final Map<String, Integer> stringCharacterCountMapping = strings.stream()
                .collect(Collectors.toMap(Function.identity(), String::length));

        stringCharacterCountMapping.entrySet().stream().forEach(System.out::println);
    }
}

现在，我个人发现相当混乱或样板式的是使用函数将每个流条目显式映射为映射的键。identity()（在1处），但是我找到的每个映射收集器都需要一个键映射器--是不是我忽略了某个收集器，或者如果需要的话，我必须定义我自己的收集器？如果有类似的东西

Collectors.toMap(String::length);

Answer 1

在Collectors类中没有不采用KeyMapper的Tomap方法。我怀疑这是因为在映射中传递的单个参数可能是非常模糊的。值还是键？

你可以很容易地自己写出这样一个收集器：

class MyCollectors {
    public static <K, U> Collector<K, ?, Map<K,U>> toMapWithValueWrapper(Function<? super K, ? extends U> valueMapper) {
        return Collectors.toMap(Function.identity(), valueMapper);
    }
}

注意，签名与collectors.tomap有点不同。只有2个泛型参数。原始的collectors.tomap有3个-T（我们正在收集的类型）、K（键的类型）和U（值的类型）。由于您正在为密钥使用identity函数，因此密钥的类型始终与我们正在收集的类型相同，因此t==k。因此，我们可以消除t。

class MyCollectors {
    public static <T extends K, K, U> Collector<T, ?, Map<K,U>> toMap(Function<? super T, ? extends U> valueMapper) {
        return Collectors.toMap(Function.identity(), valueMapper);
    }
}

如果您希望允许收集到一个映射，该映射的键类型是您正在收集的类型的超类。