From b331c9990a0396301e934daffe095f99d62d1c89 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期三, 27 九月 2023 15:58:49 +0800
Subject: [PATCH] 任务获取修改,状态2只返回已开始和已结束的

---
 api/v1/task.go |   73 ++++++++++++++++--------------------
 1 files changed, 32 insertions(+), 41 deletions(-)

diff --git a/api/v1/task.go b/api/v1/task.go
index e93f396..a03a4f0 100644
--- a/api/v1/task.go
+++ b/api/v1/task.go
@@ -20,6 +20,7 @@
 	"github.com/gin-gonic/gin"
 	"gorm.io/gorm"
 	"sort"
+	"sync"
 	"time"
 )
 
@@ -69,17 +70,22 @@
 
 	params.Page = 1
 
-	if params.Page <= 0 {
+	if params.PageSize <= 0 {
 		params.PageSize = channelAmount
 	}
 
-	taskResponse, code := service.NewTaskService().GetTask(params.Page, params.PageSize, service.TaskModeCurrent, nil) //鍙栬繘琛屼腑鐨勬垨鏈紑濮嬬殑
+	taskMode := constvar.TaskModeCurrent
+	if params.TaskMode == constvar.TaskModeUnStarted {
+		taskMode = params.TaskMode
+	}
+
+	taskResponse, code := service.NewTaskService().GetTask(params.Page, params.PageSize, taskMode, nil) //鍙栬繘琛屼腑鐨勬垨鏈紑濮嬬殑
 	if code != ecode.OK {
 		ctx.Fail(code)
 		return
 	}
 
-	if len(taskResponse.Tasks) < channelAmount {
+	if len(taskResponse.Tasks) < channelAmount && taskMode == constvar.TaskModeCurrent {
 		var existsChannel = make(map[int32]bool, channelAmount)
 		for _, task := range taskResponse.Tasks {
 			existsChannel[task.Channel] = true
@@ -88,7 +94,7 @@
 			if existsChannel[int32(i)] {
 				continue
 			}
-			taskResponseTemp, code := service.NewTaskService().GetTask(params.Page, 1, service.TaskModeLastFinished, []int32{int32(i)}) //鍙栦笂涓�涓畬鎴愮殑
+			taskResponseTemp, code := service.NewTaskService().GetTask(params.Page, 1, constvar.TaskModeLastFinished, []int32{int32(i)}) //鍙栦笂涓�涓畬鎴愮殑
 			if code != ecode.OK {
 				ctx.Fail(code)
 				return
@@ -102,7 +108,7 @@
 	})
 
 	for _, task := range taskResponse.Tasks {
-		if task.Procedure.Status == model.ProcedureStatusWaitProcess {
+		if !service.TaskFlagGet(task.Channel) {
 			task.CanStarted = true
 		}
 	}
@@ -110,34 +116,6 @@
 	taskResponse.Prompt = conf.Conf.Prompt
 	taskResponse.ChannelAmount = channelAmount
 
-	ctx.OkWithDetailed(taskResponse)
-}
-
-// TaskGetUnStarted
-// @Tags      Task
-// @Summary   鑾峰彇鏈紑濮嬬殑浠诲姟
-// @Produce   application/json
-// @Param     object  query    request.TaskList true  "鏌ヨ鍙傛暟"
-// @Success   200   {object}  contextx.Response{data=response.TaskData}  "鎴愬姛"
-// @Router    /v1/task/get/unStarted [get]
-func (slf *TaskApi) TaskGetUnStarted(c *gin.Context) {
-	var params request.TaskList
-	ctx, ok := contextx.NewContext(c, &params)
-	if !ok {
-		return
-	}
-	if params.Page <= 0 {
-		params.Page = 1
-	}
-	if params.PageSize <= 0 {
-		params.PageSize = 100
-	}
-
-	taskResponse, code := service.NewTaskService().GetTask(params.Page, params.PageSize, service.TaskModeUnStarted, nil) //鏃堕棿鍒颁簡鏈紑濮嬬殑
-	if code != ecode.OK {
-		ctx.Fail(code)
-		return
-	}
 	ctx.OkWithDetailed(taskResponse)
 }
 
@@ -167,13 +145,13 @@
 
 	order, err := service.NewTaskService().GetOrderByWorkOrderId(procedure.WorkOrderID)
 	if err != nil {
-		ctx.Fail(ecode.UnknownErr)
+		ctx.Fail(ecode.DBErr)
 		return
 	}
 
 	processModel, err := service.NewTaskService().GetProcessParams(procedure, order)
 	if err != nil || processModel == nil || processModel.ParamsMap == nil {
-		ctx.FailWithMsg(ecode.ParamsErr, "璇峰厛閰嶇疆宸ヨ壓鍙傛暟")
+		ctx.FailWithMsg(ecode.ParamsErr, "鏈幏鍙栧埌宸ヨ壓鍙傛暟锛岃鍦ㄥ伐鑹烘ā鍨嬪簱涓笂浼狅紒")
 		return
 	}
 
@@ -224,6 +202,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())
@@ -245,8 +231,12 @@
 		logx.Errorf("send task status update msg error:%v", err.Error())
 	}
 
+	service.TaskFlagUnset(procedure.Channel)
+
 	ctx.Ok()
 }
+
+var mutex sync.Mutex
 
 // TaskStart
 // @Tags      Task
@@ -256,6 +246,8 @@
 // @Success   200   {object}  contextx.Response{service.GetProcessModel}  "鎴愬姛"
 // @Router    /v1/task/sendProcessParams [post]
 func (slf *TaskApi) TaskStart(c *gin.Context) {
+	mutex.Lock()
+	defer mutex.Unlock()
 	var params request.SendProcessParams
 	ctx, ok := contextx.NewContext(c, &params)
 	if !ok {
@@ -274,10 +266,9 @@
 	}
 
 	if procedure.Status == model.ProcedureStatusProcessing {
-		ctx.FailWithMsg(ecode.ParamsErr, "璇ュ伐搴忓凡寮�濮嬬敓浜�")
+		ctx.Ok()
 		return
 	}
-
 	processModel, err := taskService.GetProcessParams(procedure, order)
 	if err != nil || processModel == nil || processModel.ParamsMap == nil {
 		ctx.Fail(ecode.UnknownErr)
@@ -298,7 +289,7 @@
 	})
 	if err != nil {
 		logx.Errorf("SendProcessParams update order and procedure status error:%v", err.Error())
-		ctx.FailWithMsg(ecode.NeedConfirmedErr, "鏇存敼宸ュ崟鐘舵�佸け璐�")
+		ctx.FailWithMsg(ecode.DBErr, "鏇存敼宸ュ崟鐘舵�佸け璐�")
 		return
 	}
 	plcConfig, code := service.NewDevicePlcService().GetDevicePlc()
@@ -322,7 +313,7 @@
 			}
 			return nil
 		})
-		ctx.FailWithMsg(ecode.NeedConfirmedErr, "绯熺硶锛屽伐鑹轰笅鍙戝け璐ャ��")
+		ctx.FailWithMsg(ecode.NeedConfirmedErr, "PLC璇锋眰澶辫触锛岃妫�鏌LC閰嶇疆锛�")
 		return
 	}
 	if code != ecode.OK {
@@ -332,7 +323,7 @@
 	plcConfig.CurrentTryTimes = 0
 	err = service.PlcWrite(plcConfig, constvar.PlcStartAddressTypeTotalNumber, params.Channel, order.Amount.IntPart())
 	if err != nil {
-		ctx.FailWithMsg(ecode.NeedConfirmedErr, "绯熺硶锛屽伐鑹轰笅鍙戝け璐ャ��")
+		ctx.FailWithMsg(ecode.NeedConfirmedErr, "PLC璇锋眰澶辫触锛岃妫�鏌LC閰嶇疆")
 		return
 	}
 
@@ -349,7 +340,7 @@
 	if err != nil {
 		logx.Errorf("send task status update msg error:%v", err.Error())
 	}
-
+	service.TaskFlagSet(procedure.Channel)
 	ctx.Ok()
 }
 

--
Gitblit v1.8.0