liujiandao
2023-10-12 17aaed743ab7a3520ec8eb9e4ab7c776ca807cdd
model/invoice.go
@@ -53,12 +53,30 @@
   }
}
func (slf *InvoiceSearch) SetSourceType(sourceType constvar.InvoiceSourceType) *InvoiceSearch {
   slf.SourceType = sourceType
   return slf
}
func (slf *InvoiceSearch) SetSourceId(sourceId int) *InvoiceSearch {
   slf.SourceId = sourceId
   return slf
}
func (slf *InvoiceSearch) build() *gorm.DB {
   var db = slf.Orm.Model(&Invoice{})
   if slf.Id != 0 {
      db = db.Where("id = ?", slf.Id)
   }
   if slf.SourceType > 0 {
      db = db.Where("source_type = ?", slf.SourceType)
   }
   if slf.SourceId > 0 {
      db = db.Where("source_id = ?", slf.SourceId)
   }
   return db
}