压缩历史至 95081aff

2026-10-03 07:06:36 +08:00 · 2026-06-08 20:34:27 +08:00
commit 8a782525de
435 changed files with 71146 additions and 0 deletions
@@ -0,0 +1,24 @@
+/*
+Copyright 2025 linux.do
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package task
+
+const (
+	InvalidTaskType    = "无效的任务类型"
+	InvalidTimeRange   = "无效的时间范围"
+	TaskDispatchFailed = "任务下发失败"
+	UserIDRequired     = "用户ID必填"
+)
@@ -0,0 +1,120 @@
+/*
+Copyright 2025 linux.do
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package task
+
+import (
+	"encoding/json"
+	"fmt"
+	"net/http"
+	"time"
+
+	"github.com/gin-gonic/gin"
+	"github.com/hibiken/asynq"
+	"github.com/linux-do/credit/internal/task"
+	"github.com/linux-do/credit/internal/task/scheduler"
+	"github.com/linux-do/credit/internal/util"
+)
+
+// ListTaskTypes 获取支持的任务类型列表
+// @Tags admin
+// @Produce json
+// @Success 200 {object} util.ResponseAny
+// @Router /api/v1/admin/tasks/types [get]
+func ListTaskTypes(c *gin.Context) {
+	c.JSON(http.StatusOK, util.OK(task.DispatchableTasks))
+}
+
+// DispatchTaskRequest 下发任务请求
+type DispatchTaskRequest struct {
+	TaskType  string     `json:"task_type" binding:"required"`
+	StartTime *time.Time `json:"start_time"`
+	EndTime   *time.Time `json:"end_time"`
+	UserID    *uint64    `json:"user_id"`
+}
+
+// DispatchTask 下发任务
+// @Tags admin
+// @Accept json
+// @Produce json
+// @Param request body DispatchTaskRequest true "request body"
+// @Success 200 {object} util.ResponseAny
+// @Router /api/v1/admin/tasks/dispatch [post]
+func DispatchTask(c *gin.Context) {
+	var req DispatchTaskRequest
+	if err := c.ShouldBindJSON(&req); err != nil {
+		c.JSON(http.StatusBadRequest, util.Err(err.Error()))
+		return
+	}
+
+	meta := task.GetTaskMeta(req.TaskType)
+	if meta == nil {
+		c.JSON(http.StatusBadRequest, util.Err(InvalidTaskType))
+		return
+	}
+
+	var taskInfo *asynq.Task
+	var taskID string
+
+	switch req.TaskType {
+	case task.TaskTypeOrderSync:
+		if req.StartTime != nil && req.EndTime != nil {
+			if req.EndTime.Before(*req.StartTime) {
+				c.JSON(http.StatusBadRequest, util.Err(InvalidTimeRange))
+				return
+			}
+			payload, _ := json.Marshal(map[string]interface{}{
+				"start_time": req.StartTime,
+				"end_time":   req.EndTime,
+			})
+			taskInfo = asynq.NewTask(meta.AsynqTask, payload)
+			taskID = fmt.Sprintf("manual_%s_%s_%s", req.TaskType,
+				req.StartTime.Format("20060102150405"),
+				req.EndTime.Format("20060102150405"))
+		} else {
+			taskInfo = asynq.NewTask(meta.AsynqTask, nil)
+			taskID = fmt.Sprintf("manual_%s", req.TaskType)
+		}
+
+	case task.TaskTypeUserGamification:
+		if req.UserID == nil || *req.UserID == 0 {
+			taskInfo = asynq.NewTask(task.UpdateUserGamificationScoresTask, nil)
+			taskID = fmt.Sprintf("manual_%s", req.TaskType)
+		} else {
+			payload, _ := json.Marshal(map[string]interface{}{
+				"user_id": *req.UserID,
+			})
+			taskInfo = asynq.NewTask(meta.AsynqTask, payload)
+			taskID = fmt.Sprintf("manual_%s_user_%d", req.TaskType, *req.UserID)
+		}
+	default:
+		taskInfo = asynq.NewTask(meta.AsynqTask, nil)
+		taskID = fmt.Sprintf("manual_%s", req.TaskType)
+	}
+
+	_, err := scheduler.AsynqClient.Enqueue(
+		taskInfo,
+		asynq.TaskID(taskID),
+		asynq.MaxRetry(meta.MaxRetry),
+		asynq.Queue(meta.Queue),
+	)
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, util.Err(fmt.Sprintf("%s: %v", TaskDispatchFailed, err)))
+		return
+	}
+
+	c.JSON(http.StatusOK, util.OKNil())
+}