3

我正在使用 MySQLdb 使用 python 连接到 MySQL。我的表都是 InnoDB 并且我正在使用事务。

我正在努力想出一种跨职能“共享”交易的方法。考虑以下伪代码：

def foo():
    db = connect()
    cur = db.cursor()
    try:
        cur.execute(...)
        conn.commit()
    except:
        conn.rollback()

def bar():
    db = connect()
    cur = db.cursor()
    try:
        cur.execute(...)
        foo()  # note this call
        conn.commit()
    except:
        conn.rollback()

在我的代码中的某些时候，我需要调用foo()并且在某些时候我需要调用bar(). 这里的最佳做法是什么？如果在外部而不是内部调用，我将如何告诉调用foo()到？如果有多个线程调用并且调用不返回相同的连接对象，这显然会更复杂。commit()bar()bar()foo()bar()connect()

更新

我找到了一个适合我的解决方案。我已经包装connect()好在调用时增加一个值。调用commit()会减少该值。如果commit()被调用并且该计数器的 > 0，则不会发生提交并且值会递减。因此，您会得到：

def foo():
    db = connect()  # internal counter = 1
    ...
    db.commit()  # internal counter = 0, so commit


def bar():
    db = connect()  # internal counter = 1
    ...
    foo()  # internal counter goes to 2, then to 1 when commit() is called, so no commit happens
    db.commit() # internal counter = 0, so commit
4

3 回答 3

0

IMO 最干净的方法是将连接对象传递给foo和bar

于 2013-02-19T21:37:13.347 回答
0

声明函数外部的连接并将它们作为参数传递给函数

foo(cur, conn)
bar(cur, conn)
于 2013-02-19T21:38:10.823 回答
0

在这种情况下，您可以利用 Python 的默认函数参数：

def foo(cur=None):
    inside_larger_transaction = False
    if cursor is None:
        db = connect()
        cur = db.cursor()
        inside_larger_transaction = True
    try:
        cur.execute(...)
        if not inside_larger_transaction:
             conn.commit()
    except:

        conn.rollback()

因此，如果bar正在调用foo，它只是将游标对象作为参数传入。

并不是说我认为为每个小函数创建不同的光标对象没有多大意义——您应该将几个函数编写为对象的方法，并具有光标属性——或者始终显式传递光标（在这种情况下，使用另一个命名参数来指示当前函数是否是主要事务的一部分）

另一种选择是创建一个上下文管理器类来进行提交，并在其中封装所有事务-因此，您的任何函数都不应该进行事务提交-您将保留对__exit__this 方法的transaction.commit 和 transaction.rollback 调用目的。

class Transaction(object):
    def __enter__(self):
       self.db = connect()
       cursor = self.db.cursor()
       return cursor
   def __exit__(self, exc_type, exc_value, traceback):
       if exc_type is None:
           self.db.commit()
       else:
           self.db.rollback()

就像这样使用它：

def foo(cursor):
    ...

def foo(cur):
        cur.execute(...)


def bar(cur):
    cur.execute(...)
    foo(cur)

with Transaction() as cursor:
    foo(cursor)


with Transaction() as cursor:
    bar(cursor)
于 2013-02-20T02:18:33.250 回答