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

fix: 修复 druid 数据源的某些场景下，数据加密无法使用的问题

Browse Source
This commit is contained in:
开源海哥 2023-08-18 09:30:21 +08:00
parent 6eadba4e77
commit 725b3efcd0
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
mybatis-flex-core/src/main/java/com/mybatisflex/core/datasource/DataSourceManager.java
View File

@ -15,6 +15,7 @@
*/ */
package com.mybatisflex.core.datasource; package com.mybatisflex.core.datasource;
import com.mybatisflex.core.exception.FlexExceptions;
import com.mybatisflex.core.util.ClassUtil; import com.mybatisflex.core.util.ClassUtil;
import org.apache.ibatis.logging.LogFactory; import org.apache.ibatis.logging.LogFactory;
@ -42,6 +43,8 @@ public class DataSourceManager {
return; return;
} }
restartDataSource(dataSource);
for (DataSourceProperty property : DataSourceProperty.values()) { for (DataSourceProperty property : DataSourceProperty.values()) {
Method getterMethod = ClassUtil.getAnyMethod(dataSource.getClass(), property.getGetterMethods()); Method getterMethod = ClassUtil.getAnyMethod(dataSource.getClass(), property.getGetterMethods());
if (getterMethod != null) { if (getterMethod != null) {
@ -57,6 +60,18 @@ public class DataSourceManager {
} }
} }
static void restartDataSource(DataSource dataSource) {
Method restartMethod = ClassUtil.getFirstMethod(ClassUtil.getUsefulClass(dataSource.getClass())
, method -> "restart".equals(method.getName()));
if (restartMethod != null) {
try {
restartMethod.invoke(dataSource);
} catch (Exception e) {
throw FlexExceptions.wrap(e);
}
}
}
static String invokeMethod(Method method, Object object, Object... params) { static String invokeMethod(Method method, Object object, Object... params) {
try { try {