parent
8f5b0adfe2
commit
d68b4dd9e0
2 changed files with 43 additions and 20 deletions
@ -0,0 +1,43 @@ |
|||||||
|
package net.mingsoft.config; |
||||||
|
|
||||||
|
import com.baomidou.mybatisplus.annotation.DbType; |
||||||
|
import com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor; |
||||||
|
import com.baomidou.mybatisplus.extension.plugins.handler.TenantLineHandler; |
||||||
|
import com.baomidou.mybatisplus.extension.plugins.inner.TenantLineInnerInterceptor; |
||||||
|
import com.baomidou.mybatisplus.extension.toolkit.JdbcUtils; |
||||||
|
import net.mingsoft.interceptor.DMInnerInterceptor; |
||||||
|
import net.mingsoft.interceptor.MysqlInnerInterceptor; |
||||||
|
import org.springframework.beans.factory.annotation.Autowired; |
||||||
|
import org.springframework.context.annotation.Bean; |
||||||
|
import org.springframework.context.annotation.Configuration; |
||||||
|
|
||||||
|
import javax.sql.DataSource; |
||||||
|
import java.sql.SQLException; |
||||||
|
|
||||||
|
/** |
||||||
|
* Mybatis 拦截器配置 |
||||||
|
*/ |
||||||
|
@Configuration |
||||||
|
public class MybatisInterceptorConfig { |
||||||
|
|
||||||
|
@Bean |
||||||
|
public MybatisPlusInterceptor mybatisPlusInterceptor(DataSource dataSource, @Autowired(required = false) TenantLineHandler tenantLineHandler) { |
||||||
|
MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor(); |
||||||
|
if(tenantLineHandler!=null){ |
||||||
|
interceptor.addInnerInterceptor(new TenantLineInnerInterceptor(tenantLineHandler)); |
||||||
|
} |
||||||
|
try { |
||||||
|
//mysql 添加转换sql
|
||||||
|
DbType dbType = JdbcUtils.getDbType(dataSource.getConnection().getMetaData().getURL()); |
||||||
|
if(DbType.MYSQL==dbType){ |
||||||
|
interceptor.addInnerInterceptor(new MysqlInnerInterceptor()); |
||||||
|
}else if(DbType.DM==dbType){ |
||||||
|
interceptor.addInnerInterceptor( new DMInnerInterceptor()); |
||||||
|
} |
||||||
|
} catch (SQLException e) { |
||||||
|
e.printStackTrace(); |
||||||
|
} |
||||||
|
|
||||||
|
return interceptor; |
||||||
|
} |
||||||
|
} |
||||||
Loading…
Reference in new issue