Search in sources :

Example 1 with TenantLineInnerInterceptor

use of com.albedo.java.plugins.database.interceptor.TenantLineInnerInterceptor in project albedo by somowhere.

the class BaseMybatisConfiguration method mybatisPlusInterceptor.

/**
 * 新的分页插件,一缓和二缓遵循mybatis的规则,需要设置 MybatisConfiguration#useDeprecatedExecutor = false 避免缓存出现问题(该属性会在旧插件移除后一同移除)
 * <p>
 * 注意:
 * 如果内部插件都是使用,需要注意顺序关系,建议使用如下顺序
 * 多租户插件,动态表名插件
 * 分页插件,乐观锁插件
 * sql性能规范插件,防止全表更新与删除插件
 * 总结: 对sql进行单次改造的优先放入,不对sql进行改造的最后放入
 * <p>
 * 参考:
 * https://mybatis.plus/guide/interceptor.html#%E4%BD%BF%E7%94%A8%E6%96%B9%E5%BC%8F-%E4%BB%A5%E5%88%86%E9%A1%B5%E6%8F%92%E4%BB%B6%E4%B8%BE%E4%BE%8B
 */
@Bean
@Order(5)
@ConditionalOnMissingBean
public MybatisPlusInterceptor mybatisPlusInterceptor() {
    MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
    log.info("detected application.database.multiTenantType={},{} mode enabled", databaseProperties.getMultiTenantType().name(), databaseProperties.getMultiTenantType().getDescribe());
    if (StrUtil.equalsAny(databaseProperties.getMultiTenantType().name(), MultiTenantType.SCHEMA.name(), MultiTenantType.SCHEMA_COLUMN.name())) {
        // SCHEMA 动态表名插件
        SchemaInterceptor schemaInterceptor = new SchemaInterceptor(databaseProperties.getTenantDatabasePrefix());
        interceptor.addInnerInterceptor(schemaInterceptor);
    }
    if (StrUtil.equalsAny(databaseProperties.getMultiTenantType().name(), MultiTenantType.COLUMN.name(), MultiTenantType.SCHEMA_COLUMN.name(), MultiTenantType.DATASOURCE_COLUMN.name())) {
        // COLUMN 模式 多租户插件
        TenantLineInnerInterceptor tli = new TenantLineInnerInterceptor();
        tli.setTenantLineHandler(new TenantLineHandler() {

            @Override
            public String getTenantIdColumn() {
                return databaseProperties.getTenantIdColumn();
            }

            @Override
            public boolean ignoreTable(String tableName) {
                return databaseProperties.getIgnoreTables() != null && databaseProperties.getIgnoreTables().contains(tableName);
            }

            @Override
            public boolean ignoreMapId(String mapperId) {
                return databaseProperties.getIgnoreMapperIds() != null && databaseProperties.getIgnoreMapperIds().contains(mapperId);
            }

            @Override
            public Expression getTenantId() {
                return new StringValue(ContextUtil.getTenant());
            }
        });
        interceptor.addInnerInterceptor(tli);
    }
    List<InnerInterceptor> beforeInnerInterceptor = getPaginationBeforeInnerInterceptor();
    if (!beforeInnerInterceptor.isEmpty()) {
        beforeInnerInterceptor.forEach(interceptor::addInnerInterceptor);
    }
    // 分页插件
    PaginationInnerInterceptor paginationInterceptor = new PaginationInnerInterceptor();
    // 单页分页条数限制
    paginationInterceptor.setMaxLimit(databaseProperties.getMaxLimit());
    // 数据库类型
    paginationInterceptor.setDbType(databaseProperties.getDbType());
    // 溢出总页数后是否进行处理
    paginationInterceptor.setOverflow(databaseProperties.getOverflow());
    // 生成 countSql 优化掉 join 现在只支持 left join
    paginationInterceptor.setOptimizeJoin(databaseProperties.getOptimizeJoin());
    interceptor.addInnerInterceptor(paginationInterceptor);
    List<InnerInterceptor> afterInnerInterceptor = getPaginationAfterInnerInterceptor();
    if (!afterInnerInterceptor.isEmpty()) {
        afterInnerInterceptor.forEach(interceptor::addInnerInterceptor);
    }
    // 防止全表更新与删除插件
    if (databaseProperties.getIsBlockAttack()) {
        interceptor.addInnerInterceptor(new BlockAttackInnerInterceptor());
    }
    // sql性能规范插件
    if (databaseProperties.getIsIllegalSql()) {
        interceptor.addInnerInterceptor(new IllegalSQLInnerInterceptor());
    }
    return interceptor;
}
Also used : TenantLineInnerInterceptor(com.albedo.java.plugins.database.interceptor.TenantLineInnerInterceptor) TenantLineHandler(com.albedo.java.plugins.database.handler.TenantLineHandler) BlockAttackInnerInterceptor(com.baomidou.mybatisplus.extension.plugins.inner.BlockAttackInnerInterceptor) IllegalSQLInnerInterceptor(com.baomidou.mybatisplus.extension.plugins.inner.IllegalSQLInnerInterceptor) MybatisPlusInterceptor(com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor) PaginationInnerInterceptor(com.baomidou.mybatisplus.extension.plugins.inner.PaginationInnerInterceptor) SchemaInterceptor(com.albedo.java.plugins.database.interceptor.SchemaInterceptor) ConditionalOnExpression(org.springframework.boot.autoconfigure.condition.ConditionalOnExpression) Expression(net.sf.jsqlparser.expression.Expression) StringValue(net.sf.jsqlparser.expression.StringValue) InnerInterceptor(com.baomidou.mybatisplus.extension.plugins.inner.InnerInterceptor) IllegalSQLInnerInterceptor(com.baomidou.mybatisplus.extension.plugins.inner.IllegalSQLInnerInterceptor) PaginationInnerInterceptor(com.baomidou.mybatisplus.extension.plugins.inner.PaginationInnerInterceptor) TenantLineInnerInterceptor(com.albedo.java.plugins.database.interceptor.TenantLineInnerInterceptor) BlockAttackInnerInterceptor(com.baomidou.mybatisplus.extension.plugins.inner.BlockAttackInnerInterceptor) Order(org.springframework.core.annotation.Order) ConditionalOnMissingBean(org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean) ConditionalOnMissingBean(org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean) Bean(org.springframework.context.annotation.Bean)

Aggregations

TenantLineHandler (com.albedo.java.plugins.database.handler.TenantLineHandler)1 SchemaInterceptor (com.albedo.java.plugins.database.interceptor.SchemaInterceptor)1 TenantLineInnerInterceptor (com.albedo.java.plugins.database.interceptor.TenantLineInnerInterceptor)1 MybatisPlusInterceptor (com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor)1 BlockAttackInnerInterceptor (com.baomidou.mybatisplus.extension.plugins.inner.BlockAttackInnerInterceptor)1 IllegalSQLInnerInterceptor (com.baomidou.mybatisplus.extension.plugins.inner.IllegalSQLInnerInterceptor)1 InnerInterceptor (com.baomidou.mybatisplus.extension.plugins.inner.InnerInterceptor)1 PaginationInnerInterceptor (com.baomidou.mybatisplus.extension.plugins.inner.PaginationInnerInterceptor)1 Expression (net.sf.jsqlparser.expression.Expression)1 StringValue (net.sf.jsqlparser.expression.StringValue)1 ConditionalOnExpression (org.springframework.boot.autoconfigure.condition.ConditionalOnExpression)1 ConditionalOnMissingBean (org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean)1 Bean (org.springframework.context.annotation.Bean)1 Order (org.springframework.core.annotation.Order)1