代码优化

smartboot · Aug 20, 2024 · f445238 · f445238
1 parent 9267fc2
commit f445238
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/servlet-core/src/main/java/tech/smartboot/servlet/impl/HttpServletRequestImpl.java b/servlet-core/src/main/java/tech/smartboot/servlet/impl/HttpServletRequestImpl.java
@@ -265,11 +265,10 @@ public boolean isUserInRole(String role) {
     @Override
     public Principal getUserPrincipal() {
         if (principal == null) {
-//            try {
-//                principal = runtime.getSecurityProvider().getUser(this);
-//            } catch (ServletException e) {
-//                throw new RuntimeException(e);
-//            }
+            Object o = getSession().getAttribute("principal");
+            if (o != null) {
+                principal = (LoginAccount) o;
+            }
         }
         return principal;
     }
@@ -395,7 +394,7 @@ public boolean authenticate(HttpServletResponse response) {
     public void login(String username, String password) throws ServletException {
         SecurityAccount securityAccount = runtime.getSecurityProvider().login(username, password);
         if (securityAccount != null) {
-            principal = new LoginAccount(securityAccount.getUsername(), securityAccount.getPassword(), securityAccount.getRoles());
+            setLoginAccount(new LoginAccount(securityAccount.getUsername(), securityAccount.getPassword(), securityAccount.getRoles()));
         }
     }
 
@@ -847,6 +846,7 @@ public void setAsyncSupported(boolean supported) {
     @Override
     public void setLoginAccount(LoginAccount loginAccount) {
         this.principal = loginAccount;
+        getSession().setAttribute("principal", loginAccount);
     }
 
     @Override