添加账号注销功能

bookshelfplus/src/main/java/plus/bookshelf/Controller/Controller/UserController.java

@@ -56,7 +56,7 @@ public class UserController extends BaseController {
         }
         String encryptPwd = DigestUtils.sha1Hex(password);
 
-        if(!userService.userRegister(username, encryptPwd)){
+        if (!userService.userRegister(username, encryptPwd)) {
             throw new BusinessException(BusinessErrorCode.UNKNOWN_ERROR, "未知错误，注册失败");
         }
         // 注册成功后，进行登录
@@ -91,6 +91,35 @@ public class UserController extends BaseController {
         return CommonReturnType.create(userVO);
     }
 
+    @ApiOperation(value = "账号注销", notes = "传入用户 token ，以及密码明文，后台计算密码SHA1值，进行注销")
+    @RequestMapping(value = "cancelAccount", method = {RequestMethod.POST}, consumes = {CONTENT_TYPE_FORMED})
+    @ResponseBody
+    public CommonReturnType cancelAccount(@RequestParam(value = "token", required = false) String token,
+                                          @RequestParam(value = "password", required = false) String password) throws BusinessException {
+        // 已经在 getUserByToken 方法中判断了 token 为空、不合法；用户不存在情况，此处无需再判断
+        UserModel userModel = userService.getUserByToken(redisTemplate, token);
+
+        if(password == null || "".equals(password)){
+            throw new BusinessException(BusinessErrorCode.PARAMETER_VALIDATION_ERROR, "参数不合法，缺少密码");
+        }
+
+        // 验证用户密码是否输入正确
+        if (!userModel.getEncriptPwd().equals(DigestUtils.sha1Hex(password))) {
+            throw new BusinessException(BusinessErrorCode.PARAMETER_VALIDATION_ERROR, "密码错误");
+        }
+
+        // 进行用户账号注销流程
+        Boolean isSuccess = userService.cancelAccount(userModel);
+        if (!isSuccess) {
+            throw new BusinessException(BusinessErrorCode.UNKNOWN_ERROR, "系统未知错误，注销失败，请联系网站管理员");
+        }
+
+        // 用户退出登录
+        onLogout(token);
+
+        return CommonReturnType.create("success");
+    }
+
     public static UserVO convertFromService(UserModel userModel) {
         if (userModel == null) {
             return null;

bookshelfplus/src/main/java/plus/bookshelf/Service/Impl/ThirdPartyUserServiceImpl.java

@@ -22,7 +22,6 @@ import plus.bookshelf.Service.Service.ThirdPartyUserService;
 import java.lang.reflect.InvocationTargetException;
 import java.util.ArrayList;
 import java.util.List;
-import java.util.Locale;
 
 @Service
 public class ThirdPartyUserServiceImpl implements ThirdPartyUserService {

bookshelfplus/src/main/java/plus/bookshelf/Service/Impl/UserServiceImpl.java

@@ -10,10 +10,14 @@ import plus.bookshelf.Common.SessionManager.RedisSessionManager;
 import plus.bookshelf.Controller.VO.UserVO;
 import plus.bookshelf.Dao.DO.ThirdPartyUserDO;
 import plus.bookshelf.Dao.DO.UserDO;
+import plus.bookshelf.Dao.Mapper.ThirdPartyUserAuthDOMapper;
+import plus.bookshelf.Dao.Mapper.ThirdPartyUserDOMapper;
 import plus.bookshelf.Dao.Mapper.UserDOMapper;
 import plus.bookshelf.Service.Model.UserModel;
 import plus.bookshelf.Service.Service.UserService;
 
+import java.util.ArrayList;
+import java.util.List;
 import java.util.Map;
 
 @Service
@@ -22,6 +26,12 @@ public class UserServiceImpl implements UserService {
     @Autowired
     UserDOMapper userDOMapper;
 
+    @Autowired
+    ThirdPartyUserDOMapper thirdPartyUserDOMapper;
+
+    @Autowired
+    ThirdPartyUserAuthDOMapper thirdPartyUserAuthDOMapper;
+
     @Override
     public UserModel userLogin(String username, String encryptPwd) {
         UserDO userDO = userDOMapper.selectByUsernameAndEncryptpwd(username, encryptPwd);
@@ -79,6 +89,44 @@ public class UserServiceImpl implements UserService {
         return userDOMapper.insertSelective(userDO) > 0;
     }
 
+    @Override
+    @Transactional
+    public Boolean cancelAccount(UserModel userModel) throws BusinessException {
+
+        // 用户Id
+        Integer userId = userModel.getId();
+
+        ThirdPartyUserDO[] userBindThirdParties = thirdPartyUserDOMapper.getUserBindThirdParties(userModel.getId());
+
+        List<Integer> thirdPartyIds = new ArrayList<>();
+        for (ThirdPartyUserDO thirdPartyUserDO : userBindThirdParties) {
+            Integer thirdPartyUserId = thirdPartyUserDO.getId();
+            // 删除第三方账号与用户的关联
+            // 首先在 Auth 表中删除
+            int affectRows = thirdPartyUserAuthDOMapper.deleteByUserIdAndThirdPartyUserId(userId, thirdPartyUserId);
+            if (affectRows == 0) {
+                // 删除失败
+                return false;
+            }
+            // 第三方账号与用户关联删除成功，删除第三方账号信息
+            int affectRows2 = thirdPartyUserDOMapper.deleteByPrimaryKey(thirdPartyUserId);
+            if (affectRows2 == 0) {
+                // 删除失败
+                return false;
+            }
+        }
+
+        // 删除用户信息
+        int affectRows3 = userDOMapper.deleteByPrimaryKey(userModel.getId());
+        if (affectRows3 == 0) {
+            // 删除失败
+            return false;
+        }
+
+        // 注销成功
+        return true;
+    }
+
     private UserModel convertFromDataObject(UserDO userDO) {
         if (userDO == null) {
             return null;

bookshelfplus/src/main/java/plus/bookshelf/Service/Service/UserService.java

@@ -38,4 +38,13 @@ public interface UserService {
      * @return 注册成功返回true，否则返回false
      */
     Boolean userRegister(String username, String encryptPwd) throws BusinessException;
+
+    /**
+     * 账号注销
+     *
+     * @param userModel
+     * @return
+     * @throws BusinessException
+     */
+    Boolean cancelAccount(UserModel userModel) throws BusinessException;
 }