我正在实现复合设计模式,我注意到我有很多代码重复的遍历。
例如:
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::foo
或this::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);
}