From 115bd9b51f5d8eade4658f844de37516486c60e7 Mon Sep 17 00:00:00 2001 From: liujiandao <274878379@qq.com> Date: 星期六, 18 十一月 2023 17:25:25 +0800 Subject: [PATCH] crm获取aps项目模块信息 --- model/receipt.go | 65 ++++++++++++++++++++++++++++---- 1 files changed, 56 insertions(+), 9 deletions(-) diff --git a/model/receipt.go b/model/receipt.go index 9ec62f4..ff0a169 100644 --- a/model/receipt.go +++ b/model/receipt.go @@ -28,23 +28,33 @@ BankAccount BankAccount `gorm:"foreignKey:BankAccountId" json:"bankAccount"` Remark string `gorm:"column:remark;type:varchar(255);not null;default '';comment:澶囨敞" json:"remark"` // 澶囨敞 FileId int `gorm:"column:file_id;type:int;not null;default 0;comment:闄勪欢id" json:"fileId"` // 闄勪欢id - gorm.Model `json:"-"` + CrmModel } // 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 } ) -func (Receipt) TableName() string { +func (slf *Receipt) TableName() string { return "receipt" +} + +func (slf *Receipt) AfterFind(db *gorm.DB) (err error) { + err = slf.CrmModel.AfterFind(db) + if slf.CrmModel.ID == 0 { + slf.CrmModel.ID = uint(slf.Id) + } + slf.CrmModel.SetNumber(constvar.NumberPrefixOfReceipt) + return nil } func NewReceiptSearch() *ReceiptSearch { @@ -53,10 +63,42 @@ } } +func (slf *ReceiptSearch) SetSourceType(sourceType constvar.ReceiptSourceType) *ReceiptSearch { + slf.SourceType = sourceType + return slf +} + +func (slf *ReceiptSearch) SetSourceId(sourceId int) *ReceiptSearch { + slf.SourceId = sourceId + 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 +} + func (slf *ReceiptSearch) build() *gorm.DB { var db = slf.Orm.Model(&Receipt{}) 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) + } + 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 @@ -91,6 +133,11 @@ func (slf *ReceiptSearch) SetId(id int) *ReceiptSearch { slf.Id = id + return slf +} + +func (slf *ReceiptSearch) SetClientId(clientId int) *ReceiptSearch { + slf.ClientId = clientId return slf } @@ -135,7 +182,7 @@ db = db.Limit(slf.PageSize).Offset((slf.PageNum - 1) * slf.PageSize) } - err := db.Preload("Principal").Preload("Client").Find(&records).Error + err := db.Preload("Principal").Preload("Client").Preload("PaymentType").Preload("BankAccount").Order("created_at desc").Find(&records).Error return records, total, err } -- Gitblit v1.8.0