tongtong/youlai-boot
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: 发送验证码代码重构优化；扩展Spring Security 支持短信验证码；

Browse Source
This commit is contained in:
Ray.Hao
2025-01-13 18:14:52 +08:00
parent b107bb5315
commit 4ecb25147f
39 changed files with 457 additions and 362 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
src/main/java/com/youlai/boot/common/constant/RedisConstants.java
View File

@@ -9,40 +9,32 @@ package com.youlai.boot.common.constant;
public interface RedisConstants {
/**
* 系统配置Redis-key
* 系统配置 Redis 键
*/
String SYSTEM_CONFIG_KEY = "system:config";
/**
* IP限流Redis-key
* IP 限流 Redis 键
*/
String IP_RATE_LIMITER_KEY = "ip:rate:limiter:";
String IP_RATE_LIMITER_KEY = "rate:limiter:ip:";
/**
* 防重复提交Redis-key
* 防重复提交 Redis 键前缀
*/
String RESUBMIT_LOCK_PREFIX = "resubmit:lock:";
String RESUBMIT_LOCK_PREFIX = "lock:resubmit:";
/**
* 单个IP请求的最大每秒查询数（QPS）阈值Key
* 登录手机验证码 Redis 键前缀
*/
String IP_QPS_THRESHOLD_LIMIT_KEY = "IP_QPS_THRESHOLD_LIMIT";
String SMS_LOGIN_CODE_PREFIX= "code:sms:login:";
/**
* 手机验证码缓存前缀
* 绑定或更换手机号验证码 Redis 键前缀
*/
String SMS_LOGIN_VERIFY_CODE_PREFIX = "sms_login:mobile:";
String SMS_CHANGE_CODE_PREFIX = "code:sms:change:";
/**
* 重置密码验证码缓存前缀
* 绑定或更换邮箱验证码 Redis 键前缀
*/
String SMS_RESET_PASSWORD_VERIFY_CODE_PREFIX = "sms_reset_password:mobile:";
/**
* 邮箱验证码缓存前缀
*/
String EMAIL_VERIFICATION_CODE_PREFIX = "VERIFICATION_CODE:EMAIL:";
String EMAIL_CHANGE_CODE_PREFIX = "code:email:change:";
}

6
src/main/java/com/youlai/boot/common/constant/SystemConstants.java
View File

@@ -3,7 +3,7 @@ package com.youlai.boot.common.constant;
/**
* 系统常量
*
* @author haoxr
* @author Ray.Hao
* @since 1.0.0
*/
public interface SystemConstants {
@@ -24,5 +24,9 @@ public interface SystemConstants {
String ROOT_ROLE_CODE = "ROOT";
/**
* 系统配置 IP的QPS限流的KEY
*/
String SYSTEM_CONFIG_IP_QPS_LIMIT_KEY = "IP_QPS_THRESHOLD_LIMIT";
}

3
src/main/java/com/youlai/boot/common/result/Result.java
View File

@@ -1,5 +1,6 @@
package com.youlai.boot.common.result;
import cn.hutool.core.util.StrUtil;
import lombok.Data;
import java.io.Serializable;
@@ -52,7 +53,7 @@ public class Result<T> implements Serializable {
}
public static <T> Result<T> failed(IResultCode resultCode, String msg) {
return result(resultCode.getCode(), msg, null);
return result(resultCode.getCode(), StrUtil.isNotBlank(msg) ? msg : resultCode.getMsg(), null);
}
private static <T> Result<T> result(IResultCode resultCode, T data) {

46
src/main/java/com/youlai/boot/common/util/ResponseUtils.java
View File

@@ -21,19 +21,15 @@ import java.nio.charset.StandardCharsets;
@Slf4j
public class ResponseUtils {
/**
* 异常消息返回(适用过滤器中处理异常响应)
*
* @param response HttpServletResponse
* @param response HttpServletResponse
* @param resultCode 响应结果码
*/
public static void writeErrMsg(HttpServletResponse response, ResultCode resultCode) {
// 根据不同的结果码设置HTTP状态
int status = switch (resultCode) {
case ACCESS_UNAUTHORIZED, ACCESS_TOKEN_INVALID , REFRESH_TOKEN_INVALID
-> HttpStatus.UNAUTHORIZED.value();
default -> HttpStatus.BAD_REQUEST.value();
};
int status = getHttpStatus(resultCode);
response.setStatus(status);
response.setContentType(MediaType.APPLICATION_JSON_VALUE);
@@ -48,4 +44,40 @@ public class ResponseUtils {
}
}
/**
* 异常消息返回(适用过滤器中处理异常响应)
*
* @param response HttpServletResponse
* @param resultCode 响应结果码
*/
public static void writeErrMsg(HttpServletResponse response, ResultCode resultCode, String message) {
int status = getHttpStatus(resultCode);
response.setStatus(status);
response.setContentType(MediaType.APPLICATION_JSON_VALUE);
response.setCharacterEncoding(StandardCharsets.UTF_8.name());
try (PrintWriter writer = response.getWriter()) {
String jsonResponse = JSONUtil.toJsonStr(Result.failed(resultCode, message));
writer.print(jsonResponse);
writer.flush(); // 确保将响应内容写入到输出流
} catch (IOException e) {
log.error("响应异常处理失败", e);
}
}
/**
* 根据结果码获取HTTP状态码
*
* @param resultCode 结果码
* @return HTTP状态码
*/
private static int getHttpStatus(ResultCode resultCode) {
return switch (resultCode) {
case ACCESS_UNAUTHORIZED, ACCESS_TOKEN_INVALID, REFRESH_TOKEN_INVALID -> HttpStatus.UNAUTHORIZED.value();
default -> HttpStatus.BAD_REQUEST.value();
};
}
}