重试逻辑调整

This commit is contained in:
caozehui
2026-04-09 13:39:10 +08:00
parent 0a8627b440
commit fc6fc9642b
39 changed files with 690 additions and 93 deletions

View File

@@ -72,7 +72,7 @@
<!-- Swagger 注解,用于 API 文档生成(@Schema、@Operation 等) -->
<dependency>
<groupId>io.swagger.core.v3</groupId>
<artifactId>swagger-annotations</artifactId>
<artifactId>swagger-annotations-jakarta</artifactId>
</dependency>
<!-- RPC 远程调用相关 -->

View File

@@ -1,5 +1,7 @@
package com.njcn.msgpush.framework.common.exception.enums;
import com.njcn.msgpush.framework.common.exception.ErrorCode;
/**
* 业务异常的错误码区间,解决:解决各模块错误码定义,避免重复,在此只声明不做实际使用
*
@@ -27,7 +29,7 @@ package com.njcn.msgpush.framework.common.exception.enums;
*
* @author hongawen
*/
public class ServiceErrorCodeRange {
public interface ServiceErrorCodeRange {
// 模块 infra 错误码区间 [1-001-000-000 ~ 1-002-000-000)
// 模块 system 错误码区间 [1-002-000-000 ~ 1-003-000-000)
@@ -43,6 +45,7 @@ public class ServiceErrorCodeRange {
// 模块 crm 错误码区间 [1-020-000-000 ~ 1-021-000-000)
// 模块 ai 错误码区间 [1-022-000-000 ~ 1-023-000-000)
// 模块 push 错误码区间 [1-022-000-000 ~ 1-023-000-000)
ErrorCode VALIDATE_SYSTEM_SECRET_FAIL = new ErrorCode(1-022-000-000, "校验系统密钥失败");
ErrorCode GENERATE_CREDENTIAL_FAIL = new ErrorCode(1-022-000-001, "生成凭证失败");
}

View File

@@ -93,9 +93,10 @@ public class MsgpushSwaggerAutoConfiguration {
}
public static GroupedOpenApi buildGroupedOpenApi(String group, String path) {
String pathSuffix = path == null || path.isBlank() ? "" : "/" + path;
return GroupedOpenApi.builder()
.group(group)
.pathsToMatch("/admin-api/" + path + "/**", "/app-api/" + path + "/**")
.pathsToMatch("/admin-api" + pathSuffix + "/**", "/app-api" + pathSuffix + "/**")
.addOperationCustomizer((operation, handlerMethod) -> operation
.addParametersItem(buildSecurityHeaderParameter()))
.addOperationCustomizer(buildOperationIdCustomizer())

View File

@@ -19,7 +19,7 @@ public class BlacklistReqVO extends PageParam {
@Schema(description = "主键 ID", example = "123444")
private Long id;
@Schema(description = "渠道类型sms/email/app_push", example = "sms")
@Schema(description = "渠道类型sms/email/app", example = "sms")
@InEnum(value = com.njcn.msgpush.module.push.enums.ChannelTypeEnum.class, message = "渠道类型必须是 {value}")
private String channel;

View File

@@ -13,7 +13,7 @@ public class ChannelProviderConfigReqVO extends PageParam {
@Schema(description = "渠道ID")
private String id;
@Schema(description = "渠道类型sms/email/app_push", example = "sms")
@Schema(description = "渠道类型sms/email/app", example = "sms")
@InEnum(value = com.njcn.msgpush.module.push.enums.ChannelTypeEnum.class, message = "渠道类型必须是 {value}")
private String channel;

View File

@@ -0,0 +1,35 @@
package com.njcn.msgpush.module.push.controller.admin.credential;
import com.njcn.msgpush.framework.common.pojo.CommonResult;
import com.njcn.msgpush.module.push.dal.dataobject.credential.dto.CredentialReqDTO;
import com.njcn.msgpush.module.push.dal.dataobject.credential.dto.CredentialRespDTO;
import com.njcn.msgpush.module.push.service.credential.ICredentialService;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;
import jakarta.validation.Valid;
import org.springframework.beans.factory.annotation.Autowired;
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;
/**
* 系统凭证 API 接口
*
* @author msgpush
*/
@Tag(name = "系统凭证管理")
@RestController
@RequestMapping("/credential")
public class CredentialController {
@Autowired
private ICredentialService credentialService;
@PostMapping("/generate")
@Operation(summary = "生成系统凭证")
public CommonResult<CredentialRespDTO> generateCredential(@Valid @RequestBody CredentialReqDTO reqDTO) {
CredentialRespDTO respDTO = credentialService.generateCredential(reqDTO);
return CommonResult.success(respDTO);
}
}

View File

@@ -7,6 +7,7 @@ import com.njcn.msgpush.framework.idempotent.core.annotation.Idempotent;
import com.njcn.msgpush.module.push.controller.admin.message.vo.MessageRecordReqVO;
import com.njcn.msgpush.module.push.controller.admin.message.vo.MessageSendResultVO;
import com.njcn.msgpush.module.push.dal.dataobject.message.MessageRecordDO;
import com.njcn.msgpush.module.push.enums.ChannelTypeEnum;
import com.njcn.msgpush.module.push.service.message.MessageRecordService;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.Parameter;
@@ -32,11 +33,27 @@ public class MessageRecordController {
private MessageRecordService messageRecordService;
@PermitAll
@PostMapping("/send")
@Operation(summary = "消息推送")
@PostMapping("/send/sms")
@Operation(summary = "短信推送")
@Idempotent(timeout = 2)
public CommonResult<List<MessageSendResultVO>> send(@Valid @RequestBody List<MessageRecordReqVO> reqVOList) {
return success(messageRecordService.send(reqVOList));
public CommonResult<List<MessageSendResultVO>> sendSms(@Valid @RequestBody List<MessageRecordReqVO> reqVOList) {
return success(messageRecordService.send(reqVOList, ChannelTypeEnum.SMS));
}
@PermitAll
@PostMapping("/send/email")
@Operation(summary = "邮箱推送")
@Idempotent(timeout = 2)
public CommonResult<List<MessageSendResultVO>> sendEmail(@Valid @RequestBody List<MessageRecordReqVO> reqVOList) {
return success(messageRecordService.send(reqVOList, ChannelTypeEnum.EMAIL));
}
@PermitAll
@PostMapping("/send/app")
@Operation(summary = "app推送")
@Idempotent(timeout = 2)
public CommonResult<List<MessageSendResultVO>> sendApp(@Valid @RequestBody List<MessageRecordReqVO> reqVOList) {
return success(messageRecordService.send(reqVOList, ChannelTypeEnum.APP));
}
@PostMapping("/page")

View File

@@ -19,11 +19,7 @@ public class MessageRecordReqVO extends PageParam {
@Schema(description = "主键ID")
private Long id;
@Schema(description = "应用名称/来源系统标识", requiredMode = Schema.RequiredMode.REQUIRED, example = "NPQS-9500")
@NotBlank(message = "应用名称/来源系统标识不能为空")
private String appName;
@Schema(description = "渠道类型sms/email/app_push", example = "sms")
@Schema(description = "渠道类型sms/email/app", example = "sms")
@InEnum(value = com.njcn.msgpush.module.push.enums.ChannelTypeEnum.class, message = "渠道类型必须是 {value}")
private String channel;
@@ -48,13 +44,6 @@ public class MessageRecordReqVO extends PageParam {
@Schema(description = "模板参数")
private String templateParams;
@Schema(description = "服务商类型", requiredMode = Schema.RequiredMode.REQUIRED, example = "telecom/cmcc/aliyun/twilio")
@InEnum(value = com.njcn.msgpush.module.push.enums.ProviderTypeEnum.class, message = "服务商类型必须是 {value}")
private String providerType;
@Schema(description = "第三方消息ID")
private String thirdPartyId;
@Schema(description = "额外信息")
private String extraInfo;
}

View File

@@ -18,7 +18,7 @@ public class RateLimitConfigReqVO extends PageParam {
@Schema(description = "主键 ID")
private Long id;
@Schema(description = "渠道类型sms/email/app_push", example = "sms")
@Schema(description = "渠道类型sms/email/app", example = "sms")
@InEnum(value = com.njcn.msgpush.module.push.enums.ChannelTypeEnum.class, message = "渠道类型必须是 {value}")
private String channel;

View File

@@ -16,7 +16,7 @@ public class MessageRetryQueueReqVO extends PageParam {
@Schema(description = "消息ID")
private Long messageId;
@Schema(description = "渠道类型", example = "sms/email/app_push")
@Schema(description = "渠道类型", example = "sms/email/app")
@NotBlank(message = "渠道类型不能为空")
private String channel;

View File

@@ -17,7 +17,7 @@ public class RetryStrategyConfigReqVO {
@Schema(description = "主键 ID")
private Long id;
@Schema(description = "渠道类型sms/email/app_push", example = "sms")
@Schema(description = "渠道类型sms/email/app", example = "sms")
@InEnum(value = com.njcn.msgpush.module.push.enums.ChannelTypeEnum.class, message = "渠道类型必须是 {value}")
private String channel;

View File

@@ -19,7 +19,7 @@ public class BlacklistDO extends BaseDO {
private Long id;
/**
* 渠道类型sms/email/app_push
* 渠道类型sms/email/app
*/
private String channel;

View File

@@ -22,7 +22,7 @@ public class ChannelProviderConfigDO extends BaseDO {
private Long id;
/**
* 渠道类型sms/email/app_push
* 渠道类型sms/email/app
*/
private String channel;

View File

@@ -25,7 +25,7 @@ public class ProviderErrorCodeMappingDO extends BaseDO {
private String providerType;
/**
* 渠道类型sms/email/app_push
* 渠道类型sms/email/app
*/
private String channel;

View File

@@ -0,0 +1,14 @@
package com.njcn.msgpush.module.push.dal.dataobject.credential;
import lombok.Data;
/**
* @author caozehui
* @data 2026-04-09
*/
@Data
public class SystemSecretDO {
private String id;
private String systemName;
private String secret;
}

View File

@@ -0,0 +1,35 @@
package com.njcn.msgpush.module.push.dal.dataobject.credential.dto;
/**
* @author caozehui
* @data 2026-03-31
*/
import jakarta.validation.constraints.NotEmpty;
import lombok.Data;
import java.io.Serializable;
/**
* 系统凭证请求 DTO
*
* @author msgpush
*/
@Data
public class CredentialReqDTO implements Serializable {
private static final long serialVersionUID = 1L;
/**
* 上游系统名称
*/
@NotEmpty(message = "上游系统名称不能为空")
private String systemName;
/**
* 密钥(用于生成凭证)
*/
@NotEmpty(message = "密钥不能为空")
private String secretKey;
}

View File

@@ -0,0 +1,37 @@
package com.njcn.msgpush.module.push.dal.dataobject.credential.dto;
/**
* @author caozehui
* @data 2026-03-31
*/
import lombok.Data;
import lombok.experimental.Accessors;
import java.io.Serializable;
import java.time.LocalDateTime;
/**
* 系统凭证响应 DTO
*
* @author msgpush
*/
@Data
public class CredentialRespDTO implements Serializable {
private static final long serialVersionUID = 1L;
/**
* 凭证令牌
*/
private String credentialToken;
/**
* 上游系统名称
*/
private String systemName;
/**
* 过期时间
*/
private LocalDateTime expiresTime;
}

View File

@@ -1,5 +1,6 @@
package com.njcn.msgpush.module.push.dal.dataobject.message;
import com.baomidou.mybatisplus.annotation.TableField;
import com.baomidou.mybatisplus.annotation.TableName;
import com.njcn.msgpush.framework.mybatis.core.dataobject.BaseDO;
import lombok.Data;
@@ -27,7 +28,7 @@ public class MessageRecordDO extends BaseDO {
private String appName;
/**
* 渠道类型sms/email/app_push
* 渠道类型sms/email/app
*/
private String channel;

View File

@@ -19,7 +19,7 @@ public class SystemQuotaConfigDO extends BaseDO {
private Long id;
/**
* 渠道类型sms/email/app_push
* 渠道类型sms/email/app
*/
private String channel;

View File

@@ -17,7 +17,7 @@ public class RateLimitConfigDO extends BaseDO {
private Long id;
/**
* 渠道类型sms/email/app_push
* 渠道类型sms/email/app
*/
private String channel;

View File

@@ -61,4 +61,9 @@ public class MessageRetryHistoryDO extends BaseDO {
* 第三方消息ID
*/
private String thirdPartyId;
/**
* 重试来源AUTO_RETRY-系统自动重试MANUAL_RETRY-人工手动重试
*/
private String retrySource;
}

View File

@@ -19,7 +19,7 @@ public class RetryStrategyConfigDO extends BaseDO {
private Long id;
/**
* 渠道类型sms/email/app_push
* 渠道类型sms/email/app
*/
private String channel;

View File

@@ -0,0 +1,11 @@
package com.njcn.msgpush.module.push.dal.mysql.credential;
import com.njcn.msgpush.framework.mybatis.core.mapper.BaseMapperX;
import com.njcn.msgpush.module.push.dal.dataobject.credential.SystemSecretDO;
/**
* @author caozehui
* @data 2026-04-09
*/
public interface SystemSecretMappper extends BaseMapperX<SystemSecretDO> {
}

View File

@@ -18,7 +18,7 @@ import java.util.Arrays;
public enum ChannelTypeEnum implements ArrayValuable<String> {
SMS("sms", "短信"),
EMAIL("email", "邮箱"),
APP_PUSH("app_push", "APP 推送"),;
APP("app", "APP 推送"),;
public static final String[] ARRAYS = Arrays.stream(values()).map(ChannelTypeEnum::getCode).toArray(String[]::new);

View File

@@ -0,0 +1,30 @@
package com.njcn.msgpush.module.push.enums;
import lombok.AllArgsConstructor;
import lombok.Getter;
/**
* @author caozehui
* @data 2026-04-07
*/
@Getter
@AllArgsConstructor
public enum RetrySourceEnum {
AUTO_RETRY("AUTO_RETRY", "系统自动重试"),
MANUAL_RETRY("MANUAL_RETRY", "人工手动重试");
private final String code;
private final String desc;
public static RetrySourceEnum getByCode(String code) {
if (code == null) {
return null;
}
for (RetrySourceEnum value : values()) {
if (value.getCode().equals(code)) {
return value;
}
}
return null;
}
}

View File

@@ -0,0 +1,115 @@
package com.njcn.msgpush.module.push.filter;
/**
* @author caozehui
* @data 2026-03-31
*/
import cn.hutool.core.util.StrUtil;
import com.njcn.msgpush.framework.common.exception.enums.GlobalErrorCodeConstants;
import com.njcn.msgpush.framework.common.pojo.CommonResult;
import com.njcn.msgpush.framework.common.util.servlet.ServletUtils;
import com.njcn.msgpush.framework.web.config.WebProperties;
import com.njcn.msgpush.framework.web.core.filter.ApiRequestFilter;
import com.njcn.msgpush.module.push.service.credential.CredentialServiceImpl;
import com.njcn.msgpush.module.push.service.credential.ICredentialService;
import jakarta.servlet.FilterChain;
import jakarta.servlet.ServletException;
import jakarta.servlet.http.HttpServletRequest;
import jakarta.servlet.http.HttpServletResponse;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.core.Ordered;
import org.springframework.stereotype.Component;
import java.io.IOException;
/**
* 凭证认证过滤器
*
* @author msgpush
*/
@Component
@Slf4j
public class CredentialAuthenticationFilter extends ApiRequestFilter implements Ordered {
@Autowired
private ICredentialService credentialService;
/**
* 凭证 Header 名称
*/
private static final String CREDENTIAL_HEADER = "X-Credential-Token";
/**
* 需要凭证认证的接口路径
*/
private static final String[] NOT_IGNORE_PATHS = {
"/admin-api/push/message/send/**",
};
private static final String[] IGNORE_PATHS = {
"/admin-api/push/credential/generate",
};
public CredentialAuthenticationFilter(WebProperties webProperties) {
super(webProperties);
}
@Override
protected boolean shouldNotFilter(HttpServletRequest request) {
// 检查是否在排除列表中
String path = request.getRequestURI().substring(request.getContextPath().length());
for (String notExcludePath : NOT_IGNORE_PATHS) {
if (path.equals(notExcludePath)) {
return false;
}
if (notExcludePath.endsWith("/**") && path.startsWith(notExcludePath.substring(0, notExcludePath.length() - 3))) {
return false;
}
}
for (String excludePath : IGNORE_PATHS){
if (path.equals(excludePath)) {
return true;
}
if (excludePath.endsWith("/**") && path.startsWith(excludePath.substring(0, excludePath.length() - 3))) {
return true;
}
}
return super.shouldNotFilter(request);
}
@Override
protected void doFilterInternal(HttpServletRequest request, HttpServletResponse response, FilterChain filterChain) throws ServletException, IOException {
// 1. 获取凭证 token
String credentialToken = request.getHeader(CREDENTIAL_HEADER);
if (StrUtil.isEmpty(credentialToken)) {
credentialToken = request.getParameter("credential");
}
// 2. 如果没有凭证,继续过滤链
if (StrUtil.isEmpty(credentialToken)) {
filterChain.doFilter(request, response);
return;
}
// 3. 验证凭证
CredentialServiceImpl.CredentialInfo credentialInfo = credentialService.verifyCredential(credentialToken);
if (credentialInfo == null) {
// 校验失败
ServletUtils.writeJSON(response, CommonResult.error(GlobalErrorCodeConstants.UNAUTHORIZED.getCode(), "凭证无效或已过期"));
return;
}
filterChain.doFilter(request, response);
}
@Override
public int getOrder() {
return 1000;
}
}

View File

@@ -1,5 +1,6 @@
package com.njcn.msgpush.module.push.job;
import com.njcn.msgpush.module.push.enums.ChannelTypeEnum;
import com.njcn.msgpush.module.push.service.retry.MessageRetryQueueService;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
@@ -25,8 +26,7 @@ public class MessageRetryJob {
*/
@Scheduled(fixedRate = 10000)
public void processSmsRetryQueue() {
log.info("开始处理短信重试队列:{}", LocalDateTime.now());
messageRetryQueueService.processRetryBatch("sms");
messageRetryQueueService.processRetryBatch(ChannelTypeEnum.SMS.getCode());
}
/**
@@ -34,7 +34,7 @@ public class MessageRetryJob {
*/
//@Scheduled(fixedRate = 10000)
public void processEmailRetryQueue() {
messageRetryQueueService.processRetryBatch("email");
messageRetryQueueService.processRetryBatch(ChannelTypeEnum.EMAIL.getCode());
}
/**
@@ -42,7 +42,7 @@ public class MessageRetryJob {
*/
//@Scheduled(fixedRate = 10000)
public void processAppPushRetryQueue() {
messageRetryQueueService.processRetryBatch("app_push");
messageRetryQueueService.processRetryBatch(ChannelTypeEnum.APP.getCode());
}
}

View File

@@ -32,4 +32,6 @@ public interface ChannelProviderConfigService extends IService<ChannelProviderCo
void failureUpdate(String providerType, String channel);
void successUpdate(String providerType, String channel);
List<ChannelProviderConfigDO> getEnabledProviders(String channel);
}

View File

@@ -61,4 +61,9 @@ public class ChannelProviderConfigServiceImpl extends ServiceImpl<ChannelProvide
byTypeAndChannel.setHealthStatus(1);
this.updateById(byTypeAndChannel);
}
@Override
public List<ChannelProviderConfigDO> getEnabledProviders(String channel) {
return this.lambdaQuery().eq(ChannelProviderConfigDO::getChannel, channel).list();
}
}

View File

@@ -0,0 +1,212 @@
package com.njcn.msgpush.module.push.service.credential;
import cn.hutool.core.codec.Base64;
import cn.hutool.core.util.StrUtil;
import cn.hutool.crypto.SecureUtil;
import cn.hutool.crypto.symmetric.AES;
import com.njcn.msgpush.framework.common.exception.ServiceException;
import com.njcn.msgpush.framework.common.exception.enums.ServiceErrorCodeRange;
import com.njcn.msgpush.framework.common.util.json.JsonUtils;
import com.njcn.msgpush.module.push.dal.dataobject.credential.SystemSecretDO;
import com.njcn.msgpush.module.push.dal.dataobject.credential.dto.CredentialReqDTO;
import com.njcn.msgpush.module.push.dal.dataobject.credential.dto.CredentialRespDTO;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.stereotype.Service;
import java.nio.charset.StandardCharsets;
import java.time.Duration;
import java.time.LocalDateTime;
import java.util.concurrent.TimeUnit;
/**
* @author caozehui
* @data 2026-03-31
*/
@Service
@Slf4j
public class CredentialServiceImpl implements ICredentialService {
/**
* 凭证加密密钥(生产环境应通过配置中心或环境变量注入)
*/
private String credentialSecretKey = "88888888";
@Autowired
private ISystemSecretService systemSecretService;
private final StringRedisTemplate stringRedisTemplate;
public CredentialServiceImpl(StringRedisTemplate stringRedisTemplate) {
this.stringRedisTemplate = stringRedisTemplate;
}
@Override
public CredentialRespDTO generateCredential(CredentialReqDTO reqDTO) {
try {
// 1. 验证系统密钥
boolean validatedSecretRes = validateSecret(reqDTO.getSystemName(), reqDTO.getSecretKey());
if (!validatedSecretRes) {
throw new ServiceException(ServiceErrorCodeRange.VALIDATE_SYSTEM_SECRET_FAIL);
}
// 2. 创建凭证信息
CredentialInfo credentialInfo = new CredentialInfo(
reqDTO.getSystemName(),
LocalDateTime.now().plusHours(24)
);
// 3. 生成凭证 token加密后的 JSON
String token = encryptCredential(credentialInfo);
// 4. 缓存凭证信息到 Redis
cacheCredential(token, credentialInfo);
// 5. 构建响应
return new CredentialRespDTO()
.setCredentialToken(token)
.setSystemName(reqDTO.getSystemName())
.setExpiresTime(credentialInfo.getExpiresTime());
} catch (Exception e) {
throw new ServiceException(ServiceErrorCodeRange.GENERATE_CREDENTIAL_FAIL);
}
}
public CredentialInfo verifyCredential(String token) {
if (StrUtil.isEmpty(token)) {
return null;
}
try {
// 1. 从 Redis 获取凭证信息
CredentialInfo credentialInfo = getCredentialFromRedis(token);
if (credentialInfo == null) {
return null;
}
// 2. 检查是否过期
if (credentialInfo.getExpiresTime().isBefore(LocalDateTime.now())) {
// 删除过期的凭证
deleteCredential(token);
return null;
}
return credentialInfo;
} catch (Exception e) {
log.warn("[verifyCredential] 验证凭证失败token={}", token, e);
return null;
}
}
/**
* 验证系统密钥
*
* @param systemName 系统名称
* @param secretKey 密钥
*/
private boolean validateSecret(String systemName, String secretKey) {
SystemSecretDO systemSecretDO = systemSecretService.getBySystemName(systemName);
String storedSecret = systemSecretDO.getSecret();
if (!storedSecret.equals(secretKey)) {
return false;
} else {
return true;
}
}
/**
* 加密凭证信息
*
* @param info 凭证信息
* @return 加密后的 token
*/
private String encryptCredential(CredentialInfo info) {
AES aes = SecureUtil.aes(credentialSecretKey.getBytes(StandardCharsets.UTF_8));
String json = JsonUtils.toJsonString(info);
byte[] encrypted = aes.encrypt(json.getBytes(StandardCharsets.UTF_8));
return Base64.encode(encrypted);
}
/**
* 解密凭证信息
*
* @param token 凭证 token
* @return 凭证信息
*/
private CredentialInfo decryptCredential(String token) {
try {
AES aes = SecureUtil.aes(credentialSecretKey.getBytes(StandardCharsets.UTF_8));
byte[] decrypted = aes.decrypt(Base64.decode(token));
String json = new String(decrypted, StandardCharsets.UTF_8);
return JsonUtils.parseObject(json, CredentialInfo.class);
} catch (Exception e) {
log.error("[decryptCredential] 解密凭证失败", e);
return null;
}
}
/**
* 缓存凭证信息到 Redis
*
* @param token 凭证 token
* @param info 凭证信息
*/
private void cacheCredential(String token, CredentialInfo info) {
String redisKey = formatRedisKey(token);
String jsonValue = JsonUtils.toJsonString(info);
// 计算剩余过期时间(秒)
long remainingSeconds = Duration.between(LocalDateTime.now(), info.getExpiresTime()).getSeconds();
if (remainingSeconds > 0) {
stringRedisTemplate.opsForValue().set(redisKey, jsonValue, remainingSeconds, TimeUnit.SECONDS);
}
}
/**
* 从 Redis 获取凭证信息
*
* @param token 凭证 token
* @return 凭证信息
*/
private CredentialInfo getCredentialFromRedis(String token) {
String redisKey = formatRedisKey(token);
String jsonValue = stringRedisTemplate.opsForValue().get(redisKey);
if (StrUtil.isEmpty(jsonValue)) {
return null;
}
return JsonUtils.parseObject(jsonValue, CredentialInfo.class);
}
/**
* 删除凭证
*
* @param token 凭证 token
*/
private void deleteCredential(String token) {
String redisKey = formatRedisKey(token);
stringRedisTemplate.delete(redisKey);
}
/**
* 格式化 Redis Key
*
* @param token 凭证 token
* @return Redis Key
*/
private String formatRedisKey(String token) {
return "credential:token:" + token;
}
/**
* 凭证信息内部类
*/
@Data
@AllArgsConstructor
public static class CredentialInfo {
private String systemName;
private LocalDateTime expiresTime;
}
}

View File

@@ -0,0 +1,26 @@
package com.njcn.msgpush.module.push.service.credential;
import com.njcn.msgpush.module.push.dal.dataobject.credential.dto.CredentialReqDTO;
import com.njcn.msgpush.module.push.dal.dataobject.credential.dto.CredentialRespDTO;
/**
* @author caozehui
* @data 2026-03-31
*/
public interface ICredentialService {
/**
* 生成系统凭证
*
* @param reqDTO 请求参数
* @return 凭证响应
*/
CredentialRespDTO generateCredential(CredentialReqDTO reqDTO);
/**
* 验证凭证是否有效
*
* @param token 凭证 token
* @return 凭证信息
*/
CredentialServiceImpl.CredentialInfo verifyCredential(String token);
}

View File

@@ -0,0 +1,12 @@
package com.njcn.msgpush.module.push.service.credential;
import com.baomidou.mybatisplus.extension.service.IService;
import com.njcn.msgpush.module.push.dal.dataobject.credential.SystemSecretDO;
/**
* @author caozehui
* @data 2026-04-09
*/
public interface ISystemSecretService extends IService<SystemSecretDO> {
SystemSecretDO getBySystemName(String systemName);
}

View File

@@ -0,0 +1,18 @@
package com.njcn.msgpush.module.push.service.credential;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.njcn.msgpush.module.push.dal.dataobject.credential.SystemSecretDO;
import com.njcn.msgpush.module.push.dal.mysql.credential.SystemSecretMappper;
import org.springframework.stereotype.Service;
/**
* @author caozehui
* @data 2026-04-09
*/
@Service
public class SystemSecretServiceImpl extends ServiceImpl<SystemSecretMappper, SystemSecretDO> implements ISystemSecretService {
@Override
public SystemSecretDO getBySystemName(String systemName) {
return this.lambdaQuery().eq(SystemSecretDO::getSystemName, systemName).one();
}
}

View File

@@ -5,29 +5,30 @@ import com.baomidou.mybatisplus.extension.service.IService;
import com.njcn.msgpush.module.push.controller.admin.message.vo.MessageRecordReqVO;
import com.njcn.msgpush.module.push.controller.admin.message.vo.MessageSendResultVO;
import com.njcn.msgpush.module.push.dal.dataobject.message.MessageRecordDO;
import com.njcn.msgpush.module.push.enums.ChannelTypeEnum;
import com.njcn.msgpush.module.push.enums.RetrySourceEnum;
import java.io.Serializable;
import java.time.LocalDateTime;
import java.util.Collection;
import java.util.List;
public interface MessageRecordService extends IService<MessageRecordDO> {
/**
* 发送消息包括email、sms、app_push
* 发送消息包括email、sms、app
*
* @param reqVOList
* @return channelTypeEnum
* @return 发送的结果
*/
List<MessageSendResultVO> send(List<MessageRecordReqVO> reqVOList);
List<MessageSendResultVO> send(List<MessageRecordReqVO> reqVOList, ChannelTypeEnum channelTypeEnum);
/**
* 处理发送消息
*
* @param messageRecordDOList
* @param retrySource
* @return 发送的结果
*/
List<MessageSendResultVO> processSendMsg(List<MessageRecordDO> messageRecordDOList);
List<MessageSendResultVO> processSendMsg(List<MessageRecordDO> messageRecordDOList, RetrySourceEnum retrySource);
/**
* 添加消息记录

View File

@@ -1,6 +1,7 @@
package com.njcn.msgpush.module.push.service.message;
import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.collection.CollectionUtil;
import cn.hutool.core.util.ObjectUtil;
import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
@@ -23,6 +24,7 @@ import com.njcn.msgpush.module.push.dal.redis.MessageConfirmRedisDAO;
import com.njcn.msgpush.module.push.dal.redis.RateLimitRedisDAO;
import com.njcn.msgpush.module.push.dal.redis.SystemQuotaRedisDAO;
import com.njcn.msgpush.module.push.enums.ChannelTypeEnum;
import com.njcn.msgpush.module.push.enums.RetrySourceEnum;
import com.njcn.msgpush.module.push.service.channel.ChannelProviderConfigService;
import com.njcn.msgpush.module.push.service.retry.MessageRetryHistoryService;
import com.njcn.msgpush.module.push.service.retry.MessageRetryQueueService;
@@ -69,15 +71,16 @@ public class MessageRecordServiceImpl extends ServiceImpl<MessageRecordMapper, M
private MessageConfirmRedisDAO messageConfirmRedisDAO;
@Override
public List<MessageSendResultVO> send(List<MessageRecordReqVO> reqVOList) {
List<MessageRecordDO> messageRecordDOList = this.createMessageRecords(reqVOList);
return this.processSendMsg(messageRecordDOList);
public List<MessageSendResultVO> send(List<MessageRecordReqVO> reqVOList, ChannelTypeEnum channelTypeEnum) {
List<MessageRecordDO> messageRecordDOList = this.createMessageRecords(reqVOList, channelTypeEnum);
return this.processSendMsg(messageRecordDOList, null);
}
@Transactional(rollbackFor = Exception.class)
public List<MessageRecordDO> createMessageRecords(List<MessageRecordReqVO> reqVOList) {
public List<MessageRecordDO> createMessageRecords(List<MessageRecordReqVO> reqVOList, ChannelTypeEnum channelTypeEnum) {
List<MessageRecordDO> messageRecordDOList = BeanUtil.copyToList(reqVOList, MessageRecordDO.class);
messageRecordDOList.forEach(messageRecordDO -> {
messageRecordDO.setChannel(channelTypeEnum.getCode());
messageRecordDO.setStatus(MsgStatusConstant.PENDING);
// messageRecordDO.setRetryCount(0);
});
@@ -90,12 +93,12 @@ public class MessageRecordServiceImpl extends ServiceImpl<MessageRecordMapper, M
* 这里统一接收 sender 返回的 SendResult再决定消息状态、重试队列和服务商健康度如何变化。
*/
@Override
public List<MessageSendResultVO> processSendMsg(List<MessageRecordDO> messageRecordDOList) {
public List<MessageSendResultVO> processSendMsg(List<MessageRecordDO> messageRecordDOList, RetrySourceEnum retrySource) {
msgPushGuardChain.checkAll(messageRecordDOList);
List<MessageSendResultVO> resultList = new ArrayList<>();
for (MessageRecordDO messageRecordDO : messageRecordDOList) {
try {
MessageSendResultVO resultVO = this.processSingleMessage(messageRecordDO);
MessageSendResultVO resultVO = this.processSingleMessage(messageRecordDO, retrySource);
resultList.add(resultVO);
} catch (Exception e) {
MessageSendResultVO failedVO = new MessageSendResultVO();
@@ -108,7 +111,7 @@ public class MessageRecordServiceImpl extends ServiceImpl<MessageRecordMapper, M
return resultList;
}
private MessageSendResultVO processSingleMessage(MessageRecordDO messageRecordDO) {
private MessageSendResultVO processSingleMessage(MessageRecordDO messageRecordDO, RetrySourceEnum retrySource) {
MessageSendResultVO resultVO = new MessageSendResultVO();
resultVO.setMessageId(messageRecordDO.getId());
@@ -118,13 +121,34 @@ public class MessageRecordServiceImpl extends ServiceImpl<MessageRecordMapper, M
this.updateById(messageRecordDO);
return resultVO;
}
List<ChannelProviderConfigDO> enabledProviders = channelProviderConfigService.getEnabledProviders(messageRecordDO.getChannel());
SendResult sendResult = null;
if (CollectionUtil.isNotEmpty(enabledProviders)) {
ChannelProviderConfigDO channelProviderConfigDO = enabledProviders.get(0);
MessageProviderFactory messageProviderFactory = messageProviderFactoryMap.get(channelProviderConfigDO.getProviderType());
sendResult = this.sendMessage(messageRecordDO, channelProviderConfigDO, messageProviderFactory);
this.applySendResult(messageRecordDO, sendResult);
this.recordRetryHistory(messageRecordDO, retrySource);
} else {
sendResult = this.sendMessage(messageRecordDO, null, null);
this.applySendResult(messageRecordDO, sendResult);
}
if (CollectionUtil.isNotEmpty(enabledProviders)) {
ChannelProviderConfigDO channelProviderConfigDO = enabledProviders.get(0);
MessageProviderFactory messageProviderFactory = messageProviderFactoryMap.get(channelProviderConfigDO.getProviderType());
sendResult = this.validateProviderAndChannel(messageRecordDO, channelProviderConfigDO, messageProviderFactory);
if (sendResult == null) {
sendResult = this.sendMessage(messageRecordDO, channelProviderConfigDO, messageProviderFactory);
}
this.applySendResult(messageRecordDO, sendResult);
this.recordRetryHistory(messageRecordDO, retrySource);
} else {
sendResult = this.validateProviderAndChannel(messageRecordDO, null, null);
this.applySendResult(messageRecordDO, sendResult);
}
ChannelProviderConfigDO channelProviderConfigDO = channelProviderConfigService
.getByTypeAndChannel(messageRecordDO.getProviderType(), messageRecordDO.getChannel());
MessageProviderFactory messageProviderFactory = messageProviderFactoryMap.get(messageRecordDO.getProviderType());
SendResult sendResult = this.sendMessage(messageRecordDO, channelProviderConfigDO, messageProviderFactory);
this.applySendResult(messageRecordDO, sendResult);
this.recordRetryHistory(messageRecordDO);
// 在这里修改接口返回的是投递结果,还是调用接口结果
resultVO.setResult(this.isSuccessOutcome(sendResult.getOutcome()));
@@ -132,26 +156,29 @@ public class MessageRecordServiceImpl extends ServiceImpl<MessageRecordMapper, M
return resultVO;
}
private SendResult sendMessage(MessageRecordDO messageRecordDO,
ChannelProviderConfigDO channelProviderConfigDO,
MessageProviderFactory messageProviderFactory) {
private SendResult validateProviderAndChannel(MessageRecordDO messageRecordDO, ChannelProviderConfigDO channelProviderConfigDO, MessageProviderFactory messageProviderFactory) {
ChannelTypeEnum channelType = ChannelTypeEnum.getByCode(messageRecordDO.getChannel());
if (channelType == null) {
return SendResult.unsupported("当前发送渠道类型暂不支持");
}
if (ObjectUtil.isNull(messageProviderFactory)) {
return SendResult.configInvalid("当前服务商未启用或未注册");
}
if (ObjectUtil.isNull(channelProviderConfigDO)) {
return SendResult.configInvalid("当前服务商渠道配置不存在");
}
return null;
}
private SendResult sendMessage(MessageRecordDO messageRecordDO,
ChannelProviderConfigDO channelProviderConfigDO,
MessageProviderFactory messageProviderFactory) {
ChannelTypeEnum channelType = ChannelTypeEnum.getByCode(messageRecordDO.getChannel());
if (channelType == null) {
return SendResult.unsupported("当前发送渠道类型暂不支持");
}
return switch (channelType) {
case SMS -> messageProviderFactory.createSmsSender(channelProviderConfigDO, sender).sendSms(messageRecordDO);
case EMAIL -> messageProviderFactory.createEmailSender(channelProviderConfigDO, sender)
.sendEmail(messageRecordDO, new HashMap<>());
case APP_PUSH -> messageProviderFactory.createAppPushSender(channelProviderConfigDO, sender).appPush(messageRecordDO);
case APP -> messageProviderFactory.createAppPushSender(channelProviderConfigDO, sender).appPush(messageRecordDO);
};
}
@@ -192,11 +219,12 @@ public class MessageRecordServiceImpl extends ServiceImpl<MessageRecordMapper, M
}
@Transactional(propagation = Propagation.REQUIRES_NEW, rollbackFor = Exception.class)
public void recordRetryHistory(MessageRecordDO messageRecordDO) {
public void recordRetryHistory(MessageRecordDO messageRecordDO, RetrySourceEnum retrySource) {
MessageRetryHistoryDO messageRetryHistoryDO = BeanUtil.copyProperties(messageRecordDO, MessageRetryHistoryDO.class, "id");
messageRetryHistoryDO.setMessageId(messageRecordDO.getId());
messageRetryHistoryDO.setStatus(messageRecordDO.getStatus());
messageRetryHistoryDO.setRetrySequence(messageRetryHistoryService.getMaxRetrySequence(messageRecordDO.getId()));
messageRetryHistoryDO.setRetrySource(retrySource != null ? retrySource.getCode() : null);
messageRetryHistoryService.add(messageRetryHistoryDO);
}
@@ -256,9 +284,7 @@ public class MessageRecordServiceImpl extends ServiceImpl<MessageRecordMapper, M
public Page<MessageRecordDO> getPage(MessageRecordReqVO reqVO) {
QueryWrapper<MessageRecordDO> wrapper = new QueryWrapper<>();
wrapper.lambda()
.eq(StrUtil.isNotBlank(reqVO.getChannel()), MessageRecordDO::getChannel, reqVO.getChannel())
.eq(StrUtil.isNotBlank(reqVO.getProviderType()), MessageRecordDO::getProviderType, reqVO.getProviderType())
.eq(StrUtil.isNotBlank(reqVO.getAppName()), MessageRecordDO::getAppName, reqVO.getAppName());
.eq(StrUtil.isNotBlank(reqVO.getChannel()), MessageRecordDO::getChannel, reqVO.getChannel());
return this.page(new Page<>(PageUtils.getPageNum(reqVO), PageUtils.getPageSize(reqVO)), wrapper);
}
}

View File

@@ -7,7 +7,6 @@ import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.njcn.msgpush.framework.common.pojo.PageResult;
import com.njcn.msgpush.module.push.constant.MsgStatusConstant;
import com.njcn.msgpush.module.push.controller.admin.message.vo.MessageSendResultVO;
import com.njcn.msgpush.module.push.controller.admin.retry.vo.MessageRetryQueueReqVO;
import com.njcn.msgpush.module.push.dal.dataobject.message.MessageRecordDO;
import com.njcn.msgpush.module.push.dal.dataobject.retry.MessageRetryQueueDO;
@@ -16,6 +15,7 @@ import com.njcn.msgpush.module.push.dal.mysql.retry.MessageRetryQueueMapper;
import com.njcn.msgpush.module.push.dal.redis.MessageConfirmRedisDAO;
import com.njcn.msgpush.module.push.dal.redis.MessageRetryRedisDAO;
import com.njcn.msgpush.module.push.enums.ChannelTypeEnum;
import com.njcn.msgpush.module.push.enums.RetrySourceEnum;
import com.njcn.msgpush.module.push.service.message.MessageRecordService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
@@ -50,7 +50,7 @@ public class MessageRetryQueueServiceImpl extends ServiceImpl<MessageRetryQueueM
private static final List<String> CHANNELS = Arrays.asList(
ChannelTypeEnum.SMS.getCode(),
ChannelTypeEnum.EMAIL.getCode(),
ChannelTypeEnum.APP_PUSH.getCode());
ChannelTypeEnum.APP.getCode());
private final AtomicBoolean startupRebuildCompleted = new AtomicBoolean(false);
@@ -136,7 +136,7 @@ public class MessageRetryQueueServiceImpl extends ServiceImpl<MessageRetryQueueM
plusSeconds = 60 * 60 * 4;
}
}
case APP_PUSH -> {
case APP -> {
if (retryCount == 1) {
plusSeconds = 60;
} else {
@@ -176,7 +176,7 @@ public class MessageRetryQueueServiceImpl extends ServiceImpl<MessageRetryQueueM
List<MessageRecordDO> messageRecordDOList = messageRecordService.listByIds(claimedMessageIds);
for (MessageRecordDO messageRecordDO : messageRecordDOList) {
processSingleRetry(messageRecordDO);
processSingleRetry(messageRecordDO, RetrySourceEnum.AUTO_RETRY);
}
}
@@ -188,18 +188,16 @@ public class MessageRetryQueueServiceImpl extends ServiceImpl<MessageRetryQueueM
*
* @param messageRecordDO 消息记录
*/
private void processSingleRetry(MessageRecordDO messageRecordDO) {
messageRecordDO.setStatus(MsgStatusConstant.PENDING);
messageRecordService.processSendMsg(Collections.singletonList(messageRecordDO)).get(0);
messageRecordService.updateRetryCount(messageRecordDO.getId());
}
/**
* 删除重试记录(数据库逻辑删除 + Redis 已提前删除)
*/
private void deleteRetryRecord(MessageRecordDO message) {
this.deleteByMessageIds(Collections.singletonList(message.getId()));
messageRecordService.updateById(message);
private void processSingleRetry(MessageRecordDO messageRecordDO, RetrySourceEnum retrySource) {
// 校验一下消息的状态是否为 retryable_failed防止误推送已经成功的消息
if (MsgStatusConstant.RETRYABLE_FAILED.equals(messageRecordDO.getStatus())) {
messageRecordDO.setStatus(MsgStatusConstant.PENDING);
messageRecordService.processSendMsg(Collections.singletonList(messageRecordDO), retrySource);
messageRecordService.updateRetryCount(messageRecordDO.getId());
} else {
// 从retryQueue中删除(已经从redis中移除过了所以不用再次移除了)
this.deleteByMessageIds(Collections.singletonList(messageRecordDO.getId()));
}
}
@Override
@@ -208,7 +206,7 @@ public class MessageRetryQueueServiceImpl extends ServiceImpl<MessageRetryQueueM
if (CollUtil.isNotEmpty(messageIds)) {
List<MessageRecordDO> messageRecordDOList = messageRecordService.listByIds(messageIds);
for (MessageRecordDO messageRecordDO : messageRecordDOList) {
processSingleRetry(messageRecordDO);
processSingleRetry(messageRecordDO, RetrySourceEnum.MANUAL_RETRY);
}
}
}

View File

@@ -21,7 +21,6 @@ import java.time.LocalDateTime;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.UUID;
/**
* @author caozehui
@@ -60,14 +59,12 @@ public class MsgPushClientTest {
for (int i = 0; i < 1; i++) {
MessageRecordReqVO message = new MessageRecordReqVO();
message.setId(1234567890L);
message.setAppName("NPQS-9000");
message.setChannel(ChannelTypeEnum.SMS.getMsg());
message.setReceiver("18839431215");
message.setContent("【南京灿能电力】测试短信" + i + ",请忽略。" + LocalDateTime.now().format(java.time.format.DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss")));
message.setProviderType(ProviderTypeEnum.TELECOM.getCode());
messageIdList.add(message);
}
List<MessageSendResultVO> sendResult = messageRecordService.send(messageIdList);
List<MessageSendResultVO> sendResult = messageRecordService.send(messageIdList, ChannelTypeEnum.SMS);
Thread.sleep(9000);
System.out.println(JSON.toJSONString(sendResult));

View File

@@ -24,7 +24,7 @@
<!-- Web 相关 -->
<dependency>
<groupId>io.swagger.core.v3</groupId>
<artifactId>swagger-annotations</artifactId>
<artifactId>swagger-annotations-jakarta</artifactId>
</dependency>
<!-- 参数校验 -->

View File

@@ -27,6 +27,7 @@
<!-- maven-surefire-plugin 暂时无法通过 bom 的依赖读取(兼容老版本 IDEA 2024 及以前版本) -->
<lombok.version>1.18.42</lombok.version>
<spring.boot.version>3.5.9</spring.boot.version>
<swagger.jakarta.version>2.2.38</swagger.jakarta.version>
<mapstruct.version>1.6.3</mapstruct.version>
<project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
</properties>
@@ -119,6 +120,12 @@
<artifactId>msgpush-spring-boot-starter-websocket</artifactId>
<version>${revision}</version>
</dependency>
<dependency>
<groupId>io.swagger.core.v3</groupId>
<artifactId>swagger-annotations-jakarta</artifactId>
<version>${swagger.jakarta.version}</version>
</dependency>
</dependencies>
</dependencyManagement>