报表功能开发

liujiandao

2023-09-27 b3a318b7f707ca49fa9127881bbe709654eaa761

 models/operation.go

@@ -41,6 +41,7 @@
      Orm      *gorm.DB
      Preload  bool
      Disuse   bool
      Ids      []int
   }
)

@@ -102,6 +103,15 @@
   return slf
}

func (slf *OperationSearch) SetIds(ids []int) *OperationSearch {
   slf.Ids = ids
   return slf
}
func (slf *OperationSearch) SetStatus(status constvar.OperationStatus) *OperationSearch {
   slf.Status = status
   return slf
}

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

@@ -134,6 +144,14 @@

   if slf.Disuse {
      db = db.Where("operation_type_id = ?", 0)
   }

   if len(slf.Ids) > 0 {
      db = db.Where("id in (?)", slf.Ids)
   }

   if slf.Status > 0 {
      db = db.Where("status = ?", slf.Status)
   }

   return db
@@ -236,7 +254,7 @@
   if slf.PageNum*slf.PageSize > 0 {
      db = db.Offset((slf.PageNum - 1) * slf.PageSize).Limit(slf.PageSize)
   }
   if err := db.Find(&records).Error; err != nil {
   if err := db.Preload("FromLocation").Preload("ToLocation").Find(&records).Error; err != nil {
      return records, total, fmt.Errorf("find records err: %v", err)
   }

@@ -252,7 +270,7 @@
   if slf.PageNum*slf.PageSize > 0 {
      db = db.Offset((slf.PageNum - 1) * slf.PageSize).Limit(slf.PageSize)
   }
   if err := db.Find(&records).Error; err != nil {
   if err := db.Preload("FromLocation").Preload("ToLocation").Find(&records).Error; err != nil {
      return records, fmt.Errorf("find records err: %v", err)
   }