Spring Boot + Mybatis Plus 拦截器实现数据权限

分析

需求是需要根据公司 id 做数据权限，那么可以利用 Mybatis 的拦截器在新增和查询的时候进行干预，在新增的时候将上下文中的 公司id 丢进去，在查询的时候增加查询条件

自动填充

由于使用了 Mybatis Plus，那么可以使用他的插件，实现 MetaObjectHandler，重写 insertFill 方法，updateFill 方法，分别对应 新增 和 修改

    /**
     * 插入时的填充策略
     */
    @Override
    public void insertFill(MetaObject metaObject) {
        log.info("start insert fill...");
        // 这里实际使用中动态替换为当前 token 的公司id
        this.setFieldValByName("companyId", 9527, metaObject);
    }

注解指定字段在哪个时机，比如这个是在 INSERT

@TableField(value = "company_id", fill = FieldFill.INSERT)

查询拦截

实现 InnerInterceptor，重写 beforeQuery 方法，表示在查询前执行，可以在这里拼接查询条件，还有个方法 beforeUpdate，表示在更新前执行，暂时用不上

log.info("查询Mapper路径：{}", mappedStatement.getId());
log.info("原查询SQL：{}", PluginUtils.mpBoundSql(boundSql));
log.info("原查询参数：{}", parameter);
log.info("新查询SQL：{}", newSql);

代码

https://gitee.com/zhaomingjian/workspace_demo/tree/master/mybatis-plus-datascope