From 10c65da3d2af7056f48d9301e83f53f102f76e18 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期一, 30 十月 2023 14:52:11 +0800
Subject: [PATCH] fix

---
 model/procedures.go |   65 ++++++++++++++++++++++++++------
 1 files changed, 52 insertions(+), 13 deletions(-)

diff --git a/model/procedures.go b/model/procedures.go
index 90b71b5..7f8c8b9 100644
--- a/model/procedures.go
+++ b/model/procedures.go
@@ -11,15 +11,17 @@
 
 type (
 	Procedures struct {
-		gorm.Model         `json:"-"`
-		WorkOrderID        string `gorm:"index;type:varchar(191);not null;comment:宸ュ崟ID" json:"-"`
-		OrderID            string `gorm:"index;type:varchar(191);not null;comment:璁㈠崟ID" json:"-"`
-		DeviceID           string `gorm:"index;type:varchar(191);comment:璁惧ID" json:"deviceId"`
-		ProcedureID        string `gorm:"index;type:varchar(191);comment:宸ュ簭ID" json:"procedureId"`
-		Channel            int32  `gorm:"index;comment:閫氶亾" json:"channel"`                //閫氶亾
-		ProcessModelNumber string `gorm:"index;comment:宸ヨ壓妯″瀷缂栧彿" json:"processModelNumber"` //宸ヨ壓妯″瀷缂栧彿
-		StartTime          int64  `gorm:"comment:璁″垝寮�濮嬫椂闂�" json:"startTime"`
-		EndTime            int64  `gorm:"comment:璁″垝缁撴潫鏃堕棿" json:"endTime"`
+		gorm.Model
+		WorkOrderID        string `gorm:"index;type:varchar(191);not null" json:"-"`
+		OrderID            string `gorm:"index;type:varchar(191);not null" json:"-"`
+		DeviceID           string `gorm:"index;type:varchar(191)" json:"deviceId"`
+		ProcedureID        string `gorm:"index;type:varchar(191)" json:"procedureId"`
+		Channel            int32  `gorm:"index;" json:"channel"`            //閫氶亾
+		ProcessModelNumber string `gorm:"index;" json:"processModelNumber"` //宸ヨ壓妯″瀷缂栧彿
+		StartTime          int64  `json:"startTime"`
+		EndTime            int64  `json:"endTime"`
+		RealStartTime      int64  `json:"realStartTime"`
+		RealEndTime        int64  `json:"realEndTime"`
 		Status             ProcedureStatus
 		ProcedureData      string                  `json:"-"`                  //common.ProductProcedure  json涓�
 		ProceduresInfo     common.ProductProcedure `json:"procedure" gorm:"-"` //common.ProductProcedure  瀵硅薄
@@ -32,11 +34,15 @@
 		PageSize     int
 		Orm          *gorm.DB
 		Preload      bool
+		StartTimeMin int64
 		StartTimeMax int64
 		EndTimeMin   int64
+		EndTimeMax   int64
 		StatusNot    ProcedureStatus
 		ProcedureIds []string
 		Channels     []int32
+		Offset       int
+		Limit        int
 	}
 )
 
@@ -79,6 +85,11 @@
 	return slf
 }
 
+func (slf *ProceduresSearch) SetOffset(offset, limit int) *ProceduresSearch {
+	slf.Offset, slf.Limit = offset, limit
+	return slf
+}
+
 func (slf *ProceduresSearch) SetOrder(order string) *ProceduresSearch {
 	slf.Order = order
 	return slf
@@ -94,6 +105,11 @@
 	return slf
 }
 
+func (slf *ProceduresSearch) SetStartTimeMin(ts int64) *ProceduresSearch {
+	slf.StartTimeMin = ts
+	return slf
+}
+
 func (slf *ProceduresSearch) SetStartTimeMax(ts int64) *ProceduresSearch {
 	slf.StartTimeMax = ts
 	return slf
@@ -101,6 +117,11 @@
 
 func (slf *ProceduresSearch) SetEndTimeMin(ts int64) *ProceduresSearch {
 	slf.EndTimeMin = ts
+	return slf
+}
+
+func (slf *ProceduresSearch) SetEndTimeMax(ts int64) *ProceduresSearch {
+	slf.EndTimeMax = ts
 	return slf
 }
 
@@ -139,6 +160,11 @@
 	return slf
 }
 
+func (slf *ProceduresSearch) SetChannel(channel int32) *ProceduresSearch {
+	slf.Channel = channel
+	return slf
+}
+
 func (slf *ProceduresSearch) build() *gorm.DB {
 	var db = slf.Orm.Model(&Procedures{})
 
@@ -166,8 +192,16 @@
 		db = db.Where("start_time <= ?", slf.StartTimeMax)
 	}
 
+	if slf.StartTimeMin != 0 {
+		db = db.Where("start_time >= ?", slf.EndTimeMin)
+	}
+
 	if slf.EndTimeMin != 0 {
 		db = db.Where("end_time > ?", slf.EndTimeMin)
+	}
+
+	if slf.EndTimeMax != 0 {
+		db = db.Where("end_time < ?", slf.StartTimeMax)
 	}
 
 	if slf.Status != 0 {
@@ -207,11 +241,11 @@
 // CreateBatch 鎵归噺鎻掑叆
 func (slf *ProceduresSearch) CreateBatch(records []*Procedures) error {
 	var db = slf.build()
-
-	if err := db.Create(&records).Error; err != nil {
-		return fmt.Errorf("create batch err: %v, records: %+v", err, records)
+	for _, record := range records {
+		if err := db.Create(record).Error; err != nil {
+			return fmt.Errorf("create batch err: %v, records: %+v", err, records)
+		}
 	}
-
 	return nil
 }
 
@@ -301,6 +335,11 @@
 	if slf.PageNum*slf.PageSize > 0 {
 		db = db.Offset((slf.PageNum - 1) * slf.PageSize).Limit(slf.PageSize)
 	}
+
+	if slf.Limit > 0 {
+		db = db.Offset(slf.Offset).Limit(slf.Limit)
+	}
+
 	if err := db.Find(&records).Error; err != nil {
 		return records, total, fmt.Errorf("find records err: %v", err)
 	}

--
Gitblit v1.8.0