zhangqian
2023-08-08 d5c533de0c2ccb5614ea3c600ede9c13da77e127
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:负责人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
}
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:负责人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:备注"`
      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
}