使用MySQL Connector自定义上下文管理器时如何处理事务提交与回滚

问题描述 #

使用MySQL Connector通过自定义上下文管理器类处理数据库变更时，如何正确执行事务提交或回滚？现有如下自定义DatabaseConnection类：

class DatabaseConnection:
    def __init__(self, host, user, password, database_name):
        self.host = host
        self.user = user
        self.passwd = password
        self.database_name = database_name
        self.connection = None

    def __enter__(self) -> Union[mysql.cursor.MySQLCursor, None]:
       try:
          if self.connection is None or not self.connection.is_connected():
              self.connection = mysql.connect(
                  host=self.host, user=self.user, passwd=self.passwd, db=self.database_name
              )
          cursor = self.connection.cursor()
          return cursor
      except Exception:
          return None

    def __exit__(self, exc_type, exc_val, exc_tb):
        if self.connection is not None: 
            self.connection.close()

    def commit(self):
        if self.connection is not None and self.connection.is_connected():
             try:
                 self.connection.commit()
                 return True
             except:
                 self.connection.rollback()

    def rollback(self):
        if self.connection is not None:
           self.connection.rollback()

当通过with语句获取游标执行操作时，如何管理提交与回滚，尤其是异常触发的场景？示例使用代码如下：

test = True
database_connection = DatabaseConnection(....)
try:
    with database_connection as cursor:
        # 执行数据库操作

        if test:
            database_connection.rollback()
        else:
            database_connection.commit()
except Exception as e:
     database_connection.rollback()

暂不考虑错误处理的不规范之处，该模式下提交与回滚是否符合预期？若可行，其原理是什么？

模式有效性与原理分析 #

是否符合预期 #

该模式基本符合事务管理的预期：

• 无异常场景：with块内执行完操作后，根据test变量手动触发提交或回滚，能正确持久化或撤销变更。
• 异常场景：with块内抛出异常时，会进入外部except块执行回滚，未提交的变更会被撤销，之后上下文管理器的__exit__方法关闭连接，避免资源泄漏。

核心原理 #

• MySQL Connector默认事务规则：MySQL Connector默认关闭自动提交（autocommit=False），同一连接下的所有SQL操作都会被纳入同一个事务，只有手动调用commit()才会把变更持久化到数据库；调用rollback()则会撤销所有未提交的变更。
• 上下文管理器的连接共享：DatabaseConnection实例的__enter__方法负责初始化或复用连接，并返回绑定该连接的游标，with块内的所有操作都基于同一个连接的事务上下文。
• 手动事务控制逻辑：不管是在with块内还是外部except块中，调用实例的commit()/rollback()方法，本质是操作共享连接的事务状态，因此能正确作用于当前正在执行的事务。

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