feat: 分库分表
This commit is contained in:
胥员员
@@ -1,81 +0,0 @@
|
||||
package com.yupi.project.service;
|
||||
|
||||
import com.baomidou.mybatisplus.extension.service.IService;
|
||||
import com.yupi.project.model.entity.User;
|
||||
import javax.servlet.http.HttpServletRequest;
|
||||
|
||||
/**
|
||||
* 用户服务
|
||||
*
|
||||
* @author yupi
|
||||
*/
|
||||
public interface UserService extends IService<User> {
|
||||
|
||||
/**
|
||||
* 用户注册
|
||||
*
|
||||
* @param userAccount 用户账户
|
||||
* @param userPassword 用户密码
|
||||
* @param checkPassword 校验密码
|
||||
* @return 新用户 id
|
||||
*/
|
||||
long userRegister(String userAccount, String userPassword, String checkPassword);
|
||||
|
||||
/**
|
||||
* 用户登录
|
||||
*
|
||||
* @param userAccount 用户账户
|
||||
* @param userPassword 用户密码
|
||||
* @param request
|
||||
* @return 脱敏后的用户信息
|
||||
*/
|
||||
User userLogin(String userAccount, String userPassword, HttpServletRequest request);
|
||||
|
||||
/**
|
||||
* 用户脱敏
|
||||
*
|
||||
* @param originUser
|
||||
* @return
|
||||
*/
|
||||
User getSafetyUser(User originUser);
|
||||
|
||||
/**
|
||||
* 用户注销
|
||||
*
|
||||
* @param request
|
||||
* @return
|
||||
*/
|
||||
int userLogout(HttpServletRequest request);
|
||||
|
||||
/**
|
||||
* 是否为管理员
|
||||
*
|
||||
* @param request
|
||||
* @return
|
||||
*/
|
||||
boolean isAdmin(HttpServletRequest request);
|
||||
|
||||
/**
|
||||
* 是否为管理员
|
||||
*
|
||||
* @param user
|
||||
* @return
|
||||
*/
|
||||
boolean isAdmin(User user);
|
||||
|
||||
/**
|
||||
* 断言是管理员
|
||||
*
|
||||
* @param request
|
||||
*/
|
||||
void assertAdmin(HttpServletRequest request);
|
||||
|
||||
/**
|
||||
* 获取登录用户（查缓存）
|
||||
*
|
||||
* @param request
|
||||
* @return
|
||||
* @throws com.yupi.project.exception.BusinessException 未登录则抛异常
|
||||
*/
|
||||
User getLoginUser(HttpServletRequest request);
|
||||
}
|
||||
@@ -1,189 +0,0 @@
|
||||
package com.yupi.project.service.impl;
|
||||
|
||||
import static com.yupi.project.constant.UserConstant.USER_LOGIN_STATE;
|
||||
|
||||
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
||||
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
||||
import com.yupi.project.common.ErrorCode;
|
||||
import com.yupi.project.constant.UserConstant;
|
||||
import com.yupi.project.exception.BusinessException;
|
||||
import com.yupi.project.mapper.UserMapper;
|
||||
import com.yupi.project.model.entity.User;
|
||||
import com.yupi.project.service.UserService;
|
||||
import java.util.regex.Matcher;
|
||||
import java.util.regex.Pattern;
|
||||
import javax.annotation.Resource;
|
||||
import javax.servlet.http.HttpServletRequest;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.apache.commons.lang3.StringUtils;
|
||||
import org.springframework.stereotype.Service;
|
||||
import org.springframework.util.DigestUtils;
|
||||
|
||||
/**
|
||||
* 用户服务实现类
|
||||
*
|
||||
* @author yupi
|
||||
*/
|
||||
@Service
|
||||
@Slf4j
|
||||
public class UserServiceImpl extends ServiceImpl<UserMapper, User>
|
||||
implements UserService {
|
||||
|
||||
@Resource
|
||||
private UserMapper userMapper;
|
||||
|
||||
/**
|
||||
* 盐值，混淆密码
|
||||
*/
|
||||
private static final String SALT = "yupi";
|
||||
|
||||
@Override
|
||||
public long userRegister(String userAccount, String userPassword, String checkPassword) {
|
||||
// 1. 校验
|
||||
if (StringUtils.isAnyBlank(userAccount, userPassword, checkPassword)) {
|
||||
throw new BusinessException(ErrorCode.PARAMS_ERROR, "参数为空");
|
||||
}
|
||||
if (userAccount.length() < 4) {
|
||||
throw new BusinessException(ErrorCode.PARAMS_ERROR, "用户账号过短");
|
||||
}
|
||||
if (userPassword.length() < 8 || checkPassword.length() < 8) {
|
||||
throw new BusinessException(ErrorCode.PARAMS_ERROR, "用户密码过短");
|
||||
}
|
||||
// 账户不能包含特殊字符
|
||||
String validPattern = "[`~!@#$%^&*()+=|{}':;',\\\\[\\\\].<>/?~！@#￥%……&*（）——+|{}【】‘；：”“’。，、？]";
|
||||
Matcher matcher = Pattern.compile(validPattern).matcher(userAccount);
|
||||
if (matcher.find()) {
|
||||
return -1;
|
||||
}
|
||||
// 密码和校验密码相同
|
||||
if (!userPassword.equals(checkPassword)) {
|
||||
return -1;
|
||||
}
|
||||
// 账户不能重复
|
||||
QueryWrapper<User> queryWrapper = new QueryWrapper<>();
|
||||
queryWrapper.eq("userAccount", userAccount);
|
||||
long count = userMapper.selectCount(queryWrapper);
|
||||
if (count > 0) {
|
||||
throw new BusinessException(ErrorCode.PARAMS_ERROR, "账号重复");
|
||||
}
|
||||
// 2. 加密
|
||||
String encryptPassword = DigestUtils.md5DigestAsHex((SALT + userPassword).getBytes());
|
||||
// 3. 插入数据
|
||||
User user = new User();
|
||||
user.setUserAccount(userAccount);
|
||||
user.setUserPassword(encryptPassword);
|
||||
boolean saveResult = this.save(user);
|
||||
if (!saveResult) {
|
||||
return -1;
|
||||
}
|
||||
return user.getId();
|
||||
}
|
||||
|
||||
@Override
|
||||
public User userLogin(String userAccount, String userPassword, HttpServletRequest request) {
|
||||
// 1. 校验
|
||||
if (StringUtils.isAnyBlank(userAccount, userPassword)) {
|
||||
return null;
|
||||
}
|
||||
if (userAccount.length() < 4) {
|
||||
return null;
|
||||
}
|
||||
if (userPassword.length() < 8) {
|
||||
return null;
|
||||
}
|
||||
// 账户不能包含特殊字符
|
||||
String validPattern = "[`~!@#$%^&*()+=|{}':;',\\\\[\\\\].<>/?~！@#￥%……&*（）——+|{}【】‘；：”“’。，、？]";
|
||||
Matcher matcher = Pattern.compile(validPattern).matcher(userAccount);
|
||||
if (matcher.find()) {
|
||||
return null;
|
||||
}
|
||||
// 2. 加密
|
||||
String encryptPassword = DigestUtils.md5DigestAsHex((SALT + userPassword).getBytes());
|
||||
// 查询用户是否存在
|
||||
QueryWrapper<User> queryWrapper = new QueryWrapper<>();
|
||||
queryWrapper.eq("userAccount", userAccount);
|
||||
queryWrapper.eq("userPassword", encryptPassword);
|
||||
User user = userMapper.selectOne(queryWrapper);
|
||||
// 用户不存在
|
||||
if (user == null) {
|
||||
log.info("user login failed, userAccount cannot match userPassword");
|
||||
return null;
|
||||
}
|
||||
// 3. 用户脱敏
|
||||
User safetyUser = getSafetyUser(user);
|
||||
// 4. 记录用户的登录态
|
||||
request.getSession().setAttribute(USER_LOGIN_STATE, safetyUser);
|
||||
return safetyUser;
|
||||
}
|
||||
|
||||
/**
|
||||
* 用户脱敏
|
||||
*
|
||||
* @param originUser
|
||||
* @return
|
||||
*/
|
||||
@Override
|
||||
public User getSafetyUser(User originUser) {
|
||||
if (originUser == null) {
|
||||
return null;
|
||||
}
|
||||
User safetyUser = new User();
|
||||
safetyUser.setId(originUser.getId());
|
||||
safetyUser.setUsername(originUser.getUsername());
|
||||
safetyUser.setUserAccount(originUser.getUserAccount());
|
||||
safetyUser.setAvatarUrl(originUser.getAvatarUrl());
|
||||
safetyUser.setGender(originUser.getGender());
|
||||
safetyUser.setPhone(originUser.getPhone());
|
||||
safetyUser.setEmail(originUser.getEmail());
|
||||
safetyUser.setUserRole(originUser.getUserRole());
|
||||
safetyUser.setUserStatus(originUser.getUserStatus());
|
||||
safetyUser.setCreateTime(originUser.getCreateTime());
|
||||
return safetyUser;
|
||||
}
|
||||
|
||||
/**
|
||||
* 用户注销
|
||||
*
|
||||
* @param request
|
||||
*/
|
||||
@Override
|
||||
public int userLogout(HttpServletRequest request) {
|
||||
// 移除登录态
|
||||
request.getSession().removeAttribute(USER_LOGIN_STATE);
|
||||
return 1;
|
||||
}
|
||||
|
||||
@Override
|
||||
public boolean isAdmin(HttpServletRequest request) {
|
||||
Object userObj = request.getSession().getAttribute(USER_LOGIN_STATE);
|
||||
User user = (User) userObj;
|
||||
return user != null && user.getUserRole() == UserConstant.ADMIN_ROLE;
|
||||
}
|
||||
|
||||
@Override
|
||||
public boolean isAdmin(User user) {
|
||||
return user != null && user.getUserRole() == UserConstant.ADMIN_ROLE;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void assertAdmin(HttpServletRequest request) {
|
||||
if (!isAdmin(request)) {
|
||||
throw new BusinessException(ErrorCode.NO_AUTH);
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public User getLoginUser(HttpServletRequest request) {
|
||||
Object userObj = request.getSession().getAttribute(USER_LOGIN_STATE);
|
||||
User loginUser = (User) userObj;
|
||||
if (loginUser == null) {
|
||||
throw new BusinessException(ErrorCode.NOT_LOGIN);
|
||||
}
|
||||
return loginUser;
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user