在访客模式中组合访客

Combine visitors in VisitorPattern

本文关键字:访客 组合 模式      更新时间:2023-10-16

我根据 cpppaterns.com 实现了访问者模式,以遍历树并根据节点类型应用特定操作。我总是需要以相同的顺序遍历树,因此我将决定下一个要访问哪个孩子的逻辑移动到基类Visitor中。单个访问者在树节点上执行特定操作,然后在基Visitor中调用visit方法。到目前为止,这效果很好。

在某些情况下,需要在一棵树上执行在多个访问者之间实现的操作(例如,VisitorRewriteXVisitorPrintX(。天真的方法是按顺序执行访问者。但是,这需要多次遍历树,这是低效的。

当然,我也可以创建一个新的访问者(例如,VisitorRewritePrintX(,它只是调用其他两个访问者 - 但我认为这不是干净的代码,也不灵活。是否有一些更通用的方法(例如,一种模式(可以帮助我以某种方式允许在不同访问者中实施的操作的"堆叠"?

有多种方法可以做到这一点,一种方法是拥有一个通用的访问者,您可以为其注册每次访问调用的处理程序:

struct visitor {
visitor(std::initializer_list<std::function<void(int)>> cbs) : callbacks(cbs) {
}
void visit(int val) {
for( auto &cb: callbacks ) {
cb(val);
}
}
std::vector<std::function<void(val)>> callbacks;
};

由于您visit不同元素的函数,因此可能需要将其更改为具有相应成员函数的类,而不是使用std::function

另一种方法是利用可变参数模板参数:

struct handler_a {
static void visit(int val) {
std::cout << "handler_a: " << val << std::endl;
}
};
struct handler_b {
static void visit(int val) {
std::cout << "handler_b: " << val << std::endl;
}
};
template <class ...Handlers>
struct visitor {
void visit(int val) {
(Handlers::visit(val), ...);
}
};
int main() {
visitor<handler_a,handler_b> v;
v.visit(10);
return 0;
}

可变参数方法的优点是没有循环,因为编译器在编译时为调用创建代码。但它的缺点是,您无法在遍历时使用这些处理程序保存临时信息,因为这些是静态函数。但是,您可以使用与遍历对象一起传递的上下文对象来解决这个问题。

如果需要,您可以使用using visitor_a_b = visitor<handler_a,handler_b>来别名您的访问者集。

我省略了继承以保持代码最小,因此您需要重新添加它们才能virtual void accept(visitor& v) override工作。

基本上,我会做这样的事情:

struct IVisitor
{
virtual ~IVisitor() = default;
virtual void visit(NodeA&) = 0;
virtual void visit(NodeB&) = 0;
// ...
};

struct ITraversal : IVisitor
{
// Traversal logic
void visit(NodeA& n) final { visit(n.left); action(n); visit(n.right); }
void visit(NodeB& n) final { action(n); visit(n.next); }
// Actions
virtual void action(NodeA&) = 0;
virtual void action(NodeB&) = 0;
// ...
};
struct VisitorRewriteX : ITraversal
{
void action(NodeA&) override;
void action(NodeB&) override;
// ...
};
struct VisitorPrintX : ITraversal
{
void action(NodeA&) override;
void action(NodeB&) override;
// ...
};
// To allow type T with generic action
template <typename T>
struct VisitorT : ITraversal
{
VisitorT() = default;
VisitorT(const T& t) : t(t) {}
void action(NodeA& n) override { t.action(n); }
void action(NodeB& n) override { t.action(n); }
// ...
T t;
};
template <typename ... Ts>
struct Combine
{
Combine() = default;
Combine(Ts... args) : visitors{args...} {}
template <typename Node>
void action(Node& n) override {
std::apply([&](auto&...visitor){ (visitor.action(n), ...); }, visitors);
}
private:
std::tuple<Ts...> visitors;
};

,然后使用其中之一

VisitorRewriteXVisitorPrintXVisitorT<Combine<VisitorRewriteX, VisitorPrintX>>