From d5c533de0c2ccb5614ea3c600ede9c13da77e127 Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期二, 08 八月 2023 09:37:51 +0800 Subject: [PATCH] 合并服务管理 --- model/serviceContract.go | 374 ++++++++++++++++++++++++++++------------------------ 1 files changed, 202 insertions(+), 172 deletions(-) diff --git a/model/serviceContract.go b/model/serviceContract.go index 5b20512..5c5a497 100644 --- a/model/serviceContract.go +++ b/model/serviceContract.go @@ -1,172 +1,202 @@ -package model - -import ( - "aps_crm/constvar" - "aps_crm/pkg/mysqlx" - "gorm.io/gorm" - "time" -) - -type ( - ServiceContract struct { - Id int `json:"id" gorm:"column:id;primary_key;AUTO_INCREMENT"` - ClientId int `json:"clientId" gorm:"column:client_id;type:int;comment:瀹㈡埛id"` - Number string `json:"number" gorm:"column:number;type:varchar(255);comment:鍚堝悓缂栧彿"` - MemberId int `json:"memberId" gorm:"column:member_id;type:int;comment:璐熻矗浜篿d"` - ContactId int `json:"contactId" gorm:"column:contact_id;type:int;comment:鑱旂郴浜篿d"` - SaleChanceId int `json:"saleChanceId" gorm:"column:sale_chance_id;type:int;comment:閿�鍞満浼歩d"` - ContractId int `json:"contractId" gorm:"column:contract_id;type:int;comment:鍚堝悓id"` - QuotationId int `json:"quotationId" gorm:"column:quotation_id;type:int;comment:鎶ヤ环鍗昳d"` - ServiceContractTypeId int `json:"serviceContractTypeId" gorm:"column:service_contract_type_id;type:int;comment:鍚堝悓绫诲瀷id"` - SignTime time.Time `json:"signTime" gorm:"column:sign_time;type:datetime;comment:绛剧害鏃堕棿"` - StartTime time.Time `json:"startTime" gorm:"column:start_time;type:datetime;comment:寮�濮嬫椂闂�"` - EndTime time.Time `json:"endTime" gorm:"column:end_time;type:datetime;comment:缁撴潫鏃堕棿"` - ServiceContractStatusId int `json:"serviceContractStatusId" gorm:"column:service_contract_status_id;type:int;comment:鍚堝悓鐘舵�乮d"` - ServiceTimes int `json:"serviceTimes" gorm:"column:service_times;type:int;comment:鏈嶅姟娆℃暟"` - Terms string `json:"terms" gorm:"column:terms;type:text;comment:鏉℃"` - Remark string `json:"remark" gorm:"column:remark;type:text;comment:澶囨敞"` - Products []Product `json:"products" gorm:"many2many:serviceContract_product;"` - gorm.Model `json:"-"` - } - - ServiceContractSearch struct { - ServiceContract - - Orm *gorm.DB - QueryClass constvar.ServiceContractQueryClass - KeywordType constvar.ServiceContractKeywordType - Keyword interface{} - OrderBy string - PageNum int - PageSize int - } -) - -func (ServiceContract) TableName() string { - return "service_contract" -} - -func NewServiceContractSearch() *ServiceContractSearch { - return &ServiceContractSearch{ - Orm: mysqlx.GetDB(), - } -} - -func (slf *ServiceContractSearch) build() *gorm.DB { - var db = slf.Orm.Model(&ServiceContract{}) - if slf.Id != 0 { - db = db.Where("id = ?", slf.Id) - } - switch slf.QueryClass { - case constvar.ServiceContractQueryClassExpireAfter30Day: - db = db.Where("end_time > ?", time.Now(), time.Now().AddDate(0, 0, 30)) - case constvar.ServiceContractQueryClassExpireAfter60Day: - db = db.Where("end_time > ?", time.Now(), time.Now().AddDate(0, 0, 60)) - case constvar.ServiceContractQueryClassExpiredBefore15Day: - db = db.Where("end_time < ?", time.Now().AddDate(0, 0, -15)) - case constvar.ServiceContractQueryClassExpiredBefore60Day: - db = db.Where("end_time < ?", time.Now().AddDate(0, 0, -60)) - - } - switch slf.KeywordType { - case constvar.ServiceContractKeywordContractNo: - db = db.Where("number = ?", slf.Keyword) - case constvar.ServiceContractKeywordCustomerName: - db = db.Where("client_id = ?", slf.Keyword) - case constvar.ServiceContractKeywordContractDate: - db = db.Where("sign_time = ?", slf.Keyword) - case constvar.ServiceContractKeywordContractType: - db = db.Where("service_contract_type_id = ?", slf.Keyword) - case constvar.ServiceContractKeywordContractStatus: - db = db.Where("service_contract_status_id = ?", slf.Keyword) - case constvar.ServiceContractKeywordPrincipal: - db = db.Where("member_id = ?", slf.Keyword) - case constvar.ServiceContractKeywordProductName: - subQuery := db.Table("service_contract_id").Select("product_id").Where("product_id = ?", slf.Keyword) - db = db.Where("id = ?", subQuery) - case constvar.ServiceContractKeywordServiceBeginDate: - db = db.Where("start_time = ?", slf.Keyword) - case constvar.ServiceContractKeywordServiceEndDate: - db = db.Where("end_time = ?", slf.Keyword) - case constvar.ServiceContractKeywordServiceTotalPrice: - //todo - - } - - return db -} - -func (slf *ServiceContractSearch) Create(record *ServiceContract) error { - var db = slf.build() - return db.Create(record).Error -} - -func (slf *ServiceContractSearch) Update(record *ServiceContract) error { - var db = slf.build() - return db.Updates(record).Error -} - -func (slf *ServiceContractSearch) Delete() error { - var db = slf.build() - return db.Delete(&ServiceContract{}).Error -} - -func (slf *ServiceContractSearch) Find() (*ServiceContract, error) { - var db = slf.build() - var record = &ServiceContract{} - err := db.First(record).Error - return record, err -} - -func (slf *ServiceContractSearch) FindAll() ([]*ServiceContract, int64, error) { - var db = slf.build() - var records = make([]*ServiceContract, 0) - var total int64 - if err := db.Count(&total).Error; err != nil { - return records, total, err - } - if slf.PageNum > 0 && slf.PageSize > 0 { - db = db.Limit(slf.PageSize).Offset((slf.PageNum - 1) * slf.PageSize) - } - - if slf.PageNum > 0 && slf.PageSize > 0 { - db = db.Limit(slf.PageSize).Offset((slf.PageNum - 1) * slf.PageSize) - } - - err := db.Preload("Products").Find(&records).Error - return records, total, err -} - -func (slf *ServiceContractSearch) SetId(id int) *ServiceContractSearch { - slf.Id = id - return slf -} - -func (slf *ServiceContractSearch) SetKeywordType(keyword constvar.ServiceContractKeywordType) *ServiceContractSearch { - slf.KeywordType = keyword - return slf -} - -func (slf *ServiceContractSearch) SetQueryClass(queryClass constvar.ServiceContractQueryClass) *ServiceContractSearch { - slf.QueryClass = queryClass - return slf -} - -func (slf *ServiceContractSearch) SetKeyword(keyword string) *ServiceContractSearch { - slf.Keyword = keyword - return slf -} - -func (slf *ServiceContractSearch) SetPage(page, size int) *ServiceContractSearch { - slf.PageNum, slf.PageSize = page, size - return slf -} - -func (slf *ServiceContractSearch) SetOrder(order string) *ServiceContractSearch { - slf.OrderBy = order - return slf -} -func (slf *ServiceContractSearch) SetIds(ids []int) *ServiceContractSearch { - slf.Orm = slf.Orm.Where("id in (?)", ids) - return slf -} +package model + +import ( + "aps_crm/constvar" + "aps_crm/pkg/mysqlx" + "fmt" + "github.com/shopspring/decimal" + "gorm.io/gorm" + "time" +) + +type ( + ServiceContract struct { + Id int `json:"id" gorm:"column:id;primary_key;AUTO_INCREMENT"` + ClientId int `json:"clientId" gorm:"column:client_id;type:int;comment:瀹㈡埛id"` + Number string `json:"number" gorm:"column:number;type:varchar(255);comment:鍚堝悓缂栧彿"` + MemberId int `json:"memberId" gorm:"column:member_id;type:int;comment:璐熻矗浜篿d"` + ContactId int `json:"contactId" gorm:"column:contact_id;type:int;comment:鑱旂郴浜篿d"` + SaleChanceId int `json:"saleChanceId" gorm:"column:sale_chance_id;type:int;comment:閿�鍞満浼歩d"` + ContractId int `json:"contractId" gorm:"column:contract_id;type:int;comment:鍚堝悓id"` + QuotationId int `json:"quotationId" gorm:"column:quotation_id;type:int;comment:鎶ヤ环鍗昳d"` + ServiceContractTypeId int `json:"serviceContractTypeId" gorm:"column:service_contract_type_id;type:int;comment:鍚堝悓绫诲瀷id"` + SignTime time.Time `json:"signTime" gorm:"column:sign_time;type:datetime;comment:绛剧害鏃堕棿"` + StartTime time.Time `json:"startTime" gorm:"column:start_time;type:datetime;comment:寮�濮嬫椂闂�"` + EndTime time.Time `json:"endTime" gorm:"column:end_time;type:datetime;comment:缁撴潫鏃堕棿"` + ServiceContractStatusId int `json:"serviceContractStatusId" gorm:"column:service_contract_status_id;type:int;comment:鍚堝悓鐘舵�乮d"` + ServiceTimes int `json:"serviceTimes" gorm:"column:service_times;type:int;comment:鏈嶅姟娆℃暟"` + Terms string `json:"terms" gorm:"column:terms;type:text;comment:鏉℃"` + Remark string `json:"remark" gorm:"column:remark;type:text;comment:澶囨敞"` + AmountReceivable decimal.Decimal `gorm:"amount_receivable" json:"amountReceivable"` // 搴旀敹閲戦 + AmountReceived decimal.Decimal `gorm:"amount_received" json:"amountReceived"` // 宸叉敹閲戦 + AmountInvoiced decimal.Decimal `gorm:"amount_invoiced" json:"amountInvoiced"` // 宸插紑绁ㄩ噾棰� + Products []Product `json:"products" gorm:"many2many:serviceContract_product;"` + gorm.Model `json:"-"` + } + + ServiceContractSearch struct { + ServiceContract + + Orm *gorm.DB + QueryClass constvar.ServiceContractQueryClass + KeywordType constvar.ServiceContractKeywordType + Keyword interface{} + OrderBy string + PageNum int + PageSize int + } +) + +func (ServiceContract) TableName() string { + return "service_contract" +} + +func NewServiceContractSearch() *ServiceContractSearch { + return &ServiceContractSearch{ + Orm: mysqlx.GetDB(), + } +} + +func (slf *ServiceContractSearch) build() *gorm.DB { + var db = slf.Orm.Model(&ServiceContract{}) + if slf.Id != 0 { + db = db.Where("id = ?", slf.Id) + } + switch slf.QueryClass { + case constvar.ServiceContractQueryClassExpireAfter30Day: + db = db.Where("end_time > ?", time.Now(), time.Now().AddDate(0, 0, 30)) + case constvar.ServiceContractQueryClassExpireAfter60Day: + db = db.Where("end_time > ?", time.Now(), time.Now().AddDate(0, 0, 60)) + case constvar.ServiceContractQueryClassExpiredBefore15Day: + db = db.Where("end_time < ?", time.Now().AddDate(0, 0, -15)) + case constvar.ServiceContractQueryClassExpiredBefore60Day: + db = db.Where("end_time < ?", time.Now().AddDate(0, 0, -60)) + + } + switch slf.KeywordType { + case constvar.ServiceContractKeywordContractNo: + db = db.Where("number = ?", slf.Keyword) + case constvar.ServiceContractKeywordCustomerName: + db = db.Where("client_id = ?", slf.Keyword) + case constvar.ServiceContractKeywordContractDate: + db = db.Where("sign_time = ?", slf.Keyword) + case constvar.ServiceContractKeywordContractType: + db = db.Where("service_contract_type_id = ?", slf.Keyword) + case constvar.ServiceContractKeywordContractStatus: + db = db.Where("service_contract_status_id = ?", slf.Keyword) + case constvar.ServiceContractKeywordPrincipal: + db = db.Where("member_id = ?", slf.Keyword) + case constvar.ServiceContractKeywordProductName: + subQuery := db.Table("service_contract_id").Select("product_id").Where("product_id = ?", slf.Keyword) + db = db.Where("id = ?", subQuery) + case constvar.ServiceContractKeywordServiceBeginDate: + db = db.Where("start_time = ?", slf.Keyword) + case constvar.ServiceContractKeywordServiceEndDate: + db = db.Where("end_time = ?", slf.Keyword) + case constvar.ServiceContractKeywordServiceTotalPrice: + //todo + + } + + return db +} + +func (slf *ServiceContractSearch) Create(record *ServiceContract) error { + var db = slf.build() + return db.Create(record).Error +} + +func (slf *ServiceContractSearch) Update(record *ServiceContract) error { + var db = slf.build() + return db.Updates(record).Error +} + +func (slf *ServiceContractSearch) Delete() error { + var db = slf.build() + return db.Delete(&ServiceContract{}).Error +} + +func (slf *ServiceContractSearch) Find() (*ServiceContract, error) { + var db = slf.build() + var record = &ServiceContract{} + err := db.First(record).Error + return record, err +} + +func (slf *ServiceContractSearch) FindAll() ([]*ServiceContract, int64, error) { + var db = slf.build() + var records = make([]*ServiceContract, 0) + var total int64 + if err := db.Count(&total).Error; err != nil { + return records, total, err + } + if slf.PageNum > 0 && slf.PageSize > 0 { + db = db.Limit(slf.PageSize).Offset((slf.PageNum - 1) * slf.PageSize) + } + + if slf.PageNum > 0 && slf.PageSize > 0 { + db = db.Limit(slf.PageSize).Offset((slf.PageNum - 1) * slf.PageSize) + } + + err := db.Preload("Products").Find(&records).Error + return records, total, err +} + +func (slf *ServiceContractSearch) SetId(id int) *ServiceContractSearch { + slf.Id = id + return slf +} + +func (slf *ServiceContractSearch) SetKeywordType(keyword constvar.ServiceContractKeywordType) *ServiceContractSearch { + slf.KeywordType = keyword + return slf +} + +func (slf *ServiceContractSearch) SetQueryClass(queryClass constvar.ServiceContractQueryClass) *ServiceContractSearch { + slf.QueryClass = queryClass + return slf +} + +func (slf *ServiceContractSearch) SetKeyword(keyword string) *ServiceContractSearch { + slf.Keyword = keyword + return slf +} + +func (slf *ServiceContractSearch) SetPage(page, size int) *ServiceContractSearch { + slf.PageNum, slf.PageSize = page, size + return slf +} + +func (slf *ServiceContractSearch) SetOrder(order string) *ServiceContractSearch { + slf.OrderBy = order + return slf +} +func (slf *ServiceContractSearch) SetIds(ids []int) *ServiceContractSearch { + slf.Orm = slf.Orm.Where("id in (?)", ids) + return slf +} + +func (slf *ServiceContractSearch) UpdateByMap(upMap map[string]interface{}) error { + var ( + db = slf.build() + ) + + if err := db.Updates(upMap).Error; err != nil { + return fmt.Errorf("update by map err: %v, upMap: %+v", err, upMap) + } + + return nil +} + +func (slf *ServiceContractSearch) First() (*ServiceContract, error) { + var ( + record = new(ServiceContract) + db = slf.build() + ) + + if err := db.First(record).Error; err != nil { + return record, err + } + + return record, nil +} -- Gitblit v1.8.0