支持强制回滚模式

easy-swoole · Sep 12, 2024 · 28a6983 · 28a6983
1 parent ca0fc4a
commit 28a6983
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 3 deletions.
diff --git a/src/Config.php b/src/Config.php
@@ -19,6 +19,8 @@ class Config extends \EasySwoole\Pool\Config
 
     protected bool $useMysqli = false;
 
+    protected bool $isForceRollback = false;
+
     /**
      * @return string
      */
@@ -172,4 +174,14 @@ public function setUseMysqli(bool $useMysqli): void
     {
         $this->useMysqli = $useMysqli;
     }
+
+    public function isIsForceRollback(): bool
+    {
+        return $this->isForceRollback;
+    }
+
+    public function setIsForceRollback(bool $isForceRollback): void
+    {
+        $this->isForceRollback = $isForceRollback;
+    }
 }
diff --git a/src/Mysql/Connection.php b/src/Mysql/Connection.php
@@ -11,11 +11,13 @@ class Connection extends Client implements ObjectInterface
     public string $connectionName;
     public bool $isInTransaction = false;
 
+    public bool $isForceRollback = false;
+
     public int $lastPingTime = 0;
 
     function gc()
     {
-        if($this->isInTransaction){
+        if($this->isInTransaction || $this->isForceRollback){
             try {
                 $this->mysqlClient()->rollback();
             }catch (\Throwable $throwable){
@@ -28,7 +30,7 @@ function gc()
 
     function objectRestore()
     {
-        if($this->isInTransaction){
+        if($this->isInTransaction || $this->isForceRollback){
             try {
                 $this->mysqlClient()->rollback();
             }catch (\Throwable $throwable){

diff --git a/src/Mysql/Pool.php b/src/Mysql/Pool.php
@@ -11,7 +11,9 @@ class Pool extends AbstractPool
     protected function createObject()
     {
         $config = new Config($this->getConfig()->toArray());
-        $con =  new Connection($config);
+        $con = new Connection($config);
+        $con->isForceRollback = $this->getConfig()->isIsForceRollback();
+
         if(!$con->connect()){
             $info = $con->mysqlClient()->connect_error;
             /** @var \EasySwoole\FastDb\Config $config */