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