如何在 SQLAlchemy for PostgreSQL 中设置事务隔离级别？

Question

我们正在使用 SQLAlchemy 声明式基础，并且我有一个方法要隔离事务级别。解释一下，有两个进程同时写入数据库，我必须让它们在事务中执行它们的逻辑。默认事务隔离级别是 READ COMMITTED，但我需要能够使用 SERIALIZABLE 隔离级别执行一段代码。

这是如何使用 SQLAlchemy 完成的？现在，我的模型中基本上有一个方法，它继承自 SQLAlchemy 的声明性基础，本质上需要以事务方式调用。

from psycopg2.extensions import ISOLATION_LEVEL_AUTOCOMMIT
from psycopg2.extensions import ISOLATION_LEVEL_READ_COMMITTED
from psycopg2.extensions import ISOLATION_LEVEL_SERIALIZABLE

class OurClass(SQLAlchemyBaseModel):

    @classmethod
    def set_isolation_level(cls, level=ISOLATION_LEVEL_SERIALIZABLE):
        cls.get_engine().connect().connection.set_isolation_level(level)

    @classmethod
    def find_or_create(cls, **kwargs):
        try:
            return cls.query().filter_by(**kwargs).one()
        except NoResultFound:
            x = cls(**kwargs)
            x.save()
            return x

我这样做是为了使用事务隔离级别来调用它，但它并没有达到我的预期。从我在 postgres 日志中看到的内容来看，隔离级别仍然是 READ COMMITTED。有人可以帮助确定我做错了什么吗？

我正在使用 SQLAlchemy 0.5.5

class Foo(OurClass):

    def insert_this(self, kwarg1=value1):
        # I am trying to set the isolation level to SERIALIZABLE
        try:
            self.set_isolation_level()
            with Session.begin():
                self.find_or_create(kwarg1=value1)
        except Exception:  # if any exception is thrown...
            print "I caught an expection."
            print sys.exc_info()
        finally:
            # Make the isolation level back to READ COMMITTED
            self.set_isolation_level(ISOLATION_LEVEL_READ_COMMITTED)

Answer 1

来自 SQLAlchemy 的维护者 Michael Bayer：

请使用“isolation_level” 参数create_engine() 并使用latest tip of SQLAlchemy 直到 0.6.4 发布，因为有 最近修复了一个特定于 psycopg2 的错误 关于隔离级别。

你下面的方法没有 影响相同的连接 后来用于查询 - 你会 而是使用设置的 PoolListener 全部设置 set_isolation_level 创建时的连接。

Answer 2

隔离级别在事务中设置，例如

try:
    Session.begin()
    Session.execute('set transaction isolation level serializable')
    self.find_or_create(kwarg1=value1)
except:
    ...

来自PostgreSQL doc：

如果 SET TRANSACTION 在没有事先 START TRANSACTION 或 BEGIN 的情况下执行，它似乎没有任何效果，因为事务将立即结束。