代码优化

smartboot · Aug 20, 2024 · 802f3bd · 802f3bd
1 parent e14f47a
commit 802f3bd
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/servlet-core/src/main/java/tech/smartboot/servlet/plugins/security/SecurityProviderImpl.java b/servlet-core/src/main/java/tech/smartboot/servlet/plugins/security/SecurityProviderImpl.java
@@ -79,8 +79,13 @@ public boolean isUserInRole(String role, LoginAccount loginAccount, HttpServletR
                 return true;
             }
         }
+        String roleLink = httpServletRequest.getServletInfo().getSecurityRoles().get(role);
+        if (roleLink == null) {
+            return loginAccount.getRoles().contains(role);
+        }
+        System.out.println(roleLink);
         System.out.println(httpServletRequest.getServletInfo().getSecurityRoles());
-        return loginAccount.getMatches().contains(role);
+        return loginAccount.getRoles().contains(roleLink) || loginAccount.getRoles().contains(role);
     }
 
     @Override