时间戳

首页 > 解决方案 > 通过包装 LinkedHashSet 实现 IdentityLinkedHashSet

java - 通过包装 LinkedHashSet 实现 IdentityLinkedHashSet

问题描述

在 Java 标准库中，aLinkedHashSet与 a 相同，HashSet但具有可预测的迭代顺序（插入顺序）。我需要一个IdentityHashSet具有可预测迭代顺序（插入顺序）的（使用对象标识或引用相等，而不是对象相等）。

虽然IdentityHashSetJava 的标准库中没有，但您可以使用IdentityHashMap以下语句轻松创建一个：

Set<T> identitySet = java.util.Collections.newSetFromMap(new IdentityHashMap<>());

这与 Guava 中使用的实现几乎相同Sets.newIdentityHashSet()；但这具有未指定的，通常是混乱的HashMap键（和HashSet元素）顺序。

我对 an 实现的搜索IdentityLinkedHashSet没有结果，所以我决定自己实现它。我发现的一个有用的结果是这个答案，它建议使用一个身份包装类与LinkedHashSet.

我试图实现这个想法。下面是我的实现的一些关键片段。在此处访问完整的要点。

public class IdentityLinkedHashSet<E> implements Set<E> {

    private LinkedHashSet<IdentityWrapper> set;

    /* ... constructors ... */

    @Override
    public boolean add(E e) {
        return set.add(new IdentityWrapper(e));
    }

    @Override
    public boolean contains(Object obj) {
        return set.contains(new IdentityWrapper((E) obj));
    }

    /* ... rest of Set methods ... */

    private class IdentityWrapper {

        public final E ELEM;

        IdentityWrapper(E elem) {
            this.ELEM = elem;
        }

        @Override
        public boolean equals(Object obj) {
            return obj != null && ELEM == obj;
        }

        @Override
        public int hashCode() {
            return System.identityHashCode(ELEM);
        }
    }
}

然后我写了一些单元测试来验证我的实现。不幸的是，一些断言失败了！这是我的测试：

String str1 = new String("test-1");
String str2 = new String("test-2");
String str3 = new String("test-2");

Set<String> identitySet = new IdentityLinkedHashSet<>();

assertTrue(idSet.add(str1));
assertFalse(idSet.add(str1));       //  <-- fails!
assertTrue(idSet.contains(str1));   //  <-- fails!
//
assertTrue(idSet.add(str2));
assertFalse(idSet.add(str2));       //  <-- fails!
assertTrue(idSet.contains(str2));   //  <-- fails!
assertFalse(idSet.contains(str3));
//
assertTrue(idSet.add(str3));
assertFalse(idSet.add(str3));       //  <-- fails!
assertTrue(idSet.contains(str3));   //  <-- fails!
assertEquals(3, idSet.size());      //  <-- fails!

我在这个实现中做错了什么？

标签： javahashset

解决方案

调用IdentityWrapper.equals()方法时。LinkedHashSet不会传递的而是ELEM一个IdentityWrapper对象。

您必须引入额外的检查 ( instanceOf)，然后打开传递的对象以比较元素：

public boolean equals(Object obj) {
    return (obj instanceof IdentityLinkedHashSet<?>.IdentityWrapper) && 
        ELEM == ((IdentityLinkedHashSet<?>.IdentityWrapper) obj).ELEM;
}

一些注意事项：

  1. instanceof将检查 null，因此您可以完全删除该检查。

  2. 你必须IdentityWrapper像这样引用：IdentityLinkedHashSet<?>.IdentityWrapper因为它不是一个static类。如评论中所述。可以将其设为静态，并且可以将其类型ELEM从 更改E为Object。您还会用更好的equals方法离开哪个：

    private static class IdentityWrapper {

    public final Object ELEM;

    IdentityWrapper(Object elem) {
        this.ELEM = elem;
    }

    @Override
    public boolean equals(Object obj) {
        return (obj instanceof IdentityWrapper) && ELEM == ((IdentityWrapper) obj).ELEM;
    }

    @Override
    public int hashCode() {
        return System.identityHashCode(ELEM);
    }
}

推荐阅读