巧用tk-mybatis插件，高效管理商品数据权限

在使用tk-mybatis处理包含公司和部门字段的商品信息时，如何根据用户所属公司和部门精准控制数据访问权限，避免繁琐的SQL条件拼接？本文提供一种基于tk-mybatis插件机制的优雅解决方案。

利用tk-mybatis插件机制

tk-mybatis强大的插件机制允许开发者自定义扩展其功能，例如常用的分页插件。我们可借鉴此模式，创建一个自定义插件来灵活控制数据权限。

自定义数据权限插件

以下代码示例展示了一个自定义插件的实现：

import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;
import com.baomidou.mybatisplus.extension.plugins.handler.TableNameHandler;
import java.sql.Statement;
import java.util.List;
import org.apache.ibatis.executor.statement.StatementHandler;

public class DataPermissionPlugin extends TableNameHandler implements MetaObjectHandler {

    @Override
    public void handleStatement(StatementHandler statementHandler, String sql, Statement statement, List<Object> parameterList) {
        String newSql = sql + " WHERE company = #{company} AND department = #{department}";
        statementHandler.parameterHandler.setParameters(parameterList);
        ((Statement) statementHandler).setSql(newSql);
    }
}

此插件重写了handleStatement方法，在原始SQL语句后追加公司和部门的筛选条件。 修改后的SQL语句随后被重新设置到StatementHandler中。

Apache Superset Dashboard and SQL Exploration Skill

Apache Superset 是一个广泛采用的开源 BI 平台，用于 SQL 探索、图表构建和仪表板交付。当代理需要查询仓库数据、组装仪表板或使用成熟的分析界面解释指标而不是临时笔记本代码时，此技能非常有用。

下载

插件注册与应用

将自定义插件注册到tk-mybatis中：

MyBatisSqlSessionFactoryBean bean = new MyBatisSqlSessionFactoryBean();
bean.setPlugins(new Interceptor[]{new DataPermissionPlugin()});

在商品实体类中定义公司和部门字段：

@Table(name = "商品表")
public class 商品 {
    private String company;
    private String department;
    private String name;

    // 省略 getter/setter
}

现在，每次使用tk-mybatis查询商品数据时，插件会自动在SQL语句中添加公司和部门的过滤条件，无需手动编写复杂的SQL语句。

总结

通过自定义tk-mybatis插件，我们可以轻松实现商品数据权限的灵活控制，显著提高开发效率，并使代码更简洁易维护。