访问者模式允许在不修改元素的前提下添加新操作。通过定义Element接口和Visitor接口，实现数据结构与行为分离。示例中文件系统使用Accept方法接收访问者，PrintVisitor打印名称，SizeVisitor统计大小，体现解耦优势。

在 Go 语言中，访问者模式（Visitor Pattern）是一种行为设计模式，它允许你在不修改结构的前提下，为复杂对象结构中的元素添加新的操作。这种模式特别适用于需要对多种类型的数据结构进行不同处理的场景，比如解析 AST（抽象语法树）、序列化、渲染等。

访问者模式核心结构

访问者模式主要包含以下几个部分：

• Element（元素）接口：定义 Accept 方法，接受一个访问者。
• ConcreteElement（具体元素）：实现 Accept 方法，调用访问者的 Visit 方法。
• Visitor（访问者）接口：定义 Visit 方法，对应每种元素类型。
• ConcreteVisitor（具体访问者）：实现 Visit 方法，执行具体逻辑。

示例：文件系统结构遍历

假设我们要遍历一个模拟的文件系统结构（包含文件和目录），并分别实现“打印名称”和“统计大小”两种操作。使用访问者模式可以将数据结构与操作解耦。

// Element 接口
type FileSystemElement interface {
  Accept(visitor Visitor)
}

// 具体元素：文件
type File struct {
  Name string
  Size int
}

func (f *File) Accept(visitor Visitor) {
  visitor.VisitFile(f)
}

// 具体元素：目录
type Directory struct {
  Name    string
  Children []FileSystemElement
}

func (d *Directory) Accept(visitor Visitor) {
  visitor.VisitDirectory(d)
  for _, child := range d.Children {
    child.Accept(visitor) // 递归访问子元素
  }
}

定义访问者接口与实现

我们定义一个访问者接口，并实现两个具体访问者：一个用于打印结构，另一个用于计算总大小。

Golang Uber Dig

使用 uber-go/dig 在 Golang 中实现依赖注入——基于反射容器，提供 Provide/Invoke、dig.In/dig.Out 参数与结果对象，支持命名依赖等

下载

立即学习“go语言免费学习笔记（深入）”；

// Visitor 接口
type Visitor interface {
  VisitFile(*File)
  VisitDirectory(*Directory)
}

// 打印访问者
type PrintVisitor struct {}

func (v *PrintVisitor) VisitFile(f *File) {
  print("File: " + f.Name + "\n")
}

func (v *PrintVisitor) VisitDirectory(d *Directory) {
  print("Dir: " + d.Name + "\n")
}

// 统计大小访问者
type SizeVisitor struct {
  TotalSize int
}

func (v *SizeVisitor) VisitFile(f *File) {
  v.TotalSize += f.Size
}

func (v *SizeVisitor) VisitDirectory(d *Directory) {
  // 目录本身不计入大小
}

使用示例

构建一个简单的文件树，并使用不同的访问者进行操作。

func main() {
  root := &Directory{
    Name: "root",
    Children: []FileSystemElement{
      &File{Name: "a.txt", Size: 100},
      &Directory{
        Name: "subdir",
        Children: []FileSystemElement{
          &File{Name: "b.txt", Size: 200},
        },
      },
    },
  }

  // 使用打印访问者
  printVisitor := &PrintVisitor{}
  root.Accept(printVisitor)

  // 使用统计访问者
  sizeVisitor := &SizeVisitor{}
  root.Accept(sizeVisitor)
  fmt.Printf("Total size: %d\n", sizeVisitor.TotalSize)
}

输出结果：

Dir: root
File: a.txt
Dir: subdir
File: b.txt
Total size: 300

基本上就这些。通过访问者模式，我们可以轻松扩展新操作（如压缩、权限检查），而无需改动现有的文件或目录结构。Go 虽然没有方法重载，但通过接口和指针类型匹配，依然能很好地实现这一模式。关键在于 Accept 和 Visit 的双向调用机制。