mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-29 14:06:36 +08:00
ryan
aee457093d
### 🛠 修复 - 修复了 CAPTCHA 验证的逻辑和安全性。 ### ⚡️ 优化与改进 - 新增了动态存储配置与多后端迁移机制。 - 新增了手动触发存储迁移的 Web GUI 操作界面。 - 实现了存储迁移的并发处理（使用 Group）与上传后 SHA-256 完整性自动校验机制。 - 实现了基于 Redis 的分布式锁与集群多节点缓存失效广播机制，防止迁移任务冲突。 - 实现了 Redis 锁续期守护协程 (watchdog) 防止超长迁移任务锁过期。 - 优化了更新存储配置时的连通性自动校验机制，防止配置错误。 - 优化了存储配置的内存缓存机制，并引入了全局共享连接池以提高 TCP 复用率。 - 优化了上传前已有同名文件的比对逻辑，跳过下载阶段，仅比对 Content-Length 以实现零网络流量跳过。 - 新增了普通用户的独立文件管理 Dashboard 与控制接口。 - 优化了文件管理接口命名空间，将全局管理迁移至管理员级 API Namespace 隔离控制。 - 优化了管理员新建用户接口，增加了邮箱（Email）字段的必填要求。 ### 💄 其他/体验 - 重构并优化了文件管理页面，引入了多标签页多维度统计数据大屏。 - 基于 shadcn 原生 UI 组件重构并优化了文件详情和仪表盘组件。 - 抽取后端核心路由注册流程，降低主路由文件圈复杂度，使路由配置更易维护。
382 lines
12 KiB
Go
382 lines
12 KiB
Go
// Copyright 2025 linux.do
|
||
// Copyright 2026 Arctel.net
|
||
// SPDX-License-Identifier: Apache-2.0
|
||
|
||
package task
|
||
|
||
import (
|
||
"context"
|
||
"errors"
|
||
"fmt"
|
||
"time"
|
||
|
||
"github.com/Rain-kl/Wavelet/internal/db/idgen"
|
||
"github.com/Rain-kl/Wavelet/internal/logger"
|
||
"github.com/Rain-kl/Wavelet/internal/model"
|
||
"github.com/Rain-kl/Wavelet/internal/otel_trace"
|
||
"github.com/hibiken/asynq"
|
||
"go.opentelemetry.io/otel/attribute"
|
||
"go.opentelemetry.io/otel/codes"
|
||
"go.opentelemetry.io/otel/trace"
|
||
)
|
||
|
||
// handlerRegistry 已注册的任务处理器
|
||
var handlerRegistry = make(map[string]TaskHandler)
|
||
|
||
// RegisterHandler 注册任务处理器
|
||
// 传入任务类型标识（对应 constants.go 中的 AsynqTask 常量）和 TaskHandler 实现
|
||
func RegisterHandler(asynqTaskType string, handler TaskHandler) {
|
||
handlerRegistry[asynqTaskType] = handler
|
||
}
|
||
|
||
// getHandler 获取已注册的处理器
|
||
func getHandler(asynqTaskType string) (TaskHandler, bool) {
|
||
h, ok := handlerRegistry[asynqTaskType]
|
||
return h, ok
|
||
}
|
||
|
||
// ValidateAndNormalizePayload 校验并标准化任务参数。
|
||
// 如果 Handler 实现了 PayloadValidator，调用其 ValidatePayload 方法；
|
||
// 否则直接返回原始 payload。
|
||
func ValidateAndNormalizePayload(asynqTaskType string, payload []byte) ([]byte, error) {
|
||
handler, ok := getHandler(asynqTaskType)
|
||
if !ok {
|
||
return payload, nil
|
||
}
|
||
if validator, ok := handler.(PayloadValidator); ok {
|
||
return validator.ValidatePayload(payload)
|
||
}
|
||
return payload, nil
|
||
}
|
||
|
||
// contextKey 用于 context 存取 taskID
|
||
type contextKey string
|
||
|
||
const taskIDKey contextKey = "task_execution_task_id"
|
||
|
||
// withTaskID 将 taskID 注入 context
|
||
func withTaskID(ctx context.Context, taskID string) context.Context {
|
||
return context.WithValue(ctx, taskIDKey, taskID)
|
||
}
|
||
|
||
// GetTaskID 从 context 中获取 taskID
|
||
func GetTaskID(ctx context.Context) string {
|
||
if v, ok := ctx.Value(taskIDKey).(string); ok {
|
||
return v
|
||
}
|
||
return ""
|
||
}
|
||
|
||
// AppendLog 追加日志到任务执行记录
|
||
// 在 TaskHandler.Execute 中调用，日志会自动追加到 TaskExecution.Log 字段
|
||
func AppendLog(ctx context.Context, format string, args ...interface{}) {
|
||
taskID := GetTaskID(ctx)
|
||
if taskID == "" {
|
||
// 上下文中没有 taskID，降级到普通日志
|
||
logger.InfoF(ctx, format, args...)
|
||
return
|
||
}
|
||
|
||
logLine := fmt.Sprintf(format, args...)
|
||
if err := model.AppendTaskExecutionLog(ctx, taskID, logLine); err != nil {
|
||
logger.ErrorF(ctx, "[TaskExecutor] 追加任务日志失败 taskID=%s: %v", taskID, err)
|
||
}
|
||
}
|
||
|
||
// DispatchTask 下发任务（创建 TaskExecution 记录 → 入队 Asynq）
|
||
func DispatchTask(ctx context.Context, taskType string, payload []byte, triggeredBy string) (string, error) {
|
||
meta := GetTaskMeta(taskType)
|
||
if meta == nil {
|
||
return "", fmt.Errorf(errUnknownTaskType, taskType)
|
||
}
|
||
|
||
// 生成唯一的 TaskID
|
||
taskID := generateTaskID(taskType, triggeredBy)
|
||
|
||
// 创建任务执行记录
|
||
execution := &model.TaskExecution{
|
||
TaskID: taskID,
|
||
TaskType: meta.AsynqTask,
|
||
TaskName: meta.Name,
|
||
Status: model.TaskExecutionStatusPending,
|
||
Retryable: meta.Retryable,
|
||
MaxRetry: meta.MaxRetry,
|
||
RetryCount: 0,
|
||
Payload: string(payload),
|
||
TriggeredBy: triggeredBy,
|
||
}
|
||
|
||
if err := model.CreateTaskExecution(ctx, execution); err != nil {
|
||
return "", fmt.Errorf(errCreateTaskExecutionFailed, err)
|
||
}
|
||
|
||
// 入队 Asynq
|
||
taskInfo := asynq.NewTask(meta.AsynqTask, payload)
|
||
if _, err := AsynqClient.Enqueue(
|
||
taskInfo,
|
||
asynq.TaskID(taskID),
|
||
asynq.MaxRetry(meta.MaxRetry),
|
||
asynq.Queue(meta.Queue),
|
||
); err != nil {
|
||
// 入队失败，更新执行记录状态
|
||
execution.Status = model.TaskExecutionStatusFailed
|
||
execution.ErrorMessage = fmt.Sprintf("入队失败: %v", err)
|
||
now := time.Now()
|
||
execution.StartedAt = &now
|
||
execution.FinishedAt = &now
|
||
_ = model.UpdateTaskExecution(ctx, execution)
|
||
return "", fmt.Errorf(errTaskEnqueueFailed, err)
|
||
}
|
||
|
||
if err := model.AppendTaskExecutionLog(ctx, taskID, fmt.Sprintf("[系统] 任务已成功入队，等待调度执行 (队列: %s, 最大重试次数: %d)", meta.Queue, meta.MaxRetry)); err != nil {
|
||
logger.ErrorF(ctx, "[TaskExecutor] 追加入队日志失败 taskID=%s: %v", taskID, err)
|
||
}
|
||
|
||
return taskID, nil
|
||
}
|
||
|
||
// RetryTask 重试失败的任务
|
||
func RetryTask(ctx context.Context, id uint64) (string, error) {
|
||
execution, err := model.GetTaskExecutionByID(ctx, id)
|
||
if err != nil {
|
||
return "", fmt.Errorf(errTaskExecutionNotFound, err)
|
||
}
|
||
|
||
if execution.Status != model.TaskExecutionStatusFailed {
|
||
return "", fmt.Errorf(errRetryOnlyFailedTask, execution.Status)
|
||
}
|
||
|
||
if !execution.Retryable {
|
||
return "", errors.New(errTaskNotRetryable)
|
||
}
|
||
|
||
if execution.RetryCount >= execution.MaxRetry {
|
||
return "", fmt.Errorf(errTaskMaxRetryExceeded, execution.MaxRetry)
|
||
}
|
||
|
||
// 生成新的 TaskID
|
||
newTaskID := generateRetryTaskID(execution.TaskID, execution.RetryCount+1)
|
||
|
||
// 创建新的执行记录
|
||
newExecution := &model.TaskExecution{
|
||
TaskID: newTaskID,
|
||
TaskType: execution.TaskType,
|
||
TaskName: execution.TaskName,
|
||
Status: model.TaskExecutionStatusPending,
|
||
Retryable: execution.Retryable,
|
||
MaxRetry: execution.MaxRetry,
|
||
RetryCount: execution.RetryCount + 1,
|
||
Payload: execution.Payload,
|
||
TriggeredBy: "retry",
|
||
}
|
||
|
||
if err := model.CreateTaskExecution(ctx, newExecution); err != nil {
|
||
return "", fmt.Errorf(errCreateRetryExecutionFailed, err)
|
||
}
|
||
|
||
meta := GetTaskMeta(execution.TaskType)
|
||
queueName := QueueDefault
|
||
if meta != nil {
|
||
queueName = meta.Queue
|
||
}
|
||
|
||
// 入队 Asynq
|
||
taskInfo := asynq.NewTask(execution.TaskType, []byte(execution.Payload))
|
||
if _, err := AsynqClient.Enqueue(
|
||
taskInfo,
|
||
asynq.TaskID(newTaskID),
|
||
asynq.MaxRetry(execution.MaxRetry),
|
||
asynq.Queue(queueName),
|
||
); err != nil {
|
||
newExecution.Status = model.TaskExecutionStatusFailed
|
||
newExecution.ErrorMessage = fmt.Sprintf("重试入队失败: %v", err)
|
||
now := time.Now()
|
||
newExecution.StartedAt = &now
|
||
newExecution.FinishedAt = &now
|
||
_ = model.UpdateTaskExecution(ctx, newExecution)
|
||
return "", fmt.Errorf(errRetryTaskEnqueueFailed, err)
|
||
}
|
||
|
||
if err := model.AppendTaskExecutionLog(ctx, newTaskID, fmt.Sprintf("[系统] 手动触发重试，已重新创建任务并入队 (原任务ID: %s, 重试次数: %d/%d)", execution.TaskID, execution.RetryCount+1, execution.MaxRetry)); err != nil {
|
||
logger.ErrorF(ctx, "[TaskExecutor] 追加重试日志失败 taskID=%s: %v", newTaskID, err)
|
||
}
|
||
|
||
return newTaskID, nil
|
||
}
|
||
|
||
// ProcessTask Asynq 实际调用的统一处理函数
|
||
// Worker 注册时统一使用此函数，内部自动分发到对应的 TaskHandler
|
||
func ProcessTask(ctx context.Context, t *asynq.Task) error {
|
||
// 初始化 Trace
|
||
ctx, span := otel_trace.Start(ctx, "TaskProcess_"+t.Type(), trace.WithSpanKind(trace.SpanKindConsumer))
|
||
defer span.End()
|
||
|
||
// 添加任务信息到 Span
|
||
span.SetAttributes(
|
||
attribute.String("task.type", t.Type()),
|
||
attribute.Int("task.payload_size", len(t.Payload())),
|
||
attribute.String("task.id", t.ResultWriter().TaskID()),
|
||
)
|
||
|
||
taskID := t.ResultWriter().TaskID()
|
||
|
||
// 注入 taskID 到 context
|
||
ctx = withTaskID(ctx, taskID)
|
||
|
||
// 查找处理器
|
||
handler, ok := getHandler(t.Type())
|
||
if !ok {
|
||
err := fmt.Errorf(errUnregisteredTaskHandler, t.Type())
|
||
logger.ErrorF(ctx, "[TaskExecutor] %v", err)
|
||
span.SetStatus(codes.Error, err.Error())
|
||
return err
|
||
}
|
||
|
||
// 加载或动态创建执行记录
|
||
now := time.Now()
|
||
execution, err := getOrCreateTaskExecution(ctx, taskID, t, now)
|
||
if err == nil && execution != nil && execution.TriggeredBy != "schedule" {
|
||
execution.Status = model.TaskExecutionStatusRunning
|
||
execution.StartedAt = &now
|
||
if updateErr := model.UpdateTaskExecution(ctx, execution); updateErr != nil {
|
||
logger.ErrorF(ctx, "[TaskExecutor] 更新执行状态失败 taskID=%s: %v", taskID, updateErr)
|
||
}
|
||
}
|
||
|
||
if execution != nil {
|
||
AppendLog(ctx, "[系统] 开始执行异步任务 [名称: %s, 类型: %s]，重试次数: %d/%d",
|
||
execution.TaskName, t.Type(), execution.RetryCount, execution.MaxRetry)
|
||
} else {
|
||
AppendLog(ctx, "[系统] 开始执行异步任务 [类型: %s]", t.Type())
|
||
}
|
||
|
||
// 开始计时
|
||
start := time.Now()
|
||
|
||
// 执行业务逻辑
|
||
result, execErr := handler.Execute(ctx, t.Payload())
|
||
|
||
// 计算耗时并归档记录
|
||
duration := time.Since(start)
|
||
finishTime := time.Now()
|
||
|
||
completeTaskExecution(ctx, execution, t, duration, finishTime, result, execErr, span)
|
||
|
||
if execution == nil && execErr != nil {
|
||
span.SetStatus(codes.Error, execErr.Error())
|
||
return execErr
|
||
}
|
||
|
||
return execErr
|
||
}
|
||
|
||
// getOrCreateTaskExecution 获取已有的任务执行记录，如果不存在则针对已知任务类型动态创建记录
|
||
func getOrCreateTaskExecution(ctx context.Context, taskID string, t *asynq.Task, now time.Time) (*model.TaskExecution, error) {
|
||
execution, err := model.GetTaskExecutionByTaskID(ctx, taskID)
|
||
if err == nil {
|
||
return execution, nil
|
||
}
|
||
|
||
meta := GetTaskMetaByAsynqTask(t.Type())
|
||
if meta == nil {
|
||
return nil, err
|
||
}
|
||
|
||
execution = &model.TaskExecution{
|
||
TaskID: taskID,
|
||
TaskType: meta.AsynqTask,
|
||
TaskName: meta.Name,
|
||
Status: model.TaskExecutionStatusRunning,
|
||
Retryable: meta.Retryable,
|
||
MaxRetry: meta.MaxRetry,
|
||
RetryCount: 0,
|
||
Payload: string(t.Payload()),
|
||
TriggeredBy: "schedule",
|
||
StartedAt: &now,
|
||
}
|
||
|
||
if createErr := model.CreateTaskExecution(ctx, execution); createErr != nil {
|
||
logger.ErrorF(ctx, "[TaskExecutor] 动态创建执行记录失败 taskID=%s: %v", taskID, createErr)
|
||
return nil, createErr
|
||
}
|
||
|
||
return execution, nil
|
||
}
|
||
|
||
// completeTaskExecution 完成并更新任务执行记录的状态和执行结果
|
||
func completeTaskExecution(ctx context.Context, execution *model.TaskExecution, t *asynq.Task, duration time.Duration, finishTime time.Time, result *TaskResult, execErr error, span trace.Span) {
|
||
if execution == nil {
|
||
return
|
||
}
|
||
|
||
execution.Duration = duration.Milliseconds()
|
||
execution.FinishedAt = &finishTime
|
||
|
||
if execErr != nil {
|
||
handleFailedTask(ctx, execution, t, duration, execErr, span)
|
||
} else {
|
||
handleSuccessfulTask(ctx, execution, t, duration, result)
|
||
}
|
||
|
||
if err := model.UpdateTaskExecution(ctx, execution); err != nil {
|
||
logger.ErrorF(ctx, "[TaskExecutor] 更新执行记录失败 taskID=%s: %v", execution.TaskID, err)
|
||
}
|
||
if shouldFlushTaskExecutionLog(ctx, execErr) {
|
||
if err := model.FlushTaskExecutionLog(ctx, execution.TaskID); err != nil {
|
||
logger.ErrorF(ctx, "[TaskExecutor] 持久化任务日志失败 taskID=%s: %v", execution.TaskID, err)
|
||
}
|
||
}
|
||
}
|
||
|
||
func shouldFlushTaskExecutionLog(ctx context.Context, execErr error) bool {
|
||
if execErr == nil {
|
||
return true
|
||
}
|
||
|
||
retryCount, hasRetryCount := asynq.GetRetryCount(ctx)
|
||
maxRetry, hasMaxRetry := asynq.GetMaxRetry(ctx)
|
||
if !hasRetryCount || !hasMaxRetry {
|
||
return true
|
||
}
|
||
return retryCount >= maxRetry
|
||
}
|
||
|
||
func handleFailedTask(ctx context.Context, execution *model.TaskExecution, t *asynq.Task, duration time.Duration, execErr error, span trace.Span) {
|
||
execution.Status = model.TaskExecutionStatusFailed
|
||
execution.ErrorMessage = execErr.Error()
|
||
logger.ErrorF(ctx, "[TaskExecutor] 任务处理失败 Type: %s TaskID: %s Duration: %d ms Error: %v", t.Type(), execution.TaskID, duration.Milliseconds(), execErr)
|
||
span.SetStatus(codes.Error, execErr.Error())
|
||
span.RecordError(execErr)
|
||
|
||
AppendLog(ctx, "[系统] 任务执行失败，耗时: %d ms，错误原因: %v", duration.Milliseconds(), execErr)
|
||
}
|
||
|
||
func handleSuccessfulTask(ctx context.Context, execution *model.TaskExecution, t *asynq.Task, duration time.Duration, result *TaskResult) {
|
||
execution.Status = model.TaskExecutionStatusSucceeded
|
||
execution.ErrorMessage = "" // 清除历史重试失败遗留的错误信息
|
||
if result != nil {
|
||
execution.Result = result.Message
|
||
if result.Detail != "" {
|
||
execution.Result = fmt.Sprintf("%s\n%s", result.Message, result.Detail)
|
||
}
|
||
}
|
||
logger.InfoF(ctx, "[TaskExecutor] 任务处理完成 Type: %s TaskID: %s Duration: %d ms", t.Type(), execution.TaskID, duration.Milliseconds())
|
||
|
||
resultMsg := "成功"
|
||
if result != nil {
|
||
resultMsg = result.Message
|
||
}
|
||
AppendLog(ctx, "[系统] 任务执行成功，耗时: %d ms，执行结果: %s", duration.Milliseconds(), resultMsg)
|
||
}
|
||
|
||
// generateTaskID 生成任务 ID
|
||
func generateTaskID(taskType string, triggeredBy string) string {
|
||
uniqueID := idgen.NextUint64ID()
|
||
return fmt.Sprintf("%s_%s_%d", triggeredBy, taskType, uniqueID)
|
||
}
|
||
|
||
// generateRetryTaskID 生成重试任务 ID
|
||
func generateRetryTaskID(originalTaskID string, retryCount int) string {
|
||
return fmt.Sprintf("retry_%d_%s", retryCount, originalTaskID)
|
||
}
|