Java-如何实现自定义映射（Function<? super T, ? extends R> mapper），它可以在给定函数的情况下转换 Map 的值

Question

我有一个自定义的排序地图。我想使用 map 函数转换地图中的每个值。我知道我可以使用 java 8 stream().map(function) 来实现这一点。但我想了解上面的 map(function) 是如何工作的并编写一个自定义函数。

我已经阅读了几篇文章，但我没有找到一篇解释清除的文章。

我是怎么做的：

SortedMap<String, Integer> items = new TreeMap();//TreeMap is a custom implementation
        items.put("item1", 2);
        items.put("item2", 1);

      SortedMap<String, Integer> transformedItems=items.entrySet().stream().collect(
             Collectors.toMap(
                   ( Map.Entry<String, Integer> entry)->entry.getKey(),//pick each key as it is
                     ( Map.Entry<String, Integer> entry)->entry.getValue()*2, //transform the value here.
                     (val1, val2) -> { throw new RuntimeException("Not expecting duplicate keys"); },
                     ()->new TreeMap<>()

我想怎么做：

SortedMap<String, Integer> items = new TreeMap();
        items.put("item1", 2);
        items.put("item2", 1);

      SortedMap<String, Integer> transformedItems=items.customMap(value->value*2)

这里的挑战是写customMap(function)

Answer 1

您可以创建一个独立的static 方法，该方法采用值映射函数。该方法几乎可以完成您的代码所做的事情：

public static <K, V, R> TreeMap<K, R> mapValues(Map<K, V> map, Function<V, R> valueMapper) {
    return map.entrySet().stream().collect(
            Collectors.toMap(Map.Entry::getKey,
                             e -> valueMapper.apply(e.getValue()),
                             (a, b) -> { throw new AssertionError("Not expecting duplicate keys"); },
                             TreeMap::new));
}

测试

SortedMap<String, Integer> items = new TreeMap<>();
items.put("item1", 2);
items.put("item2", 1);

SortedMap<String, Integer> transformedItems = mapValues(items, value -> value * 2);

System.out.println(transformedItems);

输出

{item1=4, item2=2}

注意映射值如何具有不同的类型，例如

SortedMap<String, Double> transformedItems = mapValues(items, value -> value * 2.5);

System.out.println(transformedItems);

输出

{item1=5.0, item2=2.5}