package model
|
|
import (
|
"aps_crm/constvar"
|
"aps_crm/pkg/mysqlx"
|
"fmt"
|
"github.com/shopspring/decimal"
|
"gorm.io/gorm"
|
"gorm.io/gorm/clause"
|
"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"`
|
Client Client `json:"client" gorm:"foreignKey:ClientId"`
|
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"`
|
Contact Contact `json:"contact" gorm:"foreignKey:ContactId"`
|
SaleChanceId int `json:"saleChanceId" gorm:"column:sale_chance_id;type:int;comment:销售机会id"`
|
SaleChance SaleChance `json:"SaleChance" gorm:"foreignKey:SaleChanceId"`
|
SalesDetailsId int `json:"salesDetailsId" gorm:"column:sales_details_id;type:int;comment:合同订单id"`
|
SalesDetails SalesDetails `json:"salesDetails" gorm:"foreignKey:SalesDetailsId"`
|
QuotationId int `json:"quotationId" gorm:"column:quotation_id;type:int;comment:报价单id"`
|
Quotation Quotation `json:"quotation" gorm:"foreignKey:QuotationId"`
|
ServiceContractTypeId int `json:"serviceContractTypeId" gorm:"column:service_contract_type_id;type:int;comment:合同类型id"`
|
ServiceContractType ServiceContractType `json:"serviceContractType" gorm:"foreignKey:ServiceContractTypeId"`
|
SignTime string `json:"signTime" gorm:"column:sign_time;type:varchar(255);comment:签约时间"`
|
StartTime string `json:"startTime" gorm:"column:start_time;type:varchar(255);comment:开始时间"`
|
EndTime string `json:"endTime" gorm:"column:end_time;type:varchar(255);comment:结束时间"`
|
ServiceContractStatusId int `json:"serviceContractStatusId" gorm:"column:service_contract_status_id;type:int;comment:合同状态id"`
|
ServiceContractStatus ServiceContractStatus `json:"serviceContractStatus" gorm:"foreignKey:ServiceContractStatusId"`
|
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:"column:amount_receivable;type:decimal(12,2);comment:应收金额" json:"amountReceivable"` // 应收金额
|
AmountReceived decimal.Decimal `gorm:"column:amount_received;type:decimal(12,2);comment:已收金额" json:"amountReceived"` // 已收金额
|
AmountInvoiced decimal.Decimal `gorm:"column:amount_invoiced;type:decimal(12,2);comment:已开票金额" json:"amountInvoiced"` // 已开票金额
|
AmountUnInvoiced decimal.Decimal `gorm:"column:amount_not_invoiced;type:decimal(12,2);comment:未开票金额" json:"amountUnInvoiced"` // 未开票金额
|
AmountTotal decimal.Decimal `gorm:"column:amount_total;type:decimal(12,2);comment:价税合计" json:"amountTotal"` // 价税合计
|
Products []*Product `json:"products" gorm:"many2many:service_contract_product;"`
|
gorm.Model `json:"-"`
|
}
|
|
ServiceContractSearch struct {
|
ServiceContract
|
Orm *gorm.DB
|
QueryClass constvar.ServiceContractQueryClass
|
KeywordType constvar.ServiceContractKeywordType
|
Keyword interface{}
|
OrderBy string
|
PageNum int
|
PageSize int
|
Preload bool
|
}
|
)
|
|
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().AddDate(0, 0, 30).Format("2006-01-02"))
|
case constvar.ServiceContractQueryClassExpireAfter60Day:
|
db = db.Where("end_time > ?", time.Now().AddDate(0, 0, 60).Format("2006-01-02"))
|
case constvar.ServiceContractQueryClassExpiredBefore15Day:
|
db = db.Where("end_time < ?", time.Now().AddDate(0, 0, -15).Format("2006-01-02"))
|
case constvar.ServiceContractQueryClassExpiredBefore60Day:
|
db = db.Where("end_time < ?", time.Now().AddDate(0, 0, -60).Format("2006-01-02"))
|
}
|
switch slf.KeywordType {
|
case constvar.ServiceContractKeywordContractNo:
|
db = db.Where("number like ?", fmt.Sprintf("%%%s%%", slf.Keyword))
|
case constvar.ServiceContractKeywordCustomerName:
|
db = db.Joins("Client", clause.LeftJoin).Where("Client.name like ?", fmt.Sprintf("%%%s%%", slf.Keyword))
|
case constvar.ServiceContractKeywordContractDate:
|
db = db.Where("sign_time = ?", slf.Keyword)
|
case constvar.ServiceContractKeywordContractType:
|
db = db.Joins("left join service_contract_type on service_contract_type.id = service_contract.service_contract_type_id")
|
db = db.Where("service_contract_type.name like ?", fmt.Sprintf("%%%s%%", slf.Keyword))
|
case constvar.ServiceContractKeywordContractStatus:
|
db = db.Joins("left join service_contract_status on service_contract_status.id = service_contract.service_contract_status_id")
|
db = db.Where("service_contract_status.name like ?", fmt.Sprintf("%%%s%%", slf.Keyword))
|
case constvar.ServiceContractKeywordPrincipal:
|
db = db.Joins("left join user on user.id = service_contract.member_id").Where("user.username like ?", fmt.Sprintf("%%%s%%", slf.Keyword))
|
case constvar.ServiceContractKeywordProductName:
|
db = db.Joins("left join service_contract_product scp on scp.service_contract_id = service_contract.id left join products on scp.product_id = products.id").Where("products.name like ?", fmt.Sprintf("%%%s%%", slf.Keyword))
|
case constvar.ServiceContractKeywordServiceBeginDate:
|
db = db.Where("start_time = ?", slf.Keyword)
|
case constvar.ServiceContractKeywordServiceEndDate:
|
db = db.Where("end_time = ?", slf.Keyword)
|
case constvar.ServiceContractKeywordServiceTotalPrice:
|
db = db.Where("amount_receivable = ?", slf.Keyword)
|
|
}
|
if slf.Preload {
|
db = db.
|
Preload("SaleChance").
|
Preload("SalesDetails").
|
Preload("Quotation").
|
Preload("ServiceContractType").
|
Preload("ServiceContractStatus").
|
Preload("Products").
|
Preload("Client").
|
Preload("ServiceContractType").
|
Preload("ServiceContractStatus").
|
Preload("Contact")
|
}
|
|
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, 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)
|
}
|
|
err := db.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) SetPreload(preload bool) *ServiceContractSearch {
|
slf.Preload = preload
|
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
|
}
|
|
func (slf *ServiceContractSearch) AmountReceivableAdd(tx *gorm.DB, id int, amount decimal.Decimal) error {
|
slf.Orm = tx
|
record, err := slf.SetId(id).First()
|
if err != nil {
|
return err
|
}
|
amount = record.AmountReceivable.Add(amount)
|
return slf.UpdateByMap(map[string]interface{}{"amount_receivable": amount})
|
}
|
func (slf *ServiceContractSearch) AmountReceivedAdd(tx *gorm.DB, id int, amount decimal.Decimal) error {
|
slf.Orm = tx
|
record, err := slf.SetId(id).First()
|
if err != nil {
|
return err
|
}
|
amount = record.AmountReceived.Add(amount)
|
return slf.UpdateByMap(map[string]interface{}{"amount_received": amount})
|
}
|
func (slf *ServiceContractSearch) AmountInvoicedAdd(tx *gorm.DB, id int, amount decimal.Decimal) error {
|
slf.Orm = tx
|
record, err := slf.SetId(id).First()
|
if err != nil {
|
return err
|
}
|
amount = record.AmountInvoiced.Add(amount)
|
return slf.UpdateByMap(map[string]interface{}{"amount_invoiced": amount})
|
}
|
func (slf *ServiceContractSearch) AmountNotInvoicedAdd(tx *gorm.DB, id int, amount decimal.Decimal) error {
|
slf.Orm = tx
|
record, err := slf.SetId(id).First()
|
if err != nil {
|
return err
|
}
|
amount = record.AmountUnInvoiced.Add(amount)
|
return slf.UpdateByMap(map[string]interface{}{"amount_not_invoiced": amount})
|
}
|
