pythondjango事务transaction源码分析

2019独角兽企业重金招聘Python工程师标准>>>

python Django事务

网上关于django1.6的事务资料很多&＃xff0c;但是1.8的却搜不到任何资料&＃xff0c;自己要用的时候费了不少劲就是不行&＃xff0c;现在记下要用的人少走弯路 version&＃xff1a;Django 1.8 事务官方文档事务中文文档里面介绍很多方法&＃xff0c;不一一赘述&＃xff0c;按照文档即可&＃xff0c;下面只分析下atomic方法的源码按照官方文档 transaction.atomic 有两种用法装饰器和上下文管理器

# atomic() 方法 # from django.db import transaction ################### # atomic() ################### def atomic(using&＃61;None, savepoint&＃61;True): # 装饰器和上下文管理器必须.()调用方法&＃xff0c;因为真正的处理是该方法返回的实例&＃xff0c;不是该方法本身if callable(using):return Atomic(DEFAULT_DB_ALIAS, savepoint)(using)# Decorator: &＃64;atomic(...) or context manager: with atomic(...): ...else:return Atomic(using, savepoint) ########################################## # Atomic类省略了非核心内容 ############################################ class Atomic(ContextDecorator):def __init__(self, using, savepoint):self.using &＃61; usingself.savepoint &＃61; savepointdef __enter__(self):connection &＃61; get_connection(self.using)sid &＃61; connection.savepoint() # 进入with创建一个保存点# .............dodef __exit__(self, exc_type, exc_value, traceback):if connection.in_atomic_block:# do.............if sid is not None:try:connection.savepoint_commit(sid) # 提交事务except DatabaseError:try:connection.savepoint_rollback(sid) # 捕获数据库异常回滚connection.savepoint_commit(sid)except Error:connection.needs_rollback &＃61; Trueraise## 还有一段代码是exec_type收到其他程序异常时候全局回滚&＃xff0c;此处省略# do................. ############################### # ContextDecorator ################################# class ContextDecorator(object):def __call__(self, func):def inner(*args, **kwargs):with self: # 把函数放进self的with上下文管理器&＃xff0c;效果with相同&＃xff0c;只是控制细粒度不同return func(*args, **kwargs)return inner

python MySQLdb

class Tran():def __init__(self, conn&＃61;None, close&＃61;True):if conn is None: # 创建数据库链接print &＃39;init&＃39;self.conn &＃61; conn_tbkt()self.cur &＃61; self.conn.cursor()self.sql &＃61; []def __enter__(self): # 上下文管理器返回 sql语句列表 with Tran(&＃39;tbkt_pxb&＃39;) as sqls:print &＃39;enter&＃39;return self.sql # sql.append(&＃39;select 1&＃39;)def __exit__(self, exc_type, exc_val, exc_tb):print &＃39;exit&＃39;try:print self.sql # 执行sqlfor s in self.sql:self.cur.execute(s)self.conn.commit()except: # 可以捕获所有异常(django事务如果中间出现程序异常终止无法回滚)try: # 回滚本身也是sql执行&＃xff0c;也有可能失败import tracebacktraceback.print_exc()print &＃39;rollback&＃39;self.conn.rollback()except:print u&＃39;回滚失败&＃39;finally:self.cur.close()self.conn.close()

更细粒度的回滚:

# 在事务块中&＃64;atomic() 或者 with atomic(): sid &＃61; transaction.savepoint(&＃39;tbkt_pxb&＃39;) try:# do .......... except:transaction.savepoint_rollback(sid, &＃39;tbkt_pxb&＃39;)

注意&＃xff1a;如果有多个数据库有路由&＃xff0c;则需要指定和路由返回一致的useing&＃xff1a; math2下的model需要事务&＃xff0c;即使ziyuan_new和default是同一个库,也必须使用useing&＃61;ziyuan_new

ziyuan_app &＃61; [&＃39;math2&＃39;, &＃39;ziyuan&＃39;]if model._meta.app_label in ziyuan_app:return "ziyuan_new"return &＃39;default&＃39;

调用时候必须.()方法调用

atomic块中必须注意try的使用&＃xff0c;如果手动捕获了程序错误会导致atomic包装器捕获不到异常&＃xff0c;也就不会回滚。要么try内代码不影响事务操作&＃xff0c;要么就捕获异常后raise出&＃xff0c;让atomic可以正常回滚(就是因为没有注意到这个问题&＃xff0c;导致尝试了好几天都没成功&＃xff0c;切记)