feat: Initial commit

2025-12-20 09:30:16 +08:00
commit 362237241e
189 changed files with 12035 additions and 0 deletions
@@ -0,0 +1,107 @@
+package controllers
+
+import (
+	"strconv"
+
+	"baihu/internal/database"
+	"baihu/internal/models"
+	"baihu/internal/utils"
+
+	"github.com/gin-gonic/gin"
+)
+
+type LogController struct{}
+
+func NewLogController() *LogController {
+	return &LogController{}
+}
+
+type TaskLogResponse struct {
+	ID        uint             `json:"id"`
+	TaskID    uint             `json:"task_id"`
+	TaskName  string           `json:"task_name"`
+	Command   string           `json:"command"`
+	Status    string           `json:"status"`
+	Duration  int64            `json:"duration"`
+	CreatedAt models.LocalTime `json:"created_at"`
+}
+
+func (lc *LogController) GetLogs(c *gin.Context) {
+	p := utils.ParsePagination(c)
+	taskID, _ := strconv.Atoi(c.DefaultQuery("task_id", "0"))
+	taskName := c.DefaultQuery("task_name", "")
+
+	var logs []models.TaskLog
+	var total int64
+
+	query := database.DB.Model(&models.TaskLog{})
+	if taskID > 0 {
+		query = query.Where("task_id = ?", taskID)
+	}
+
+	// 按任务名称过滤
+	if taskName != "" {
+		var taskIDs []uint
+		database.DB.Model(&models.Task{}).Where("name LIKE ?", "%"+taskName+"%").Pluck("id", &taskIDs)
+		if len(taskIDs) > 0 {
+			query = query.Where("task_id IN ?", taskIDs)
+		} else {
+			utils.PaginatedResponse(c, []TaskLogResponse{}, 0, p)
+			return
+		}
+	}
+
+	query.Count(&total)
+	query.Order("id DESC").Offset(p.Offset()).Limit(p.PageSize).Find(&logs)
+
+	taskIDList := make([]uint, 0)
+	for _, log := range logs {
+		taskIDList = append(taskIDList, log.TaskID)
+	}
+
+	var tasks []models.Task
+	database.DB.Where("id IN ?", taskIDList).Find(&tasks)
+	taskMap := make(map[uint]string)
+	for _, t := range tasks {
+		taskMap[t.ID] = t.Name
+	}
+
+	result := make([]TaskLogResponse, len(logs))
+	for i, log := range logs {
+		result[i] = TaskLogResponse{
+			ID:        log.ID,
+			TaskID:    log.TaskID,
+			TaskName:  taskMap[log.TaskID],
+			Command:   log.Command,
+			Status:    log.Status,
+			Duration:  log.Duration,
+			CreatedAt: log.CreatedAt,
+		}
+	}
+
+	utils.PaginatedResponse(c, result, total, p)
+}
+
+func (lc *LogController) GetLogDetail(c *gin.Context) {
+	id, err := strconv.Atoi(c.Param("id"))
+	if err != nil {
+		utils.BadRequest(c, "无效的日志ID")
+		return
+	}
+
+	var log models.TaskLog
+	if err := database.DB.First(&log, id).Error; err != nil {
+		utils.NotFound(c, "日志不存在")
+		return
+	}
+
+	utils.Success(c, gin.H{
+		"id":         log.ID,
+		"task_id":    log.TaskID,
+		"command":    log.Command,
+		"output":     log.Output,
+		"status":     log.Status,
+		"duration":   log.Duration,
+		"created_at": log.CreatedAt,
+	})
+}