From cab6eea89a0d82710369604ecb51f1cdb122e433 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期四, 19 十月 2023 17:26:34 +0800
Subject: [PATCH] fix

---
 service/progress.go |   53 ++++++++++++++++++++++++++++++++---------------------
 1 files changed, 32 insertions(+), 21 deletions(-)

diff --git a/service/progress.go b/service/progress.go
index 972a248..ed904f1 100644
--- a/service/progress.go
+++ b/service/progress.go
@@ -1,9 +1,10 @@
 package service
 
 import (
+	"apsClient/conf"
 	"apsClient/model"
 	"errors"
-	"gorm.io/gorm"
+	"github.com/jinzhu/gorm"
 )
 
 type ProgressService struct {
@@ -13,25 +14,29 @@
 	return &ProgressService{}
 }
 
-func (slf ProgressService) AddProgress(db *gorm.DB, procedure *model.Procedures, order *model.Order) error {
-	progress := &model.ProductionProgress{
-		WorkOrderID:   procedure.WorkOrderID,
-		OrderID:       procedure.OrderID,
-		ProcedureID:   procedure.ProceduresInfo.ProcedureID,
-		DeviceID:      procedure.DeviceID,
-		TotalQuantity: order.Amount.IntPart(),
-		Position:      procedure.Position,
+func (slf ProgressService) Add(db *gorm.DB, procedure *model.Procedures, order *model.Order) error {
+	_, err := model.NewProductionProgressSearch(db).SetProcedureId(procedure.ProcedureID).SetWorkOrderId(procedure.WorkOrderID).First()
+	if err == gorm.ErrRecordNotFound {
+		progress := &model.ProductionProgress{
+			WorkOrderID:   procedure.WorkOrderID,
+			OrderID:       procedure.OrderID,
+			ProcedureID:   procedure.ProceduresInfo.ProcedureID,
+			DeviceID:      procedure.DeviceID,
+			TotalQuantity: order.Amount.IntPart(),
+			Channel:       procedure.Channel,
+		}
+		err := model.NewProductionProgressSearch(db).Create(progress)
+		if err != nil {
+			return err
+		}
+		ProgressCacheSet(procedure.Channel, progress)
 	}
-	err := model.NewProductionProgressSearch(db).Create(progress)
-	if err != nil {
-		return err
-	}
-	ProgressCacheSet(procedure.Position, progress)
+
 	return nil
 }
 
-func (slf ProgressService) UpdateProgress(position int, finishedQuantity int64) (err error) {
-	progressCache, err := slf.GetCurrentProgress(position)
+func (slf ProgressService) UpdateProgress(channel int32, finishedQuantity int64) (err error) {
+	progressCache, err := slf.GetCurrentProgress(channel)
 	if err != nil {
 		return err
 	}
@@ -40,17 +45,17 @@
 	}
 	if finishedQuantity > progressCache.FinishedQuantity { //褰撴湁鍙樺寲鏃舵墠鏇存柊
 		progressCache.FinishedQuantity = finishedQuantity
-		ProgressCacheSet(position, progressCache)
+		ProgressCacheSet(channel, progressCache)
 		return model.NewProductionProgressSearch(nil).SetId(progressCache.ID).Save(progressCache)
 	}
 	return nil
 }
 
-func (slf ProgressService) GetCurrentProgress(position int) (progressCache *model.ProductionProgress, err error) {
+func (slf ProgressService) GetCurrentProgress(channel int32) (progressCache *model.ProductionProgress, err error) {
 	var ok bool
-	progressCache, ok = ProgressCacheGet(position)
+	progressCache, ok = ProgressCacheGet(channel)
 	if !ok {
-		progressCache, err = model.NewProductionProgressSearch(nil).SetPosition(position).SetOrder("id desc").First()
+		progressCache, err = model.NewProductionProgressSearch(nil).SetDeviceId(conf.Conf.System.DeviceId).SetChannel(channel).SetOrder("id desc").First()
 		if err == gorm.ErrRecordNotFound {
 			return nil, errors.New("progress not found")
 		}
@@ -62,8 +67,14 @@
 			progressCache = nil
 		}
 		if progressCache != nil {
-			ProgressCacheSet(position, progressCache)
+			ProgressCacheSet(channel, progressCache)
 		}
 	}
 	return
 }
+
+// GetProgressList 鑾峰彇寰呭悓姝ヨ繘搴﹀伐搴�
+func (slf ProgressService) GetProgressList() (progressList []*model.ProductionProgress, err error) {
+	progressList, err = model.NewProductionProgressSearch(nil).SetUnFinished().SetOrder("id desc").SetPage(1, 100).FindNotTotal()
+	return
+}

--
Gitblit v1.8.0