如何用Golang实现组合模式 处理树形结构的统一接口

package main

import (
    "fmt"
    "strings"
)

// DocumentComponent 是所有文档组件的通用接口
type DocumentComponent interface {
    Render(indent int) string // 渲染内容的方法
}

// Paragraph 是叶子组件，代表一个段落
type Paragraph struct {
    Content string
}

// Render 实现 DocumentComponent 接口
func (p *Paragraph) Render(indent int) string {
    return fmt.Sprintf("%s- %s\n", strings.Repeat("  ", indent), p.Content)
}

// Chapter 是组合组件，代表一个章节，可以包含多个文档组件
type Chapter struct {
    Title    string
    Children []DocumentComponent
}

// AddComponent 向章节中添加子组件
func (c *Chapter) AddComponent(comp DocumentComponent) {
    c.Children = append(c.Children, comp)
}

// Render 实现 DocumentComponent 接口
func (c *Chapter) Render(indent int) string {
    var sb strings.Builder
    sb.WriteString(fmt.Sprintf("%s+ %s\n", strings.Repeat("  ", indent), c.Title))
    for _, child := range c.Children {
        sb.WriteString(child.Render(indent + 1))
    }
    return sb.String()
}

func main() {
    // 创建叶子组件
    introPara := &Paragraph{Content: "这是一篇关于组合模式的介绍。"}
    conclusionPara := &Paragraph{Content: "总结一下，组合模式非常有用。"}
    detailPara1 := &Paragraph{Content: "组合模式允许你将对象组合成树形结构以表示“部分-整体”的层次结构。"}
    detailPara2 := &Paragraph{Content: "它使得客户端对单个对象和组合对象的使用具有一致性。"}

    // 创建组合组件
    chapter1 := &Chapter{Title: "第一章：模式概述"}
    chapter1.AddComponent(detailPara1)
    chapter1.AddComponent(detailPara2)

    subChapter1_1 := &Chapter{Title: "1.1 子章节：Go语言实现"}
    subChapter1_1.AddComponent(&Paragraph{Content: "Go语言的接口特性非常适合实现组合模式。"})
    chapter1.AddComponent(subChapter1_1)

    rootDocument := &Chapter{Title: "Go语言中的组合模式应用"}
    rootDocument.AddComponent(introPara)
    rootDocument.AddComponent(chapter1)
    rootDocument.AddComponent(conclusionPara)

    // 统一接口调用，无论叶子还是组合，都调用 Render 方法
    fmt.Println(rootDocument.Render(0))
}

DocumentComponent

if-else

Paragraph

Chapter

DocumentComponent

DocumentComponent

AddComponent

RemoveComponent

DocumentComponent

Paragraph

Composite

Chapter

Chapter

chapter.AddComponent()

Render

DocumentComponent

DocumentComponent

Paragraph

Chapter

comp.(type)