zchengo
109 lines
3.0 KiB
Go
109 lines
3.0 KiB
Go
package api
|
|
|
|
import (
|
|
"crm/models"
|
|
"crm/response"
|
|
"crm/service"
|
|
"strconv"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
type ContractApi struct {
|
|
contractService *service.ContractService
|
|
}
|
|
|
|
func NewContractApi() *ContractApi {
|
|
contractApi := ContractApi{
|
|
contractService: &service.ContractService{},
|
|
}
|
|
return &contractApi
|
|
}
|
|
|
|
// 创建合同
|
|
func (c *ContractApi) Create(context *gin.Context) {
|
|
var param models.ContractCreateParam
|
|
uid, _ := strconv.Atoi(context.Request.Header.Get("uid"))
|
|
err := context.ShouldBind(¶m)
|
|
if uid <= 0 || err != nil {
|
|
response.Result(response.ErrCodeParamInvalid, nil, context)
|
|
return
|
|
}
|
|
param.Creator = int64(uid)
|
|
errCode := c.contractService.Create(¶m)
|
|
response.Result(errCode, nil, context)
|
|
}
|
|
|
|
// 更新合同
|
|
func (c *ContractApi) Update(context *gin.Context) {
|
|
var param models.ContractUpdateParam
|
|
if err := context.ShouldBind(¶m); err != nil {
|
|
response.Result(response.ErrCodeParamInvalid, nil, context)
|
|
return
|
|
}
|
|
errCode := c.contractService.Update(¶m)
|
|
response.Result(errCode, nil, context)
|
|
}
|
|
|
|
// 删除合同
|
|
func (c *ContractApi) Delete(context *gin.Context) {
|
|
var param models.ContractDeleteParam
|
|
if err := context.ShouldBind(¶m); err != nil {
|
|
response.Result(response.ErrCodeParamInvalid, nil, context)
|
|
return
|
|
}
|
|
errCode := c.contractService.Delete(¶m)
|
|
response.Result(errCode, nil, context)
|
|
}
|
|
|
|
// 查询合同列表
|
|
func (c *ContractApi) GetList(context *gin.Context) {
|
|
var param models.ContractQueryParam
|
|
uid, _ := strconv.Atoi(context.Request.Header.Get("uid"))
|
|
err := context.ShouldBind(¶m)
|
|
if uid <= 0 || err != nil || param.Page.PageNum <= 0 || param.Page.PageSize <= 0 {
|
|
response.Result(response.ErrCodeParamInvalid, nil, context)
|
|
return
|
|
}
|
|
param.Creator = int64(uid)
|
|
productList, rows, errCode := c.contractService.GetList(¶m)
|
|
response.PageResult(errCode, productList, rows, context)
|
|
}
|
|
|
|
// 查询合同信息
|
|
func (c *ContractApi) GetInfo(context *gin.Context) {
|
|
var param models.ContractQueryParam
|
|
if err := context.ShouldBind(¶m); err != nil {
|
|
response.Result(response.ErrCodeParamInvalid, nil, context)
|
|
return
|
|
}
|
|
productInfo, errCode := c.contractService.GetInfo(¶m)
|
|
response.Result(errCode, productInfo, context)
|
|
}
|
|
|
|
// 编辑合同时，查询产品列表
|
|
func (p *ContractApi) GetProductList(context *gin.Context) {
|
|
var param models.ContractQueryParam
|
|
if err := context.ShouldBind(¶m); err != nil {
|
|
response.Result(response.ErrCodeParamInvalid, nil, context)
|
|
return
|
|
}
|
|
productList, errCode := p.contractService.GetProductList(¶m)
|
|
response.Result(errCode, productList, context)
|
|
}
|
|
|
|
// 导出Excel文件
|
|
func (c *ContractApi) Export(context *gin.Context) {
|
|
uid, _ := strconv.Atoi(context.Request.Header.Get("uid"))
|
|
if uid <= 0 {
|
|
response.Result(response.ErrCodeParamInvalid, nil, context)
|
|
return
|
|
}
|
|
file, errCode := c.contractService.Export(int64(uid))
|
|
if len(file) >= 0 && errCode != 0 {
|
|
response.Result(errCode, nil, context)
|
|
return
|
|
}
|
|
context.File(file)
|
|
}
|