java - 在这种情况下替代 instanceof 方法

Question

你好，我想知道有什么更优雅的替代方法来代替这样的东西:

class Base...

class A extends Base...

class B extends Base...

//iterator of colection containing mixed As and Bs i want to remowe Bs and do omething with As
while(iterator.hasNext()) {
    Base next = iterator.next();
    if(next instanceof A) // do something
    if(next instanceof B)
        iterator.remove();
}

播种有哪些替代方案...

谢谢你的建议。

编辑:基类可能有很多子类，不仅仅是两个，而且它们的数量可能会随着时间的推移而增长

Answer 1

您可以在 Base 中创建方法并在 A 和 B 中覆盖它们。

例如:

class Base{
    public boolean shouldRemove(){
        return false;
    }
    public void doSomething(){
    }
}

class A extends Base{
    @Override
    public void doSomething() {            
    }
}

class B extends Base{
    @Override
    public boolean shouldRemove() {
        return true;
    }
}

然后你不需要知道对象是哪个类的实例:

    while(iterator.hasNext()) {
        Base next = iterator.next();
        if(next.shouldRemove()){
            iterator.remove();
        }
        else{
            next.doSomething();
        }
    }