lbqms.git - Gitblit

			@@ -2,10 +2,14 @@

			import cn.hutool.core.annotation.AnnotationUtil;
			import cn.hutool.core.collection.CollUtil;
			import cn.hutool.core.collection.ConcurrentHashSet;
			import cn.hutool.core.util.ArrayUtil;
			import cn.hutool.core.util.ClassUtil;
			import cn.hutool.core.util.ObjectUtil;
			import lombok.extern.slf4j.Slf4j;
			import net.sf.jsqlparser.JSQLParserException;
			import net.sf.jsqlparser.expression.Expression;
			import net.sf.jsqlparser.expression.Parenthesis;
			import net.sf.jsqlparser.expression.operators.conditional.AndExpression;
			import net.sf.jsqlparser.parser.CCJSqlParserUtil;
			import org.dromara.common.core.domain.dto.RoleDTO;
			import org.dromara.common.core.domain.model.LoginUser;
			import org.dromara.common.core.exception.ServiceException;
			@@ -17,12 +21,6 @@
			import org.dromara.common.mybatis.enums.DataScopeType;
			import org.dromara.common.mybatis.helper.DataPermissionHelper;
			import org.dromara.common.satoken.utils.LoginHelper;
			import lombok.extern.slf4j.Slf4j;
			import net.sf.jsqlparser.JSQLParserException;
			import net.sf.jsqlparser.expression.Expression;
			import net.sf.jsqlparser.expression.Parenthesis;
			import net.sf.jsqlparser.expression.operators.conditional.AndExpression;
			import net.sf.jsqlparser.parser.CCJSqlParserUtil;
			import org.springframework.context.expression.BeanFactoryResolver;
			import org.springframework.expression.BeanResolver;
			import org.springframework.expression.ExpressionParser;
			@@ -51,11 +49,6 @@
			private final Map<String, DataPermission> dataPermissionCacheMap = new ConcurrentHashMap<>();

			/**
			* 无效注解方法缓存用于快速返回
			*/
			private final Set<String> invalidCacheSet = new ConcurrentHashSet<>();

			/**
			* spel 解析器
			*/
			private final ExpressionParser parser = new SpelExpressionParser();
			@@ -68,10 +61,6 @@

			public Expression getSqlSegment(Expression where, String mappedStatementId, boolean isSelect) {
			DataColumn[] dataColumns = findAnnotation(mappedStatementId);
			if (ArrayUtil.isEmpty(dataColumns)) {
			invalidCacheSet.add(mappedStatementId);
			return where;
			}
			LoginUser currentUser = DataPermissionHelper.getVariable("user");
			if (ObjectUtil.isNull(currentUser)) {
			currentUser = LoginHelper.getLoginUser();
			@@ -155,7 +144,7 @@
			return "";
			}

			private DataColumn[] findAnnotation(String mappedStatementId) {
			public DataColumn[] findAnnotation(String mappedStatementId) {
			StringBuilder sb = new StringBuilder(mappedStatementId);
			int index = sb.lastIndexOf(".");
			String clazzName = sb.substring(0, index);
			@@ -189,10 +178,4 @@
			return null;
			}

			/**
			* 是否为无效方法无数据权限
			*/
			public boolean isInvalid(String mappedStatementId) {
			return invalidCacheSet.contains(mappedStatementId);
			}
			}