From 1e409f99905c894363248239b856bce6735fd3d5 Mon Sep 17 00:00:00 2001
From: zhubaomin <zhubaomin>
Date: 星期一, 28 十月 2024 20:15:05 +0800
Subject: [PATCH] 2024-10-28 优化获取常用取水口接口

---
 pipIrr-platform/pipIrr-web/pipIrr-web-sso/src/main/java/com/dy/sso/busi/SsoCtrl.java |   33 +++++++++++++++++++++++++++++----
 1 files changed, 29 insertions(+), 4 deletions(-)

diff --git a/pipIrr-platform/pipIrr-web/pipIrr-web-sso/src/main/java/com/dy/sso/busi/SsoCtrl.java b/pipIrr-platform/pipIrr-web/pipIrr-web-sso/src/main/java/com/dy/sso/busi/SsoCtrl.java
index 46fc4db..2e349e8 100644
--- a/pipIrr-platform/pipIrr-web/pipIrr-web-sso/src/main/java/com/dy/sso/busi/SsoCtrl.java
+++ b/pipIrr-platform/pipIrr-web/pipIrr-web-sso/src/main/java/com/dy/sso/busi/SsoCtrl.java
@@ -25,9 +25,7 @@
 import org.springframework.validation.BindingResult;
 import org.springframework.web.bind.annotation.*;
 
-import java.util.List;
-import java.util.Objects;
-import java.util.UUID;
+import java.util.*;
 
 /**
  * 娉ㄨВTag 鍦ˋPI涓樉绀猴細 Tag 娉ㄨВ, 缁欐暣涓帴鍙h捣浜嗕釜鍚嶅瓧涓庢弿杩�"
@@ -45,6 +43,7 @@
     //鍦ㄥ睘鎬т笂娉ㄨВ@Autowired鏃讹紝浼氳鍛� Field injection is not recommended锛堜笉鍐嶆帹鑽愪娇鐢ㄥ瓧娈垫敞鍏ワ級
     private SsoSv sv ;
 
+    //private KaptchaConfig kaptchaConfig;
     //@Autowired
     //private CacheManager cacheManager ;
 
@@ -53,6 +52,10 @@
         this.sv = sv ;
     }
 
+    //@Autowired
+    //public void setKaptchaConfig(KaptchaConfig kaptchaConfig) {
+    //    this.kaptchaConfig = kaptchaConfig;
+    //}
 
     /**
      * 瀹㈡埛绔姹傚緱鍒版墍鏈夌粍缁囨満鏋�
@@ -98,7 +101,7 @@
             )
     })
     @PostMapping(path = "loginJson", consumes = MediaType.APPLICATION_JSON_VALUE)
-    public BaseResponse<UserVo> loginJson(@RequestBody @Parameter(description = "鐧诲綍json鏁版嵁", required = true) @Valid LoginVo vo,  @Parameter(hidden = true) BindingResult bindingResult) {
+    public BaseResponse<UserVo> loginJson(HttpServletRequest request, @RequestBody @Parameter(description = "鐧诲綍json鏁版嵁", required = true) @Valid LoginVo vo,  @Parameter(hidden = true) BindingResult bindingResult) {
         if(bindingResult != null && bindingResult.hasErrors()){
             return BaseResponseUtils.buildErrorMsg(Objects.requireNonNull(bindingResult.getFieldError()).getDefaultMessage());
         }
@@ -113,6 +116,28 @@
         //鎶婄粍缁囧崟浣嶆爣绛句綔涓烘暟鎹簮鍚嶇О
         DataSourceContext.set(vo.orgTag);
 
+        String token = Optional.ofNullable(vo.getToken()).orElse("");
+        String code = Optional.ofNullable(vo.getCode()).orElse("");
+
+        if(token.length() > 0 && code.length() > 0) {
+            // 浠巗ession涓幏鍙栭獙璇佺爜
+            //HttpSession session = (HttpSession) request.getSession();
+            //String localCode = session.getAttribute(token).toString();
+
+            Map map = sv.getCodeByToken(token);
+            Long expiration = Long.parseLong(map.get("expiration").toString());
+            Long currentTimestamp = System.currentTimeMillis();
+            if(currentTimestamp > expiration) {
+                return BaseResponseUtils.buildErrorMsg("楠岃瘉鐮佸凡瓒呮椂");
+            }
+
+            // 浠庢暟鎹簱鑾峰彇楠岃瘉鐮�
+            String localCode = map.get("code").toString();
+            if(!code.equals(localCode)) {
+                return BaseResponseUtils.buildErrorMsg("楠岃瘉鐮侀敊璇�");
+            }
+        }
+
         //寰楀埌鎵�鏈夌敤鎴疯处鍙�
         List<String> phones = sv.getPhones();
         if(!phones.contains(vo.phone)){

--
Gitblit v1.8.0