zchengo
107 lines
3.2 KiB
Go
107 lines
3.2 KiB
Go
package service
|
|
|
|
import (
|
|
"crm/global"
|
|
"crm/models"
|
|
"crm/response"
|
|
"time"
|
|
)
|
|
|
|
type ContractService struct {
|
|
}
|
|
|
|
// 创建合同
|
|
func (c *ContractService) Create(param *models.ContractCreateParam) int {
|
|
contract := models.Contract{
|
|
Name: param.Name,
|
|
Amount: param.Amount,
|
|
BeginTime: param.BeginTime,
|
|
OverTime: param.OverTime,
|
|
Remarks: param.Remarks,
|
|
Cid: param.Cid,
|
|
Productlist: param.Productlist,
|
|
Status: param.Status,
|
|
Creator: param.Creator,
|
|
Created: time.Now().Unix(),
|
|
}
|
|
if err := global.Db.Create(&contract).Error; err != nil {
|
|
return response.ErrCodeFailed
|
|
}
|
|
return response.ErrCodeSuccess
|
|
}
|
|
|
|
// 更新合同
|
|
func (c *ContractService) Update(param *models.ContractUpdateParam) int {
|
|
contract := models.Contract{
|
|
Id: param.Id,
|
|
Name: param.Name,
|
|
Amount: param.Amount,
|
|
BeginTime: param.BeginTime,
|
|
OverTime: param.OverTime,
|
|
Remarks: param.Remarks,
|
|
Cid: param.Cid,
|
|
Productlist: param.Productlist,
|
|
Status: param.Status,
|
|
Updated: time.Now().Unix(),
|
|
}
|
|
err := global.Db.Model(&contract).Updates(&contract).Error
|
|
if err != nil {
|
|
return response.ErrCodeFailed
|
|
}
|
|
return response.ErrCodeSuccess
|
|
}
|
|
|
|
// 删除合同
|
|
func (c *ContractService) Delete(param *models.ContractDeleteParam) int {
|
|
if err := global.Db.Delete(&models.Contract{}, param.Ids).Error; err != nil {
|
|
return response.ErrCodeFailed
|
|
}
|
|
return response.ErrCodeSuccess
|
|
}
|
|
|
|
// 查询合同列表
|
|
func (c *ContractService) QueryList(param *models.ContractQueryParam) ([]*models.ContractList, int64, int) {
|
|
contractList := make([]*models.ContractList, 0)
|
|
s := "contract.id, contract.name, contract.amount, contract.begin_time, contract.over_time, customer.name as cname, contract.remarks, contract.status, contract.created, contract.updated"
|
|
j := "inner join customer on contract.cid = customer.id and contract.creator = ?"
|
|
|
|
// 分页查询
|
|
offset := (param.Page.PageNum - 1) * param.Page.PageSize
|
|
mdb := global.Db.Offset(offset).Limit(param.Page.PageSize).Table(CONTRACT).Select(s)
|
|
var err error
|
|
if param.Id != 0 {
|
|
err = mdb.Joins(j+" and contract.id = ?", param.Creator, param.Id).Scan(&contractList).Error
|
|
} else {
|
|
err = mdb.Joins(j, param.Creator).Scan(&contractList).Error
|
|
}
|
|
if err != nil {
|
|
return nil, 0, response.ErrCodeFailed
|
|
}
|
|
var rows int64
|
|
global.Db.Raw("select count(*) from contract where creator = ?", param.Creator).Scan(&rows)
|
|
return contractList, rows, response.ErrCodeSuccess
|
|
}
|
|
|
|
// 查询合同信息
|
|
func (c *ContractService) QueryInfo(param *models.ContractQueryParam) (*models.ContractInfo, int) {
|
|
contract := models.Contract{
|
|
Id: param.Id,
|
|
}
|
|
contractInfo := models.ContractInfo{}
|
|
err := global.Db.Table(CONTRACT).Where(&contract).First(&contractInfo).Error
|
|
if err != nil {
|
|
return nil, response.ErrCodeFailed
|
|
}
|
|
return &contractInfo, response.ErrCodeSuccess
|
|
}
|
|
|
|
// 在编辑合同中，添加产品后，增加已添加的产品列表
|
|
func (p *ContractService) QueryPlist(param *models.ProductQueryParam) ([]*models.Products, int) {
|
|
products := make([]*models.Products, 0)
|
|
err := global.Db.Table(PRODUCT).Find(&products, param.Ids).Error
|
|
if err != nil {
|
|
return nil, response.ErrCodeFailed
|
|
}
|
|
return products, response.ErrCodeSuccess
|
|
}
|