From 55798c815327639fdc8f1874acf2ec543c596aa1 Mon Sep 17 00:00:00 2001
From: wuzeyu <1223318623@qq.com>
Date: 星期三, 28 八月 2024 17:22:45 +0800
Subject: [PATCH] 优化代码

---
 pipIrr-platform/pipIrr-web/pipIrr-web-sso/src/main/java/com/dy/sso/busi/SsoCtrl.java |   55 +++++++++++++++++++++++++++++++++++++++----------------
 1 files changed, 39 insertions(+), 16 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 2f7571a..46fc4db 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
@@ -7,7 +7,6 @@
 import com.dy.common.webUtil.BaseResponseUtils;
 import com.dy.common.webUtil.ResultCodeMsg;
 import com.dy.pipIrrGlobal.pojoBa.BaUser;
-import com.dy.pipIrrGlobal.util.DistrictLevel;
 import com.dy.pipIrrGlobal.util.Org;
 import com.mysql.cj.util.StringUtils;
 import io.swagger.v3.oas.annotations.Hidden;
@@ -22,12 +21,10 @@
 import jakarta.validation.Valid;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
-//import org.springframework.cache.CacheManager;
 import org.springframework.http.MediaType;
 import org.springframework.validation.BindingResult;
 import org.springframework.web.bind.annotation.*;
 
-import java.util.Arrays;
 import java.util.List;
 import java.util.Objects;
 import java.util.UUID;
@@ -71,7 +68,7 @@
             )
     })
     @GetMapping(path = "allOrg")
-    public BaseResponse<List<Org>> allOrg(){
+    public BaseResponse<List<Org.OrgVo>> allOrg(){
         //List<Org> list = Arrays.asList(Org.Ym, Org.Pj) ;
         return BaseResponseUtils.buildSuccess(Org.OrgList);
     }
@@ -103,18 +100,24 @@
     @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) {
         if(bindingResult != null && bindingResult.hasErrors()){
-            return BaseResponseUtils.buildFail(Objects.requireNonNull(bindingResult.getFieldError()).getDefaultMessage());
+            return BaseResponseUtils.buildErrorMsg(Objects.requireNonNull(bindingResult.getFieldError()).getDefaultMessage());
         }
         if(!vo.phone.equals("admin")){
             if(vo.phone.length() != 11){
-                return BaseResponseUtils.buildFail("鎵嬫満鍙凤紙闀垮害涓嶆槸11浣嶏級涓嶆纭�");
+                return BaseResponseUtils.buildErrorMsg("鎵嬫満鍙凤紙闀垮害涓嶆槸11浣嶏級涓嶆纭�");
             }
         }
         if(vo.orgTag == null || vo.orgTag.trim().length() == 0){
-            return BaseResponseUtils.buildFail("鏈�夋嫨缁勭粐鍗曚綅");
+            return BaseResponseUtils.buildErrorMsg("鏈�夋嫨缁勭粐鍗曚綅");
         }
         //鎶婄粍缁囧崟浣嶆爣绛句綔涓烘暟鎹簮鍚嶇О
         DataSourceContext.set(vo.orgTag);
+
+        //寰楀埌鎵�鏈夌敤鎴疯处鍙�
+        List<String> phones = sv.getPhones();
+        if(!phones.contains(vo.phone)){
+            return BaseResponseUtils.buildErrorMsg("璐﹀彿涓嶅瓨鍦�");
+        }
 
         String uuid ;
         BaUser userPo ;
@@ -139,7 +142,7 @@
             uVo.token = uuid ;
             return BaseResponseUtils.buildSuccess(uVo);
         }else{
-            return BaseResponseUtils.buildFail("鐧诲綍澶辫触");
+            return BaseResponseUtils.buildErrorMsg("瀵嗙爜閿欒");
         }
     }
 
@@ -160,10 +163,10 @@
     @PostMapping(path = "loginForm", consumes = MediaType.MULTIPART_FORM_DATA_VALUE)
     public BaseResponse<UserVo> loginForm(@Parameter(description = "form琛ㄥ崟鏁版嵁", required = true) @Valid LoginVo vo,  @Parameter(hidden = true) BindingResult bindingResult){
         if(bindingResult != null && bindingResult.hasErrors()){
-            return BaseResponseUtils.buildFail(Objects.requireNonNull(bindingResult.getFieldError()).getDefaultMessage());
+            return BaseResponseUtils.buildErrorMsg(Objects.requireNonNull(bindingResult.getFieldError()).getDefaultMessage());
         }
         if(vo.orgTag == null || vo.orgTag.trim().length() == 0){
-            return BaseResponseUtils.buildFail("鏈�夋嫨缁勭粐鍗曚綅");
+            return BaseResponseUtils.buildErrorMsg("鏈�夋嫨缁勭粐鍗曚綅");
         }
         //鎶婄粍缁囧崟浣嶆爣绛句綔涓烘暟鎹簮鍚嶇О
         DataSourceContext.set(vo.orgTag);
@@ -191,7 +194,7 @@
             uVo.token = uuid ;
             return BaseResponseUtils.buildSuccess(uVo);
         }else{
-            return BaseResponseUtils.buildFail("鐧诲綍澶辫触");
+            return BaseResponseUtils.buildErrorMsg("鐧诲綍澶辫触");
         }
     }
 
@@ -216,7 +219,7 @@
             this.sv.logout(token) ;
             return BaseResponseUtils.buildSuccess(true);
         }else{
-            return BaseResponseUtils.buildFail("鏈粠header涓緱鍒皌oken");
+            return BaseResponseUtils.buildErrorMsg("鏈粠header涓緱鍒皌oken");
         }
     }
 
@@ -232,7 +235,27 @@
         BaUser userPo = this.sv.getByUuid(token) ;
         return userPo == null ? null : userPo.id ;
     }
-
+    /**
+     * 姝ゆ柟娉曚緵瀛愭ā鍧楃郴缁熻皟鐢紝鎵�浠ヤ笉鍏紑鍦ˋPI鎺ュ彛涓�
+     * 鏂规硶鍔熻兘锛氶獙璇佹槸鍚﹀凡缁忕櫥褰�
+     * @param token 鐧诲綍鐢ㄦ埛token
+     * @return SsoVo
+     */
+    @Hidden
+    @GetMapping(path = "ssoCheck")
+    public SsoVo ssoCheck(String token){
+        BaUser userPo = this.sv.getByUuid(token) ;
+        SsoVo vo = new SsoVo();
+        if(userPo != null){
+            vo.dataSourceName = userPo.orgTag ;
+            vo.logined = true ;
+            vo.hasPower = true ;//榛樿鏈夋潈闄愩��2023-12-21 缁忓晢璁紝鐢卞墠绔壌鏉�
+        }else{
+            vo.logined = false ;
+            vo.hasPower = true ;//榛樿鏈夋潈闄愩��2023-12-21 缁忓晢璁紝鐢卞墠绔壌鏉�
+        }
+        return vo ;
+    }
     /**
      * 姝ゆ柟娉曚緵瀛愭ā鍧楃郴缁熻皟鐢紝鎵�浠ヤ笉鍏紑鍦ˋPI鎺ュ彛涓�
      * 鏂规硶鍔熻兘锛氶獙璇佹槸鍚﹀凡缁忕櫥褰曪紝濡傛灉鐧诲綍浜嗭紝鍐嶉獙璇佹潈闄�
@@ -243,8 +266,8 @@
      * @return SsoVo
      */
     @Hidden
-    @GetMapping(path = "ssoCheck")
-    public SsoVo ssoCheck(String token, String privilege, String[] allPrivilege, String[] anyPrivilege){
+    @GetMapping(path = "ssoPowerCheck")
+    public SsoVo ssoPowerCheck(String token, String privilege, String[] allPrivilege, String[] anyPrivilege){
         BaUser userPo = this.sv.getByUuid(token) ;
         SsoVo vo = new SsoVo();
         if(userPo != null){
@@ -288,7 +311,7 @@
         CurUserVo vo = new CurUserVo();
         if(userPo != null){
             vo.id = userPo.id ;
-            vo.name = userPo.name ;
+            vo.name = userPo.userName;
         }
         return vo ;
     }

--
Gitblit v1.8.0