diff --git a/src/main/java/io/linfeng/modules/admin/service/impl/AppUserServiceImpl.java b/src/main/java/io/linfeng/modules/admin/service/impl/AppUserServiceImpl.java index 30692a0..cc6a80a 100644 --- a/src/main/java/io/linfeng/modules/admin/service/impl/AppUserServiceImpl.java +++ b/src/main/java/io/linfeng/modules/admin/service/impl/AppUserServiceImpl.java @@ -143,11 +143,20 @@ public class AppUserServiceImpl extends ServiceImpl i return response; } + /** + * 注册/登录 + * @param form 手机验证码登录dto + * @param request + * @return 用户ID + */ @Override public Integer smsLogin(SmsLoginForm form, HttpServletRequest request) { AppUserEntity appUserEntity = this.lambdaQuery().eq(AppUserEntity::getMobile, form.getMobile()).one(); String codeKey = "code_" + form.getMobile(); String s = redisUtils.get(codeKey); + if (io.linfeng.common.utils.ObjectUtil.isEmpty(s)) { + throw new LinfengException("请先发送验证码"); + } if (!s.equals(form.getCode())) { throw new LinfengException("验证码错误"); } @@ -170,7 +179,13 @@ public class AppUserServiceImpl extends ServiceImpl i list.add("新人"); appUser.setTagStr(JSON.toJSONString(list)); baseMapper.insert(appUser); - AppUserEntity user = this.lambdaQuery().eq(AppUserEntity::getMobile, form.getMobile()).one(); + AppUserEntity user = this.lambdaQuery() + .eq(AppUserEntity::getMobile, form.getMobile()) + .one(); + if(ObjectUtil.isNull(user)){ + throw new LinfengException("注册失败"); + } + //其他业务处理 return user.getUid(); } @@ -329,6 +344,10 @@ public class AppUserServiceImpl extends ServiceImpl i appUser.setTagStr(JSON.toJSONString(list)); baseMapper.insert(appUser); AppUserEntity users = this.lambdaQuery().eq(AppUserEntity::getOpenid, openId).one(); + if(ObjectUtil.isNull(users)){ + throw new LinfengException("注册失败"); + } + //其他业务todo return users.getUid(); } }