Go中组合与访问者模式结合可高效处理树形结构:组合通过嵌入BaseNode统一管理子节点,访问者解耦遍历与业务逻辑,各节点实现Accept方法分发至对应VisitXxx,支持安全遍历与泛型增强。

在 Go 语言中,组合(Composition)与访问者(Visitor)模式结合,是处理复杂树形结构(如 AST、配置树、DOM-like 结构)的高效方式:组合用于构建灵活的树节点关系,访问者用于解耦遍历逻辑与业务操作,避免在节点类型中堆砌大量 if/switch 或重复遍历代码。
用组合定义可扩展的树节点
Go 不支持继承,但可通过嵌入(embedding)实现“组合即扩展”。每个节点类型内嵌一个通用接口或基础结构,统一管理子节点、父引用、类型标识等共性字段。
例如:
type Node interface {
Accept(v Visitor)
Children() []Node
Kind() string
}
type BaseNode struct {
children []Node
parent Node
}
func (b BaseNode) Children() []Node { return b.children }
func (b BaseNode) SetParent(p Node) { b.parent = p }
type BinaryOp struct {
BaseNode
Left, Right Node
Op string
}
func (n BinaryOp) Kind() string { return "BinaryOp" }
func (n BinaryOp) Accept(v Visitor) { v.VisitBinaryOp(n) }
所有具体节点(BinaryOp、NumberLit、CallExpr 等)都嵌入 BaseNode,共享子节点管理和遍历能力;同时各自实现 Accept 方法,将控制权交由访问者——这是访问者模式的关键分发点。
立即学习“go语言免费学习笔记(深入)”;
定义访问者接口并实现具体行为
访问者是一个接口,为每种节点类型声明一个 VisitXxx 方法。实际逻辑(如求值、格式化、校验、转换)放在具体访问者实现里,与节点结构完全分离。
示例接口:
type Visitor interface {
VisitBinaryOp(*BinaryOp)
VisitNumberLit(*NumberLit)
VisitCallExpr(*CallExpr)
// ... 其他节点类型方法
}
一个简单求值访问者:
type EvalVisitor struct {
env map[string]float64
}
func (v EvalVisitor) VisitBinaryOp(n BinaryOp) {
left := v.eval(n.Left)
right := v.eval(n.Right)
switch n.Op {
case "+": v.result = left + right
case "": v.result = left right
}
}
func (v EvalVisitor) VisitNumberLit(n NumberLit) {
v.result = n.Value
}
func (v *EvalVisitor) eval(n Node) float64 {
n.Accept(v) // 触发对应 Visit 方法
return v.result
}
注意:Go 中没有虚函数调用,所以 n.Accept(v) 是显式委托,必须确保每个节点的 Accept 正确调用访问者的对应方法 —— 这是手动维护的“双分派”模拟。
安全遍历:避免循环引用与空指针
树形结构常含父子双向引用或共享子节点,直接递归 Accept 可能导致栈溢出或无限循环。建议在访问者内部维护已访问节点集合(如 map[uintptr]bool),或在 BaseNode 中加入 Visited bool 字段(仅调试/单次场景)。
- 使用
unsafe.Pointer做节点唯一标识(轻量且稳定):
func (v *EvalVisitor) VisitBinaryOp(n *BinaryOp) {
ptr := uintptr(unsafe.Pointer(n))
if v.seen[ptr] { return }
v.seen[ptr] = true
// 递归处理子节点
v.eval(n.Left)
v.eval(n.Right)
}
- 对
Children()返回值做 nil 判空,避免 panic; - 若需深度优先 / 广度优先 / 后序遍历,可在访问者中自行控制子节点调用顺序,不依赖节点内部逻辑。
实用增强技巧
- 泛型访问者基类(Go 1.18+):用泛型封装公共逻辑,如错误收集、上下文传递、路径追踪:
type TracingVisitor[T any] struct {
path []string
data T
}
func (v TracingVisitor[T]) Enter(n Node) {
v.path = append(v.path, n.Kind())
}
func (v TracingVisitor[T]) Leave(n Node) {
v.path = v.path[:len(v.path)-1]
}
-
组合多个访问者:用装饰器模式链式调用,例如
&ValidationVisitor{&FormattingVisitor{}}; -
生成式访问者:配合
go:generate和 AST 分析工具,自动生成VisitXxx方法签名和空实现,减少手写样板。
不复杂但容易忽略:访问者不是银弹——若节点类型极少变动而操作逻辑频繁增删,直接方法调用更清晰;只有当“操作种类多、节点结构稳定”时,组合+访问者才真正发挥价值。










