zhangqian
2023-11-01 530fed8ec225453572d57b15c200ab062c335457
model/serviceContract.go
@@ -17,7 +17,9 @@
      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"`
      Member                  User                  `json:"member" gorm:"foreignKey:MemberId"`
      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"`
@@ -34,12 +36,14 @@
      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"`       // 已开票金额
      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:"-"`
      CodeStandID             string                `json:"codeStandID" gorm:"column:code_stand_id;type:varchar(255);comment:编码id"`
      CrmModel
   }
   ServiceContractSearch struct {
@@ -52,10 +56,11 @@
      PageNum     int
      PageSize    int
      Preload     bool
      MemberIds   []int
   }
)
func (ServiceContract) TableName() string {
func (slf *ServiceContract) TableName() string {
   return "service_contract"
}
@@ -105,8 +110,16 @@
      db = db.Where("amount_receivable = ?", slf.Keyword)
   }
   if len(slf.MemberIds) > 0 {
      db = db.Where("service_contract.member_id in ?", slf.MemberIds)
   }
   if slf.Preload {
      db = db.
         Preload("Client").
         Preload("Member").
         Preload("Contact").
         Preload("SaleChance").
         Preload("SalesDetails").
         Preload("Quotation").
@@ -115,9 +128,26 @@
         Preload("Products").
         Preload("Client").
         Preload("ServiceContractType").
         Preload("ServiceContractStatus")
         Preload("ServiceContractStatus").
         Preload("Contact")
   }
   if slf.SalesDetailsId != 0 {
      db = db.Where("sales_details_id = ?", slf.SalesDetailsId)
   }
   if slf.QuotationId != 0 {
      db = db.Where("quotation_id = ?", slf.QuotationId)
   }
   if slf.SaleChanceId != 0 {
      db = db.Where("sale_chance_id = ?", slf.SaleChanceId)
   }
   if slf.ContactId != 0 {
      db = db.Where("contact_id = ?", slf.ContactId)
   }
   if slf.Number != "" {
      db = db.Where("number = ?", slf.Number)
   }
   return db
}
@@ -136,6 +166,12 @@
   return db.Delete(&ServiceContract{}).Error
}
func (slf *ServiceContractSearch) DeleteByIds(ids []int) error {
   var db = slf.build()
   db = db.Where("id in ?", ids)
   return db.Delete(&ServiceContract{}).Error
}
func (slf *ServiceContractSearch) Find() ([]*ServiceContract, int64, error) {
   var db = slf.build()
   var records = make([]*ServiceContract, 0)
@@ -147,8 +183,37 @@
      db = db.Limit(slf.PageSize).Offset((slf.PageNum - 1) * slf.PageSize)
   }
   err := db.Find(&records).Error
   err := db.Order("id desc").Find(&records).Error
   return records, total, err
}
func (slf *ServiceContractSearch) Count() (int64, error) {
   var db = slf.build()
   var total int64
   err := db.Count(&total).Error
   return total, err
}
func (slf *ServiceContractSearch) MaxAutoIncr() (int, error) {
   type Result struct {
      Max int
   }
   var (
      result Result
      db     = slf.build()
   )
   err := db.Select("MAX(id) as max").Scan(&result).Error
   if err != nil {
      return result.Max, fmt.Errorf("max err: %v", err)
   }
   return result.Max, nil
}
func (slf *ServiceContractSearch) SetNumber(number string) *ServiceContractSearch {
   slf.Number = number
   return slf
}
func (slf *ServiceContractSearch) SetId(id int) *ServiceContractSearch {
@@ -185,6 +250,31 @@
   return slf
}
func (slf *ServiceContractSearch) SetMemberIds(memberIds []int) *ServiceContractSearch {
   slf.MemberIds = memberIds
   return slf
}
func (slf *ServiceContractSearch) SetSalesDetailsId(salesDetailsId int) *ServiceContractSearch {
   slf.SalesDetailsId = salesDetailsId
   return slf
}
func (slf *ServiceContractSearch) SetQuotationId(quotationId int) *ServiceContractSearch {
   slf.QuotationId = quotationId
   return slf
}
func (slf *ServiceContractSearch) SetSaleChanceId(saleChanceId int) *ServiceContractSearch {
   slf.SaleChanceId = saleChanceId
   return slf
}
func (slf *ServiceContractSearch) SetContactId(contactId int) *ServiceContractSearch {
   slf.ContactId = contactId
   return slf
}
func (slf *ServiceContractSearch) SetPreload(preload bool) *ServiceContractSearch {
   slf.Preload = preload
   return slf
@@ -214,3 +304,40 @@
   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})
}