add chain

Author: xlorne

springboot-starter-security/src/main/java/com/codingapi/springboot/security/AutoConfiguration.java

@@ -3,6 +3,7 @@
 import com.codingapi.springboot.security.configurer.HttpSecurityConfigurer;
 import com.codingapi.springboot.security.controller.VersionController;
 import com.codingapi.springboot.security.dto.request.LoginRequest;
+import com.codingapi.springboot.security.dto.response.LoginResponse;
 import com.codingapi.springboot.security.filter.*;
 import com.codingapi.springboot.security.gateway.Token;
 import com.codingapi.springboot.security.gateway.TokenGateway;
@@ -70,8 +71,12 @@ public void preHandle(HttpServletRequest request, HttpServletResponse response,
             }
 
             @Override
-            public void postHandle(HttpServletRequest request, HttpServletResponse response, LoginRequest handler, Token token) {
-
+            public LoginResponse postHandle(HttpServletRequest request, HttpServletResponse response, LoginRequest handler, Token token) {
+                LoginResponse loginResponse = new LoginResponse();
+                loginResponse.setToken(token.getToken());
+                loginResponse.setUsername(token.getUsername());
+                loginResponse.setAuthorities(token.getAuthorities());
+                return loginResponse;
             }
         };
     }

springboot-starter-security/src/main/java/com/codingapi/springboot/security/dto/response/LoginResponse.java

@@ -8,9 +8,8 @@
 @Setter
 @Getter
 public class LoginResponse {
-
     private String username;
     private String token;
     private List<String> authorities;
-
+    private Object data;
 }

springboot-starter-security/src/main/java/com/codingapi/springboot/security/filter/MyLoginFilter.java

@@ -76,15 +76,10 @@ protected void successfulAuthentication(HttpServletRequest request, HttpServletR
                 user.getAuthorities().stream().map(GrantedAuthority::getAuthority).collect(Collectors.toList()),
                 TokenContext.getExtra());
 
-        LoginResponse login = new LoginResponse();
-        login.setUsername(user.getUsername());
-        login.setToken(token.getToken());
-        login.setAuthorities(token.getAuthorities());
-
-        String content = JSONObject.toJSONString(SingleResponse.of(login));
+        LoginResponse loginResponse = loginHandler.postHandle(request,response,loginRequest,token);
+        String content = JSONObject.toJSONString(SingleResponse.of(loginResponse));
         IOUtils.write(content, response.getOutputStream(), StandardCharsets.UTF_8);
 
-        loginHandler.postHandle(request,response,loginRequest,token);
         LoginRequestContext.getInstance().clean();
 
     }

springboot-starter-security/src/main/java/com/codingapi/springboot/security/filter/SecurityLoginHandler.java

@@ -1,6 +1,7 @@
 package com.codingapi.springboot.security.filter;
 
 import com.codingapi.springboot.security.dto.request.LoginRequest;
+import com.codingapi.springboot.security.dto.response.LoginResponse;
 import com.codingapi.springboot.security.gateway.Token;
 import jakarta.servlet.http.HttpServletRequest;
 import jakarta.servlet.http.HttpServletResponse;
@@ -9,6 +10,6 @@ public interface SecurityLoginHandler {
 
     void preHandle(HttpServletRequest request, HttpServletResponse response, LoginRequest handler) throws Exception;
 
-    void postHandle(HttpServletRequest request, HttpServletResponse response, LoginRequest handler, Token token);
+    LoginResponse postHandle(HttpServletRequest request, HttpServletResponse response, LoginRequest handler, Token token);
 
 }