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

Add:设置全局方言

Browse Source
This commit is contained in:
farukonfly 2024-01-10 00:46:58 +08:00
parent c931e203ed
commit 1bd44a0de7
2 changed files with 21 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
mybatis-flex-core/src/main/java/com/mybatisflex/core/dialect/DialectFactory.java
View File

@ -18,6 +18,8 @@ package com.mybatisflex.core.dialect;
import java.util.EnumMap; import java.util.EnumMap;
import java.util.Map; import java.util.Map;
import com.mybatisflex.core.exception.MybatisFlexException;
import org.apache.ibatis.util.MapUtil; import org.apache.ibatis.util.MapUtil;
import com.mybatisflex.core.FlexGlobalConfig; import com.mybatisflex.core.FlexGlobalConfig;
import com.mybatisflex.core.dialect.impl.CommonsDialectImpl; import com.mybatisflex.core.dialect.impl.CommonsDialectImpl;
@ -45,6 +47,7 @@ public class DialectFactory {
* 通过设置当前线程的数据库类型，以达到在代码执行时随时切换方言的功能 * 通过设置当前线程的数据库类型，以达到在代码执行时随时切换方言的功能
*/ */
private static final ThreadLocal<DbType> dbTypeThreadLocal = new ThreadLocal<>(); private static final ThreadLocal<DbType> dbTypeThreadLocal = new ThreadLocal<>();
private static DbType dbTypeGlobal = null ;
/** /**
@ -75,6 +78,19 @@ public class DialectFactory {
return dbTypeThreadLocal.get(); return dbTypeThreadLocal.get();
} }
public static DbType getGlobalDbType() {
return dbTypeGlobal;
}
public static void setGlobalDbType(DbType dbType) {
if(dbTypeGlobal == null&&dbType!=null){
dbTypeGlobal = dbType ;
}else if(dbTypeGlobal != null){
throw new MybatisFlexException("dbTypeGlobal is only set once");
}else if(dbType==null){
throw new MybatisFlexException("dbType can not be null");
}
}
/** /**
* 清除当前线程的 dbType * 清除当前线程的 dbType

5
mybatis-flex-core/src/main/java/com/mybatisflex/core/mybatis/MapperInvocationHandler.java
View File

@ -76,6 +76,11 @@ public class MapperInvocationHandler implements InvocationHandler {
//优先获取用户自己配置的 dbType //优先获取用户自己配置的 dbType
DbType dbType = DialectFactory.getHintDbType(); DbType dbType = DialectFactory.getHintDbType();
DbType dbTypeGlobal = DialectFactory.getGlobalDbType();
//当前线程没有设置dbType,但是全局设置了dbTypeGlobal，那么就使用全局的dbTypeGlobal
if(dbTypeGlobal!=null&&dbType==null){
dbType = dbTypeGlobal ;
}
if (dbType == null) { if (dbType == null) {
if (dataSourceKey != null && dataSource != null) { if (dataSourceKey != null && dataSource != null) {
dbType = dataSource.getDbType(dataSourceKey); dbType = dataSource.getDbType(dataSourceKey);