fix

wangpengfei

2023-07-28 b4400a06b5f801bc7a85320680d0b6c45c547ff0

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
package model
 
import (
    "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
        Keyword  string
        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.Keyword != "" {
        db = db.Where("name LIKE ?", "%"+slf.Keyword+"%")
    }
    if slf.Keyword != "" {
        db = db.Where("name LIKE ?", "%"+slf.Keyword+"%")
    }
    if slf.Id != 0 {
        db = db.Where("id = ?", slf.Id)
    }
 
    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) 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
}