From a0ce911a58e9aa499b1f56c9bad06331401ac784 Mon Sep 17 00:00:00 2001 From: guoshengxiong <1923636941@qq.com> Date: Fri, 24 May 2024 16:40:35 +0800 Subject: [PATCH] =?UTF-8?q?=E6=89=93=E6=97=A5=E5=BF=97?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../zhgd/mybatis/DataScopeInterceptor.java | 19 ++++++++++--------- 1 file changed, 10 insertions(+), 9 deletions(-) diff --git a/src/main/java/com/zhgd/mybatis/DataScopeInterceptor.java b/src/main/java/com/zhgd/mybatis/DataScopeInterceptor.java index 243b450a9..f8ce85f17 100644 --- a/src/main/java/com/zhgd/mybatis/DataScopeInterceptor.java +++ b/src/main/java/com/zhgd/mybatis/DataScopeInterceptor.java @@ -60,6 +60,15 @@ public class DataScopeInterceptor extends JsqlParserSupport implements InnerInte return false; } + public static boolean isNotSqlTest() { + try { + HttpServletRequest request = ((ServletRequestAttributes) (RequestContextHolder.currentRequestAttributes())).getRequest(); + return request.getParameter("qqq") == null; + } catch (Exception e) { + return true; + } + } + public void beforePrepare(StatementHandler sh, Connection connection, Integer transactionTimeout) { PluginUtils.MPStatementHandler mpSh = PluginUtils.mpStatementHandler(sh); MappedStatement ms = mpSh.mappedStatement(); @@ -99,6 +108,7 @@ public class DataScopeInterceptor extends JsqlParserSupport implements InnerInte Method[] declaredMethods = clazz.getDeclaredMethods(); Optional dsOption = Arrays.stream(declaredMethods).filter(method -> method.getName().equals(methodName)).map(method -> method.getAnnotation(DataScope.class)).filter(Objects::nonNull).findFirst(); annotation = dsOption.orElseGet(() -> clazz.getAnnotation(DataScope.class)); + log.debug("执行mapper方法: {} # {}", clazz.getName(), methodName); if (findIgnoreDataScope(parameter, annotation) && isNotSqlTest()) { return; } @@ -112,15 +122,6 @@ public class DataScopeInterceptor extends JsqlParserSupport implements InnerInte } } - public static boolean isNotSqlTest() { - try { - HttpServletRequest request = ((ServletRequestAttributes) (RequestContextHolder.currentRequestAttributes())).getRequest(); - return request.getParameter("qqq") == null; - } catch (Exception e) { - return true; - } - } - protected void processSelect(Select select, int index, String sql, Object obj) { this.processSelectBody(select.getSelectBody(), obj); }