refactor: optimize FlexSpringTransaction

mybatis-flex-spring/src/main/java/com/mybatisflex/spring/FlexSpringTransaction.java

@@ -16,6 +16,8 @@
 package com.mybatisflex.spring;
 
 import com.mybatisflex.core.datasource.FlexDataSource;
+import com.mybatisflex.core.transaction.TransactionContext;
+import com.mybatisflex.core.util.StringUtil;
 import org.apache.ibatis.transaction.Transaction;
 
 import java.sql.Connection;
@@ -30,6 +32,9 @@ import java.sql.SQLException;
 public class FlexSpringTransaction implements Transaction {
 
     private final FlexDataSource dataSource;
+    private Boolean isConnectionTransactional;
+    private Boolean autoCommit;
+    private Connection connection;
 
     public FlexSpringTransaction(FlexDataSource dataSource) {
         this.dataSource = dataSource;
@@ -37,22 +42,41 @@ public class FlexSpringTransaction implements Transaction {
 
     @Override
     public Connection getConnection() throws SQLException {
-        return dataSource.getConnection();
+        if (isConnectionTransactional == null) {
+            connection = dataSource.getConnection();
+            isConnectionTransactional = StringUtil.isNotBlank(TransactionContext.getXID());
+            autoCommit = connection.getAutoCommit();
+            return connection;
+        }
+        // 非事务
+        else if (!isConnectionTransactional) {
+            return connection;
+        }
+        // 在事务中
+        else {
+            return dataSource.getConnection();
+        }
     }
 
     @Override
     public void commit() throws SQLException {
-        getConnection().commit();
+        if (this.connection != null && !this.isConnectionTransactional && !this.autoCommit) {
+            this.connection.commit();
+        }
     }
 
     @Override
     public void rollback() throws SQLException {
-        getConnection().rollback();
+        if (this.connection != null && !this.isConnectionTransactional && !this.autoCommit) {
+            this.connection.rollback();
+        }
     }
 
     @Override
     public void close() throws SQLException {
-        getConnection().close();
+        if (this.connection != null && !this.isConnectionTransactional) {
+            connection.close();
+        }
     }
 
     @Override