Java8: HashMap <X, Y> в HashMap <X, Z>, використовуючи Stream / Map-Reduce / Collector

Я знаю, як "перетворити" просту Java Listз Y-> Z, тобто:

List<String> x;
List<Integer> y = x.stream()
        .map(s -> Integer.parseInt(s))
        .collect(Collectors.toList());

Тепер я хотів би зробити те ж саме з картою, тобто:

INPUT:
{
  "key1" -> "41",    // "41" and "42"
  "key2" -> "42      // are Strings
}

OUTPUT:
{
  "key1" -> 41,      // 41 and 42
  "key2" -> 42       // are Integers
}

Рішення не повинно обмежуватися String-> Integer. Як і в Listнаведеному вище прикладі, я хотів би викликати будь-який метод (або конструктор).

Map<String, String> x;
Map<String, Integer> y =
    x.entrySet().stream()
        .collect(Collectors.toMap(
            e -> e.getKey(),
            e -> Integer.parseInt(e.getValue())
        ));

Це не так приємно, як код списку. Ви не можете сконструювати нові Map.Entrys у map()виклику, тому робота змішується з collect()викликом.

Ось декілька варіантів відповіді Сотіріоса Деліманоліса , який почався досить непогано (+1). Розглянемо наступне:

static <X, Y, Z> Map<X, Z> transform(Map<? extends X, ? extends Y> input,
                                     Function<Y, Z> function) {
    return input.keySet().stream()
        .collect(Collectors.toMap(Function.identity(),
                                  key -> function.apply(input.get(key))));
}

Пару балів тут. По-перше, це використання знакових символів у генериці; це робить функцію дещо гнучкішою. Підстановочний знак буде необхідним, якщо, наприклад, ви хотіли, щоб у вихідній карті був ключ, який є надкласом ключа вхідної карти:

Map<String, String> input = new HashMap<String, String>();
input.put("string1", "42");
input.put("string2", "41");
Map<CharSequence, Integer> output = transform(input, Integer::parseInt);

(Також є приклад для значень карти, але це дійсно надумано, і я визнаю, що обмеження малого підключення для Y допомагає лише у крайніх випадках.)

Другий момент полягає в тому, що замість запуску потоку над вхідною картою entrySetя провів його над keySet. Я думаю, що це робить трохи чистішим, я думаю, ціною вибору значень з карти, а не з карти. Між іншим, я спочатку мав key -> keyяк перший аргумент, toMap()і це не вдалося з помилкою виводу типу чомусь. Змінивши його на (X key) -> keyпрацюючий, як це зробили Function.identity().

Ще одна варіація полягає в наступному:

static <X, Y, Z> Map<X, Z> transform1(Map<? extends X, ? extends Y> input,
                                      Function<Y, Z> function) {
    Map<X, Z> result = new HashMap<>();
    input.forEach((k, v) -> result.put(k, function.apply(v)));
    return result;
}

Для цього використовується Map.forEach()замість потоків. Думаю, це ще простіше, тому що він обходиться з колекторами, які є дещо незграбними для використання з картами. Причина полягає в тому, що він Map.forEach()надає ключ і значення як окремі параметри, тоді як потік має лише одне значення - і ви повинні вибрати, чи використовувати цей ключ або запис карти в якості цього значення. З боку мінусу, цього не вистачає багатої, поточної доброти інших підходів. :-)

Узагальнене рішення подібне

public static <X, Y, Z> Map<X, Z> transform(Map<X, Y> input,
        Function<Y, Z> function) {
    return input
            .entrySet()
            .stream()
            .collect(
                    Collectors.toMap((entry) -> entry.getKey(),
                            (entry) -> function.apply(entry.getValue())));
}

Приклад

Map<String, String> input = new HashMap<String, String>();
input.put("string1", "42");
input.put("string2", "41");
Map<String, Integer> output = transform(input,
            (val) -> Integer.parseInt(val));

Функція Guava - Maps.transformValuesце те, що ви шукаєте, і вона чудово працює з лямбда-виразами:

Maps.transformValues(originalMap, val -> ...)

Чи абсолютно воно повинно бути на 100% функціональним та вільним? Якщо ні, то як щодо цього, який приблизно такий короткий:

Map<String, Integer> output = new HashMap<>();
input.forEach((k, v) -> output.put(k, Integer.valueOf(v));

( якщо ви можете жити з ганьбою і виною поєднувати потоки з побічними ефектами )

Моя бібліотека StreamEx, яка розширює стандартний API потоку, забезпечує EntryStreamклас, який краще підходить для трансформації карт:

Map<String, Integer> output = EntryStream.of(input).mapValues(Integer::valueOf).toMap();

Альтернатива, яка завжди існує для цілей навчання, - це створити свій власний колектор через Collector.of (), хоча toMap () колектор JDK тут є стислим (+1 тут ).

Map<String,Integer> newMap = givenMap.
                entrySet().
                stream().collect(Collector.of
               ( ()-> new HashMap<String,Integer>(),
                       (mutableMap,entryItem)-> mutableMap.put(entryItem.getKey(),Integer.parseInt(entryItem.getValue())),
                       (map1,map2)->{ map1.putAll(map2); return map1;}
               ));

Якщо ви не заперечуєте над використанням сторонніх бібліотек, у моїй бібліотеці, що реагує на циклопи, є розширення для всіх типів колекції JDK , включаючи Map . Ми можемо просто перетворити карту безпосередньо за допомогою оператора «map» (карта за замовчуванням діє на значення на карті).

   MapX<String,Integer> y = MapX.fromMap(HashMaps.of("hello","1"))
                                .map(Integer::parseInt);

bimap може використовуватися для одночасного перетворення ключів та значень

  MapX<String,Integer> y = MapX.fromMap(HashMaps.of("hello","1"))
                               .bimap(this::newKey,Integer::parseInt);

Декларативним та простішим рішенням було б:

yourMutableMap.replaceAll ((ключ, val) -> return_value_of_bi_your_function); Nb. пам'ятайте, що ви змінюєте стан своєї карти. Тож це може бути не те, що ви хочете.

Будьте здорові: http://www.deadcoderising.com/2017-02-14-java-8-declarative-ways-of-modifying-a-map-using-compute-merge-and-replace/