From a0801ab3f16bd3d967c220c5925ad02673ce1baa Mon Sep 17 00:00:00 2001 From: 彭雪彬 <1724387007@qq.com> Date: Tue, 15 Jul 2025 15:24:16 +0800 Subject: [PATCH] 骑手微信登录 --- oying-system/src/main/java/com/oying/modules/security/rest/AuthController.java | 131 +++++++++++++++++++++++++++++++++++++++---- 1 files changed, 118 insertions(+), 13 deletions(-) diff --git a/oying-system/src/main/java/com/oying/modules/security/rest/AuthController.java b/oying-system/src/main/java/com/oying/modules/security/rest/AuthController.java index 02ed3dd..67ef05d 100644 --- a/oying-system/src/main/java/com/oying/modules/security/rest/AuthController.java +++ b/oying-system/src/main/java/com/oying/modules/security/rest/AuthController.java @@ -11,6 +11,7 @@ import com.oying.modules.security.service.UserDetailsServiceImpl; import com.oying.modules.security.service.WeiXinService; import com.oying.modules.security.service.dto.AuthUserDto; +import com.oying.modules.security.service.dto.AuthUserPhoneDto; import com.oying.modules.security.service.dto.AuthUserWeixinDto; import com.oying.modules.security.service.dto.JwtUserDto; import com.oying.modules.system.domain.DictDetail; @@ -80,7 +81,7 @@ if (StringUtils.isBlank(code)) { throw new BadRequestException("验证码不存在或已过期"); } - if (StringUtils.isBlank(authUser.getCode()) || !authUser.getCode().equalsIgnoreCase(code)) { + if (!authUser.getCode().equalsIgnoreCase(code)) { throw new BadRequestException("验证码错误"); } // 获取用户信息 @@ -108,26 +109,60 @@ return ResponseEntity.ok(R.success(authInfo)); } - @Log("微信授权登录") - @ApiOperation("微信授权登录") + @Log("短信验证码登录") + @ApiOperation("短信验证码登录") + @AnonymousPostMapping(value = "/login/phone") + public ResponseEntity<Object> loginPhone(@Validated @RequestBody AuthUserPhoneDto authUser, HttpServletRequest request) { + // 查询验证码 + String code = redisUtils.get(authUser.getUuid(), String.class); + // 清除验证码 + redisUtils.del(authUser.getUuid()); + if (StringUtils.isBlank(code)) { + throw new BadRequestException("验证码不存在或已过期"); + } + if (!authUser.getCode().equalsIgnoreCase(code)) { + throw new BadRequestException("验证码错误"); + } + // 获取用户信息 + JwtUserDto jwtUser = userDetailsService.loadUserByUsername(authUser.getUsername()); + Authentication authentication = new UsernamePasswordAuthenticationToken(jwtUser, null, jwtUser.getAuthorities()); + SecurityContextHolder.getContext().setAuthentication(authentication); + // 生成令牌 + String token = tokenProvider.createToken(jwtUser); + // 返回 token 与 用户信息 + Map<String, Object> authInfo = new HashMap<String, Object>(2) {{ + put("token", properties.getTokenStartWith() + token); + put("user", jwtUser); + }}; + if (loginProperties.isSingleLogin()) { + // 踢掉之前已经登录的token + onlineUserService.kickOutForUsername(authUser.getUsername()); + } + // 保存在线信息 + onlineUserService.save(jwtUser, token, request); + // 返回登录信息 + return ResponseEntity.ok(R.success(authInfo)); + } + + public static final String OLD = "OLD"; + public static final String NEW = "NEW"; + + @Log("小程序:微信授权登录") + @ApiOperation("小程序:微信授权登录") @AnonymousPostMapping(value = "/login/weixin") public ResponseEntity<Object> loginWeixin(@Validated @RequestBody AuthUserWeixinDto authUser, HttpServletRequest request) throws Exception { JSONObject jsonObject; - switch (authUser.getCode()) { - case "OLD": + switch (authUser.getType()) { + case OLD: jsonObject = weiXinService.code2Session(authUser.getCode()); String openid = jsonObject.getString("openid"); User userDto = userService.findByOpenid(openid); if (userDto == null) { - Map<String, Object> authInfo = new HashMap<String, Object>(2) {{ - put("token", openid); - put("member", null); - }}; - return ResponseEntity.ok(authInfo); + return ResponseEntity.ok(R.success(openid)); } authUser.setUsername(userDto.getUsername()); break; - case "NEW": + case NEW: jsonObject = weiXinService.getPhoneNumber(authUser.getCode()); String phone = jsonObject.getJSONObject("phone_info").getString("purePhoneNumber"); User user1 = userService.findByName(phone); @@ -135,8 +170,10 @@ //创建用户 User user = new User(); user.setUsername(phone); - user.setNickName("LYHD-" + phone); + user.setNickName("OYING-" + phone); user.setUserType(ConstantsKey.BUYER); + user.setPhone(phone); + user.setGender("男"); user.setEnabled(true); getRole(user); user.setPassword(passwordEncoder.encode(phone.substring(phone.length() - 6))); @@ -182,11 +219,79 @@ Set<Role> roles = new HashSet<>(); Role role = new Role(); DictDetail detail = dictDetailService.getDictByName(ConstantsKey.USER_TYPE_BUYER).get(0); - role.setId(Long.valueOf(detail.getLabel())); + role.setId(Long.valueOf(detail.getValue())); roles.add(role); user1.setRoles(roles); } + @Log("骑手小程序:微信授权登录") + @ApiOperation("骑手小程序:微信授权登录") + @AnonymousPostMapping(value = "/rider/login/weixin") + public ResponseEntity<Object> riderLoginWeixin(@Validated @RequestBody AuthUserWeixinDto authUser, HttpServletRequest request) throws Exception { + JSONObject jsonObject; + switch (authUser.getType()) { + case OLD: + jsonObject = weiXinService.code2SessionRider(authUser.getCode()); + String openid = jsonObject.getString("openid"); + User userDto = userService.findByRiderOpenId(openid); + if (userDto == null) { + return ResponseEntity.ok(R.success(openid)); + } + authUser.setUsername(userDto.getUsername()); + break; + case NEW: + jsonObject = weiXinService.getPhoneNumber(authUser.getCode()); + String phone = jsonObject.getJSONObject("phone_info").getString("purePhoneNumber"); + User user1 = userService.findByName(phone); + if (user1 == null) { + //创建用户 + User user = new User(); + user.setUsername(phone); + user.setNickName("OYING-" + phone); + user.setUserType(ConstantsKey.RIDER); + user.setPhone(phone); + user.setGender("男"); + user.setEnabled(true); + getRole(user); + user.setPassword(passwordEncoder.encode(phone.substring(phone.length() - 6))); + user.setOpenid(authUser.getUsername()); + userService.create(user); + } else { + if (user1.getUserType().equals(ConstantsKey.RIDER)) { + if (user1.getRoles() == null) { + getRole(user1); + } + } + user1.setOpenid(authUser.getUsername()); + userService.update(user1); + } + authUser.setUsername(phone); + break; + default: + throw new BadRequestException("登录类型错误"); + + } + // 获取用户信息 + JwtUserDto jwtUser = userDetailsService.loadUserByUsername(authUser.getUsername()); + Authentication authentication = new UsernamePasswordAuthenticationToken(jwtUser, null, jwtUser.getAuthorities()); + SecurityContextHolder.getContext().setAuthentication(authentication); + // 生成令牌 + String token = tokenProvider.createToken(jwtUser); + // 返回 token 与 用户信息 + Map<String, Object> authInfo = new HashMap<String, Object>(2) {{ + put("token", properties.getTokenStartWith() + token); + put("user", jwtUser); + }}; + if (loginProperties.isSingleLogin()) { + // 踢掉之前已经登录的token + onlineUserService.kickOutForUsername(authUser.getUsername()); + } + // 保存在线信息 + onlineUserService.save(jwtUser, token, request); + // 返回登录信息 + return ResponseEntity.ok(R.success(authInfo)); + } + @ApiOperation("临时授权") @AnonymousGetMapping(value = "/token") public ResponseEntity<Object> loginTest(@RequestParam String username, HttpServletRequest request) { -- Gitblit v1.9.3