让数据库事务操作更准确

1. 使用事务时只选取一个数据库连接
2. 嵌套事务只在最外层提交

Author: You Ming

ThinkPHP/Library/Think/Db/Driver.class.php

@@ -28,6 +28,8 @@ abstract class Driver
     protected $lastInsID = null;
     // 返回或者影响记录数
     protected $numRows = 0;
+    // 事物操作PDO实例
+    protected $transPDO = null;
     // 事务指令数
     protected $transTimes = 0;
     // 错误信息
@@ -276,6 +278,8 @@ public function startTrans()
 
         //数据rollback 支持
         if (0 == $this->transTimes) {
+            // 记录当前操作PDO
+            $this->transPdo = $this->_linkID;
             $this->_linkID->beginTransaction();
         }
         $this->transTimes++;
@@ -289,13 +293,17 @@ public function startTrans()
      */
     public function commit()
     {
-        if ($this->transTimes > 0) {
-            $result           = $this->_linkID->commit();
+        if ($this->transTimes == 1) {
+            // 由嵌套事物的最外层进行提交
+            $result = $this->_linkID->commit();
             $this->transTimes = 0;
+            $this->transPdo = null;
             if (!$result) {
                 $this->error();
                 return false;
             }
+        } else {
+            $this->transTimes--;
         }
         return true;
     }
@@ -308,8 +316,9 @@ public function commit()
     public function rollback()
     {
         if ($this->transTimes > 0) {
-            $result           = $this->_linkID->rollback();
+            $result = $this->_linkID->rollback();
             $this->transTimes = 0;
+            $this->transPdo = null;
             if (!$result) {
                 $this->error();
                 return false;
@@ -1188,6 +1197,11 @@ protected function debug($start)
      */
     protected function initConnect($master = true)
     {
+        // 开启事物时用同一个连接进行操作
+        if ($this->transPDO) {
+            return $this->transPDO;
+        }
+
         if (!empty($this->config['deploy']))
         // 采用分布式数据库
         {

ThinkPHP/Library/Think/Db/Lite.class.php

@@ -28,6 +28,8 @@ class Lite
     protected $lastInsID = null;
     // 返回或者影响记录数
     protected $numRows = 0;
+   	// 事物操作PDO实例
+    protected $transPDO = null;
     // 事务指令数
     protected $transTimes = 0;
     // 错误信息
@@ -248,6 +250,8 @@ public function startTrans()
 
         //数据rollback 支持
         if (0 == $this->transTimes) {
+            // 记录当前操作PDO
+            $this->transPdo = $this->_linkID;
             $this->_linkID->beginTransaction();
         }
         $this->transTimes++;
@@ -261,13 +265,17 @@ public function startTrans()
      */
     public function commit()
     {
-        if ($this->transTimes > 0) {
-            $result           = $this->_linkID->commit();
+        if ($this->transTimes == 1) {
+            // 由嵌套事物的最外层进行提交
+            $result = $this->_linkID->commit();
             $this->transTimes = 0;
+            $this->transPdo = null;
             if (!$result) {
                 $this->error();
                 return false;
             }
+        } else {
+            $this->transTimes--;
         }
         return true;
     }
@@ -280,8 +288,9 @@ public function commit()
     public function rollback()
     {
         if ($this->transTimes > 0) {
-            $result           = $this->_linkID->rollback();
+            $result = $this->_linkID->rollback();
             $this->transTimes = 0;
+            $this->transPdo = null;
             if (!$result) {
                 $this->error();
                 return false;
@@ -353,7 +362,7 @@ public function error()
         // 记录错误日志
         trace($this->error, '', 'ERR');
         if ($this->config['debug']) {
-// 开启数据库调试模式
+            // 开启数据库调试模式
             E($this->error);
         } else {
             return $this->error;
@@ -421,7 +430,7 @@ public function setModel($model)
     protected function debug($start)
     {
         if ($this->config['debug']) {
-// 开启数据库调试模式
+            // 开启数据库调试模式
             if ($start) {
                 G('queryStartTime');
             } else {
@@ -442,6 +451,11 @@ protected function debug($start)
      */
     protected function initConnect($master = true)
     {
+        // 开启事物时用同一个连接进行操作
+        if ($this->transPDO) {
+            return $this->transPDO;
+        }
+
         if (!empty($this->config['deploy']))
         // 采用分布式数据库
         {