在 Java 8 中合并多个地图

Question

这里是 Java 8。我有以下课程：

public interface Animal {
  ...
}

public class Dog implements Animal {
  ...
}

public class Cat implements Animal {
  ...
}

public class Elephant implements Animal {
  ...
}

我必须实现以下方法：

void doSomething(Map<String,Dog> dogs, Map<String,Cat> cats, Map<String,Elephant> elephants) {
  // TODO:
  // * Merge all dogs, cats & elephants together into the same Map<String,Animal>,
  //     but...
  // * Do so generically (without having to create, say, a HashMap instance, etc.)
}

在我的doSomething(...) 方法中，我需要将所有映射参数合并到同一个Map<String,Animal> 映射中，但是我真的更愿意在没有我的代码的情况下这样做 必须实例化特定的地图实现（例如HashMap 等）。

意思是，我知道我可以这样做：

void doSomething(Map<String,Dog> dogs, Map<String,Cat> cats, Map<String,Elephant> elephants) {
  Map<String,Animal> animals = new HashMap<>();
  for(String dog : dogs.keySet()) {
    animals.put(dog, dogs.get(dog));
  }
  for(String cat : cats.keySet()) {
    animals.put(cat, cats.get(cat));
  }
  for(String elephant : elephants.keySet()) {
    animals.put(elephant, elephants.get(elephant));
  }

  // Now animals has all the argument maps merged into it, but is specifically
  // a HashMap...
}

如果它存在的话，我什至可以使用一些实用程序，比如Collections.merge(dogs, cats, elephants) 等。有什么想法吗？

Answer 1

一种方法是创建一组条目流，然后将其平面映射为具有条目流并将它们收集到地图中。

Map<String,Animal> animals = 
    Stream.of(dogs.entrySet(), cats.entrySet(), elephants.entrySet())
          .flatMap(Set::stream)
          .collect(toMap(Map.Entry::getKey, Map.Entry::getValue));

//Stream.of(dogs, cats, elephants).flatMap(m -> m.entrySet().stream()) could also be an option

也不是单行且没有使用Map#putAll 的流：

Map<String,Animal> animals = new HashMap<>(dogs);
animals.putAll(cats);
animals.putAll(elephants);

Answer 2

您可以使用Stream.Concat 完成手头的任务：

Map<String,Animal> animals = Stream.concat(Stream.concat(dogs.entrySet().stream(), cats.entrySet().stream()), elephants.entrySet().stream())
                                   .collect(Collectors.toMap
                                    (
                                       Map.Entry::getKey,
                                       Map.Entry::getValue
                                    )
                              );

您需要在这里小心，因为当合并并且有重复的键时，会按预期引发异常。

Answer 3

IMO 的最佳方式是在Alexis C.'s answer 中使用Map.putAll：

Map<String, Animal> animals = new HashMap<>(dogs);
animals.putAll(cats);
animals.putAll(elephants);

这个的一个变种：

Map<String, Animal> animals = new HashMap<>(dogs);
cats.forEach(animals::put);
elephants.forEach(animals::put);