admin/mybatis-flex
1
0
Fork 0
mirror of https://gitee.com/mybatis-flex/mybatis-flex.git synced 2025-12-08 09:38:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

!246 缓存数据库连接并且兼容多数据源，防止数据库连接不够造成死锁问题

Browse Source 
Merge pull request !246 from life/main
This commit is contained in:
Michael Yang 2023-08-04 08:46:48 +00:00 committed by Gitee
commit 5c89401d92
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 23 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -15,13 +15,21 @@
*/ */
package com.mybatisflex.spring; package com.mybatisflex.spring;
import com.mybatisflex.core.datasource.DataSourceKey;
import com.mybatisflex.core.datasource.FlexDataSource; import com.mybatisflex.core.datasource.FlexDataSource;
import com.mybatisflex.core.transaction.TransactionContext; import com.mybatisflex.core.transaction.TransactionContext;
import com.mybatisflex.core.util.StringUtil;
import java.sql.Connection; import java.sql.Connection;
import java.sql.SQLException; import java.sql.SQLException;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import javax.sql.DataSource; import javax.sql.DataSource;
import org.apache.ibatis.session.TransactionIsolationLevel; import org.apache.ibatis.session.TransactionIsolationLevel;
import org.apache.ibatis.transaction.Transaction; import org.apache.ibatis.transaction.Transaction;
import org.mybatis.logging.Logger;
import org.mybatis.logging.LoggerFactory;
import org.mybatis.spring.transaction.SpringManagedTransaction;
/** /**
* spring事务支持，解决issusehttps://gitee.com/mybatis-flex/mybatis-flex/issues/I7HJ4J * spring事务支持，解决issusehttps://gitee.com/mybatis-flex/mybatis-flex/issues/I7HJ4J
@ -29,8 +37,11 @@ import org.apache.ibatis.transaction.Transaction;
*/ */
public class FlexSpringTransaction implements Transaction { public class FlexSpringTransaction implements Transaction {
private static final Logger LOGGER = LoggerFactory.getLogger(FlexSpringTransaction.class);
DataSource dataSource; DataSource dataSource;
Map<String,Connection> connectionMap =new HashMap<>();
boolean isTransaction =false; boolean isTransaction =false;
TransactionIsolationLevel level; TransactionIsolationLevel level;
@ -47,16 +58,23 @@ public class FlexSpringTransaction implements Transaction {
@Override @Override
public Connection getConnection() throws SQLException { public Connection getConnection() throws SQLException {
if (dataSource instanceof FlexDataSource) { if (dataSource instanceof FlexDataSource) {
this.autoCommit = this.dataSource.getConnection().getAutoCommit(); String dataSourceKey = DataSourceKey.get();
if (StringUtil.isBlank(dataSourceKey)) {
dataSourceKey = ((FlexDataSource) dataSource).getDefaultDataSourceKey();
}
Connection connection = connectionMap.get(dataSourceKey);
if (connection == null){
connection = this.dataSource.getConnection();
connectionMap.put(dataSourceKey,connection);
}
this.autoCommit = connection.getAutoCommit();
if ( TransactionContext.getXID() != null){ if ( TransactionContext.getXID() != null){
isTransaction = true; isTransaction = true;
} }
return dataSource.getConnection(); return connection;
}else{ }else{
throw new SQLException("The datasource must be FlextDataSource"); throw new SQLException("The datasource must be FlexDataSource");
} }
} }
@Override @Override
@ -64,7 +82,6 @@ public class FlexSpringTransaction implements Transaction {
if (!isTransaction && !autoCommit){ if (!isTransaction && !autoCommit){
getConnection().commit(); getConnection().commit();
} }
} }
@Override @Override