package com.vincent.rsf.server.common.config;
|
|
import com.baomidou.mybatisplus.autoconfigure.ConfigurationCustomizer;
|
import com.baomidou.mybatisplus.core.MybatisConfiguration;
|
import com.baomidou.mybatisplus.extension.MybatisMapWrapperFactory;
|
import com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor;
|
import com.baomidou.mybatisplus.extension.plugins.handler.TenantLineHandler;
|
import com.baomidou.mybatisplus.extension.plugins.inner.PaginationInnerInterceptor;
|
import com.baomidou.mybatisplus.extension.plugins.inner.TenantLineInnerInterceptor;
|
import com.vincent.rsf.server.system.entity.User;
|
import net.sf.jsqlparser.expression.Expression;
|
import net.sf.jsqlparser.expression.LongValue;
|
import net.sf.jsqlparser.expression.NullValue;
|
import org.springframework.context.annotation.Bean;
|
import org.springframework.context.annotation.Configuration;
|
import org.springframework.security.core.Authentication;
|
import org.springframework.security.core.context.SecurityContextHolder;
|
import org.springframework.transaction.annotation.EnableTransactionManagement;
|
|
import java.util.Arrays;
|
|
/**
|
* MybatisPlus配置
|
*
|
* @author vincent
|
* @since 2018-02-22 11:29:28
|
*/
|
@Configuration
|
@EnableTransactionManagement
|
public class MybatisPlusConfig {
|
|
@Bean
|
public MybatisPlusInterceptor mybatisPlusInterceptor() {
|
MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
|
|
// 多租户插件配置
|
TenantLineHandler tenantLineHandler = new TenantLineHandler() {
|
@Override
|
public Expression getTenantId() {
|
return getLoginUserTenantId();
|
}
|
|
@Override
|
public boolean ignoreTable(String tableName) {
|
return Arrays.asList(
|
"sys_tenant",
|
"sys_host",
|
"sys_user_role",
|
"sys_role_menu",
|
"sys_menu",
|
"man_loc_type_rela"
|
).contains(tableName);
|
}
|
};
|
TenantLineInnerInterceptor tenantLineInnerInterceptor = new TenantLineInnerInterceptor(tenantLineHandler);
|
interceptor.addInnerInterceptor(tenantLineInnerInterceptor);
|
|
// 分页插件配置
|
PaginationInnerInterceptor paginationInnerInterceptor = new PaginationInnerInterceptor();
|
interceptor.addInnerInterceptor(paginationInnerInterceptor);
|
|
return interceptor;
|
}
|
|
/**
|
* 获取当前登录用户的租户id
|
*
|
* @return Integer
|
*/
|
public Expression getLoginUserTenantId() {
|
try {
|
Authentication authentication = SecurityContextHolder.getContext().getAuthentication();
|
if (authentication != null) {
|
Object object = authentication.getPrincipal();
|
if (object instanceof User) {
|
return new LongValue(((User) object).getTenantId());
|
}
|
}
|
} catch (Exception e) {
|
System.out.println(e.getMessage());
|
}
|
//TODO 需设置一个系统调度帐号
|
// return new NullValue();
|
return new LongValue(1);
|
}
|
|
|
/**
|
* 解决Map映射非驼峰
|
* @return
|
*/
|
@Bean
|
public ConfigurationCustomizer mybatisConfigurationCustomizer(){
|
return new ConfigurationCustomizer() {
|
/**
|
* Customize the given a {@link MybatisConfiguration} object.
|
*
|
* @param configuration the configuration object to customize
|
*/
|
@Override
|
public void customize(MybatisConfiguration configuration) {
|
configuration.setObjectWrapperFactory(new MybatisMapWrapperFactory());
|
}
|
};
|
}
|
|
}
|
