深入解析装饰器模式在真实场景的落地应用

2023-11-17 01:20:35

引言

在软件开发中，我们经常会遇到需要对某个对象进行扩展或修改的情况。传统的方法是直接修改对象的源代码，但这可能会带来很多问题，比如：

代码变得难以维护和扩展。
修改源代码可能会引入新的错误。
扩展后的代码可能会与其他部分的代码产生冲突。

为了解决这些问题，设计模式应运而生。装饰器模式就是其中一种非常有用的设计模式，它允许你动态地给一个对象添加一些新的功能，而不需要修改该对象的源代码。

装饰器模式的原理

装饰器模式的原理很简单，它就是在原有对象的基础上，再创建一个新的对象，并将原有对象作为参数传递给新对象。新对象可以调用原有对象的方法，并在此基础上添加一些新的功能。

class OriginalObject:
    def __init__(self):
        self.name = "Original Object"

    def say_hello(self):
        print("Hello from Original Object!")


class DecoratedObject(OriginalObject):
    def __init__(self, original_object):
        super().__init__()
        self.original_object = original_object

    def say_hello(self):
        self.original_object.say_hello()
        print("Hello from Decorated Object!")


if __name__ == "__main__":
    original_object = OriginalObject()
    decorated_object = DecoratedObject(original_object)

    original_object.say_hello()
    decorated_object.say_hello()

在上面的代码中，OriginalObject 是原有对象，它有一个 say_hello() 方法。DecoratedObject 是装饰器对象，它继承自 OriginalObject，并在 say_hello() 方法中添加了新的功能。当调用 DecoratedObject.say_hello() 方法时，它会先调用 OriginalObject.say_hello() 方法，然后再调用自己的 say_hello() 方法。这样，我们就实现了对原有对象的功能扩展，而不需要修改原有对象的源代码。