1.添加邮箱登录，注册接口

2025-08-04 21:58:55 +08:00
parent 715e6a1b70
commit f5cbe5cac2
17 changed files with 530 additions and 22 deletions
--- a/src/main/java/vvpkassistant/User/service/UserServiceImpl.java
+++ b/src/main/java/vvpkassistant/User/service/UserServiceImpl.java
@@ -0,0 +1,128 @@
+package vvpkassistant.User.service;
+
+import cn.dev33.satoken.stp.StpUtil;
+import cn.hutool.core.bean.BeanUtil;
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Service;
+import vvpkassistant.Data.WxChatParam;
+import vvpkassistant.Tools.BcryptUtils;
+import vvpkassistant.Tools.VVTools;
+import vvpkassistant.User.mapper.UserDao;
+import vvpkassistant.User.model.DTO.UserModelDTO;
+import vvpkassistant.User.model.UserModel;
+import vvpkassistant.User.model.UserModelVO;
+import vvpkassistant.common.ErrorCode;
+import vvpkassistant.exception.BusinessException;
+import vvpkassistant.mail.model.MailModel;
+
+import javax.annotation.Resource;
+
+/*
+ * @author: ziin
+ * @date: 2025/8/4 16:19
+ */
+@Service
+public class UserServiceImpl extends ServiceImpl<UserDao, UserModel> implements UserService {
+
+    @Resource
+    private UserDao userDao;
+
+    @Resource
+    private WxChatParam wxChatParam;
+
+    @Override
+    public UserModelVO loginWithMail(UserModelDTO model) {
+
+        LambdaQueryWrapper<UserModel> lambdaQueryWrapper = new LambdaQueryWrapper<>();
+        lambdaQueryWrapper.eq(UserModel::getEmail,model.getEmail())
+                .in(UserModel::getStatus, 0,2);
+
+        UserModel userModel = userDao.selectOne(lambdaQueryWrapper);
+        if (userModel == null) {
+            throw new BusinessException(ErrorCode.USER_DOES_NOT_EXIST);
+        }
+
+        String password = userModel.getPassword();
+        UserModelVO userModelVO = BeanUtil.copyProperties(userModel, UserModelVO.class);
+        if (BcryptUtils.matchPassword(password, model.getPassword())) {
+            StpUtil.login(userModel.getId());
+            userModelVO.setToken(StpUtil.getTokenValue());
+            userModelVO.setChatInfo(wxChatParam);
+            return userModelVO;
+        }else {
+            throw new BusinessException(ErrorCode.PASSWORD_ERROR);
+        }
+    }
+
+    @Override
+    public UserModelVO updateUserInfo(UserModelDTO userModelDTO) {
+
+        UserModel userInfo = userDao.selectById(userModelDTO.getId());
+        if (userInfo == null) {
+            throw new BusinessException(ErrorCode.USER_DOES_NOT_EXIST);
+        }
+//        用户没有密码的情况下设置密码
+        if (userInfo.getPassword() == null) {
+            if (!userModelDTO.getNewPassword().isEmpty()){
+                if (userModelDTO.getNewPassword().length()<6){
+                    throw new BusinessException(ErrorCode.PARAMS_ERROR,"密码长度不能小于 6 位");
+                }
+                userModelDTO.setPassword(BcryptUtils.encryptPassword(userModelDTO.getNewPassword()));
+            }
+        }
+//        用户有密码的情况下重新设置密码
+        if (userInfo.getPassword() != null && userModelDTO.getOldPassword() != null) {
+            if (BcryptUtils.matchPassword(userModelDTO.getOldPassword(), userInfo.getPassword())) {
+                userModelDTO.setPassword(BcryptUtils.encryptPassword(userModelDTO.getNewPassword()));
+            }else {
+                throw new BusinessException(ErrorCode.PASSWORD_ERROR,"旧密码不正确");
+            }
+        }
+
+        UserModel userModel = new UserModel();
+        BeanUtil.copyProperties(userModelDTO, userModel);
+        int i = userDao.updateById(userModel);
+        // 返回结果
+        userDao.selectById(userModel.getId());
+        UserModelVO userModelVO = BeanUtil.copyProperties(userModel, UserModelVO.class);
+        userModelVO.setNewAccount(false);
+        if (i == 1){
+           return userModelVO;
+        }else {
+            throw  new BusinessException(ErrorCode.SYSTEM_ERROR);
+        }
+    }
+
+    @Override
+    public UserModelVO addUserWithMail(UserModelDTO model) {
+
+        LambdaQueryWrapper<UserModel> lambdaQueryWrapper = new LambdaQueryWrapper<>();
+        lambdaQueryWrapper.eq(UserModel::getEmail,model.getEmail());
+        UserModel userModel = userDao.selectOne(lambdaQueryWrapper);
+        if (userModel != null) {
+            throw new BusinessException(ErrorCode.MAIL_ALREADY_EXIST);
+        }
+        if (model.getPassword().length() < 6 ){
+            throw new BusinessException(ErrorCode.PARAMS_ERROR,"密码长度不能小于 6 位");
+        }
+        model.setPassword(BcryptUtils.encryptPassword(model.getPassword()));
+        model.setCreateTime(VVTools.currentTimeStamp());
+        //设置状态为正常
+        model.setStatus(2);
+        //设置积分为0
+        model.setPoints(0);
+        userDao.insert(BeanUtil.copyProperties(model, UserModel.class));
+        // 判断用户是否为邀请用户
+        if (model.getInviterId() != null) {
+            UserModel oldUser = userDao.selectById(model.getInviterId());
+            oldUser.setPoints(oldUser.getPoints() + 10);
+            userDao.updateById(oldUser);
+        }
+        UserModelVO userModelVO = BeanUtil.copyProperties(model, UserModelVO.class);
+        userModelVO.setNewAccount(true);
+        userModelVO.setChatInfo(wxChatParam);
+        return userModelVO;
+    }
+}