update 使用instanceof模式匹配, 减少二次转换

This commit is contained in:
JackyTang 2023-04-20 10:49:29 +08:00
parent 0ebcac34e3
commit 5b7c787b31

View File

@ -66,8 +66,8 @@ public class PlusDataPermissionInterceptor extends JsqlParserSupport implements
@Override @Override
protected void processSelect(Select select, int index, String sql, Object obj) { protected void processSelect(Select select, int index, String sql, Object obj) {
SelectBody selectBody = select.getSelectBody(); SelectBody selectBody = select.getSelectBody();
if (selectBody instanceof PlainSelect) { if (selectBody instanceof PlainSelect plainSelect) {
this.setWhere((PlainSelect) selectBody, (String) obj); this.setWhere(plainSelect, (String) obj);
} else if (selectBody instanceof SetOperationList setOperationList) { } else if (selectBody instanceof SetOperationList setOperationList) {
List<SelectBody> selectBodyList = setOperationList.getSelects(); List<SelectBody> selectBodyList = setOperationList.getSelects();
selectBodyList.forEach(s -> this.setWhere((PlainSelect) s, (String) obj)); selectBodyList.forEach(s -> this.setWhere((PlainSelect) s, (String) obj));