it-swarm.com.ru

Карта потока Java 8 в наборе записей

Я пытаюсь выполнить операцию с картой для каждой записи в объекте Map.

Мне нужно снять префикс с ключа и преобразовать значение из одного типа в другой. Мой код берет записи конфигурации из Map<String, String> и преобразует в Map<String, AttributeType> (AttributeType - это просто класс, содержащий некоторую информацию. Дальнейшее объяснение не относится к этому вопросу.)

Лучшее, что я смог придумать, используя потоки Java 8, это следующее:

private Map<String, AttributeType> mapConfig(Map<String, String> input, String prefix) {
   int subLength = prefix.length();
   return input.entrySet().stream().flatMap((Map.Entry<String, Object> e) -> {
      HashMap<String, AttributeType> r = new HashMap<>();
      r.put(e.getKey().substring(subLength), AttributeType.GetByName(e.getValue()));
      return r.entrySet().stream();
   }).collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue));
}

Невозможность создать Map.Entry из-за того, что он является интерфейсом, вызывает создание единственного экземпляра Map и использование flatMap(), что выглядит ужасно.

Есть ли лучшая альтернатива? Кажется, лучше сделать это с помощью цикла for:

private Map<String, AttributeType> mapConfig(Map<String, String> input, String prefix) {
   Map<String, AttributeType> result = new HashMap<>(); 
   int subLength = prefix.length(); 
   for(Map.Entry<String, String> entry : input.entrySet()) {
      result.put(entry.getKey().substring(subLength), AttributeType.GetByName( entry.getValue()));
   }
   return result;
}

Должен ли я избежать Stream API для этого? Или есть более хороший способ, который я пропустил?

29
Wil Selwood

Просто переведите «старый путь цикла» в потоки:

private Map<String, String> mapConfig(Map<String, Integer> input, String prefix) {
    return input.entrySet().stream()
            .collect(Collectors.toMap(
                   entry -> entry.getKey().substring(subLength), 
                   entry -> AttributeType.GetByName(entry.getValue())));
}
69
Smutje

Вопрос может быть немного устаревшим, но вы можете просто использовать AbstractMap.SimpleEntry <> следующим образом:

private Map<String, AttributeType> mapConfig(
    Map<String, String> input, String prefix) {
       int subLength = prefix.length();
       return input.entrySet()
          .stream()
          .map(e -> new AbstractMap.SimpleEntry<>(
               e.getKey().substring(subLength),
               AttributeType.GetByName(e.getValue()))
          .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue));

любой другой объект с парным значением также будет работать (т. е. ApacheCommons Pair Tuple).

14
Dawid Pancerz

Пожалуйста, сделайте следующую часть API Collectors:

<K, V> Collector<? super Map.Entry<K, V>, ?, Map<K, V>> toMap() {
  return Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue);
}
5
Lóránt Miglécz

Вот более короткое решение от AbacusUtil

Stream.of(input).toMap(e -> e.getKey().substring(subLength), 
                       e -> AttributeType.GetByName(e.getValue()));
0
user_3380739