MyBatisPlus數(shù)據(jù)權(quán)限控制實現(xiàn)的三種方式
更新時間：2024年05月29日 10:21:56 作者：Lee_SmallNorth
數(shù)據(jù)權(quán)限是保障數(shù)據(jù)安全的重要手段,本文主要介紹了MyBatisPlus數(shù)據(jù)權(quán)限控制實現(xiàn)的三種方式,具有一定的參考價值,具有一定的參考價值,感興趣的可以了解一下
一、場景
在數(shù)據(jù)權(quán)限控制時，希望是全局的，比如每個sql都根據(jù)當前登錄人查詢，因此希望每個sql都在最后拼接 “create_user = ‘admin’”
二、實現(xiàn)方式（提供兩種，自行選擇）
注意：每張表都必須保證create_user 的字段存在，否則要重寫ignoreTable方法
2.1 模仿租戶實現(xiàn)方式
@Bean
public MybatisPlusInterceptor paginationInterceptor() {
MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
PaginationInnerInterceptor paginationInnerInterceptor = new PaginationInnerInterceptor();
paginationInnerInterceptor.setDbType(DbType.MYSQL);
paginationInnerInterceptor.setMaxLimit(1000L);
paginationInnerInterceptor.setOverflow(true);
interceptor.addInnerInterceptor(paginationInnerInterceptor);
interceptor.addInnerInterceptor(new CustomLineInnerInterceptor(new com.szls.config.DataPermissionHandler() {
@Override
public Expression getCreateUser() {
//這里直接寫死admin,實際按需獲取
return new StringValue("admin");
}
@Override
public String getCreateUserColumn() {
return DataPermissionHandler.super.getCreateUserColumn();
}
@Override
public boolean ignoreTable(String tableName) {
//這是案例，多表需要過濾時，可從配置文件或者static代碼塊中獲取
List<String> ignoreTables = Collections.singletonList("sys_model_relation");
//判斷當前表是否在過濾列表中
return ignoreTables.stream().anyMatch(tableName::equalsIgnoreCase);
}
}));
return interceptor;
}
/**
* @description:
* @author: SmallNorth_Lee
* @date: 2024/5/7 15:29
* @version: 1.0
*/
public interface DataPermissionHandler {
/**
* 獲取用戶 ID 值表達式，只支持單個 ID 值
* <p>
*
* @return 用戶 ID 值表達式
*/
Expression getCreateUser();
/**
* 獲取用戶字段名
* <p>
* 默認字段名叫: create_user
*
* @return 用戶字段名
*/
default String getCreateUserColumn() {
return "create_user";
}
/**
* 根據(jù)表名判斷是否忽略拼接多用戶條件
* <p>
* 默認都要進行解析并拼接多用戶條件
*
* @param tableName 表名
* @return 是否忽略, true:表示忽略，false:需要解析并拼接多用戶條件
*/
default boolean ignoreTable(String tableName) {
return false;
}
/**
* 忽略插入用戶字段邏輯
*
* @param columns 插入字段
* @param createUserColumn 用戶 ID 字段
* @return
*/
default boolean ignoreInsert(List<Column> columns, String createUserColumn) {
return columns.stream().map(Column::getColumnName).anyMatch(i -> i.equalsIgnoreCase(createUserColumn));
}
}
/**
* @description:
* @author: SmallNorth_Lee
* @date: 2024/5/7 14:14
* @version: 1.0
*/
@Data
@NoArgsConstructor
@AllArgsConstructor
@ToString(callSuper = true)
@EqualsAndHashCode(callSuper = true)
public class CustomLineInnerInterceptor extends BaseMultiTableInnerInterceptor implements InnerInterceptor {
private DataPermissionHandler dataPermissionHandler;
@Override
public void beforeQuery(Executor executor, MappedStatement ms, Object parameter, RowBounds rowBounds, ResultHandler resultHandler, BoundSql boundSql) throws SQLException {
if (InterceptorIgnoreHelper.willIgnoreTenantLine(ms.getId())) {
return;
}
PluginUtils.MPBoundSql mpBs = PluginUtils.mpBoundSql(boundSql);
mpBs.sql(parserSingle(mpBs.sql(), null));
}
@Override
public void beforePrepare(StatementHandler sh, Connection connection, Integer transactionTimeout) {
PluginUtils.MPStatementHandler mpSh = PluginUtils.mpStatementHandler(sh);
MappedStatement ms = mpSh.mappedStatement();
SqlCommandType sct = ms.getSqlCommandType();
if (sct == SqlCommandType.INSERT || sct == SqlCommandType.UPDATE || sct == SqlCommandType.DELETE) {
if (InterceptorIgnoreHelper.willIgnoreTenantLine(ms.getId())) {
return;
}
PluginUtils.MPBoundSql mpBs = mpSh.mPBoundSql();
mpBs.sql(parserMulti(mpBs.sql(), null));
}
}
@Override
protected void processSelect(Select select, int index, String sql, Object obj) {
final String whereSegment = (String) obj;
processSelectBody(select.getSelectBody(), whereSegment);
List<WithItem> withItemsList = select.getWithItemsList();
if (!CollectionUtils.isEmpty(withItemsList)) {
withItemsList.forEach(withItem -> processSelectBody(withItem, whereSegment));
}
}
@Override
protected void processInsert(Insert insert, int index, String sql, Object obj) {
if (dataPermissionHandler.ignoreTable(insert.getTable().getName())) {
// 過濾退出執(zhí)行
return;
}
List<Column> columns = insert.getColumns();
if (CollectionUtils.isEmpty(columns)) {
// 針對不給列名的insert 不處理
return;
}
String createUserColumn = dataPermissionHandler.getCreateUserColumn();
if (dataPermissionHandler.ignoreInsert(columns, createUserColumn)) {
// 針對已給出租戶列的insert 不處理
return;
}
columns.add(new Column(createUserColumn));
// fixed gitee pulls/141 duplicate update
List<Expression> duplicateUpdateColumns = insert.getDuplicateUpdateExpressionList();
if (CollectionUtils.isNotEmpty(duplicateUpdateColumns)) {
EqualsTo equalsTo = new EqualsTo();
equalsTo.setLeftExpression(new StringValue(createUserColumn));
equalsTo.setRightExpression(dataPermissionHandler.getCreateUser());
duplicateUpdateColumns.add(equalsTo);
}
Select select = insert.getSelect();
if (select != null && (select.getSelectBody() instanceof PlainSelect)) { //fix github issue 4998 修復升級到4.5版本的問題
this.processInsertSelect(select.getSelectBody(), (String) obj);
} else if (insert.getItemsList() != null) {
// fixed github pull/295
ItemsList itemsList = insert.getItemsList();
Expression createUser = dataPermissionHandler.getCreateUser();
if (itemsList instanceof MultiExpressionList) {
((MultiExpressionList) itemsList).getExpressionLists().forEach(el -> el.getExpressions().add(createUser));
} else {
List<Expression> expressions = ((ExpressionList) itemsList).getExpressions();
if (CollectionUtils.isNotEmpty(expressions)) {//fix github issue 4998 jsqlparse 4.5 批量insert ItemsList不是MultiExpressionList 了，需要特殊處理
int len = expressions.size();
for (int i = 0; i < len; i++) {
Expression expression = expressions.get(i);
if (expression instanceof RowConstructor) {
((RowConstructor) expression).getExprList().getExpressions().add(createUser);
} else if (expression instanceof Parenthesis) {
RowConstructor rowConstructor = new RowConstructor()
.withExprList(new ExpressionList(((Parenthesis) expression).getExpression(), createUser));
expressions.set(i, rowConstructor);
} else {
if (len - 1 == i) { // (?,?) 只有最后一個expre的時候才拼接tenantId
expressions.add(createUser);
}
}
}
} else {
expressions.add(createUser);
}
}
} else {
throw ExceptionUtils.mpe("Failed to process multiple-table update, please exclude the tableName or statementId");
}
}
/**
* update 語句處理
*/
@Override
protected void processUpdate(Update update, int index, String sql, Object obj) {
final Table table = update.getTable();
if (dataPermissionHandler.ignoreTable(table.getName())) {
// 過濾退出執(zhí)行
return;
}
ArrayList<UpdateSet> sets = update.getUpdateSets();
if (!CollectionUtils.isEmpty(sets)) {
sets.forEach(us -> us.getExpressions().forEach(ex -> {
if (ex instanceof SubSelect) {
processSelectBody(((SubSelect) ex).getSelectBody(), (String) obj);
}
}));
}
update.setWhere(this.andExpression(table, update.getWhere(), (String) obj));
}
/**
* delete 語句處理
*/
@Override
protected void processDelete(Delete delete, int index, String sql, Object obj) {
if (dataPermissionHandler.ignoreTable(delete.getTable().getName())) {
// 過濾退出執(zhí)行
return;
}
delete.setWhere(this.andExpression(delete.getTable(), delete.getWhere(), (String) obj));
}
/**
* 處理 insert into select
* <p>
* 進入這里表示需要 insert 的表啟用了多租戶,則 select 的表都啟動了
*
* @param selectBody SelectBody
*/
protected void processInsertSelect(SelectBody selectBody, final String whereSegment) {
PlainSelect plainSelect = (PlainSelect) selectBody;
FromItem fromItem = plainSelect.getFromItem();
if (fromItem instanceof Table) {
// fixed gitee pulls/141 duplicate update
processPlainSelect(plainSelect, whereSegment);
appendSelectItem(plainSelect.getSelectItems());
} else if (fromItem instanceof SubSelect) {
SubSelect subSelect = (SubSelect) fromItem;
appendSelectItem(plainSelect.getSelectItems());
processInsertSelect(subSelect.getSelectBody(), whereSegment);
}
}
/**
* 追加 SelectItem
*
* @param selectItems SelectItem
*/
protected void appendSelectItem(List<SelectItem> selectItems) {
if (CollectionUtils.isEmpty(selectItems)) {
return;
}
if (selectItems.size() == 1) {
SelectItem item = selectItems.get(0);
if (item instanceof AllColumns || item instanceof AllTableColumns) {
return;
}
}
selectItems.add(new SelectExpressionItem(new Column(dataPermissionHandler.getCreateUserColumn())));
}
/**
* 租戶字段別名設(shè)置
* <p>tenantId 或 tableAlias.tenantId</p>
*
* @param table 表對象
* @return 字段
*/
protected Column getAliasColumn(Table table) {
StringBuilder column = new StringBuilder();
// todo 該起別名就要起別名,禁止修改此處邏輯
if (table.getAlias() != null) {
column.append(table.getAlias().getName()).append(StringPool.DOT);
}
column.append(dataPermissionHandler.getCreateUserColumn());
return new Column(column.toString());
}
@Override
public void setProperties(Properties properties) {
PropertyMapper.newInstance(properties).whenNotBlank("dataPermissionHandler",
ClassUtils::newInstance, this::setDataPermissionHandler);
}
/**
* 構(gòu)建租戶條件表達式
*
* @param table 表對象
* @param where 當前where條件
* @param whereSegment 所屬Mapper對象全路徑（在原租戶攔截器功能中，這個參數(shù)并不需要參與相關(guān)判斷）
* @return 租戶條件表達式
* @see BaseMultiTableInnerInterceptor#buildTableExpression(Table, Expression, String)
*/
@Override
public Expression buildTableExpression(final Table table, final Expression where, final String whereSegment) {
if (dataPermissionHandler.ignoreTable(table.getName())) {
return null;
}
return new EqualsTo(getAliasColumn(table), dataPermissionHandler.getCreateUser());
}
}
2.2 使用DataPermissionInterceptor和MultiDataPermissionHandler
interceptor.addInnerInterceptor(new DataPermissionInterceptor(new MultiDataPermissionHandler() {
@Override
public Expression getSqlSegment(Table table, Expression where, String mappedStatementId) {
Column aliasColumn = getAliasColumn(table);
if (null == aliasColumn) {
return null;
}
return new EqualsTo(getAliasColumn(table), new StringValue(AuthUtils.getUser()));
}
}));
protected Column getAliasColumn(Table table) {
//這是案例，多表需要過濾時，可從配置文件或者static代碼塊中獲取
List<String> ignoreTables = Collections.singletonList("sys_model_relation");
//判斷當前表是否在過濾列表中
boolean match = ignoreTables.stream().anyMatch(table.getName()::equalsIgnoreCase);
if (match) {
return null;
}
StringBuilder column = new StringBuilder();
// todo 該起別名就要起別名,禁止修改此處邏輯
if (table.getAlias() != null) {
column.append(table.getAlias().getName()).append(StringPool.DOT);
}
column.append("create_user");
return new Column(column.toString());
}到此這篇關(guān)于MyBatisPlus數(shù)據(jù)權(quán)限控制實現(xiàn)的三種方式的文章就介紹到這了,更多相關(guān)MyBatisPlus權(quán)限控制內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
這篇文章主要介紹了Java 定時器的使用，幫助大家更好的理解和使用Java time類，感興趣的朋友可以了解下2020-09-09
這篇文章主要介紹了Java基于正則實現(xiàn)的日期校驗功能,涉及java文件讀取、日期轉(zhuǎn)換及字符串正則匹配相關(guān)操作技巧,需要的朋友可以參考下2017-03-03
這篇文章主要為大家詳細介紹了如何通過SpringBoot實現(xiàn)抽獎算法，文中的示例代碼簡潔易懂，具有一定的參考價值，感興趣的小伙伴可以了解一下2023-06-06
本文主要介紹了Spring的RequestBodyAdvice攔截淺析，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2023-02-02
Spring Cloud Feign的文件上傳實現(xiàn)的示例代碼
這篇文章主要介紹了Spring Cloud Feign的文件上傳實現(xiàn)的示例代碼，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2018-03-03
這篇文章主要介紹了Springboot-注解-操作日志的實現(xiàn)方式，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2021-03-03
SpringBoot 整合 Netty 多端口監(jiān)聽的操作方法
Netty提供異步的、基于事件驅(qū)動的網(wǎng)絡應用程序框架,用以快速開發(fā)高性能、高可靠性的網(wǎng)絡 IO 程序,是目前最流行的 NIO 框架,這篇文章主要介紹了SpringBoot 整和 Netty 并監(jiān)聽多端口,需要的朋友可以參考下2023-10-10
這篇文章主要介紹了使用java編程從0到1實現(xiàn)一個簡單計算器，文章中用代碼實例講解的很清晰，有感興趣的同學可以學習研究下2021-02-02
我們在開發(fā)過程中經(jīng)常會遇到前后端分離而導致的跨域問題,導致無法獲取返回結(jié)果,下面給大家介紹Java中解決跨域問題的方法匯總,感興趣的朋友跟隨小編一起看看吧2024-04-04
eclipse創(chuàng)建springboot項目的三種方式總結(jié)
這篇文章主要介紹了eclipse創(chuàng)建springboot項目的三種方式，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2021-07-07