第一步版

This commit is contained in:
2025-07-22 11:30:53 +08:00
commit 7bf8058c39
45 changed files with 2154 additions and 0 deletions

View File

@ -0,0 +1,15 @@
package org.traceability;
import org.mybatis.spring.annotation.MapperScan;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
@SpringBootApplication
@MapperScan("org.traceability.mapper")
public class TraceabilityApplication {
public static void main(String[] args) {
SpringApplication.run(TraceabilityApplication.class, args);
}
}

View File

@ -0,0 +1,42 @@
package org.traceability.common;
import io.swagger.v3.oas.annotations.media.Schema;
import lombok.Data;
import java.io.Serializable;
@Data
@Schema(description = "通用返回对象")
public class BaseResponse<T> implements Serializable {
@Schema(description = "状态码例如200 表示成功")
private int code;
@Schema(description = "返回的数据内容")
private T data;
@Schema(description = "提示消息")
private String message;
@Schema(description = "错误描述")
private String description;
public BaseResponse(int code, T data, String message, String description) {
this.code = code;
this.data = data;
this.message = message;
this.description = description;
}
public BaseResponse(int code, T data, String message) {
this(code, data, message, "");
}
public BaseResponse(int code, T data) {
this(code, data, "", "");
}
public BaseResponse(ErrorCode errorCode) {
this(errorCode.getCode(), null , errorCode.getMessage(), errorCode.getDescription());
}
}

View File

@ -0,0 +1,49 @@
package org.traceability.common;
/**
* 错误码
*/
public enum ErrorCode {
SUCCESS(0, "ok", ""),
PARAMS_ERROR(40000, "请求参数错误", ""),
NULL_ERROR(40001, "请求数据为空", ""),
NOT_LOGIN(40100, "未登录", ""),
NO_AUTH(40101, "无权限", ""),
NOT_FOUND_ERROR(40400,"请求数据不存在",""),
FORBIDDEN_ERROR(40300, "禁止访问",""),
SYSTEM_ERROR(50000, "系统内部异常", ""),
OPERATION_ERROR(50001, "操作失败", "");
/**
* 状态码
*/
private final int code;
/**
* 状态码信息
*/
private final String message;
/**
* 状态码的详细描述
*/
private final String description;
ErrorCode(int code, String message, String description) {
this.code = code;
this.message = message;
this.description = description;
}
public int getCode() {
return code;
}
public String getMessage() {
return message;
}
public String getDescription() {
return description;
}
}

View File

@ -0,0 +1,66 @@
package org.traceability.common;
/**
* 返回工具类
*/
public class ResultUtils {
/**
* 成功
*
* @param data
* @param <T>
* @return
*/
public static <T> BaseResponse<T> success(T data) {
return new BaseResponse<>(0, data, "ok");
}
public static <T> BaseResponse<T> success(T data, String message) {
return new BaseResponse<>(0, data, message);
}
/**
* 失败
*
* @param errorCode
* @return
*/
public static BaseResponse error(ErrorCode errorCode) {
return new BaseResponse<>(errorCode);
}
/**
* 失败
*
* @param code
* @param message
* @param description
* @return
*/
public static BaseResponse error(int code, String message, String description) {
return new BaseResponse(code, null, message, description);
}
/**
* 失败
*
* @param errorCode
* @return
*/
public static BaseResponse error(ErrorCode errorCode, String message, String description) {
return new BaseResponse(errorCode.getCode(), null, message, description);
}
/**
* 失败
*
* @param errorCode
* @return
*/
public static BaseResponse error(ErrorCode errorCode, String description) {
return new BaseResponse(errorCode.getCode(), errorCode.getMessage(), description);
}
}

View File

@ -0,0 +1,28 @@
package org.traceability.config;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.servlet.config.annotation.CorsRegistry;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;
/**
* 全局跨域配置
*
*
* @author xy
*/
@Configuration
public class CorsConfig implements WebMvcConfigurer {
@Override
public void addCorsMappings(CorsRegistry registry) {
// 覆盖所有请求
registry.addMapping("/**")
// 允许发送 Cookie
.allowCredentials(true)
// 放行哪些域名(必须用 patterns否则 * 会和 allowCredentials 冲突)
.allowedOriginPatterns("*")
.allowedMethods("GET", "POST", "PUT", "DELETE", "OPTIONS")
.allowedHeaders("*")
.exposedHeaders("*");
}
}

View File

@ -0,0 +1,25 @@
package org.traceability.config;
import io.swagger.v3.oas.models.OpenAPI;
import io.swagger.v3.oas.models.info.Info;
import io.swagger.v3.oas.models.info.Contact;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
/**
* 访问链接 http://localhost:7080/swagger-ui/index.html
*/
@Configuration
public class SwaggerConfig {
@Bean
public OpenAPI customOpenAPI() {
return new OpenAPI()
.info(new Info()
.title("Traceability API")
.version("1.0.0")
.description("溯源码系统 API 文档")
.contact(new Contact()
.name("X&L")));
}
}

View File

@ -0,0 +1,18 @@
package org.traceability.contant;
/**
* 通用常量
*/
public interface CommonConstant {
/**
* 升序
*/
String SORT_ORDER_ASC = "ascend";
/**
* 降序
*/
String SORT_ORDER_DESC = " descend";
}

View File

@ -0,0 +1,28 @@
package org.traceability.contant;
/**
* 正则表达式常量
*/
@SuppressWarnings("all")
public interface RegexConstant {
/**
* 手机号正则
*/
String PHONE_REGEX = "^1([38][0-9]|4[579]|5[0-3,5-9]|6[6]|7[0135678]|9[89])\\d{8}$";
/**
* 邮箱正则
*/
String EMAIL_REGEX = "^[a-zA-Z0-9_-]+@[a-zA-Z0-9_-]+(\\.[a-zA-Z0-9_-]+)+$";
/**
* 密码正则。4~32位的字母、数字、下划线
*/
String PASSWORD_REGEX = "^\\w{4,32}$";
/**
* 特殊字符校验
*/
String SPECAIL_REGEX = "[ `~!@#$%^&*()+=|{}':;',\\\\[\\\\].<>/?~@#¥%……&*()——+|{}【】‘;:”“’。,、?]";
}

View File

@ -0,0 +1,14 @@
package org.traceability.contant;
public interface UserConstant {
/**
* 盐值
*/
String USER_SALT = "tsuk";
/**
* 登录态
*/
String USER_LOGIN_STATE = "trace_user";
}

View File

@ -0,0 +1,60 @@
package org.traceability.controller;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;
import jakarta.annotation.Resource;
import org.springframework.web.bind.annotation.*;
import org.traceability.common.BaseResponse;
import org.traceability.common.ErrorCode;
import org.traceability.common.ResultUtils;
import org.traceability.exception.BusinessException;
import org.traceability.model.domain.Roles;
import org.traceability.model.dto.role.RoleAddRequest;
import org.traceability.service.RolesService;
import java.util.List;
@Tag(name = "用户角色接口")
@RestController
@RequestMapping("/roles")
public class RolesController {
@Resource
private RolesService rolesService;
/**
* 添加角色
* @param roleAddRequest 添加角色请求体
* @return 角色id
*/
@PostMapping("/add")
@Operation(summary = "添加角色")
public BaseResponse<Long> addRole(@RequestBody RoleAddRequest roleAddRequest) {
if (roleAddRequest == null) {
throw new BusinessException(ErrorCode.PARAMS_ERROR, "当前请求体为空");
}
long role = rolesService.addRole(roleAddRequest);
return ResultUtils.success(role);
}
/**
* 删除角色
* @param code 角色编码
* @return 是否删除成功
*/
@GetMapping("/delete")
@Operation(summary = "删除角色")
public BaseResponse<Boolean> deleteRole(@RequestParam String code) {
boolean b = rolesService.delRole(code);
return ResultUtils.success(b);
}
@PostMapping("/list")
public BaseResponse<List<Roles>> listRoles() {
List<Roles> list = rolesService.list();
return ResultUtils.success(list);
}
}

View File

@ -0,0 +1,69 @@
package org.traceability.controller;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;
import jakarta.annotation.Resource;
import jakarta.servlet.http.HttpServletRequest;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.traceability.common.BaseResponse;
import org.traceability.common.ErrorCode;
import org.traceability.common.ResultUtils;
import org.traceability.exception.BusinessException;
import org.traceability.model.domain.Users;
import org.traceability.model.dto.user.UserAddRequest;
import org.traceability.model.dto.user.UserLoginRequest;
import org.traceability.model.dto.user.UserUpdateRequest;
import org.traceability.service.UsersService;
@RestController
@RequestMapping("/user")
@Tag(name = "用户接口")
@Slf4j
public class UsersController {
@Resource
private UsersService usersService;
/**
* 添加用户
* @param userAddRequest 添加用户请求体
* @return 用户id
*/
@PostMapping("/add")
@Operation(summary = "添加用户")
public BaseResponse<Long> addUser(@RequestBody UserAddRequest userAddRequest) {
if (userAddRequest == null) {
throw new BusinessException(ErrorCode.PARAMS_ERROR, "当前请求体为空");
}
Long users = usersService.addUsers(userAddRequest);
return ResultUtils.success(users);
}
/**
* 登录
* @param userLoginRequest 登录请求体
* @param request 用户请求体
* @return 登录用户
*/
@PostMapping("/login")
@Operation(summary = "登录用户")
public BaseResponse<Users> userLogin (@RequestBody UserLoginRequest userLoginRequest, HttpServletRequest request) {
if (userLoginRequest == null) {
throw new BusinessException(ErrorCode.PARAMS_ERROR, "当前请求体为空");
}
String userAccount = userLoginRequest.getUserAccount();
String password = userLoginRequest.getPassword();
Users users = usersService.loginUser(userAccount, password, request);
return ResultUtils.success(users);
}
// @PostMapping("/update")
// public BaseResponse<Users> updateUser (@RequestBody UserUpdateRequest updateRequest, HttpServletRequest) {
//
// }
}

View File

@ -0,0 +1,41 @@
package org.traceability.exception;
import lombok.Getter;
import org.traceability.common.ErrorCode;
/**
* 自定义异常类
*
*/
@Getter
public class BusinessException extends RuntimeException {
/**
* 异常码
*/
private final int code;
/**
* 描述
*/
private final String description;
public BusinessException(String message, int code, String description) {
super(message);
this.code = code;
this.description = description;
}
public BusinessException(ErrorCode errorCode) {
super(errorCode.getMessage());
this.code = errorCode.getCode();
this.description = errorCode.getDescription();
}
public BusinessException(ErrorCode errorCode, String description) {
super(errorCode.getMessage());
this.code = errorCode.getCode();
this.description = description;
}
}

View File

@ -0,0 +1,31 @@
package org.traceability.exception;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.RestControllerAdvice;
import org.traceability.common.BaseResponse;
import org.traceability.common.ErrorCode;
import org.traceability.common.ResultUtils;
/**
* 全局异常处理器
*
*/
@RestControllerAdvice
@Slf4j
public class GlobalExceptionHandler {
@ExceptionHandler(BusinessException.class)
public BaseResponse<?> businessExceptionHandler(BusinessException e) {
log.error("businessException: {}", e.getMessage(), e);
return ResultUtils.error(e.getCode(), e.getMessage(), e.getDescription());
}
@ExceptionHandler(RuntimeException.class)
public BaseResponse<?> runtimeExceptionHandler(RuntimeException e) {
log.error("runtimeException", e);
return ResultUtils.error(ErrorCode.SYSTEM_ERROR, e.getMessage(), "");
}
}

View File

@ -0,0 +1,44 @@
package org.traceability.exception;
import org.traceability.common.ErrorCode;
/**
* 抛异常工具类
*/
@SuppressWarnings("all")
public class ThrowUtils {
/**
* 条件成立则抛异常
*
* @param condition 条件
* @param runtimeException 运行时异常
*/
public static void throwIf(boolean condition, RuntimeException runtimeException) {
if (condition) {
throw runtimeException;
}
}
/**
* 条件成立则抛异常
*
* @param condition 条件
* @param errorCode 自定义异常
*/
public static void throwIf(boolean condition, ErrorCode errorCode) {
throwIf(condition, new BusinessException(errorCode));
}
/**
* 条件成立则抛异常
*
* @param condition 条件
* @param errorCode 自定义异常
* @param message 报错信息
*/
public static void throwIf(boolean condition, ErrorCode errorCode, String message) {
throwIf(condition, new BusinessException(errorCode, message));
}
}

View File

@ -0,0 +1,17 @@
package org.traceability.mapper;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import org.apache.ibatis.annotations.Mapper;
import org.traceability.model.domain.Roles;
/**
* @author xy
*/
@Mapper
public interface RolesMapper extends BaseMapper<Roles> {
}

View File

@ -0,0 +1,17 @@
package org.traceability.mapper;
import org.apache.ibatis.annotations.Mapper;
import org.traceability.model.domain.UserRoles;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
/**
* @author xy
*/
@Mapper
public interface UserRolesMapper extends BaseMapper<UserRoles> {
}

View File

@ -0,0 +1,17 @@
package org.traceability.mapper;
import org.apache.ibatis.annotations.Mapper;
import org.traceability.model.domain.Users;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
/**
* @author xy
*/
@Mapper
public interface UsersMapper extends BaseMapper<Users> {
}

View File

@ -0,0 +1,65 @@
package org.traceability.model.domain;
import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.annotation.TableField;
import com.baomidou.mybatisplus.annotation.TableId;
import com.baomidou.mybatisplus.annotation.TableName;
import java.io.Serializable;
import java.util.Date;
import java.util.List;
import com.baomidou.mybatisplus.extension.handlers.JacksonTypeHandler;
import lombok.Data;
/**
*
* @TableName roles
*/
@TableName(value ="roles")
@Data
public class Roles implements Serializable {
/**
*
*/
@TableId(type = IdType.AUTO)
private Long id;
/**
* 角色名称
*/
private String name;
/**
* 角色编码
*/
private String code;
/**
* 角色描述
*/
private String description;
/**
* 角色权限配置,如:["user:read", "user:write", "product:manage"]
*/
@TableField(typeHandler = JacksonTypeHandler.class)
private List<String> permissions;
/**
* 状态:0-禁用,1-启用
*/
private Integer status;
/**
*
*/
private Date createdTime;
/**
*
*/
private Date updatedTime;
@TableField(exist = false)
private static final long serialVersionUID = 1L;
}

View File

@ -0,0 +1,41 @@
package org.traceability.model.domain;
import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.annotation.TableField;
import com.baomidou.mybatisplus.annotation.TableId;
import com.baomidou.mybatisplus.annotation.TableName;
import java.io.Serializable;
import java.util.Date;
import lombok.Data;
/**
*
* @TableName user_roles
*/
@TableName(value ="user_roles")
@Data
public class UserRoles implements Serializable {
/**
*
*/
@TableId(type = IdType.AUTO)
private Long id;
/**
*
*/
private Long userId;
/**
*
*/
private Long roleId;
/**
*
*/
private Date createdTime;
@TableField(exist = false)
private static final long serialVersionUID = 1L;
}

View File

@ -0,0 +1,75 @@
package org.traceability.model.domain;
import com.baomidou.mybatisplus.annotation.*;
import lombok.Data;
import java.io.Serializable;
import java.util.Date;
/**
*
* @TableName xy
*/
@TableName(value ="users")
@Data
public class Users implements Serializable {
/**
*
*/
@TableId(type = IdType.AUTO)
private Long id;
/**
* 用户名
*/
private String username;
/**
* 用户账号
*/
private String userAccount;
/**
* 密码(加密)
*/
private String password;
/**
* 邮箱
*/
private String email;
/**
* 手机号
*/
private String phone;
/**
* 状态:0-禁用,1-启用
*/
private Integer status;
/**
* 最后登录时间
*/
private Date lastLoginTime;
/**
*
*/
private Date createdTime;
/**
*
*/
private Date updateTime;
/**
* 逻辑删除
*/
@TableLogic
private Integer isdelete;
@TableField(exist = false)
private static final long serialVersionUID = 1L;
}

View File

@ -0,0 +1,32 @@
package org.traceability.model.dto;
import lombok.Data;
import org.traceability.contant.CommonConstant;
/**
* 分页请求
*
*/
@Data
public class PageRequest {
/**
* 当前页号
*/
private long current = 1;
/**
* 页面大小
*/
private long pageSize = 10;
/**
* 排序字段
*/
private String sortField;
/**
* 排序顺序(默认升序)
*/
private String sortOrder = CommonConstant.SORT_ORDER_ASC;
}

View File

@ -0,0 +1,29 @@
package org.traceability.model.dto.role;
import lombok.Data;
import java.util.List;
@Data
public class RoleAddRequest {
/**
* 角色编码
*/
private String code;
/**
* 角色名称
*/
private String name;
/**
* 角色描述
*/
private String description;
/**
* 角色权限配置,如:["user:read", "user:write", "product:manage"]
*/
private List<String> permissions;
}

View File

@ -0,0 +1,37 @@
package org.traceability.model.dto.user;
import lombok.Data;
@Data
public class UserAddRequest {
/**
* 用户名
*/
private String username;
/**
* 用户账号
*/
private String userAccount;
/**
* 密码(加密)
*/
private String password;
/**
* 邮箱
*/
private String email;
/**
* 手机号
*/
private String phone;
/**
* 用户角色
*/
private String UserCode;
}

View File

@ -0,0 +1,17 @@
package org.traceability.model.dto.user;
import lombok.Data;
@Data
public class UserLoginRequest {
/**
* 用户账号
*/
private String userAccount;
/**
* 密码(加密)
*/
private String password;
}

View File

@ -0,0 +1,32 @@
package org.traceability.model.dto.user;
import lombok.Data;
@Data
public class UserUpdateRequest {
/**
* 用户名
*/
private String username;
/**
* 用户账号
*/
private String userAccount;
/**
* 密码(加密)
*/
private String password;
/**
* 邮箱
*/
private String email;
/**
* 手机号
*/
private String phone;
}

View File

@ -0,0 +1,15 @@
package org.traceability.service;
import com.baomidou.mybatisplus.extension.service.IService;
import org.traceability.model.domain.Roles;
import org.traceability.model.dto.role.RoleAddRequest;
/**
* @author xy
*/
public interface RolesService extends IService<Roles> {
long addRole(RoleAddRequest roleAddRequest);
boolean delRole(String code);
}

View File

@ -0,0 +1,11 @@
package org.traceability.service;
import org.traceability.model.domain.UserRoles;
import com.baomidou.mybatisplus.extension.service.IService;
/**
* @author xy
*/
public interface UserRolesService extends IService<UserRoles> {
}

View File

@ -0,0 +1,17 @@
package org.traceability.service;
import jakarta.servlet.http.HttpServletRequest;
import org.traceability.model.domain.Users;
import com.baomidou.mybatisplus.extension.service.IService;
import org.traceability.model.dto.user.UserAddRequest;
/**
* @author xy
*/
public interface UsersService extends IService<Users> {
Long addUsers (UserAddRequest userAddRequest);
Users loginUser (String userAccount, String password, HttpServletRequest request);
}

View File

@ -0,0 +1,65 @@
package org.traceability.service.impl;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.springframework.beans.BeanUtils;
import org.traceability.common.ErrorCode;
import org.traceability.exception.ThrowUtils;
import org.traceability.model.domain.Roles;
import org.traceability.model.dto.role.RoleAddRequest;
import org.traceability.service.RolesService;
import org.traceability.mapper.RolesMapper;
import org.springframework.stereotype.Service;
import org.traceability.utils.RegexUtils;
import java.util.List;
/**
* @author xy
*/
@Service
public class RolesServiceImpl extends ServiceImpl<RolesMapper, Roles>
implements RolesService{
@Override
public long addRole(RoleAddRequest roleAddRequest) {
Roles roles = new Roles();
String code = roleAddRequest.getCode();
String name = roleAddRequest.getName();
String description = roleAddRequest.getDescription();
List<String> permissions = roleAddRequest.getPermissions();
boolean invalid = RegexUtils.isUserAccountInvalid(code);
boolean invalid1 = RegexUtils.isUserAccountInvalid(name);
boolean invalid2 = RegexUtils.isUserAccountInvalid(description);
// 权限列表验证 - 检查每个权限字符串是否符合规范
boolean invalid3 = permissions == null ||
permissions.isEmpty();
ThrowUtils.throwIf(!invalid || !invalid1 || !invalid2 || invalid3,
ErrorCode.PARAMS_ERROR, "添加的权限不符合规范");
long count = this.count(Wrappers.<Roles>lambdaQuery()
.eq(Roles::getCode, code));
ThrowUtils.throwIf(count > 0, ErrorCode.PARAMS_ERROR, "当前角色code已重复");
BeanUtils.copyProperties(roleAddRequest, roles);
boolean save = this.save(roles);
ThrowUtils.throwIf(!save, ErrorCode.OPERATION_ERROR, "添加角色失败");
return roles.getId();
}
@Override
public boolean delRole(String code) {
boolean invalid = RegexUtils.isUserAccountInvalid(code);
ThrowUtils.throwIf(!invalid, ErrorCode.PARAMS_ERROR, "当前角色编码不符合规范");
Roles roles = this.getOne(Wrappers.<Roles>lambdaQuery()
.eq(Roles::getCode, code));
ThrowUtils.throwIf(roles == null, ErrorCode.PARAMS_ERROR, "当前角色编码不正确");
return this.removeById(roles);
}
}

View File

@ -0,0 +1,20 @@
package org.traceability.service.impl;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.traceability.model.domain.UserRoles;
import org.traceability.service.UserRolesService;
import org.traceability.mapper.UserRolesMapper;
import org.springframework.stereotype.Service;
/**
* @author xy
*/
@Service
public class UserRolesServiceImpl extends ServiceImpl<UserRolesMapper, UserRoles>
implements UserRolesService{
}

View File

@ -0,0 +1,118 @@
package org.traceability.service.impl;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import jakarta.annotation.Resource;
import jakarta.servlet.http.HttpServletRequest;
import jakarta.servlet.http.HttpSession;
import org.springframework.beans.BeanUtils;
import org.springframework.util.DigestUtils;
import org.traceability.common.ErrorCode;
import org.traceability.exception.BusinessException;
import org.traceability.exception.ThrowUtils;
import org.traceability.model.domain.Roles;
import org.traceability.model.domain.UserRoles;
import org.traceability.model.domain.Users;
import org.traceability.model.dto.user.UserAddRequest;
import org.traceability.service.RolesService;
import org.traceability.service.UserRolesService;
import org.traceability.service.UsersService;
import org.traceability.mapper.UsersMapper;
import org.springframework.stereotype.Service;
import org.traceability.utils.RegexUtils;
import static org.traceability.contant.UserConstant.USER_LOGIN_STATE;
import static org.traceability.contant.UserConstant.USER_SALT;
/**
* @author xy
*/
@Service
public class UsersServiceImpl extends ServiceImpl<UsersMapper, Users>
implements UsersService{
@Resource
private UserRolesService userRolesService;
@Resource
private RolesService rolesService;
@Override
public Long addUsers(UserAddRequest userAddRequest) {
String userAccount = userAddRequest.getUserAccount();
String password = userAddRequest.getPassword();
String email = userAddRequest.getEmail();
String phone = userAddRequest.getPhone();
String userCode = userAddRequest.getUserCode();
Users users = new Users();
//校验账号信息
this.validUser(userAccount, password);
boolean invalid = RegexUtils.isUserAccountInvalid(userAccount);
ThrowUtils.throwIf(!invalid, ErrorCode.PARAMS_ERROR, "账号不符合规范");
boolean emailInvalid = RegexUtils.isEmailInvalid(email);
ThrowUtils.throwIf(!emailInvalid, ErrorCode.PARAMS_ERROR, "邮箱不符合规范");
boolean phoneInvalid = RegexUtils.isPhoneInvalid(phone);
ThrowUtils.throwIf(!phoneInvalid, ErrorCode.PARAMS_ERROR, "手机号不符合规范");
LambdaQueryWrapper<Users> queryWrapper = new LambdaQueryWrapper<>();
queryWrapper.eq(Users::getUserAccount, userAccount);
long count = this.count(queryWrapper);
ThrowUtils.throwIf(count > 0 ,ErrorCode.PARAMS_ERROR, "当前账号重复");
//校验用户角色
Roles roles = rolesService.getOne(Wrappers.<Roles>lambdaQuery()
.eq(Roles::getCode, userCode));
ThrowUtils.throwIf(roles == null, ErrorCode.PARAMS_ERROR, "角色编码不正确");
//密码加密
DigestUtils.md5DigestAsHex((USER_SALT + password).getBytes());
BeanUtils.copyProperties(userAddRequest, users);
//存入数据库
boolean save = this.save(users);
ThrowUtils.throwIf(!save, ErrorCode.OPERATION_ERROR, "存储用户失败");
//关联用户与角色
UserRoles userRoles = new UserRoles();
userRoles.setUserId(users.getId());
userRoles.setRoleId(roles.getId());
boolean save1 = userRolesService.save(userRoles);
ThrowUtils.throwIf(!save1, ErrorCode.OPERATION_ERROR, "关联用户角色失败");
return users.getId();
}
@Override
public Users loginUser(String userAccount, String password, HttpServletRequest request) {
this.validUser(userAccount, password);
String newPassword = DigestUtils.md5DigestAsHex((USER_SALT + password).getBytes());
LambdaQueryWrapper<Users> wrapper = new LambdaQueryWrapper<>();
wrapper.eq(Users::getUserAccount, userAccount);
wrapper.eq(Users::getPassword, newPassword);
Users users = this.getOne(wrapper);
ThrowUtils.throwIf(users == null, ErrorCode.PARAMS_ERROR, "账号或密码出错");
HttpSession session = request.getSession();
session.setAttribute(USER_LOGIN_STATE,users);
return users;
}
private void validUser(String userAccount, String password){
//校验用户名密码是否合规
if (userAccount.length() <= 4 || userAccount.length() >= 20) {
throw new BusinessException(ErrorCode.PARAMS_ERROR, "用户名长度应在4~20位之间");
}
if (password.length() <= 8 || password.length() >= 20) {
throw new BusinessException(ErrorCode.PARAMS_ERROR, "密码长度应在8~16位");
}
}
}

View File

@ -0,0 +1,50 @@
package org.traceability.utils;
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
import static org.traceability.contant.RegexConstant.*;
/**
* @author xy
*/
@SuppressWarnings("all")
public class RegexUtils {
/**
* 是否是无效手机格式
*
* @param phone 要校验的手机号
* @return true:符合false不符合
*/
public static boolean isPhoneInvalid(String phone) {
return mismatch(phone, PHONE_REGEX);
}
/**
* 是否是无效邮箱格式
*
* @param email 要校验的邮箱
* @return true:符合false不符合
*/
public static boolean isEmailInvalid(String email) {
return mismatch(email, EMAIL_REGEX);
}
/**
* 是否是无效账号
* @param userAccount 要检验的账号
* @return true:符合false不符合
*/
public static boolean isUserAccountInvalid(String userAccount) {
return mismatch(userAccount, SPECAIL_REGEX);
}
// 校验是否不符合正则格式
private static boolean mismatch(String str, String regex) {
if (StringUtils.isBlank(str)) {
return true;
}
return !str.matches(regex);
}
}

View File

@ -0,0 +1,25 @@
package org.traceability.utils;
import org.apache.commons.lang3.StringUtils;
/**
* SQL 工具
*
*/
@SuppressWarnings("all")
public class SqlUtils {
/**
* 校验排序字段是否合法(防止 SQL 注入)
*
* @param sortField
* @return
*/
public static boolean validSortField(String sortField) {
if (StringUtils.isBlank(sortField)) {
return false;
}
return !StringUtils.containsAny(sortField, "=", "(", ")", " ");
}
}