admin/mybatis-flex
1
0
Fork 0
mirror of https://gitee.com/mybatis-flex/mybatis-flex.git synced 2025-12-07 17:18:24 +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;
import com.mybatisflex.core.datasource.DataSourceKey;
import com.mybatisflex.core.datasource.FlexDataSource;
import com.mybatisflex.core.transaction.TransactionContext;
import com.mybatisflex.core.util.StringUtil;
import java.sql.Connection;
import java.sql.SQLException;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import javax.sql.DataSource;
import org.apache.ibatis.session.TransactionIsolationLevel;
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
@ -29,8 +37,11 @@ import org.apache.ibatis.transaction.Transaction;
*/
public class FlexSpringTransaction implements Transaction {
private static final Logger LOGGER = LoggerFactory.getLogger(FlexSpringTransaction.class);
DataSource dataSource;
Map<String,Connection> connectionMap =new HashMap<>();
boolean isTransaction =false;
TransactionIsolationLevel level;
@ -47,16 +58,23 @@ public class FlexSpringTransaction implements Transaction {
@Override
public Connection getConnection() throws SQLException {
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){
isTransaction = true;
}
return dataSource.getConnection();
return connection;
}else{
throw new SQLException("The datasource must be FlextDataSource");
throw new SQLException("The datasource must be FlexDataSource");
}
}
@Override
@ -64,7 +82,6 @@ public class FlexSpringTransaction implements Transaction {
if (!isTransaction && !autoCommit){
getConnection().commit();
}
}
@Override