admin/wisdomisite-java
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

打日志

Browse Source
This commit is contained in:
guoshengxiong 2024-05-24 16:40:35 +08:00
parent 753ca35894
commit a0ce911a58
1 changed files with 10 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
src/main/java/com/zhgd/mybatis/DataScopeInterceptor.java
View File

@ -60,6 +60,15 @@ public class DataScopeInterceptor extends JsqlParserSupport implements InnerInte
return false; 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) { public void beforePrepare(StatementHandler sh, Connection connection, Integer transactionTimeout) {
PluginUtils.MPStatementHandler mpSh = PluginUtils.mpStatementHandler(sh); PluginUtils.MPStatementHandler mpSh = PluginUtils.mpStatementHandler(sh);
MappedStatement ms = mpSh.mappedStatement(); MappedStatement ms = mpSh.mappedStatement();
@ -99,6 +108,7 @@ public class DataScopeInterceptor extends JsqlParserSupport implements InnerInte
Method[] declaredMethods = clazz.getDeclaredMethods(); Method[] declaredMethods = clazz.getDeclaredMethods();
Optional<DataScope> dsOption = Arrays.stream(declaredMethods).filter(method -> method.getName().equals(methodName)).map(method -> method.getAnnotation(DataScope.class)).filter(Objects::nonNull).findFirst(); Optional<DataScope> 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)); annotation = dsOption.orElseGet(() -> clazz.getAnnotation(DataScope.class));
log.debug("执行mapper方法: {} # {}", clazz.getName(), methodName);
if (findIgnoreDataScope(parameter, annotation) && isNotSqlTest()) { if (findIgnoreDataScope(parameter, annotation) && isNotSqlTest()) {
return; 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) { protected void processSelect(Select select, int index, String sql, Object obj) {
this.processSelectBody(select.getSelectBody(), obj); this.processSelectBody(select.getSelectBody(), obj);
} }