使用标识键将 Stream 映射到 Map 的快捷方式

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);
    }
}

现在，我个人觉得相当混乱或样板式的是使用Function.identity()（1）将每个流条目显式映射为映射的键，但到目前为止我发现的每个映射收集器都需要一个键映射器 -是否有一些我忽略的收集器，或者如果我需要这个，我是否必须定义我自己的收集器？有类似的东西会很方便

Collectors.toMap(String::length);

或类似的。有什么想法吗？

Answer 1

There is no toMap method in the Collectors class that does not take a keyMapper. 我怀疑这是因为您传递的单个参数映射到的内容可能非常模棱两可。值还是键？

你可以自己轻松写出这样的收集器：

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（值的类型）。由于您使用密钥的标识函数，密钥将始终具有与我们正在收集的类型相同的类型，因此T == K。因此，我们可以消除T。

编辑：

正如 Holger 在 cmets 中建议的那样，我们不必强制 T == K，因为 T 可能是 K 的子类，您仍然可以使用标识函数从 T 映射到 K .因此，您可以使用：

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);
    }
}

如果您想允许收集到其键类型是您正在收集的类型的超类的地图。