一文搞懂Go实现装饰者模式

原创

闫同学

发布于 2025-01-05 21:59:17

4900

装饰者模式（Decorator Pattern）作为结构型设计模式之一，允许在不改变对象自身的情况下，动态地给对象添加新的功能。本文将深入探讨装饰者模式，并通过Go语言实现该模式，结合实际案例和UML图，帮助你更好地理解和应用装饰者模式。

装饰者模式简介

装饰者模式允许在不改变原有对象结构的情况下，动态地给对象添加新的职责。它通过创建一个包装对象，将原有对象包裹起来，并在保持原有对象接口的同时，扩展其功能。这种模式的核心在于透明地扩展对象的功能，而无需继承。

装饰者模式的主要组成

组件（Component）：定义一个对象接口，可以给这些对象动态地添加职责。
具体组件（ConcreteComponent）：实现组件接口的具体类，代表被装饰的对象。
装饰者（Decorator）：持有一个组件对象的引用，并定义与组件接口一致的接口。
具体装饰者（ConcreteDecorator）：实现装饰者接口，负责向组件添加新的职责。

Go语言实现装饰者模式

Go语言虽然不支持传统的类继承，但通过接口和组合（embedding），同样可以实现装饰者模式。下面通过一个具体的例子来说明。

代码示例

假设我们有一个简单的通知系统，支持发送消息。我们希望在发送消息时，可以动态地添加日志记录和消息加密的功能。

package main

import (
    "fmt"
)

// Component 定义通知接口
type Notifier interface {
    Send(message string) error
}

// ConcreteComponent 具体的通知实现
type EmailNotifier struct{}

func (e *EmailNotifier) Send(message string) error {
    fmt.Println("Sending email with message:", message)
    return nil
}

// Decorator 装饰者基类，持有一个Notifier
type NotifierDecorator struct {
    notifier Notifier
}

func (d *NotifierDecorator) Send(message string) error {
    return d.notifier.Send(message)
}

// ConcreteDecoratorA 添加日志功能
type LoggingDecorator struct {
    NotifierDecorator
}

func NewLoggingDecorator(n Notifier) Notifier {
    return &LoggingDecorator{
        NotifierDecorator{notifier: n},
    }
}

func (l *LoggingDecorator) Send(message string) error {
    fmt.Println("Logging: About to send message.")
    return l.notifier.Send(message)
}

// ConcreteDecoratorB 添加加密功能
type EncryptionDecorator struct {
    NotifierDecorator
}

func NewEncryptionDecorator(n Notifier) Notifier {
    return &EncryptionDecorator{
        NotifierDecorator{notifier: n},
    }
}

func (e *EncryptionDecorator) Send(message string) error {
    encryptedMessage := encrypt(message)
    fmt.Println("Encrypting message.")
    return e.notifier.Send(encryptedMessage)
}

func encrypt(message string) string {
    // 简单的加密示例，实际应用中应使用更安全的加密方式
    return fmt.Sprintf("Encrypted(%s)", message)
}

func main() {
    // 创建基本的邮件通知
    notifier := &EmailNotifier{}

    // 添加日志功能
    notifierWithLogging := NewLoggingDecorator(notifier)

    // 添加加密功能
    notifierWithLoggingAndEncryption := NewEncryptionDecorator(notifierWithLogging)

    // 发送消息，具有日志记录和加密功能
    notifierWithLoggingAndEncryption.Send("Hello, Decorator Pattern!")
}

代码解析

Notifier接口：定义了发送通知的Send方法。
EmailNotifier：实现了Notifier接口，代表具体的通知方式。
NotifierDecorator：装饰者基类，持有一个Notifier接口的引用，并实现了Send方法，调用被装饰对象的Send方法。
LoggingDecorator：具体装饰者，添加日志记录功能。在发送消息前打印日志。
EncryptionDecorator：具体装饰者，添加消息加密功能。对消息进行简单加密后发送。
main函数：创建了一个EmailNotifier实例，并依次添加日志和加密功能，最终发送消息。

运行上述代码，输出如下：

Logging: About to send message.
Encrypting message.
Sending email with message: Encrypted(Hello, Decorator Pattern!)

可以看到，消息发送前先记录了日志，并对消息进行了加密。

实际案例

场景描述

假设我们在开发一个图形绘制应用，需要绘制不同形状（如圆形、矩形）。为了增强图形的功能，如添加边框、阴影等，我们可以使用装饰者模式动态地为图形对象添加这些功能，而无需修改原有的图形类。

实现步骤

定义组件接口：Shape接口，包含Draw方法。
具体组件：如Circle、Rectangle，实现Shape接口。
装饰者：ShapeDecorator，持有一个Shape对象的引用，并实现Shape接口。
具体装饰者：如BorderDecorator、ShadowDecorator，在Draw方法中添加额外的功能。

代码示例

package main

import (
    "fmt"
)

// Shape 定义图形接口
type Shape interface {
    Draw()
}

// Circle 具体图形实现
type Circle struct{}

func (c *Circle) Draw() {
    fmt.Println("Drawing a Circle.")
}

// Rectangle 具体图形实现
type Rectangle struct{}

func (r *Rectangle) Draw() {
    fmt.Println("Drawing a Rectangle.")
}

// ShapeDecorator 装饰者基类
type ShapeDecorator struct {
    shape Shape
}

func (d *ShapeDecorator) Draw() {
    d.shape.Draw()
}

// BorderDecorator 添加边框功能
type BorderDecorator struct {
    ShapeDecorator
}

func NewBorderDecorator(s Shape) Shape {
    return &BorderDecorator{
        ShapeDecorator{shape: s},
    }
}

func (b *BorderDecorator) Draw() {
    b.shape.Draw()
    b.addBorder()
}

func (b *BorderDecorator) addBorder() {
    fmt.Println("Adding a border.")
}

// ShadowDecorator 添加阴影功能
type ShadowDecorator struct {
    ShapeDecorator
}

func NewShadowDecorator(s Shape) Shape {
    return &ShadowDecorator{
        ShapeDecorator{shape: s},
    }
}

func (sdc *ShadowDecorator) Draw() {
    sdc.shape.Draw()
    sdc.addShadow()
}

func (sdc *ShadowDecorator) addShadow() {
    fmt.Println("Adding a shadow.")
}

func main() {
    // 创建一个圆形
    circle := &Circle{}

    // 添加边框功能
    circleWithBorder := NewBorderDecorator(circle)

    // 添加阴影功能
    circleWithBorderAndShadow := NewShadowDecorator(circleWithBorder)

    // 绘制带有边框和阴影的圆形
    circleWithBorderAndShadow.Draw()

    fmt.Println("-----")

    // 创建一个矩形并直接添加阴影
    rectangle := &Rectangle{}
    rectangleWithShadow := NewShadowDecorator(rectangle)
    rectangleWithShadow.Draw()
}

运行结果

Drawing a Circle.
Adding a border.
Adding a shadow.
-----
Drawing a Rectangle.
Adding a shadow.

通过装饰者模式，我们可以灵活地为不同的图形对象添加各种功能，而无需修改原有的图形类。

UML图

下面是装饰者模式的UML类图示意：

+----------------+           +-------------------+
|    Notifier    |<--------- |  NotifierDecorator |
+----------------+           +-------------------+
| +Send(message) |           | +Send(message)    |
+----------------+           | -notifier: Notifier|
                             +-------------------+
                                     ^
                                     |
           +-------------------------+-------------------------+
           |                           |                       |
+-------------------+     +--------------------------+   +-----------------------+
| LoggingDecorator  |     |   EncryptionDecorator    |   | 其他具体装饰者类       |
+-------------------+     +--------------------------+   +-----------------------+
| +Send(message)    |     | +Send(message)           |   
+-------------------+     +--------------------------+

说明

Notifier（组件接口）：定义了Send方法。
EmailNotifier（具体组件）：实现了Notifier接口。
NotifierDecorator（装饰者）：实现了Notifier接口，并持有一个Notifier对象的引用。
LoggingDecorator & EncryptionDecorator（具体装饰者）：继承自NotifierDecorator，在Send方法中添加额外的功能。

小总结

装饰者模式通过组合和接口实现了对对象功能的动态扩展，避免了继承带来的类爆炸问题。在Go语言中，虽然没有类的概念，但通过接口和结构体的组合，同样可以优雅地实现装饰者模式。通过本文的介绍和示例，相信你已经对装饰者模式有了更深的理解，并能够在实际开发中灵活应用这一模式，为你的代码增加更多的灵活性和可维护性。

原创声明：本文系作者授权腾讯云开发者社区发表，未经许可，不得转载。

如有侵权，请联系 cloudcommunity@tencent.com 删除。

腾讯技术创作特训营S11#重启人生