在Set of Derived类中搜索基类的对象

时间:2016-12-16 19:02:38

标签: java hashset

我有一个名为Skill的课程:

public class Skill {
    private final int type;
    private final int level;

    public Skill(int type, int level) {
        this.type = type;
        this.level = level;
    }

    // Getters

    @Override
    public int hashCode() {
        int h = 17;
        h = 31 * h + type;
        h = 31 * h + level;
        return h;
    }

    @Override
    public boolean equals(Object obj) {
        if (this == obj)
            return true;
        if (obj instanceof Skill) {
            Skill that = (Skill) obj;
            return this.type == that.type && this.level == that.level;
        }
        return false;
    }
}

另一个人称PSkill为:

public class PSkill extends Skill {
    private final int preferenceLevel;
    private final boolean mandatory;

    public PSkill(int type, int level, int preferenceLevel, boolean mandatory) {
        super(type, level);
        this.preferenceLevel = preferenceLevel;
        this.mandatory = mandatory;
    }

    // Getters

    @Override
    public int hashCode() {
        return super.hashCode();
    }

    @Override
    public boolean equals(Object obj) {
        if (this == obj)
            return true;
        if (obj instanceof PSkill) {
            PSkill that = (PSkill) obj;
            return super.equals(obj) 
                && this.preferenceLevel == that.preferenceLevel
                && this.mandatory == that.mandatory;
        }
        return false;
    }
}

我的要求: 搜索一组PSkill个对象以查找Skill个对象的匹配项。 例如:Skill - type:1, level:2匹配PSkill - type:1, level:2, preferenceLevel: any, mandatory: any

当我运行以下内容时,它可以正常工作。

public class Invoker {
    public static void main(String[] args) {
        Set<PSkill> skills = new HashSet<>(Arrays.asList(new PSkill(1, 1, 1, true), new PSkill(1, 2, 1, true)));
        System.out.println(skills.contains(new Skill(1, 1))); // prints true
        System.out.println(skills.contains(new Skill(1, 3))); // prints false
    }
}

我知道为什么因为两种类型的hashCode()实现相同而且在HashSet实现中使用了key.equals(k),并且在我的情况下密钥是Skill对象,因此,平等有效。

来自HashSet实现

final Node<K,V> getNode(int hash, Object key) {
    Node<K,V>[] tab; Node<K,V> first, e; int n; K k;
    if ((tab = table) != null && (n = tab.length) > 0 &&
        (first = tab[(n - 1) & hash]) != null) {
        if (first.hash == hash && // always check first node
            ((k = first.key) == key || (key != null && key.equals(k))))  // here
            return first;
        if ((e = first.next) != null) {
            if (first instanceof TreeNode)
                return ((TreeNode<K,V>)first).getTreeNode(hash, key);
            do {
                if (e.hash == hash &&
                    ((k = e.key) == key || (key != null && key.equals(k)))) // and here
                    return e;
            } while ((e = e.next) != null);
        }
    }
    return null;
}

我知道我违反了hashCode()equals()合同。但是代码正常运行,即检查Skill是否匹配集合中的任何PSkill

我的问题是: key.equals(k)实现中的相等性检查HashSet是否依赖,可以在将来的版本k.equals(key)中反转,代码将停止工作? 还有,任何更好的方法可以使它不那么脆弱而不简单地循环集合?感谢

1 个答案:

答案 0 :(得分:0)

这种行为肯定无法保证。合同是,如果您提供hashCode和equals的正确实现,您将获得正确的hashset实现。如果您没有遵循合同的终止,则无法保证集合的运作情况。