|  |  | 
 |  |  | 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 在API中显示: Tag 注解, 给整个接口起了个名字与描述" | 
 |  |  | 
 |  |  | @RequestMapping(path="sso") | 
 |  |  | @SuppressWarnings("unchecked")//java版本越高,对泛型约束越严,所以配置SuppressWarnings("unchecked") | 
 |  |  | public class SsoCtrl { | 
 |  |  |     //万用token | 
 |  |  |     private static final String UniversalUserToken = "0000-0000-1234-9876-5"; | 
 |  |  |  | 
 |  |  |     //在属性上注解@Autowired时,会警告 Field injection is not recommended(不再推荐使用字段注入) | 
 |  |  |     private SsoSv sv ; | 
 |  |  |  | 
 |  |  |     //private KaptchaConfig kaptchaConfig; | 
 |  |  |     //@Autowired | 
 |  |  |     //private CacheManager cacheManager ; | 
 |  |  |  | 
 |  |  | 
 |  |  |         this.sv = sv ; | 
 |  |  |     } | 
 |  |  |  | 
 |  |  |     //@Autowired | 
 |  |  |     //public void setKaptchaConfig(KaptchaConfig kaptchaConfig) { | 
 |  |  |     //    this.kaptchaConfig = kaptchaConfig; | 
 |  |  |     //} | 
 |  |  |  | 
 |  |  |     /** | 
 |  |  |      * 客户端请求得到所有组织机构 | 
 |  |  | 
 |  |  |             ) | 
 |  |  |     }) | 
 |  |  |     @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) { | 
 |  |  |             // 从session中获取验证码 | 
 |  |  |             //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)){ | 
 |  |  |                 /* | 
 |  |  |                 如果前端进行了base64加密 | 
 |  |  | 
 |  |  |                 */ | 
 |  |  |                 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()) ; | 
 |  |  | 
 |  |  |  | 
 |  |  |         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("登录失败"); | 
 |  |  |         } | 
 |  |  |     } | 
 |  |  |  | 
 |  |  | 
 |  |  |     @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); | 
 |  |  | 
 |  |  |             uVo.token = uuid ; | 
 |  |  |             return BaseResponseUtils.buildSuccess(uVo); | 
 |  |  |         }else{ | 
 |  |  |             return BaseResponseUtils.buildFail("登录失败"); | 
 |  |  |             return BaseResponseUtils.buildErrorMsg("登录失败"); | 
 |  |  |         } | 
 |  |  |     } | 
 |  |  |  | 
 |  |  | 
 |  |  |             this.sv.logout(token) ; | 
 |  |  |             return BaseResponseUtils.buildSuccess(true); | 
 |  |  |         }else{ | 
 |  |  |             return BaseResponseUtils.buildFail("未从header中得到token"); | 
 |  |  |             return BaseResponseUtils.buildErrorMsg("未从header中得到token"); | 
 |  |  |         } | 
 |  |  |     } | 
 |  |  |  | 
 |  |  | 
 |  |  |     @Hidden | 
 |  |  |     @GetMapping(path = "ssoCheck") | 
 |  |  |     public SsoVo ssoCheck(String token){ | 
 |  |  |         BaUser userPo = this.sv.getByUuid(token) ; | 
 |  |  |         BaUser userPo = null ; | 
 |  |  |         if(token.equals(UniversalUserToken)){ | 
 |  |  |             //调试阶段,用的万用token | 
 |  |  |             userPo = new BaUser() ; | 
 |  |  |             Org.OrgVo orgVo = Org.OrgList.get(0) ; | 
 |  |  |             userPo.orgTag = orgVo.tag ; | 
 |  |  |         }else{ | 
 |  |  |             userPo = this.sv.getByUuid(token) ; | 
 |  |  |         } | 
 |  |  |         SsoVo vo = new SsoVo(); | 
 |  |  |         if(userPo != null){ | 
 |  |  |             vo.dataSourceName = userPo.orgTag ; |