feat(M3): 申请审批 + 邮件队列 — 公开申请、管理员审批自动建号、mail_logs 落库重试
This commit is contained in:
@@ -0,0 +1,84 @@
|
||||
package mail
|
||||
|
||||
import (
|
||||
"context"
|
||||
"log/slog"
|
||||
|
||||
"gorm.io/gorm"
|
||||
|
||||
"ws_usernode/internal/model"
|
||||
)
|
||||
|
||||
// 邮件状态(mail_logs.status)。
|
||||
const (
|
||||
StatusPending = "pending" // 已入队,等待直发
|
||||
StatusSent = "sent" // 发送成功
|
||||
StatusFailed = "failed" // 发送失败,等待重试
|
||||
)
|
||||
|
||||
// MaxRetries 单封邮件最大重试次数;超出后停止重试,保留失败记录(PLAN F5)。
|
||||
const MaxRetries = 5
|
||||
|
||||
// Retryable 可由 cron 定时调用的邮件队列重试接口。
|
||||
type Retryable interface {
|
||||
Retry(ctx context.Context, limit int) error
|
||||
}
|
||||
|
||||
// QueuedMailer 邮件队列实现(PLAN F5):每封邮件先落 mail_logs 再同步尝试直发;
|
||||
// 失败记录 error 与 retry_count,由 cron 定时重试;成功标记 sent。
|
||||
// Send 恒返回 nil(除 DB 落库失败):邮件失败不阻断业务
|
||||
// (OTP 双通道、审批结果等,见 AuthService.UserOTPSend 的既有约定)。
|
||||
type QueuedMailer struct {
|
||||
base Mailer
|
||||
db *gorm.DB
|
||||
log *slog.Logger
|
||||
}
|
||||
|
||||
// NewQueuedMailer 创建邮件队列,包装底层直发实现(SMTP 或 LogMailer)。
|
||||
func NewQueuedMailer(base Mailer, db *gorm.DB, log *slog.Logger) *QueuedMailer {
|
||||
return &QueuedMailer{base: base, db: db, log: log}
|
||||
}
|
||||
|
||||
// Send 入队并尝试直发。返回 nil 表示已受理;实际送达与否见 mail_logs。
|
||||
func (m *QueuedMailer) Send(ctx context.Context, to, subject, body string) error {
|
||||
entry := model.MailLog{To: to, Subject: subject, Body: body, Status: StatusPending}
|
||||
if err := m.db.WithContext(ctx).Create(&entry).Error; err != nil {
|
||||
m.log.Error("mail: 记录 mail_log 失败", "err", err)
|
||||
return err
|
||||
}
|
||||
if err := m.base.Send(ctx, to, subject, body); err != nil {
|
||||
_ = m.db.WithContext(ctx).Model(&entry).Updates(map[string]any{
|
||||
"status": StatusFailed, "error": err.Error(), "retry_count": 1,
|
||||
}).Error
|
||||
m.log.Warn("mail: 直发失败,进入重试队列", "to", to, "subject", subject, "err", err)
|
||||
return nil // 非阻断
|
||||
}
|
||||
_ = m.db.WithContext(ctx).Model(&entry).Update("status", StatusSent).Error
|
||||
return nil
|
||||
}
|
||||
|
||||
// Retry 重试失败邮件(retry_count < limit,按原内容重发)。成功标记 sent,
|
||||
// 失败累加 retry_count 并刷新 error;达到上限后不再挑选,保留失败记录待人工处理。
|
||||
func (m *QueuedMailer) Retry(ctx context.Context, limit int) error {
|
||||
if limit <= 0 {
|
||||
return nil
|
||||
}
|
||||
var rows []model.MailLog
|
||||
if err := m.db.WithContext(ctx).
|
||||
Where("status = ? AND retry_count < ?", StatusFailed, limit).
|
||||
Find(&rows).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
for _, r := range rows {
|
||||
if err := m.base.Send(ctx, r.To, r.Subject, r.Body); err != nil {
|
||||
_ = m.db.WithContext(ctx).Model(&model.MailLog{}).Where("id = ?", r.ID).Updates(map[string]any{
|
||||
"retry_count": r.RetryCount + 1, "error": err.Error(),
|
||||
}).Error
|
||||
m.log.Warn("mail: 重试仍失败", "id", r.ID, "to", r.To, "retry_count", r.RetryCount+1, "err", err)
|
||||
continue
|
||||
}
|
||||
_ = m.db.WithContext(ctx).Model(&model.MailLog{}).Where("id = ?", r.ID).Update("status", StatusSent).Error
|
||||
m.log.Info("mail: 重试成功", "id", r.ID, "to", r.To)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
Reference in New Issue
Block a user