From ccd758bcd9b357b365c28f62efde203faa83fb93 Mon Sep 17 00:00:00 2001 From: luoyisheng <yangdongji@argo-ai.cn> Date: 星期一, 24 二月 2025 14:45:10 +0800 Subject: [PATCH] bug --- medicalWaste-admin/src/main/java/com/sinata/web/controller/applet/AppLoginController.java | 102 +++++++++++++++++++++++++++++++++++++++++++-------- 1 files changed, 86 insertions(+), 16 deletions(-) diff --git a/medicalWaste-admin/src/main/java/com/sinata/web/controller/applet/AppLoginController.java b/medicalWaste-admin/src/main/java/com/sinata/web/controller/applet/AppLoginController.java index 42a86cb..e38fe84 100644 --- a/medicalWaste-admin/src/main/java/com/sinata/web/controller/applet/AppLoginController.java +++ b/medicalWaste-admin/src/main/java/com/sinata/web/controller/applet/AppLoginController.java @@ -2,7 +2,7 @@ import com.alibaba.fastjson2.JSON; import com.alibaba.fastjson2.JSONObject; -import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; +import com.baomidou.mybatisplus.core.toolkit.CollectionUtils; import com.baomidou.mybatisplus.core.toolkit.Wrappers; import com.sinata.common.constant.Constants; import com.sinata.common.core.domain.AjaxResult; @@ -12,6 +12,7 @@ import com.sinata.common.core.domain.entity.SysUser; import com.sinata.common.core.domain.model.LoginBody; import com.sinata.common.core.domain.model.LoginUser; +import com.sinata.common.core.redis.RedisCache; import com.sinata.common.utils.SecurityUtils; import com.sinata.common.utils.StringUtils; import com.sinata.framework.web.service.SysLoginService; @@ -31,11 +32,13 @@ import io.swagger.annotations.Api; import io.swagger.annotations.ApiOperation; import org.springframework.beans.factory.annotation.Autowired; -import org.springframework.util.CollectionUtils; -import org.springframework.web.bind.annotation.*; +import org.springframework.web.bind.annotation.GetMapping; +import org.springframework.web.bind.annotation.PostMapping; +import org.springframework.web.bind.annotation.RequestBody; +import org.springframework.web.bind.annotation.RequestMapping; +import org.springframework.web.bind.annotation.RestController; import javax.annotation.Resource; -import java.time.LocalDateTime; import java.util.List; import java.util.Map; import java.util.Random; @@ -72,6 +75,8 @@ private AliSmsService aliSmsService; @Resource private ISysUserService userService; + @Autowired + private RedisCache redisCache; @PostMapping("/getCode") @@ -79,11 +84,29 @@ public AjaxResult getCode(String phone) { Random random = new Random(); - + List<SysUser> list = userService.lambdaQuery().eq(SysUser::getUserName, phone).list(); + if (list.isEmpty()){ + return AjaxResult.error("当前手机号没有对应用户,请联系管理员"); + } int i = 100000 + random.nextInt(900000); aliSmsService.sendLoginCode(phone, String.valueOf(i)); + redisCache.setCacheObject(String.valueOf(phone), String.valueOf(i)); return AjaxResult.success(); } + + + @PostMapping("/set/getCode") + @ApiOperation("获取验证码") + public AjaxResult setgetCode(String phone) + { + Random random = new Random(); + int i = 100000 + random.nextInt(900000); + aliSmsService.sendLoginCode(phone, String.valueOf(i)); + redisCache.setCacheObject(String.valueOf(phone), String.valueOf(i)); + return AjaxResult.success(); + } + + @Resource private WeChatUtil weChatUtil; @@ -101,15 +124,34 @@ String openid = map.get("openid").toString(); String sessionKey = map.get("sessionKey").toString(); - String decrypt = WXCore.decrypt(EncryptedData_phone, sessionKey, Iv_phone); if (StringUtils.isEmpty(decrypt)) { return R.fail("获取手机信息失败"); } JSONObject phone = JSON.parseObject(decrypt); String purePhoneNumber = phone.getString("purePhoneNumber"); - return R.ok(purePhoneNumber); + + // 查询手机号对应的用户 + SysUser user = userService.lambdaQuery() + .eq(SysUser::getUserName, purePhoneNumber) + .one(); + if(user == null) { + return R.fail("该手机号未注册"); + } + // 创建LoginUser对象 + LoginUser loginUser = new LoginUser(); + loginUser.setUserId(user.getUserId()); + loginUser.setUser(user); + + // 生成token + String token = tokenService.createToken(loginUser); + + JSONObject result = new JSONObject(); + result.put("phone", purePhoneNumber); + result.put(Constants.TOKEN, token); + + return R.ok(result); } /** * 登录方法 @@ -117,27 +159,55 @@ * @param loginBody 登录信息 * @return 结果 */ +// @PostMapping("/login") +// @ApiOperation("验证码登录") +// public AjaxResult login(@RequestBody LoginBody loginBody) +// { +// AjaxResult ajax = AjaxResult.success(); +// // 生成令牌 +// LoginUser loginUser = loginService.login1(loginBody.getUsername(), loginBody.getPassword(), loginBody.getCode(), +// loginBody.getUuid()); +// ajax.put(Constants.TOKEN, tokenService.createToken(loginUser)); +// // List<SysRole> roles = loginUser.getUser().getRoles(); +// // if (CollectionUtils.isEmpty(roles)) { +// // return AjaxResult.error("请关联角色!"); +// // } + +// List<SysMenu> menus = roleService.roleInfoFromUserId(loginUser.getUserId()); +// String string = redisCache.getCacheObject(loginBody.getUsername()).toString(); +// if (StringUtils.isBlank(string)) { +// return AjaxResult.error("验证码已失效"); +// } +// if (!loginBody.getCode().equals(string)){ +// return AjaxResult.error("验证码错误,请重试"); +// } +// // ajax.put("menus", menus); +// ajax.put("userInfo", loginUser); +// return ajax; +// } + @PostMapping("/login") - @ApiOperation("验证码登录") + @ApiOperation("管理后台登录") public AjaxResult login(@RequestBody LoginBody loginBody) { AjaxResult ajax = AjaxResult.success(); // 生成令牌 - LoginUser loginUser = loginService.login1(loginBody.getUsername(), loginBody.getPassword(), loginBody.getCode(), + LoginUser loginUser = loginService.login(loginBody.getUsername(), loginBody.getPassword(), loginBody.getCode(), loginBody.getUuid()); ajax.put(Constants.TOKEN, tokenService.createToken(loginUser)); -// List<SysRole> roles = loginUser.getUser().getRoles(); -// if (CollectionUtils.isEmpty(roles)) { -// return AjaxResult.error("请关联角色!"); -// } - + List<SysRole> roles = loginUser.getUser().getRoles(); + if (CollectionUtils.isEmpty(roles)) { + return AjaxResult.error("请关联角色!"); + } List<SysMenu> menus = roleService.roleInfoFromUserId(loginUser.getUserId()); - -// ajax.put("menus", menus); + Set<Long> permissions = roleService.getMenuIdSets(loginUser.getUserId()); + ajax.put("menus", menus); ajax.put("userInfo", loginUser); + ajax.put("permissions", permissions); return ajax; } + @PostMapping("/get/department") @ApiOperation("登录后获取部门") public R<SysDepartment> department() -- Gitblit v1.7.1