Golang如何使用装饰者模式_Golang 装饰者模式实践_技术教程

装饰者模式通过接口和组合动态扩展功能，适用于日志、权限等场景。定义统一接口，装饰器持有接口实例并叠加行为。Go中用结构体嵌入实现，如消息处理器可包装日志与重试逻辑，NewRetryProcessor(p, 3)层层增强。

装饰者模式在Go语言中可以通过接口和组合的方式优雅实现。它允许你在不修改原有代码的前提下，动态地给对象添加新功能。这种模式特别适合日志、权限校验、缓存、监控等横切关注点的处理。

理解装饰者模式的核心思想

装饰者模式的关键在于：

定义一个统一的接口，所有具体实现和装饰器都遵循这个接口
装饰器持有一个该接口的实例，可以在调用前后插入额外逻辑
通过层层包装，实现功能叠加，而不是继承扩展

在Go中，由于没有类的概念，我们依靠结构体嵌入和函数式编程技巧来实现这一模式。

基于接口的装饰者实现

假设我们要构建一个消息处理器，支持普通处理、日志记录和重试机制：

type MessageProcessor interface {
    Process(msg string) error
}

type BasicProcessor struct{}

func (p *BasicProcessor) Process(msg string) error {
    fmt.Println("Processing:", msg)
    return nil
}

现在添加日志装饰器：

type LoggingProcessor struct {
    processor MessageProcessor
}

func NewLoggingProcessor(p MessageProcessor) MessageProcessor {
    return &LoggingProcessor{processor: p}
}

func (l *LoggingProcessor) Process(msg string) error {
    fmt.Println("Log: Start processing", msg)
    err := l.processor.Process(msg)
    fmt.Println("Log: Finished processing", msg)
    return err
}

再添加重试装饰器：

type RetryProcessor struct {
    processor MessageProcessor
    retries   int
}

func NewRetryProcessor(p MessageProcessor, retries int) MessageProcessor {
    return &RetryProcessor{processor: p, retries: retries}
}

func (r *RetryProcessor) Process(msg string) error {
    for i := 0; i <= r.retries; i++ {
        err := r.processor.Process(msg)
        if err == nil {
            return nil
        }
        fmt.Printf("Retry %d failed: %v\n", i+1, err)
    }
    return fmt.Errorf("failed after %d retries", r.retries)
}

使用方式：

processor := &BasicProcessor{}
processor = NewLoggingProcessor(processor)
processor = NewRetryProcessor(processor, 2)
processor.Process("Hello")

函数式装饰器的简洁写法

Go中还可以用函数类型实现更灵活的装饰者模式：

type HandlerFunc func(string) error

func (f HandlerFunc) Serve(msg string) error {
    return f(msg)
}

func WithLogging(next HandlerFunc) HandlerFunc {
    return func(msg string) error {
        fmt.Println("Log start:", msg)
        err := next(msg)
        fmt.Println("Log end:", msg)
        return err
    }
}

func WithMetrics(next HandlerFunc) HandlerFunc {
    return func(msg string) error {
        start := time.Now()
        err := next(msg)
        fmt.Printf("Duration: %v\n", time.Since(start))
        return err
    }
}

链式调用：

handler := HandlerFunc(func(msg string) error {
    fmt.Println("Handling:", msg)
    return nil
})

handler = WithLogging(WithMetrics(handler))
handler.Serve("test message")

这种方式更轻量，适合中间件场景，如HTTP处理链。