访问者模式Golang实现与封装解析

package main

import "fmt"

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

// 文本段落
type Paragraph struct {
    Content string
}

func (p *Paragraph) Accept(v Visitor) {
    v.VisitParagraph(p)
}

// 图片
type Image struct {
    URL string
}

func (i *Image) Accept(v Visitor) {
    v.VisitImage(i)
}

// Visitor 接口：定义访问不同类型的方法
type Visitor interface {
    VisitParagraph(*Paragraph)
    VisitImage(*Image)
}

// 渲染访问者
type RenderVisitor struct{}

func (v *RenderVisitor) VisitParagraph(p *Paragraph) {
    fmt.Println("渲染段落:", p.Content)
}

func (v *RenderVisitor) VisitImage(i *Image) {
    fmt.Println("渲染图片:", i.URL)
}

// JSON导出访问者
type JSONExportVisitor struct{}

func (v *JSONExportVisitor) VisitParagraph(p *Paragraph) {
    fmt.Printf("JSON: {\"type\": \"paragraph\", \"content\": \"%s\"}\n", p.Content)
}

func (v *JSONExportVisitor) VisitImage(i *Image) {
    fmt.Printf("JSON: {\"type\": \"image\", \"url\": \"%s\"}\n", i.URL)
}

// 文档结构，包含多个元素
type Document struct {
    Elements []Element
}

func (d *Document) Add(e Element) {
    d.Elements = append(d.Elements, e)
}

func (d *Document) Accept(v Visitor) {
    for _, e := range d.Elements {
        e.Accept(v)
    }
}

func main() {
    doc := &Document{}
    doc.Add(&Paragraph{Content: "Hello World"})
    doc.Add(&Image{URL: "https://example.com/image.png"})

    fmt.Println("=== 渲染 ===")
    doc.Accept(&RenderVisitor{})

    fmt.Println("=== 导出为JSON ===")
    doc.Accept(&JSONExportVisitor{})
}