设计模式：依赖注入，为什么使用依赖注入，依赖注入带来的代码结构的改变和优点

最新推荐文章于 2024-09-12 17:23:35 发布

李默_jefferli

最新推荐文章于 2024-09-12 17:23:35 发布

阅读量898

点赞数 13

文章标签：设计模式 python 网络

本文链接：https://blog.csdn.net/weixin_44402083/article/details/140956656

版权

依赖注入

20240806

**依赖注入（Dependency Injection, DI）**是一种设计模式，用于将对象的依赖关系从内部管理转移到外部管理，从而提高代码的可维护性和可测试性。下面是一个简单的依赖注入案例，使用 Python 来演示。

步骤 1：定义接口

首先，我们定义一个消息发送的接口：

from abc import ABC, abstractmethod

class MessageSender(ABC):
    @abstractmethod
"""
抽象基类强制继承类实现
"""
    def send(self, message: str, recipient: str):
        pass

步骤 2：实现具体的消息发送类

接下来，我们实现两个具体的消息发送类：一个用于发送电子邮件，另一个用于发送短信。

class EmailSender(MessageSender):
    def send(self, message: str, recipient: str):
        print(f"Sending email to {recipient}: {message}")

class SMSSender(MessageSender):
    def send(self, message: str, recipient: str):
        print(f"Sending SMS to {recipient}: {message}")

步骤 3：创建消息服务类

然后，我们创建一个消息服务类，该类依赖于 MessageSender 接口。我们通过构造函数注入依赖关系。

class MessageService:
    def __init__(self, sender: MessageSender):
        self.sender = sender

    def send_message(self, message: str, recipient: str):
        self.sender.send(message, recipient)

步骤 4：使用依赖注入

最后，我们创建具体的消息发送类实例，并将其注入到消息服务类中。

if __name__ == "__main__":
    email_sender = EmailSender()
    sms_sender = SMSSender()

    email_service = MessageService(email_sender)
    sms_service = MessageService(sms_sender)

    email_service.send_message("Hello via Email!", "email@example.com")
    sms_service.send_message("Hello via SMS!", "123-456-7890"

解释

接口定义：MessageSender 是一个抽象基类，定义了 send 方法。
具体实现：EmailSender 和 SMSSender 实现了 MessageSender 接口。
依赖注入：MessageService 类通过构造函数接受一个 MessageSender 实例，并在 send_message 方法中使用它。
使用：在主程序中，我们创建了 EmailSender 和 SMSSender 的实例，并将它们注入到 MessageService 中。

通过这种方式，我们可以轻松地更换消息发送的实现，而无需修改 MessageService 类的代码。这就是依赖注入的核心思想：将依赖关系从内部管理转移到外部管理。

然而，在实际的代码设计中可能我们更多的会使用类似入下代码：

class SMSSender:
    def send(self, message: str, recipient: str):
        print(f"Sending SMS to {recipient}: {message}")

class EmailSender:
    def send(self, message: str, recipient: str):
        print(f"Sending email to {recipient}: {message}")

class MessageService:
    def __init__(self, method: str):
        if method == 'sms':
            self.sender = SMSSender()
        elif method == 'email':
            self.sender = EmailSender()
        else:
            raise ValueError("Unsupported message sending method")

    def send_message(self, message: str, recipient: str):
        self.sender.send(message, recipient)

if __name__ == "__main__":
    # 使用 SMS 发送
    sms_service = MessageService('sms')
    sms_service.send_message("Hello via SMS!", "123-456-7890")

    # 使用 Email 发送
    email_service = MessageService('email')
    email_service.send_message("Hello via Email!", "email@example.com")

将 MessageService 设计为直接实例化 email_service 或 email_service 可能看起来更简单，但这种设计会带来一些问题，特别是在代码的可维护性、可测试性和灵活性方面。

虽然这种设计方法简单直接，但它有一些缺点：

紧耦合：MessageService 与具体的消息发送实现紧密耦合，难以更换消息发送实现。比如后续要新增一个IMSender,还需要改动MessageService类
可测试性差：难以为 MessageService 编写单元测试，因为无法轻松地注入模拟或伪造的消息发送实现。
违反单一职责原则：MessageService 负责选择和管理消息发送实现，职责过多。

尽管如此，如果你的应用程序非常简单，并且不需要频繁更换消息发送实现，这种设计方法也是可以接受的。

我们不难发现依赖注入的优点

1. 松耦合

依赖注入使得 MessageService 与具体的消息发送实现解耦。这样，如果你需要更换消息发送的实现（例如，从 SMSSender 切换到 EmailSender），你只需要更改注入的依赖，而不需要修改 MessageService 的代码。

2. 可测试性

依赖注入使得单元测试更加容易。你可以轻松地为 MessageService 注入一个模拟（mock）或伪造（fake）的 MessageSender，从而测试 MessageService 的行为，而不依赖于实际的消息发送实现。

class MockSender(MessageSender):
    def send(self, message: str, recipient: str):
        print(f"Mock send to {recipient}: {message}")

def test_message_service():
    mock_sender = MockSender()
    service = MessageService(mock_sender)
    service.send_message("Test message", "test@example.com")

3. 单一职责原则

根据单一职责原则（Single Responsibility Principle），一个类应该只有一个引起其变化的原因。MessageService 的职责是发送消息，而不是管理消息发送的具体实现。
如果 MessageService 直接实例化 SMSSender 或 EmailSender，它就承担了选择和管理消息发送实现的职责，这违反了单一职责原则。

4. 灵活性和扩展性

依赖注入使得系统更具灵活性和扩展性。如果你需要添加新的消息发送实现（例如，推送通知），你只需要实现 MessageSender 接口，并在需要时注入新的实现，而不需要修改现有的 MessageService 代码。

class PushNotificationSender(MessageSender):
    def send(self, message: str, recipient: str):
        print(f"Sending push notification to {recipient}: {message}")

# 使用新的实现
push_service = MessageService(PushNotificationSender())
push_service.send_message("Hello via Push Notification!", "user123")