逻辑修改

zhangqian

2023-08-27 c6173b62e0fea3116075d56692b73cc0bc624815

 model/work_order.go

@@ -23,6 +23,7 @@
      StartTime   int64           `gorm:"comment:计划开始时间" json:"startTime"`
      EndTime     int64           `gorm:"comment:计划结束时间" json:"endTime"`
      Status      OrderStatus
      StatusList  []OrderStatus
   }

   OrderSearch struct {
@@ -40,8 +41,9 @@
type OrderStatus int

const (
   OrderStatusUnFinished OrderStatus = 1
   OrderStatusFinished   OrderStatus = 2
   OrderStatusWaitProcess OrderStatus = 1
   OrderStatusProcessing  OrderStatus = 2
   OrderStatusFinished    OrderStatus = 3
)

func (slf *Order) TableName() string {
@@ -84,6 +86,11 @@
   return slf
}

func (slf *OrderSearch) SetStatusList(statusList []OrderStatus) *OrderSearch {
   slf.StatusList = statusList
   return slf
}

func (slf *OrderSearch) SetPreload(preload bool) *OrderSearch {
   slf.Preload = preload
   return slf
@@ -94,11 +101,20 @@
   return slf
}

func (slf *OrderSearch) SetId(id uint) *OrderSearch {
   slf.ID = id
   return slf
}

func (slf *OrderSearch) build() *gorm.DB {
   var db = slf.Orm.Model(&Order{})

   if slf.OrderBy != "" {
      db = db.Order(slf.OrderBy)
   }

   if slf.ID != 0 {
      db = db.Where("id = ?", slf.ID)
   }

   if slf.WorkOrderID != "" {
@@ -109,6 +125,10 @@
      db = db.Where("status = ?", slf.Status)
   }

   if len(slf.StatusList) != 0 {
      db = db.Where("status != ?", slf.StatusList)
   }

   if slf.StartTimeMax != 0 {
      db = db.Where("start_time <= ?", slf.StartTimeMax)
   }