按类别将一个 Set 拆分为多个 Set

Question

如果这个问题已在其他地方得到解答，我深表歉意，但我还没有找到好的答案！

---

我有一个 Set 对象，需要按其底层 类进行拆分 类型，将每个对象放入一个 Set 中（显然这意味着初始集合中的每个对象只会出现在新 Set 的一个中代码>对象）。

我当前的方法如下：

public static Map<Class,Set<Foo>> splitByClass(Set<Foo> foos) {

    // Create a map for the result
    Map<Class,Set<Foo>> FooMap = new HashMap<Class,Set<Foo>>();

    for (Foo foo: foos) {
        Class type = foo.getClass();

        // If a set for this key exists, add the item to that.
        if (FooMap.containsKey(type)) {
            FooMap.get(type).add(foo);
        } else {
            // Otherwise make a new set, add the item and the set to the map.
            Set<Foo> set = new HashSet<Foo>();
            set.add(foo);
            FooMap.put(type, set);
        }
    }        
    return FooMap;
}

我的问题：是否有一种更通用的方法可以根据某些评估方法（例如检查类类型）将Set拆分为子集？

Answer 1

您可以使用 Guava Multimap ，它会让生活变得更加简单：

Multimap<Class<? extends Foo>, Foo> mmap = HashMultimap.create();
for (Foo foo : foos) {
    mmap.put(foo.getClass(), foo);
}
return mmap;

此代码与上面的代码几乎相同，只是您必须将返回类型更改为 Multimap, Foo>。

这是一个适用于任何提供的接口类型的通用版本：

public static <X> Multimap<Class<X>, X> splitByClass(
    final Iterable<? extends X> implementations, final Class<X> interfaceType) {

    final Multimap<Class<X>, X> mmap = HashMultimap.create();
    for (final X implementation : implementations) {
        @SuppressWarnings("unchecked")
        // this should be safe:
        final Class<X> implementationType = (Class<X>) implementation.getClass();
        mmap.put(implementationType, implementation);
    }
    return mmap;
}