Как установить уровень изоляции транзакции в SQLAlchemy для PostgreSQL? - PullRequest
Новая
       21

Как установить уровень изоляции транзакции в SQLAlchemy для PostgreSQL?

9 голосов
/ 19 августа 2010

Мы используем декларативную базу 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

Я делаю это, чтобы вызвать это, используя уровень изоляции транзакции, но это не то, что я ожидаю. Уровень изоляции все еще READ COMMITTED из того, что я вижу в журналах postgres. Может ли кто-нибудь помочь определить, что я делаю неправильно?

Я использую 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)

Ответы [ 2 ]

9 голосов
/ 26 августа 2010

От Майкла Байера, сопровождающего SQLAlchemy:

Пожалуйста, используйте "изоляционный уровень" аргумент create_engine () и используйте последний совет по SQLAlchemy пока не выйдет 0.6.4, как было исправлена ​​ошибка, специфичная для psycopg2 относительно уровня изоляции.

Подход, предложенный вами ниже, не влияет на то же соединение, которое позже используется для запросов - вы бы вместо этого используйте PoolListener, который устанавливает up set_isolation_level для всех соединения по мере их создания.

0 голосов
/ 22 августа 2010

Уровень изоляции устанавливается внутри транзакции, например, 

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

С PostgreSQL doc :

Если SET TRANSACTION выполняется без предварительной START TRANSACTION или BEGIN, это, похоже, не имеет никакого эффекта, поскольку транзакция немедленно завершится.

...