From 73fdf8b815f0fb6a0decd50bee315901a86bea7f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=BC=80=E6=BA=90=E6=B5=B7=E5=93=A5?= <fuhai999@gmail.com>
Date: Wed, 23 Aug 2023 15:44:56 +0800
Subject: [PATCH] refactor: optimize FlexSpringTransaction.java

---
 .../mybatisflex/spring/FlexSpringTransaction.java   | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

diff --git a/mybatis-flex-spring/src/main/java/com/mybatisflex/spring/FlexSpringTransaction.java b/mybatis-flex-spring/src/main/java/com/mybatisflex/spring/FlexSpringTransaction.java
index 644e174a..94321fb3 100644
--- a/mybatis-flex-spring/src/main/java/com/mybatisflex/spring/FlexSpringTransaction.java
+++ b/mybatis-flex-spring/src/main/java/com/mybatisflex/spring/FlexSpringTransaction.java
@@ -48,14 +48,15 @@ public class FlexSpringTransaction implements Transaction {
             autoCommit = connection.getAutoCommit();
             return connection;
         }
-        // 非事务
-        else if (!isConnectionTransactional) {
-            return connection;
-        }
-        // 在事务中
-        else {
+        // 在事务中，通过 FlexDataSource 去获取
+        // FlexDataSource 内部会进行 connection 缓存以及多数据源下的 key 判断
+        else if (isConnectionTransactional) {
             return dataSource.getConnection();
         }
+        // 非事务，返回当前链接
+        else {
+            return connection;
+        }
     }
 
     @Override