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:负责人id"`
|
ContactId int `json:"contactId" gorm:"column:contact_id;type:int;comment:联系人id"`
|
SaleChanceId int `json:"saleChanceId" gorm:"column:sale_chance_id;type:int;comment:销售机会id"`
|
ContractId int `json:"contractId" gorm:"column:contract_id;type:int;comment:合同id"`
|
QuotationId int `json:"quotationId" gorm:"column:quotation_id;type:int;comment:报价单id"`
|
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:合同状态id"`
|
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
|
}
|
