数据权限判断对象类型

5 years ago · 75332929ac
parent 290b568cf4
commit 75332929ac
1 changed files with 6 additions and 2 deletions
--- a/ruoyi-common/ruoyi-common-datascope/src/main/java/com/ruoyi/common/datascope/aspect/DataScopeAspect.java
+++ b/ruoyi-common/ruoyi-common-datascope/src/main/java/com/ruoyi/common/datascope/aspect/DataScopeAspect.java
@ -145,10 +145,14 @@ public class DataScopeAspect

        if (StringUtils.isNotBlank(sqlString.toString()))
        {
-            BaseEntity baseEntity = (BaseEntity) joinPoint.getArgs()[0];
+            Object params = joinPoint.getArgs()[0];
+            if (StringUtils.isNotNull(params) && params instanceof BaseEntity)
+            {
+                BaseEntity baseEntity = (BaseEntity) params;
                baseEntity.getParams().put(DATA_SCOPE, " AND (" + sqlString.substring(4) + ")");
            }
        }
+    }

    /**
     * 是否存在注解，如果存在就获取