Examples with TenantLineHandler com.baomidou.mybatisplus.extension.plugins.handler.TenantLineHandler used on opensource projects

Example 1 with TenantLineHandler

use of com.baomidou.mybatisplus.extension.plugins.handler.TenantLineHandler in project jeecg-boot by jeecgboot.

the class MybatisPlusSaasConfig method mybatisPlusInterceptor.

@Bean
public MybatisPlusInterceptor mybatisPlusInterceptor() {
    MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
    // 先 add TenantLineInnerInterceptor 再 add PaginationInnerInterceptor
    interceptor.addInnerInterceptor(new TenantLineInnerInterceptor(new TenantLineHandler() {

        @Override
        public Expression getTenantId() {
            String tenant_id = oConvertUtils.getString(TenantContext.getTenant(), "0");
            return new LongValue(tenant_id);
        }

        @Override
        public String getTenantIdColumn() {
            return TENANT_FIELD_NAME;
        }

        // 返回 true 表示不走租户逻辑
        @Override
        public boolean ignoreTable(String tableName) {
            for (String temp : tenantTable) {
                if (temp.equalsIgnoreCase(tableName)) {
                    return false;
                }
            }
            return true;
        }
    }));
    interceptor.addInnerInterceptor(new PaginationInnerInterceptor());
    return interceptor;
}

Example 2 with TenantLineHandler

use of com.baomidou.mybatisplus.extension.plugins.handler.TenantLineHandler in project lamp-util by zuihou.

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("检测到 lamp.database.multiTenantType={}，已启用 {} 模式", 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 Expression getTenantId() {
                return MultiTenantType.COLUMN.eq(databaseProperties.getMultiTenantType()) ? new StringValue(ContextUtil.getTenant()) : new StringValue(ContextUtil.getSubTenant());
            }
        });
        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;
}

Example 3 with TenantLineHandler

use of com.baomidou.mybatisplus.extension.plugins.handler.TenantLineHandler in project mybatis-plus-samples by baomidou.

the class MybatisPlusConfig method mybatisPlusInterceptor.

/**
 * 新多租户插件配置,一缓和二缓遵循mybatis的规则,需要设置 MybatisConfiguration#useDeprecatedExecutor = false 避免缓存万一出现问题
 */
@Bean
public MybatisPlusInterceptor mybatisPlusInterceptor() {
    MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
    interceptor.addInnerInterceptor(new TenantLineInnerInterceptor(new TenantLineHandler() {

        @Override
        public Expression getTenantId() {
            return new LongValue(1);
        }

        // 这是 default 方法,默认返回 false 表示所有表都需要拼多租户条件
        @Override
        public boolean ignoreTable(String tableName) {
            return !"user".equalsIgnoreCase(tableName);
        }
    }));
    // interceptor.addInnerInterceptor(new PaginationInnerInterceptor());
    return interceptor;
}

Example 4 with TenantLineHandler

use of com.baomidou.mybatisplus.extension.plugins.handler.TenantLineHandler in project sakura-boot by yanjingfan.

the class MybatisPlusConfig method tenantPaginationInterceptor.

/**
 * 多租户分页
 * @return
 */
@ConditionalOnProperty(prefix = "tenant", name = "column")
@Bean
public MybatisPlusInterceptor tenantPaginationInterceptor() {
    MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
    TenantLineInnerInterceptor tenantInnerInterceptor = new TenantLineInnerInterceptor();
    tenantInnerInterceptor.setTenantLineHandler(new TenantLineHandler() {

        @Override
        public Expression getTenantId() {
            // 可将租户id和租户字段放置在请求头中
            try {
                String tenantId = request.getHeader("tenantId");
                if (tenantId == null || "".equals(tenantId.trim())) {
                    return new LongValue(-1);
                }
                return new LongValue(Long.parseLong(tenantId));
            } catch (Exception e) {
                return new LongValue(-1);
            }
        }

        @Override
        public String getTenantIdColumn() {
            return environment.getProperty("tenant.column");
        // 可将租户id和租户字段放置在请求头中
        // return request.getHeader("tenantColumn");
        }

        @Override
        public boolean ignoreTable(String tableName) {
            // 忽略uid-generator模块相关表
            if ("worker_node".equalsIgnoreCase(tableName)) {
                return true;
            }
            return false;
        }
    });
    interceptor.addInnerInterceptor(tenantInnerInterceptor);
    interceptor.addInnerInterceptor(new PaginationInnerInterceptor(DbType.MYSQL));
    return interceptor;
}