对虚拟方法的递归回调



我正在实现复合设计模式,我注意到我有很多代码重复的遍历。

例如:

class Composite extends BaseComposite{
    LinkedList<BaseComposite> objs;
    public BaseComposite foo(int id){
        for(BaseComposite obj : objs){
            if(obj.getId() == id){
                return obj;
            }
            BaseComposite tmp = obj.foo(id);
            if(tmp != null){
                return tmp;
            }
        }
        return null;
    }
    public BaseComposite bar(int id){
        for(BaseComposite obj : objs){
            if(obj.getId() == id){
                return obj;
            }
            BaseComposite tmp = obj.bar(id);
            if(tmp != null){
                return tmp;
            }
        }
        return null;
    }
}
class Leaf extends BaseComposite{
    public BaseComposite foo(int id){
        if(this.id == id){
            return this;
        }
        return null;
    }
    public BaseComposite bar(int id){
        return null;
    }
}

有没有办法防止复合类中的这种代码重复?有没有办法有一个虚拟方法的"回调"?

我宁愿避免反思。

在Java 8中,你可以只传递一个递归方法,并将一个函数作为第二个参数传递。然后你会传递this::foothis::bar作为参数。大致如下:

public BaseComposite traverse(BiFunction<Integer, BaseComposite, BaseComposite> f, int id){
        for(BaseComposite obj : objs){
            if(obj.getId() == id){
                return obj;
            }
            BaseComposite tmp = f.apply(obj, id);
            if(tmp != null){
                return tmp;
            }
        }
        return null;
    }
    public BaseComposite foo(int id){
        return traverse(this::foo, id);
    }
    public BaseComposite bar(int id){
        return traverse(this::bar, id);
    }

最新更新