最美POETIZE个人博客系统源码

package com.ld.poetry.service.impl; import cn.hutool.crypto.SecureUtil; import com.alibaba.fastjson.JSON; import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import com.ld.poetry.config.PoetryResult; import com.ld.poetry.constants.CommonConst; import com.ld.poetry.dao.UserMapper; import com.ld.poetry.entity.User; import com.ld.poetry.entity.WebInfo; import com.ld.poetry.entity.WeiYan; import com.ld.poetry.enums.PoetryEnum; import com.ld.poetry.handle.PoetryRuntimeException; import com.ld.poetry.im.http.dao.ImChatGroupUserMapper; import com.ld.poetry.im.http.dao.ImChatUserFriendMapper; import com.ld.poetry.im.http.entity.ImChatGroupUser; import com.ld.poetry.im.http.entity.ImChatUserFriend; import com.ld.poetry.im.websocket.ImConfigConst; import com.ld.poetry.im.websocket.TioUtil; import com.ld.poetry.im.websocket.TioWebsocketStarter; import com.ld.poetry.service.UserService; import com.ld.poetry.service.WeiYanService; import com.ld.poetry.utils.*; import com.ld.poetry.utils.cache.PoetryCache; import com.ld.poetry.utils.mail.MailUtil; import com.ld.poetry.vo.BaseRequestVO; import com.ld.poetry.vo.UserVO; import lombok.extern.slf4j.Slf4j; import org.springframework.beans.BeanUtils; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Value; import org.springframework.stereotype.Service; import org.springframework.util.CollectionUtils; import org.springframework.util.DigestUtils; import org.springframework.util.StringUtils; import org.tio.core.Tio; import java.nio.charset.StandardCharsets; import java.util.ArrayList; import java.util.List; import java.util.Random; import java.util.UUID; import java.util.concurrent.atomic.AtomicInteger; import java.util.stream.Collectors; /** * <p> * 用户信息表 服务实现类 * </p> * * @author sara * @since 2021-08-12 */ @Service @Slf4j public class UserServiceImpl extends ServiceImpl<UserMapper, User> implements UserService { @Autowired private WeiYanService weiYanService; @Autowired private ImChatGroupUserMapper imChatGroupUserMapper; @Autowired private ImChatUserFriendMapper imChatUserFriendMapper; @Autowired private MailUtil mailUtil; @Value("${user.code.format}") private String codeFormat; @Override public PoetryResult<UserVO> login(String account, String password, Boolean isAdmin) { password = new String(SecureUtil.aes(CommonConst.CRYPOTJS_KEY.getBytes(StandardCharsets.UTF_8)).decrypt(password)); User one = lambdaQuery().and(wrapper -> wrapper .eq(User::getUsername, account) .or() .eq(User::getEmail, account) .or() .eq(User::getPhoneNumber, account)) .eq(User::getPassword, DigestUtils.md5DigestAsHex(password.getBytes())) .one(); if (one == null) { return PoetryResult.fail("账号/密码错误，请重新输入！"); } if (!one.getUserStatus()) { return PoetryResult.fail("账号被冻结！"); } String adminToken = ""; String userToken = ""; if (isAdmin) { if (one.getUserType() != PoetryEnum.USER_TYPE_ADMIN.getCode() && one.getUserType() != PoetryEnum.USER_TYPE_DEV.getCode()) { return PoetryResult.fail("请输入管理员账号！"); } if (PoetryCache.get(CommonConst.ADMIN_TOKEN + one.getId()) != null) { adminToken = (String) PoetryCache.get(CommonConst.ADMIN_TOKEN + one.getId()); } } else { if (PoetryCache.get(CommonConst.USER_TOKEN + one.getId()) != null) { userToken = (String) PoetryCache.get(CommonConst.USER_TOKEN + one.getId()); } } if (isAdmin && !StringUtils.hasText(adminToken)) { String uuid = UUID.randomUUID().toString().replaceAll("-", ""); adminToken = CommonConst.ADMIN_ACCESS_TOKEN + uuid; PoetryCache.put(adminToken, one, CommonConst.TOKEN_EXPIRE); PoetryCache.put(CommonConst.ADMIN_TOKEN + one.getId(), adminToken, CommonConst.TOKEN_EXPIRE); } else if (!isAdmin && !StringUtils.hasText(userToken)) { String uuid = UUID.randomUUID().toString().replaceAll("-", ""); userToken = CommonConst.USER_ACCESS_TOKEN + uuid; PoetryCache.put(userToken, one, CommonConst.TOKEN_EXPIRE); PoetryCache.put(CommonConst.USER_TOKEN + one.getId(), userToken, CommonConst.TOKEN_EXPIRE); } UserVO userVO = new UserVO(); BeanUtils.copyProperties(one, userVO); userVO.setPassword(null); if (isAdmin && one.getUserType() == PoetryEnum.USER_TYPE_ADMIN.getCode()) { userVO.setIsBoss(true); } if (isAdmin) { userVO.setAccessToken(adminToken); } else { userVO.setAccessToken(userToken); } return PoetryResult.success(userVO); } @Override public PoetryResult exit() { String token = PoetryUtil.getToken(); Integer userId = PoetryUtil.getUserId(); if (token.contains(CommonConst.USER_ACCESS_TOKEN)) { PoetryCache.remove(CommonConst.USER_TOKEN + userId); TioWebsocketStarter tioWebsocketStarter = TioUtil.getTio(); if (tioWebsocketStarter != null) { Tio.removeUser(tioWebsocketStarter.getServerTioConfig(), String.valueOf(userId), "remove user"); } } else if (token.contains(CommonConst.ADMIN_ACCESS_TOKEN)) { PoetryCache.remove(CommonConst.ADMIN_TOKEN + userId); } PoetryCache.remove(token); return PoetryResult.success(); } @Override public PoetryResult<UserVO> regist(UserVO user) { String regex = "\\d{11}"; if (user.getUsername().matches(regex)) { return PoetryResult.fail("用户名不能为11位数字！"); } if (user.getUsername().contains("@")) { return PoetryResult.fail("用户名不能包含@！"); } if (StringUtils.hasText(user.getPhoneNumber()) && StringUtils.hasText(user.getEmail())) { return PoetryResult.fail("手机号与邮箱只能选择其中一个！"); } if (StringUtils.hasText(user.getPhoneNumber())) { Integer codeCache = (Integer) PoetryCache.get(CommonConst.FORGET_PASSWORD + user.getPhoneNumber() + "_1"); if (codeCache == null || codeCache != Integer.parseInt(user.getCode())) { return PoetryResult.fail("验证码错误！"); } PoetryCache.remove(CommonConst.FORGET_PASSWORD + user.getPhoneNumber() + "_1"); } else if (StringUtils.hasText(user.getEmail())) { Integer codeCache = (Integer) PoetryCache.get(CommonConst.FORGET_PASSWORD + user.getEmail() + "_2"); if (codeCache == null || codeCache != Integer.parseInt(user.getCode())) { return PoetryResult.fail("验证码错误！"); } PoetryCache.remove(CommonConst.FORGET_PASSWORD + user.getEmail() + "_2"); } else { return PoetryResult.fail("请输入邮箱或手机号！"); } user.setPassword(new String(SecureUtil.aes(CommonConst.CRYPOTJS_KEY.getBytes(StandardCharsets.UTF_8)).decrypt(user.getPassword()))); Integer count = lambdaQuery().eq(User::getUsername, user.getUsername()).count(); if (count != 0) { return PoetryResult.fail("用户名重复！"); } if (StringUtils.hasText(user.getPhoneNumber())) { Integer phoneNumberCount = lambdaQuery().eq(User::getPhoneNumber, user.getPhoneNumber()).count(); if