分析
需求是需要根据公司 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