From f05f4cca1340f0ddec7261d4dbe65dd331224423 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期四, 28 九月 2023 15:59:30 +0800
Subject: [PATCH] 接收设备信息改变

---
 api/v1/task.go |    9 +++++++++
 1 files changed, 9 insertions(+), 0 deletions(-)

diff --git a/api/v1/task.go b/api/v1/task.go
index 558b127..419316b 100644
--- a/api/v1/task.go
+++ b/api/v1/task.go
@@ -115,6 +115,7 @@
 
 	taskResponse.Prompt = conf.Conf.Prompt
 	taskResponse.ChannelAmount = channelAmount
+	taskResponse.TaskCount = service.NewTaskService().NewTaskCount()
 
 	ctx.OkWithDetailed(taskResponse)
 }
@@ -202,6 +203,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