test-version22
/
MeterSphere
mirror of https://gitee.com/fit2cloud-feizhiyun/MeterSphere.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

简化配置

This commit is contained in:
Captain.B 2020-02-25 14:29:01 +08:00
parent 7aca1d5163
commit fe12305ec2
4 changed files with 5 additions and 57 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
backend/src/main/java/io/metersphere/commons/utils/MybatisInterceptorConfigHolder.java
View File

@ -1,10 +0,0 @@
package io.metersphere.commons.utils;
import java.util.ArrayList;
import java.util.List;
public interface MybatisInterceptorConfigHolder {
default List<MybatisInterceptorConfig> interceptorConfig() {
return new ArrayList<>();
}
}

18
backend/src/main/java/io/metersphere/config/DBEncryptConfig.java
View File

@ -1,18 +0,0 @@
package io.metersphere.config;
import io.metersphere.commons.utils.MybatisInterceptorConfig;
import io.metersphere.commons.utils.MybatisInterceptorConfigHolder;
import org.springframework.context.annotation.Configuration;
import java.util.Arrays;
import java.util.List;
@Configuration
public class DBEncryptConfig implements MybatisInterceptorConfigHolder {
@Override
public List<MybatisInterceptorConfig> interceptorConfig() {
return Arrays.asList(
new MybatisInterceptorConfig("io.metersphere.base.domain.FileContent", "file", "io.metersphere.commons.utils.CompressUtils", "zip", "unzip")
);
}
}

30
backend/src/main/java/io/metersphere/config/MybatisConfig.java
View File

@ -1,24 +1,17 @@
package io.metersphere.config; package io.metersphere.config;
import com.github.pagehelper.PageInterceptor; import com.github.pagehelper.PageInterceptor;
import io.metersphere.commons.utils.LogUtil;
import io.metersphere.commons.utils.MybatisInterceptor;
import io.metersphere.commons.utils.MybatisInterceptorConfig; import io.metersphere.commons.utils.MybatisInterceptorConfig;
import io.metersphere.commons.utils.MybatisInterceptorConfigHolder; import io.metersphere.interceptor.MybatisInterceptor;
import org.mybatis.spring.annotation.MapperScan; import org.mybatis.spring.annotation.MapperScan;
import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean; import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
import org.springframework.context.ApplicationContext;
import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration; import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.PropertySource; import org.springframework.context.annotation.PropertySource;
import org.springframework.context.event.ContextRefreshedEvent;
import org.springframework.context.event.EventListener;
import org.springframework.transaction.annotation.EnableTransactionManagement; import org.springframework.transaction.annotation.EnableTransactionManagement;
import org.springframework.util.CollectionUtils;
import java.util.ArrayList; import java.util.ArrayList;
import java.util.List; import java.util.List;
import java.util.Map;
import java.util.Properties; import java.util.Properties;
@Configuration @Configuration
@ -46,27 +39,8 @@ public class MybatisConfig {
public MybatisInterceptor dbInterceptor() { public MybatisInterceptor dbInterceptor() {
MybatisInterceptor interceptor = new MybatisInterceptor(); MybatisInterceptor interceptor = new MybatisInterceptor();
List<MybatisInterceptorConfig> configList = new ArrayList<>(); List<MybatisInterceptorConfig> configList = new ArrayList<>();
configList.add(new MybatisInterceptorConfig("io.metersphere.base.domain.FileContent", "file", "io.metersphere.commons.utils.CompressUtils", "zip", "unzip"));
interceptor.setInterceptorConfigList(configList); interceptor.setInterceptorConfigList(configList);
return interceptor; return interceptor;
} }
/**
* 等到ApplicationContext 加载完成之后 装配MybatisInterceptorConfigHolder
*/
@EventListener
public void handleContextRefresh(ContextRefreshedEvent event) {
try {
ApplicationContext context = event.getApplicationContext();
MybatisInterceptor dBEncryptInterceptor = context.getBean(MybatisInterceptor.class);
Map<String, MybatisInterceptorConfigHolder> beansOfType = context.getBeansOfType(MybatisInterceptorConfigHolder.class);
for (MybatisInterceptorConfigHolder config : beansOfType.values()) {
if (!CollectionUtils.isEmpty(config.interceptorConfig())) {
dBEncryptInterceptor.getInterceptorConfigList().addAll(config.interceptorConfig());
}
}
} catch (Exception e) {
LogUtil.error("装配Mybatis插件拦截配置错误，错误：" + e.getMessage());
}
}
} }

4
backend/src/main/java/io/metersphere/commons/utils/MybatisInterceptor.java → backend/src/main/java/io/metersphere/interceptor/MybatisInterceptor.java
View File

@ -1,5 +1,7 @@
package io.metersphere.commons.utils; package io.metersphere.interceptor;
import io.metersphere.commons.utils.BeanUtils;
import io.metersphere.commons.utils.MybatisInterceptorConfig;
import org.apache.commons.lang3.StringUtils; import org.apache.commons.lang3.StringUtils;
import org.apache.ibatis.cache.CacheKey; import org.apache.ibatis.cache.CacheKey;
import org.apache.ibatis.executor.Executor; import org.apache.ibatis.executor.Executor;