sqlalchemy 实现select for update

sqlalchemy 对于行级锁有两种实现方式,with_lockmode(self, mode):with_for_update(self, read=False, nowait=False, of=None),前者在sqlalchemy 0.9.0 被废弃,用后者代替。所以我们使用with_for_update !

看下函数的定义:

 @_generative()
    def with_for_update(self, read=False, nowait=False, of=None):
        """return a new :class:`.Query` with the specified options for the
        ``FOR UPDATE`` clause.

        The behavior of this method is identical to that of
        :meth:`.SelectBase.with_for_update`.  When called with no arguments,
        the resulting ``SELECT`` statement will have a ``FOR UPDATE`` clause
        appended.  When additional arguments are specified, backend-specific
        options such as ``FOR UPDATE NOWAIT`` or ``LOCK IN SHARE MODE``
        can take effect.

        E.g.::

            q = sess.query(User).with_for_update(nowait=True, of=User)

        The above query on a Postgresql backend will render like::

            SELECT users.id AS users_id FROM users FOR UPDATE OF users NOWAIT

        .. versionadded:: 0.9.0 :meth:`.Query.with_for_update` supersedes
           the :meth:`.Query.with_lockmode` method.

        .. seealso::

            :meth:`.GenerativeSelect.with_for_update` - Core level method with
            full argument and behavioral description.

        """
        
read
    是标识加互斥锁还是共享锁. 当为 True 时, 即 for share 的语句, 是共享锁. 多个事务可以获取共享锁, 互斥锁只能一个事务获取. 有"多个地方"都希望是"这段时间我获取的数据不能被修改, 我也不会改", 那么只能使用共享锁. 
nowait
    其它事务碰到锁, 是否不等待直接"报错". 
of
    指明上锁的表, 如果不指明, 则查询中涉及的所有表(行)都会加锁.

  

q = sess.query(User).with_for_update(nowait=True, of=User)
对应于sql:
SELECT users.id AS users_id FROM users FOR UPDATE OF users NOWAIT
 

 

 mysql 不支持这几个参数,转成sql都是:
SELECT users.id AS users_id FROM users FOR UPDATE

范例:
def query_city_for_update():
    session = get_session()
    with session.begin():
        query = session.query(City).with_for_update().filter(City.ID == 8)
        print 'SQL : %s' % str(query)
        print_city_info(query.first())

结果:

SQL : SELECT city."ID" AS "city_ID", city."Name" AS "city_Name", city."CountryCode" AS "city_CountryCode", city."District" AS "city_District", city."Population" AS "city_Population" 
FROM city 
WHERE city."ID" = :ID_1 FOR UPDATE

{'city': {'population': 234323, 'district': u'Utrecht', 'id': 8, 'country_code': u'NLD', 'name': u'Utrecht'}}
SELECT ... FOR UPDATE 的用法,不过锁定(Lock)的数据是判别就得要注意一下了。由于InnoDB 预设是Row-Level Lock,所以只有「明确」的指定主键,MySQL 才会执行Row lock (只锁住被选取的数据) ,否则mysql 将会执行Table Lock (将整个数据表单给锁住)。


作者:笨手笨脚越,链接:https://www.jianshu.com/p/7e4de9ab942c

posted @ 2020-09-24 11:16  傻白甜++  阅读(2554)  评论(0编辑  收藏  举报
TOP