From c073c915b23db5a3bde659ceed21849b5d0e484c Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期日, 08 十月 2023 10:34:31 +0800
Subject: [PATCH] 获取任务去掉时间限制
---
api/v1/task.go | 17 +++++++++++++++--
1 files changed, 15 insertions(+), 2 deletions(-)
diff --git a/api/v1/task.go b/api/v1/task.go
index 558b127..2e2e8ea 100644
--- a/api/v1/task.go
+++ b/api/v1/task.go
@@ -68,10 +68,14 @@
return
}
+ taskCount := service.NewTaskService().NewTaskCount()
params.Page = 1
-
if params.PageSize <= 0 {
- params.PageSize = channelAmount
+ if params.TaskMode == constvar.TaskModeUnStarted {
+ params.PageSize = int(taskCount)
+ } else {
+ params.PageSize = channelAmount
+ }
}
taskMode := constvar.TaskModeCurrent
@@ -115,6 +119,7 @@
taskResponse.Prompt = conf.Conf.Prompt
taskResponse.ChannelAmount = channelAmount
+ taskResponse.TaskCount = taskCount
ctx.OkWithDetailed(taskResponse)
}
@@ -202,6 +207,14 @@
ctx.Fail(code)
return
}
+ if procedure.Status == model.ProcedureStatusFinished {
+ ctx.Ok()
+ return
+ }
+ if procedure.Status != model.ProcedureStatusProcessing {
+ ctx.Fail(ecode.ParamsErr)
+ return
+ }
err := service.NewTaskService().UpdateProcedureStatus(nil, id, model.ProcedureStatusFinished, procedure.Channel)
if err != nil {
logx.Errorf("UpdateProcedureStatus err: %v", err.Error())
--
Gitblit v1.8.0