zhangqian
2023-11-01 530fed8ec225453572d57b15c200ab062c335457
model/receipt.go
@@ -34,12 +34,13 @@
   // ReceiptSearch 收款单搜索条件
   ReceiptSearch struct {
      Receipt
      Orm         *gorm.DB
      QueryClass  constvar.ReceiptQueryClass
      KeywordType constvar.ReceiptKeywordType
      Keyword     string
      PageNum     int
      PageSize    int
      Orm          *gorm.DB
      QueryClass   constvar.ReceiptQueryClass
      KeywordType  constvar.ReceiptKeywordType
      Keyword      string
      PageNum      int
      PageSize     int
      PrincipalIds []int
   }
)
@@ -72,6 +73,11 @@
   return slf
}
func (slf *ReceiptSearch) SetPrincipalIds(principalIds []int) *ReceiptSearch {
   slf.PrincipalIds = principalIds
   return slf
}
func (slf *ReceiptSearch) SetPage(page, size int) *ReceiptSearch {
   slf.PageNum, slf.PageSize = page, size
   return slf
@@ -90,6 +96,9 @@
   }
   if slf.ClientId != 0 {
      db = db.Where("client_id = ?", slf.ClientId)
   }
   if len(slf.PrincipalIds) != 0 {
      db = db.Where("principal_id in ?", slf.PrincipalIds)
   }
   return db
@@ -173,7 +182,7 @@
      db = db.Limit(slf.PageSize).Offset((slf.PageNum - 1) * slf.PageSize)
   }
   err := db.Preload("Principal").Preload("Client").Preload("PaymentType").Preload("BankAccount").Find(&records).Error
   err := db.Preload("Principal").Preload("Client").Preload("PaymentType").Preload("BankAccount").Order("created_at desc").Find(&records).Error
   return records, total, err
}