From 7634d7ff15b1fa84ea84a51a1ba6e45b11a4aa21 Mon Sep 17 00:00:00 2001 From: liurunyu <lry9898@163.com> Date: 星期日, 27 四月 2025 11:17:34 +0800 Subject: [PATCH] SSO登录逻辑再修改 --- pipIrr-platform/pipIrr-web/pipIrr-web-sso/src/main/java/com/dy/sso/busi/SsoCtrl.java | 76 ++++++++++++++++++++++++-------------- 1 files changed, 48 insertions(+), 28 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 dda4c8b..f475e27 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,15 +21,11 @@ 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; +import java.util.*; /** * 娉ㄨВTag 鍦ˋPI涓樉绀猴細 Tag 娉ㄨВ, 缁欐暣涓帴鍙h捣浜嗕釜鍚嶅瓧涓庢弿杩�" @@ -48,6 +43,7 @@ //鍦ㄥ睘鎬т笂娉ㄨВ@Autowired鏃讹紝浼氳鍛� Field injection is not recommended锛堜笉鍐嶆帹鑽愪娇鐢ㄥ瓧娈垫敞鍏ワ級 private SsoSv sv ; + //private KaptchaConfig kaptchaConfig; //@Autowired //private CacheManager cacheManager ; @@ -56,6 +52,10 @@ this.sv = sv ; } + //@Autowired + //public void setKaptchaConfig(KaptchaConfig kaptchaConfig) { + // this.kaptchaConfig = kaptchaConfig; + //} /** * 瀹㈡埛绔姹傚緱鍒版墍鏈夌粍缁囨満鏋� @@ -71,7 +71,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); } @@ -101,26 +101,46 @@ ) }) @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.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); - String uuid ; - BaUser userPo ; + if(vo.token != null && vo.token.trim().length() > 0) { + // 浠巗ession涓幏鍙栭獙璇佺爜 + //HttpSession session = (HttpSession) request.getSession(); + //String localCode = session.getAttribute(token).toString(); + // 浠庢暟鎹簱鑾峰彇楠岃瘉鐮� + Map map = sv.getCodeByToken(vo.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(!vo.code.equals(localCode)) { + return BaseResponseUtils.buildErrorMsg("楠岃瘉鐮侀敊璇�"); + } + } + + if(!sv.existPhone(vo.phone)){ + return BaseResponseUtils.buildErrorMsg("璐﹀彿涓嶅瓨鍦�"); + } + + String uuid4Token = null; + BaUser userPo = null ; try { - //Boolean flag = cacheManager.getCacheNames().isEmpty() ; - uuid = UUID.randomUUID().toString(); + uuid4Token = UUID.randomUUID().toString(); if(!StringUtils.isNullOrEmpty(vo.password)){ /* 濡傛灉鍓嶇杩涜浜哹ase64鍔犲瘑 @@ -128,7 +148,7 @@ */ vo.password = MD5.encrypt(vo.password) ; } - userPo = this.sv.loginWithMapperXml(uuid, vo.phone, vo.password); + userPo = this.sv.loginWithMapperXml(uuid4Token, vo.phone, vo.password); } catch (Exception e) { log.error("鐢ㄦ埛鐧诲綍寮傚父", e); return BaseResponseUtils.buildException(e.getMessage()) ; @@ -136,10 +156,10 @@ if(userPo != null){ UserVo uVo = UserVoMapper.INSTANCT.po2vo(userPo); - uVo.token = uuid ; + uVo.token = uuid4Token ; return BaseResponseUtils.buildSuccess(uVo); }else{ - return BaseResponseUtils.buildFail("鐧诲綍澶辫触"); + return BaseResponseUtils.buildErrorMsg("鐧诲綍澶辫触"); } } @@ -160,10 +180,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 +211,7 @@ uVo.token = uuid ; return BaseResponseUtils.buildSuccess(uVo); }else{ - return BaseResponseUtils.buildFail("鐧诲綍澶辫触"); + return BaseResponseUtils.buildErrorMsg("鐧诲綍澶辫触"); } } @@ -216,7 +236,7 @@ this.sv.logout(token) ; return BaseResponseUtils.buildSuccess(true); }else{ - return BaseResponseUtils.buildFail("鏈粠header涓緱鍒皌oken"); + return BaseResponseUtils.buildErrorMsg("鏈粠header涓緱鍒皌oken"); } } @@ -234,7 +254,7 @@ } /** * 姝ゆ柟娉曚緵瀛愭ā鍧楃郴缁熻皟鐢紝鎵�浠ヤ笉鍏紑鍦ˋPI鎺ュ彛涓� - * 鏂规硶鍔熻兘锛氶獙璇佹槸鍚﹀凡缁忕櫥褰曪紝濡傛灉鐧诲綍浜嗭紝鍐嶉獙璇佹潈闄� + * 鏂规硶鍔熻兘锛氶獙璇佹槸鍚﹀凡缁忕櫥褰� * @param token 鐧诲綍鐢ㄦ埛token * @return SsoVo */ @@ -246,10 +266,10 @@ if(userPo != null){ vo.dataSourceName = userPo.orgTag ; vo.logined = true ; - vo.hasPower = true ;//榛樿鏈夋潈闄� + vo.hasPower = true ;//榛樿鏈夋潈闄愩��2023-12-21 缁忓晢璁紝鐢卞墠绔壌鏉� }else{ vo.logined = false ; - vo.hasPower = true ;//榛樿鏈夋潈闄� + vo.hasPower = true ;//榛樿鏈夋潈闄愩��2023-12-21 缁忓晢璁紝鐢卞墠绔壌鏉� } return vo ; } @@ -263,8 +283,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){ @@ -308,7 +328,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