Golang访问者模式操作对象结构实践

Element

Visitor

Element

Accept

Visitor

Circle

Square

Accept

Visitor

Visit

Visitor

Element

Visit

Accept

package main

import (
    "fmt"
    "math"
    "strings" // 用于处理JSON输出的逗号
)

// Element 接口：定义了一个接受访问者的方法
type Element interface {
    Accept(v Visitor)
}

// Visitor 接口：为每种具体元素类型定义一个访问方法
// 如果有更多元素类型，这里需要扩展
type Visitor interface {
    VisitCircle(c *Circle)
    VisitSquare(s *Square)
}

// Concrete Elements
type Circle struct {
    Radius float64
}

func (c *Circle) Accept(v Visitor) {
    v.VisitCircle(c)
}

type Square struct {
    Side float64
}

func (s *Square) Accept(v Visitor) {
    v.VisitSquare(s)
}

// Concrete Visitors
type AreaCalculator struct {
    TotalArea float64
}

func (ac *AreaCalculator) VisitCircle(c *Circle) {
    ac.TotalArea += math.Pi * c.Radius * c.Radius
}

func (ac *AreaCalculator) VisitSquare(s *Square) {
    ac.TotalArea += s.Side * s.Side
}

type JSONExporter struct {
    Output strings.Builder // 使用strings.Builder更高效
}

func (je *JSONExporter) VisitCircle(c *Circle) {
    je.Output.WriteString(fmt.Sprintf(`{"type": "circle", "radius": %.2f},`, c.Radius))
}

func (je *JSONExporter) VisitSquare(s *Square) {
    je.Output.WriteString(fmt.Sprintf(`{"type": "square", "side": %.2f},`, s.Side))
}

func main() {
    elements := []Element{
        &Circle{Radius: 5},
        &Square{Side: 10},
        &Circle{Radius: 3},
    }

    fmt.Println("--- 计算总面积 ---")
    areaCalc := &AreaCalculator{}
    for _, elem := range elements {
        elem.Accept(areaCalc)
    }
    fmt.Printf("总面积: %.2f\n", areaCalc.TotalArea)

    fmt.Println("\n--- 导出为 JSON ---")
    jsonExp := &JSONExporter{}
    jsonExp.Output.WriteString("[") // 开始JSON数组
    for _, elem := range elements {
        elem.Accept(jsonExp)
    }
    // 移除末尾逗号（如果有），然后关闭数组
    finalOutput := jsonExp.Output.String()
    if len(finalOutput) > 1 && finalOutput[len(finalOutput