fix: agent workdir error

This commit is contained in:
engigu
2025-12-29 19:52:39 +08:00
parent 78c3707cce
commit 6c666d998c
6 changed files with 84 additions and 13 deletions
+10 -1
View File
@@ -96,8 +96,17 @@ func (tc *TaskController) CreateTask(c *gin.Context) {
func (tc *TaskController) GetTasks(c *gin.Context) {
p := utils.ParsePagination(c)
name := c.DefaultQuery("name", "")
agentIDStr := c.DefaultQuery("agent_id", "")
var agentID *uint
if agentIDStr != "" {
if id, err := strconv.ParseUint(agentIDStr, 10, 32); err == nil {
uid := uint(id)
agentID = &uid
}
}
tasks, total := tc.taskService.GetTasksWithPagination(p.Page, p.PageSize, name)
tasks, total := tc.taskService.GetTasksWithPagination(p.Page, p.PageSize, name, agentID)
utils.PaginatedResponse(c, tasks, total, p)
}
+4 -1
View File
@@ -39,7 +39,7 @@ func (ts *TaskService) GetTasks() []models.Task {
}
// GetTasksWithPagination 分页获取任务列表
func (ts *TaskService) GetTasksWithPagination(page, pageSize int, name string) ([]models.Task, int64) {
func (ts *TaskService) GetTasksWithPagination(page, pageSize int, name string, agentID *uint) ([]models.Task, int64) {
var tasks []models.Task
var total int64
@@ -47,6 +47,9 @@ func (ts *TaskService) GetTasksWithPagination(page, pageSize int, name string) (
if name != "" {
query = query.Where("name LIKE ?", "%"+name+"%")
}
if agentID != nil {
query = query.Where("agent_id = ?", *agentID)
}
query.Count(&total)
query.Order("id DESC").Offset((page - 1) * pageSize).Limit(pageSize).Find(&tasks)