Python：数据库通信代码结构设计——Stock类与DBConnection协作疑问

针对你遇到的Stock类与数据库连接的依赖问题，下面是几种可行的设计方案及各自的利弊：

方案一：将DBConnection传入Stock类 #

直接在Stock的构造方法中接收数据库连接实例，让Stock类自行处理数据交互逻辑。

优点 #

• 依赖注入：明确Stock类对数据库的依赖，方便替换不同的DBConnection实现（比如测试用的内存数据库连接），大幅提升可测试性。
• 调用直观：调用Stock的方法时无需额外传递连接，比如stock.is_valid()直接内部使用传入的连接，代码更简洁。

示例代码 #

class Stock:
    def __init__(self, ticker: str, db_conn: DBConnection):
        self.ticker = ticker
        self.db_conn = db_conn

    def is_valid(self) -> bool:
        # 使用self.db_conn查询数据库，验证ticker是否存在
        query = f"SELECT EXISTS(SELECT 1 FROM stocks WHERE ticker = '{self.ticker}')"
        return self.db_conn.execute(query).fetchone()[0]

    def fetch_price(self) -> float:
        query = f"SELECT price FROM stocks WHERE ticker = '{self.ticker}'"
        result = self.db_conn.execute(query).fetchone()
        return result[0] if result else 0.0

缺点 #

• 每次创建Stock实例都需要传递DBConnection，若项目中连接是全局单例，会显得冗余。
• Stock类同时承担了股票属性封装和数据交互的职责，违反单一职责原则。

方案二：创建辅助类处理数据交互 #

拆分职责：Stock类仅封装股票的核心属性（如ticker），创建专门的辅助类（比如StockDataService）处理与数据库相关的方法。

优点 #

• 单一职责：Stock类专注于股票本身的属性，辅助类专注于数据存取，代码结构更清晰，后期维护更方便。
• 复用性高：辅助类可以集中处理所有股票相关的数据库操作，避免在Stock类中重复编写SQL逻辑。

示例代码 #

class Stock:
    def __init__(self, ticker: str):
        self.ticker = ticker

class StockDataService:
    def __init__(self, db_conn: DBConnection):
        self.db_conn = db_conn

    def is_valid(self, stock: Stock) -> bool:
        query = f"SELECT EXISTS(SELECT 1 FROM stocks WHERE ticker = '{stock.ticker}')"
        return self.db_conn.execute(query).fetchone()[0]

    def fetch_price(self, stock: Stock) -> float:
        query = f"SELECT price FROM stocks WHERE ticker = '{stock.ticker}'"
        result = self.db_conn.execute(query).fetchone()
        return result[0] if result else 0.0

缺点 #

• 调用时需要同时持有Stock实例和StockDataService实例，比如service.fetch_price(stock)，相比方案一多了一层调用，略显繁琐。

方案三：重构结构（混合优化） #

如果觉得前两种方案都有缺陷，可以结合两者的优点做优化：

1. 全局/单例DBConnection：如果项目中数据库连接是全局唯一的，可以让Stock类内部使用单例的DBConnection（但要注意单例的测试问题，可通过注入方式兼容）。
2. Stock类作为数据载体+可选注入：在Stock类的方法中预留可选的db_conn参数，优先使用传入的连接，否则调用单例连接，兼顾灵活性和简洁性。

示例（单例连接+可选注入） #

class DBConnection:
    _instance = None

    @classmethod
    def get_instance(cls):
        if not cls._instance:
            cls._instance = cls()
        return cls._instance

class Stock:
    def __init__(self, ticker: str):
        self.ticker = ticker

    def is_valid(self, db_conn: DBConnection = None) -> bool:
        used_conn = db_conn or DBConnection.get_instance()
        query = f"SELECT EXISTS(SELECT 1 FROM stocks WHERE ticker = '{self.ticker}')"
        return used_conn.execute(query).fetchone()[0]

注意点 #

• 这种方式既保留了单例连接的简洁性，又通过可选注入支持测试场景，平衡了耦合度和灵活性。

选择建议 #

• 若项目需要多数据库支持、频繁单元测试，优先选方案一（依赖注入），灵活性最高。
• 若追求代码职责清晰、长期维护性，优先选方案二（辅助类），符合单一职责原则。
• 若项目规模小、数据库连接单一，方案三的混合方式可以快速实现，同时预留扩展空间。

内容的提问来源于stack exchange，提问作者Mihir Sampat