设计模式之装饰器模式：精湛代码技巧的必备武器

闲谈

2023-04-27 05:30:29

装饰器模式：为代码增添调味

在软件开发的世界里，代码就像一道料理，而装饰器模式就像一种调味品，让你能够在不改变原有代码的情况下，为代码添加新功能或调整现有功能。就像在咖啡中加入牛奶、糖或奶油，装饰器模式让你可以灵活地对代码进行定制，赋予其新的风味。

装饰器模式的魅力

装饰器模式的优势在于它能让你：

代码重用： 把共用的功能封装成装饰器类，你可以轻松地将这些功能应用到不同的类或对象，避免重复编码。
代码可读性： 把复杂的代码逻辑分解成更小、更易理解的单元，让代码更易于理解和维护。
代码灵活性： 允许你动态地添加或修改功能，而无需修改原有代码，提升代码的灵活性。
代码可扩展性： 通过添加新的装饰器来扩展现有功能，让代码更易于扩展，而无需修改原有代码。

实战演练

理论固然重要，但实践出真知。让我们用一个实际场景来了解装饰器模式的妙用。

场景： 开发一个日志记录系统，记录不同级别的日志信息，比如错误日志、警告日志和调试日志。

传统方法： 通常的做法是为每种类型的日志信息创建一个单独的日志记录类，如 ErrorLogger、WarningLogger 和 DebugLogger。每当需要记录日志时，你需要根据日志信息的类型选择相应的日志记录类。这种方法存在的主要问题是代码重复，因为每种类型的日志记录类都包含了相同的基本日志记录功能。

装饰器模式： 使用装饰器模式，我们可以把日志记录的基本功能封装到一个装饰器类中，比如 LoggerDecorator。然后，我们可以创建不同的日志记录装饰器来扩展基本功能，比如 ErrorLoggerDecorator、WarningLoggerDecorator 和 DebugLoggerDecorator。每当需要记录日志时，只需要将相应的日志记录装饰器应用到基本日志记录器即可。

这种方法不仅避免了代码重复，还提高了代码的可读性和可维护性。你可以轻松地添加新的日志记录装饰器来扩展日志记录功能，而无需修改原有代码。

代码示例

# 基本日志记录器
class Logger:
    def log(self, message):
        print(message)

# 装饰器基类
class LoggerDecorator(Logger):
    def __init__(self, logger):
        self.logger = logger

    def log(self, message):
        self.logger.log(message)

# 错误日志记录装饰器
class ErrorLoggerDecorator(LoggerDecorator):
    def log(self, message):
        self.logger.log("[Error] " + message)

# 警告日志记录装饰器
class WarningLoggerDecorator(LoggerDecorator):
    def log(self, message):
        self.logger.log("[Warning] " + message)

# 调试日志记录装饰器
class DebugLoggerDecorator(LoggerDecorator):
    def log(self, message):
        self.logger.log("[Debug] " + message)

# 使用装饰器
logger = Logger()
error_logger = ErrorLoggerDecorator(logger)
warning_logger = WarningLoggerDecorator(logger)
debug_logger = DebugLoggerDecorator(logger)

error_logger.log("Error message")
warning_logger.log("Warning message")
debug_logger.log("Debug message")