怎样实现Golang的访问者模式 动态添加操作的设计方案

package main

import (
    "fmt"
    "math"
)

// Element 接口定义了数据结构接受访问者的方法
type Element interface {
    Accept(Visitor)
}

// Circle 是一个具体的图形元素
type Circle struct {
    Radius float64
}

// Accept 方法让 Circle 能够接受一个访问者
func (c *Circle) Accept(v Visitor) {
    v.VisitCircle(c) // Circle 调用访问者的 VisitCircle 方法
}

// Rectangle 是另一个具体的图形元素
type Rectangle struct {
    Width  float64
    Height float64
}

// Accept 方法让 Rectangle 能够接受一个访问者
func (r *Rectangle) Accept(v Visitor) {
    v.VisitRectangle(r) // Rectangle 调用访问者的 VisitRectangle 方法
}

// Visitor 接口定义了对每种具体 Element 的操作方法
// 注意：如果新增 Element 类型，这个接口就需要修改
type Visitor interface {
    VisitCircle(*Circle)
    VisitRectangle(*Rectangle)
}

// AreaCalculatorVisitor 是一个具体的访问者，用于计算面积
type AreaCalculatorVisitor struct {
    TotalArea float64
}

// VisitCircle 实现对 Circle 的面积计算
func (ac *AreaCalculatorVisitor) VisitCircle(c *Circle) {
    area := math.Pi * c.Radius * c.Radius
    ac.TotalArea += area
    fmt.Printf("计算圆形面积: %.2f\n", area)
}

// VisitRectangle 实现对 Rectangle 的面积计算
func (ac *AreaCalculatorVisitor) VisitRectangle(r *Rectangle) {
    area := r.Width * r.Height
    ac.TotalArea += area
    fmt.Printf("计算矩形面积: %.2f\n", area)
}

// DrawVisitor 是另一个具体的访问者，用于模拟绘制操作
type DrawVisitor struct{}

// VisitCircle 实现对 Circle 的绘制
func (dv *DrawVisitor) VisitCircle(c *Circle) {
    fmt.Printf("绘制圆形，半径: %.2f\n", c.Radius)
}

// VisitRectangle 实现对 Rectangle 的绘制
func (dv *DrawDrawVisitor) VisitRectangle(r *Rectangle) {
    fmt.Printf("绘制矩形，宽度: %.2f, 高度: %.2f\n", r.Width, r.Height)
}

func main() {
    // 创建一些图形元素
    shapes := []Element{
        &Circle{Radius: 5},
        &Rectangle{Width: 4, Height: 6},
        &Circle{Radius: 3},
    }

    // 使用面积计算访问者
    areaCalc := &AreaCalculatorVisitor{}
    fmt.Println("--- 计算总面积 ---")
    for _, shape := range shapes {
        shape.Accept(areaCalc)
    }
    fmt.Printf("所有图形的总面积: %.2f\n\n", areaCalc.TotalArea)

    // 使用绘制访问者
    drawVisitor := &DrawVisitor{}
    fmt.Println("--- 绘制图形 ---")
    for _, shape := range shapes {
        shape.Accept(drawVisitor)
    }

    // 动态添加操作的体现：
    // 如果我们现在想新增一个“导出为JSON”的操作，
    // 我们只需要创建一个新的 JSONExportVisitor，而无需修改 Circle 或 Rectangle 的代码。
    fmt.Println("\n--- 新增操作：导出为JSON ---")
    jsonExporter := &JSONExportVisitor{} // 假设这是新加的访问者
    for _, shape := range shapes {
        shape.Accept(jsonExporter)
    }
}

// JSONExportVisitor 是一个新加入的访问者，展示了如何“动态”添加操作
type JSONExportVisitor struct{}

func (jev *JSONExportVisitor) VisitCircle(c *Circle) {
    fmt.Printf("将圆形导出为JSON: {\"type\": \"circle\", \"radius\": %.2f}\n", c.Radius)
}

func (jev *JSONExportVisitor) VisitRectangle(r *Rectangle) {
    fmt.Printf("将矩形导出为JSON: {\"type\": \"rectangle\", \"width\": %.2f, \"height\": %.2f}\n", r.Width, r.Height)
}

Circle

Rectangle

JSONExportVisitor

Visitor

Circle

Rectangle

switch type

if _, ok := element.(*Circle); ok

Accept

Element

Visitor

Accept(Visitor)

Visitor

Element

VisitCircle(*Circle)

VisitRectangle(*Rectangle)

Visitor

Element

Visit

Triangle

Triangle

Accept

Visitor

VisitTriangle(*Triangle)

Visitor

AreaCalculatorVisitor

DrawVisitor

JSONExportVisitor

VisitTriangle

shape.Accept(visitor)

shape

Circle

Rectangle

Accept

Accept

element.Accept

visitor.VisitConcreteElement(element)

Visitor

Element

*Circle

*Rectangle

Visitor

interface{}

Visitor

Visit

interface{}

AreaCalculatorVisitor

DrawVisitor

JSONExportVisitor

Visitor

Element

Element