java - 使用 Java 8 流将 Map 转换为 Map>

Question

我有一些保存在 map 中的属性文件。示例:

Map<String, String> map = new HashMap<>();
map.put("1", "One");
map.put("2", "Two");
map.put("3", "Two");
map.put("4", "One"); 

我想转换 Map<String, String>到

Map<String, List<String>> map = new HashMap<>(); 

应该是

<"One", ("1", "4")>
<"Two", ("2", "3")>

我有一些代码想用 Java 8 风格重写。

  private Map<File, List<File>> getAllFiles(Set<File> files) {
   Map<File, File> inputFilesWithTskFile = 
   AppStorage.getInstance().getApplicationBean().getInputFilesWithTskFile();

    List<File> tsks = new ArrayList<>();
    for (Map.Entry<File, File> entry : inputFilesWithTskFile.entrySet()) {
        if (files.contains(entry.getKey())) {
            tsks.add(entry.getValue());
        }
    }
    Map<File, List<File>> listTsk = new HashMap<>();
    for (Map.Entry<File, File> entry : inputFilesWithTskFile.entrySet()) {
        if (tsks.contains(entry.getValue())) {
            List<File> files1 = listTsk.get(entry.getValue());
            if (files1 == null) {
                files1 = new ArrayList<>();
            }
            files1.add(entry.getKey());
            listTsk.put(entry.getValue(), files1);
        }
    }
    return listTsk;
  }

感谢您的帮助。也许你知道一些解释如何从左侧应该有值的 Map 创建 Map 的教程，右侧应该是按值分组的键列表。

Answer 1

类似下面的内容:

private Map<File, List<File>> getAllFiles(List<File> files) {
    return AppStorage.getInstance().getApplicationBean().getInputFilesWithTskFile()
            .entrySet()
            .stream()
            .filter(e -> files.contains(e.getKey()))
            .collect(groupingBy(Entry::getValue, mapping(Entry::getKey, toList())));
}

即获取所有filesWithTskFile，过滤掉Map中key不在files中的条目。然后按 value 对 Map 进行分组。

---

使用

import static java.util.stream.Collectors.groupingBy;
import static java.util.stream.Collectors.mapping;
import static java.util.stream.Collectors.toList;

---

一个更通用的示例，包括:

public static Map<String, List<String>> groupByValue(final Map<String, String> input) {
    return input.entrySet().stream()
            .collect(groupingBy(Entry::getValue, mapping(Entry::getKey, toList())));
}

运行:

final Map<String, String> example = Map.of(
        "1", "A",
        "2", "B",
        "3", "A",
        "4", "B"
        );

groupByValue(example).forEach((k, v) -> System.out.printf("%s->%s%n", k, v));

给予:

A->[3, 1]
B->[2, 4]

扩展到:

public static Map<String, List<String>> groupByValue(final Map<String, String> input, final Set<String> take) {
    return input.entrySet().stream()
            .filter(e -> take.contains(e.getKey()))
            .collect(groupingBy(Entry::getValue, mapping(Entry::getKey, toList())));
}

并运行:

groupByValue(example, Set.of("1", "3")).forEach((k, v) -> System.out.printf("%s->%s%n", k, v));

给予:

A->[1, 3]

---

注意:我使用Set 而不是List 作为过滤器。 Set 保证其 contains 方法将在 O(1) 中运行，而 List 将在 O(1) 中运行O(n)。所以 List 将非常效率低下。根据输入的大小，您实际上最好将 List 复制到 Set before 过滤中:

private Map<File, List<File>> getAllFiles(List<File> files) {
    final Set<File> filter = Set.copyOf(files);
    return AppStorage.getInstance().getApplicationBean().getInputFilesWithTskFile()
            .entrySet()
            .stream()
            .filter(e -> filter.contains(e.getKey()))
            .collect(groupingBy(Entry::getValue, mapping(Entry::getKey, toList())));
}