fms_api/controller/census/invoice_payment.go

package census

import (
	"fmt"
	"github.com/gin-gonic/gin"
	"github.com/go-playground/validator/v10"
	"hongze/fms_api/controller/resp"
	"hongze/fms_api/global"
	"hongze/fms_api/models/base"
	"hongze/fms_api/models/fms"
	fmsService "hongze/fms_api/services/fms"
	"hongze/fms_api/utils"
	"strings"
)

// InvoicePaymentController 商品到款统计
type InvoicePaymentController struct{}

// List
// @Title 商品到款统计列表
// @Description 商品到款统计列表
// @Param   Keyword			query	string	false	"关键词"
// @Param   SellGroupId		query	int		false	"销售组别ID"
// @Param   ServiceType		query	int		false	"套餐类型"
// @Param   StartDate		query	string	false	"合同开始日期"
// @Param   EndDate			query	string	false	"合同结束日期"
// @Param   TimeType		query	int		false	"时间类型: 1-开票时间; 2-到款时间"
// @Param   HasInvoice		query	int		false	"已开票"
// @Param   HasPayment		query	int		false	"已到款"
// @Success 200 {object} fms.ContractRegisterItem
// @router /census/invoice_payment/list [get]
func (ct *InvoicePaymentController) List(c *gin.Context) {
	var req fms.InvoicePaymentCensusListReq
	if e := c.BindQuery(&req); e != nil {
		err, ok := e.(validator.ValidationErrors)
		if !ok {
			resp.FailData("参数解析失败", "Err:"+e.Error(), c)
			return
		}
		resp.FailData("参数解析失败", err.Translate(global.Trans), c)
		return
	}

	cond := `1 = 1`
	pars := make([]interface{}, 0)
	// 合同编号/客户姓名/销售/实际使用方
	if req.Keyword != "" {
		kw := "%" + req.Keyword + "%"
		cond += ` b.company_name LIKE ? OR a.seller_name LIKE ?`
		pars = append(pars, kw, kw)
	}
	if req.SellGroupId > 0 {
		cond += ` AND a.seller_group_id = ?`
		pars = append(pars, req.SellGroupId)
	}
	// 套餐筛选
	if req.ServiceType != 0 {
		registerIds, e := fms.GetContractRegisterIdsByTempId(req.ServiceType)
		if e != nil {
			resp.FailMsg("获取失败", "获取合同登记IDs失败, Err: "+e.Error(), c)
			return
		}
		if len(registerIds) > 0 {
			cond += ` AND contract_register_id IN ?`
			pars = append(pars, registerIds)
		} else {
			cond += ` AND 1 = 2`
		}
	}
	// 开票到款日期
	if req.TimeType > 0 && req.StartDate != "" && req.EndDate != "" {
		st := fmt.Sprint(req.StartDate, " 00:00:00")
		ed := fmt.Sprint(req.EndDate, " 23:59:59")
		cond += ` a.invoice_type = ? AND (invoice_time BETWEEN ? AND ?)`
		pars = append(pars, req.TimeType, st, ed)
	}
	// 已开票
	if req.HasInvoice == 1 && req.HasPayment == 0 {
		cond += ` AND a.invoice_type = 1`
	}
	// 已到款
	if req.HasInvoice == 0 && req.HasPayment == 1 {
		cond += ` AND a.invoice_type = 2`
	}

	page := new(base.Page)
	page.SetPageSize(req.PageSize)
	page.SetCurrent(req.Current)
	page.AddOrderItem(base.OrderItem{Column: "a.create_time", Asc: false})

	total, list, e := fms.GetInvoicePaymentCensusPageList(page, cond, pars)
	if e != nil {
		resp.FailMsg("获取失败", "获取商品到款统计列表失败, Err: "+e.Error(), c)
		return
	}
	registerIds := make([]int, 0)
	for i := range list {
		registerIds = append(registerIds, list[i].ContractRegisterId)
	}

	results := new(fms.InvoicePaymentCensusResp)
	if len(registerIds) > 0 {
		// 获取开票到款列表
		ivCond := `contract_register_id IN ?`
		ivPars := make([]interface{}, 0)
		ivPars = append(ivPars, registerIds)
		iv := new(fms.ContractInvoice)
		invoiceList, e := iv.List(ivCond, ivPars, "")
		if e != nil {
			resp.FailMsg("获取失败", "获取开票到款列表失败, Err: "+e.Error(), c)
			return
		}
		// 取出开票、到款
		invoiceMap := make(map[int][]*fms.ContractInvoice, 0)
		paymentMap := make(map[int][]*fms.ContractInvoice, 0)
		paymentIds := make([]int, 0)
		for i := range invoiceList {
			if invoiceMap[invoiceList[i].ContractRegisterId] == nil {
				invoiceMap[invoiceList[i].ContractRegisterId] = make([]*fms.ContractInvoice, 0)
			}
			if paymentMap[invoiceList[i].ContractRegisterId] == nil {
				paymentMap[invoiceList[i].ContractRegisterId] = make([]*fms.ContractInvoice, 0)
			}
			if invoiceList[i].InvoiceType == fms.ContractInvoiceTypeMake {
				invoiceMap[invoiceList[i].ContractRegisterId] = append(invoiceMap[invoiceList[i].ContractRegisterId], invoiceList[i])
			}
			if invoiceList[i].InvoiceType == fms.ContractInvoiceTypePay {
				paymentMap[invoiceList[i].ContractRegisterId] = append(paymentMap[invoiceList[i].ContractRegisterId], invoiceList[i])
				paymentIds = append(paymentIds, invoiceList[i].ContractInvoiceId)
			}
		}

		// 合同套餐
		contractServiceCond := `contract_register_id IN ?`
		contractServicePars := make([]interface{}, 0)
		contractServicePars = append(contractServicePars, registerIds)
		contractServiceOB := new(fms.ContractService)
		contractServiceList, e := contractServiceOB.List(contractServiceCond, contractServicePars)
		if e != nil {
			resp.FailMsg("获取失败", "获取合同套餐列表失败, Err:"+e.Error(), c)
			return
		}
		contractServiceMap := make(map[int][]*fms.ContractService, 0)
		servicesNameMap := make(map[int][]string, 0)
		for i := range contractServiceList {
			if contractServiceMap[contractServiceList[i].ContractRegisterId] == nil {
				contractServiceMap[contractServiceList[i].ContractRegisterId] = make([]*fms.ContractService, 0)
			}
			contractServiceMap[contractServiceList[i].ContractRegisterId] = append(contractServiceMap[contractServiceList[i].ContractRegisterId], contractServiceList[i])
			servicesNameMap[contractServiceList[i].ContractRegisterId] = append(servicesNameMap[contractServiceList[i].ContractRegisterId], contractServiceList[i].Title)
		}

		// 到款套餐分配
		serviceAmountMap := make(map[int][]*fms.ContractPaymentServiceAmount, 0)
		if len(paymentIds) > 0 {
			serviceAmountCond := `contract_invoice_id IN ?`
			serviceAmountPars := make([]interface{}, 0)
			serviceAmountPars = append(serviceAmountPars, paymentIds)
			serviceAmountOB := new(fms.ContractPaymentServiceAmount)
			serviceAmountList, e := serviceAmountOB.List(serviceAmountCond, serviceAmountPars)
			if e != nil {
				resp.FailMsg("获取失败", "获取到款套餐分配列表失败, Err:"+e.Error(), c)
				return
			}
			for i := range serviceAmountList {
				if serviceAmountMap[serviceAmountList[i].ContractRegisterId] == nil {
					serviceAmountMap[serviceAmountList[i].ContractRegisterId] = make([]*fms.ContractPaymentServiceAmount, 0)
				}
				serviceAmountMap[serviceAmountList[i].ContractRegisterId] = append(serviceAmountMap[serviceAmountList[i].ContractRegisterId], serviceAmountList[i])
			}
		}

		// 整合响应数据
		respList := make([]*fms.InvoicePaymentCensusItem, 0)
		for i := range list {
			v := new(fms.InvoicePaymentCensusItem)
			v.ContractRegisterId = list[i].ContractRegisterId
			v.CompanyName = list[i].CompanyName
			v.NewCompany = list[i].NewCompany
			v.StartDate = list[i].StartDate.Format(utils.FormatDate)
			v.StartDate = list[i].StartDate.Format(utils.FormatDate)
			svList := servicesNameMap[list[i].ContractRegisterId]
			v.ServicesName = strings.Join(svList, ",")
			// 格式化(合并)开票到款数据
			v.InvoicePaymentList = fmsService.MergeInvoiceList2InvoicePaymentCensusInfo(invoiceMap[list[i].ContractRegisterId], paymentMap[list[i].ContractRegisterId],
				contractServiceMap[list[i].ContractRegisterId], serviceAmountMap[list[i].ContractRegisterId])
			respList = append(respList, v)
		}

		// 开票到款金额合计
		amountTotalCond := `contract_register_id IN ?`
		amountTotalPars := make([]interface{}, 0)
		amountTotalPars = append(amountTotalPars, registerIds)
		amountTotalList, e := fms.GetContractInvoiceAmountTotal(amountTotalCond, amountTotalPars)
		if e != nil {
			resp.FailMsg("获取失败", "获取开票到款金额合计失败, Err:"+e.Error(), c)
			return
		}
		amountTotalMap := make(map[int]float64)
		for i := range amountTotalList {
			amountTotalMap[amountTotalList[i].InvoiceType] = amountTotalList[i].TotalAmount
		}

		results.DataList = respList
		results.InvoiceTotal = amountTotalMap[fms.ContractInvoiceTypeMake]
		results.PaymentTotal = amountTotalMap[fms.ContractInvoiceTypePay]
	}

	page.SetTotal(total)
	baseData := new(base.BaseData)
	baseData.SetPage(page)
	baseData.SetList(results)
	resp.OkData("获取成功", baseData, c)
}