Browse Source

单点登录

tianwu.sun 9 months ago
parent
commit
76356ced9d

+ 11 - 0
src/main/java/com/bootdo/common/token/Header.java

@@ -9,6 +9,9 @@ import java.io.Serializable;
 public class Header implements Serializable {
     private long timestamp;
 
+    //系统标志:0后台管理,1数据备案,2风险评估
+    private Integer sysFlag;
+
     public long getTimestamp() {
         return timestamp;
     }
@@ -16,4 +19,12 @@ public class Header implements Serializable {
     public void setTimestamp(long timestamp) {
         this.timestamp = timestamp;
     }
+
+    public Integer getSysFlag() {
+        return sysFlag;
+    }
+
+    public void setSysFlag(Integer sysFlag) {
+        this.sysFlag = sysFlag;
+    }
 }

+ 1 - 0
src/main/java/com/bootdo/common/token/TokenUtil.java

@@ -77,6 +77,7 @@ public final class TokenUtil {
         tokenObj.setPayload(payload);
         Header header = new Header();
         header.setTimestamp(System.currentTimeMillis() / 1000);
+        header.setSysFlag(1);
         tokenObj.setHeader(header);
         String token = createTokenString(tokenObj);
         return token;

+ 4 - 13
src/main/java/com/bootdo/system/controller/LoginController.java

@@ -7,17 +7,13 @@ import com.bootdo.common.controller.BaseController;
 import com.bootdo.common.domain.FileDO;
 import com.bootdo.common.domain.Tree;
 import com.bootdo.common.service.FileService;
-import com.bootdo.common.token.Header;
 import com.bootdo.common.token.Payload;
 import com.bootdo.common.token.Token;
 import com.bootdo.common.token.TokenUtil;
 import com.bootdo.common.utils.*;
 import com.bootdo.system.domain.MenuDO;
-import com.bootdo.system.domain.UserDO;
 import com.bootdo.system.service.MenuService;
-import com.bootdo.system.service.ThirdMsgLogDOService;
 import org.apache.shiro.SecurityUtils;
-import org.apache.shiro.authc.AuthenticationException;
 import org.apache.shiro.authc.UsernamePasswordToken;
 import org.apache.shiro.subject.Subject;
 import org.slf4j.Logger;
@@ -55,11 +51,6 @@ public class LoginController extends BaseController {
     @Autowired
     private ThirdPartSsoConfig thirdPartSsoConfig;
 
-    /**
-     * 第三方日志记录
-     */
-    @Resource
-    private ThirdMsgLogDOService thirdMsgLogDOService;
 
     @Resource
     private TokenUtil tokenUtil;
@@ -167,9 +158,9 @@ public class LoginController extends BaseController {
         }
     }
 
-    @Log("使用token登录")
-    @GetMapping("/loginWithToken")
-    String loginWithToken(String token) {
+    @Log("单点登录")
+    @PostMapping("/ssoLogin")
+    String ssoLogin(String token) {
         try {
             if(ObjectUtils.isEmpty(token)){
                 throw new Exception("token不能为空");
@@ -186,7 +177,7 @@ public class LoginController extends BaseController {
             UsernamePasswordToken usernamePasswordToken = new UsernamePasswordToken(username, password);
             Subject subject = SecurityUtils.getSubject();
             subject.login(usernamePasswordToken);
-            return "loginWithToken";
+            return "ssoLogin";
         } catch (Exception e) {
             e.printStackTrace();
             return "login";

src/main/resources/templates/loginWithToken.html → src/main/resources/templates/ssoLogin.html