lbqms.git - Gitblit

			@@ -2,14 +2,15 @@

			import com.baomidou.mybatisplus.core.plugins.InterceptorIgnoreHelper;
			import com.baomidou.mybatisplus.core.toolkit.PluginUtils;
			import com.baomidou.mybatisplus.extension.parser.JsqlParserSupport;
			import com.baomidou.mybatisplus.extension.plugins.handler.MultiDataPermissionHandler;
			import com.baomidou.mybatisplus.extension.plugins.inner.BaseMultiTableInnerInterceptor;
			import com.baomidou.mybatisplus.extension.plugins.inner.InnerInterceptor;
			import lombok.extern.slf4j.Slf4j;
			import net.sf.jsqlparser.expression.Expression;
			import net.sf.jsqlparser.schema.Table;
			import net.sf.jsqlparser.statement.delete.Delete;
			import net.sf.jsqlparser.statement.select.PlainSelect;
			import net.sf.jsqlparser.statement.select.Select;
			import net.sf.jsqlparser.statement.select.SelectBody;
			import net.sf.jsqlparser.statement.select.SetOperationList;
			import net.sf.jsqlparser.statement.update.Update;
			import org.apache.ibatis.executor.Executor;
			@@ -32,7 +33,7 @@
			* @version 3.5.0
			*/
			@Slf4j
			public class PlusDataPermissionInterceptor extends JsqlParserSupport implements InnerInterceptor {
			public class PlusDataPermissionInterceptor extends BaseMultiTableInnerInterceptor implements InnerInterceptor {

			private final PlusDataPermissionHandler dataPermissionHandler;

			@@ -75,11 +76,10 @@

			@Override
			protected void processSelect(Select select, int index, String sql, Object obj) {
			SelectBody selectBody = select.getSelectBody();
			if (selectBody instanceof PlainSelect plainSelect) {
			this.setWhere(plainSelect, (String) obj);
			} else if (selectBody instanceof SetOperationList setOperationList) {
			List<SelectBody> selectBodyList = setOperationList.getSelects();
			if (select instanceof PlainSelect) {
			this.setWhere((PlainSelect) select, (String) obj);
			} else if (select instanceof SetOperationList setOperationList) {
			List<Select> selectBodyList = setOperationList.getSelects();
			selectBodyList.forEach(s -> this.setWhere((PlainSelect) s, (String) obj));
			}
			}
			@@ -113,5 +113,11 @@
			}
			}

			@Override
			public Expression buildTableExpression(Table table, Expression where, String whereSegment) {
			// 只有新版数据权限处理器才会执行到这里
			final MultiDataPermissionHandler handler = (MultiDataPermissionHandler) dataPermissionHandler;
			return handler.getSqlSegment(table, where, whereSegment);
			}
			}