opms_backend/opms_parent/app/service/contract/ctr_contract_collection_plan.go

package service

import (
	"context"
	"database/sql"
	"fmt"

	dao "dashoo.cn/micro/app/dao/contract"
	model "dashoo.cn/micro/app/model/contract"

	"dashoo.cn/opms_libary/micro_srv"
	"dashoo.cn/opms_libary/myerrors"
	"dashoo.cn/opms_libary/request"
	"github.com/gogf/gf/database/gdb"
	"github.com/gogf/gf/os/gtime"
	"github.com/gogf/gf/util/gvalid"
)

type CtrContractCollectionPlanService struct {
	Dao         *dao.CtrContractCollectionPlanDao
	ContractDao *dao.CtrContractDao
	ctrSrv      *CtrContractService

	Tenant   string
	userInfo request.UserInfo
}

func NewCtrContractCollectionPlanService(ctx context.Context) (*CtrContractCollectionPlanService, error) {
	tenant, err := micro_srv.GetTenant(ctx)
	if err != nil {
		return nil, fmt.Errorf("获取组合码异常：%s", err.Error())
	}
	// 获取用户信息
	userInfo, err := micro_srv.GetUserInfo(ctx)
	if err != nil {
		return nil, fmt.Errorf("获取用户信息异常：%s", err.Error())
	}
	ctrSrv, err := NewCtrContractService(ctx)
	if err != nil {
		return nil, err
	}
	return &CtrContractCollectionPlanService{
		Dao:         dao.NewCtrContractCollectionPlanDao(tenant),
		ContractDao: dao.NewCtrContractDao(tenant),
		ctrSrv:      ctrSrv,
		Tenant:      tenant,
		userInfo:    userInfo,
	}, nil
}

func (s CtrContractCollectionPlanService) List(ctx context.Context, req *model.CtrContractCollectionPlanListReq) (int, []*model.CtrContractCollectionPlan, error) {
	dao := &s.Dao.CtrContractCollectionPlanDao
	if req.SearchText != "" {
		likestr := fmt.Sprintf("%%%s%%", req.SearchText)
		dao = dao.Where("(cust_name LIKE ? || contract_code LIKE ?)", likestr, likestr)
	}
	if req.CustId != 0 {
		dao = dao.Where("cust_id = ?", req.CustId)
	}
	if req.CustName != "" {
		likestr := fmt.Sprintf("%%%s%%", req.CustName)
		dao = dao.Where("cust_name like ?", likestr)
	}
	if req.ContractId != 0 {
		dao = dao.Where("contract_id = ?", req.ContractId)
	}
	if req.ContractCode != "" {
		likestr := fmt.Sprintf("%%%s%%", req.ContractCode)
		dao = dao.Where("contract_code like ?", likestr)
	}
	if req.ContractStatus != "" {
		dao = dao.Where("contract_status = ?", req.ContractStatus)
	}
	if req.BeginTime != "" {
		dao = dao.Where("created_time > ?", req.BeginTime)
	}
	if req.EndTime != "" {
		dao = dao.Where("created_time < ?", req.EndTime)
	}

	total, err := dao.Count()
	if err != nil {
		return 0, nil, err
	}
	if req.PageNum != 0 {
		dao = dao.Page(req.GetPage())
	}
	orderby := "created_time desc"
	if req.OrderBy != "" {
		orderby = req.OrderBy
	}
	dao = dao.Order(orderby)

	ents := []*model.CtrContractCollectionPlan{}
	err = dao.Structs(&ents)
	if err != nil && err != sql.ErrNoRows {
		return 0, nil, err
	}
	return total, ents, err
}

func (s CtrContractCollectionPlanService) Add(ctx context.Context, req *model.CtrContractCollectionPlanAddReq) (int, error) {
	validErr := gvalid.CheckStruct(ctx, req, nil)
	if validErr != nil {
		return 0, myerrors.NewMsgError(nil, validErr.Current().Error())
	}

	c, err := s.ContractDao.Where("id = ?", req.ContractId).One()
	if err != nil {
		return 0, err
	}
	if c == nil {
		return 0, myerrors.NewMsgError(nil, fmt.Sprintf("合同:%d 不存在", req.ContractId))
	}

	id, err := s.Dao.InsertAndGetId(model.CtrContractCollectionPlan{
		CustId:         c.CustId,
		CustName:       c.CustName,
		ContractId:     req.ContractId,
		ContractCode:   c.ContractCode,
		ContractStatus: "10",
		PlanAmount:     req.PlanAmount,
		PlanDatetime:   req.PlanDatetime,
		CashedAmount:   req.CashedAmount,
		CashedDatetime: req.CashedDatetime,
		Remark:         req.Remark,
		CreatedBy:      s.userInfo.Id,
		CreatedName:    s.userInfo.NickName,
		CreatedTime:    gtime.Now(),
		UpdatedBy:      s.userInfo.Id,
		UpdatedName:    s.userInfo.NickName,
		UpdatedTime:    gtime.Now(),
		DeletedTime:    gtime.Now(),
	})
	if err != nil {
		return 0, err
	}
	return int(id), err
}

func (s CtrContractCollectionPlanService) Update(ctx context.Context, req *model.CtrContractCollectionPlanUpdateReq) error {
	validErr := gvalid.CheckStruct(ctx, req, nil)
	if validErr != nil {
		return myerrors.NewMsgError(nil, validErr.Current().Error())
	}

	p, err := s.Dao.Where("id = ?", req.Id).One()
	if err != nil {
		return err
	}
	if p == nil {
		return myerrors.NewMsgError(nil, fmt.Sprintf("回款计划:%d 不存在", req.Id))
	}

	// var c *model.CtrContract
	// if req.ContractId != 0 {
	// 	c, err = s.ContractDao.Where("id = ?", req.ContractId).One()
	// 	if err != nil {
	// 		return err
	// 	}
	// 	if c == nil {
	// 		return myerrors.NewMsgError(nil, fmt.Sprintf("合同:%d 不存在", req.ContractId))
	// 	}
	// }

	dao := &s.Dao.CtrContractCollectionPlanDao
	toupdate := map[string]interface{}{}
	// if req.ContractId != 0 {
	// 	toupdate["cust_id"] = c.CustId
	// 	toupdate["cust_name"] = c.CustName
	// 	toupdate["contract_id"] = req.ContractId
	// 	toupdate["contract_code"] = c.ContractCode
	// }
	if req.PlanAmount != nil {
		toupdate["plan_amount"] = *req.PlanAmount
	}
	if req.PlanDatetime != nil {
		toupdate["plan_datetime"] = req.PlanDatetime
	}
	// if req.CashedAmount != nil {
	// 	toupdate["cashed_amount"] = *req.CashedAmount
	// }
	// if req.CashedDatetime != nil {
	// 	toupdate["cashed_datetime"] = req.CashedDatetime
	// }
	if req.Remark != nil {
		toupdate["remark"] = *req.Remark
	}
	if len(toupdate) != 0 {
		toupdate["updated_by"] = s.userInfo.Id
		toupdate["updated_name"] = s.userInfo.NickName
		toupdate["updated_time"] = gtime.Now()
		_, err = dao.Where("Id", req.Id).Data(toupdate).Update()
		if err != nil {
			return err
		}
	}
	return nil
}

func (s CtrContractCollectionPlanService) UpdateCashedAmount(tx *gdb.TX, id int) error {
	plan := model.CtrContractCollectionPlan{}
	err := tx.GetStruct(&plan, "select * from ctr_contract_collection_plan where id = ?", id)
	if err == sql.ErrNoRows {
		return myerrors.NewMsgError(err, fmt.Sprintf("回款计划不存在 %d", id))
	}
	if err != nil {
		return err
	}

	v, err := tx.GetValue("select sum(collection_amount) from ctr_contract_collection where plan_id=? and appro_status='20' and deleted_time is null", id)
	if err != nil {
		return err
	}
	amount := v.Float64()

	var contractStatus string
	if amount < plan.PlanAmount {
		contractStatus = "20"
	} else {
		contractStatus = "30"
	}
	if amount == 0 {
		contractStatus = "10"
	}
	_, err = tx.Update("ctr_contract_collection_plan",
		map[string]interface{}{
			"contract_status": contractStatus,
			"cashed_amount":   amount,
			"cashed_datetime": gtime.Now(),
		}, "id = ?", plan.Id)
	if err != nil {
		return err
	}
	return nil
}

func (s CtrContractCollectionPlanService) Delete(ctx context.Context, id []int) error {
	if len(id) == 0 {
		return nil
	}
	return s.Dao.DB.Transaction(ctx, func(ctx context.Context, tx *gdb.TX) error {
		contractId := map[int]bool{}
		for _, i := range id {
			ent := model.CtrContractCollectionPlan{}
			err := tx.GetStruct(&ent, "select * from ctr_contract_collection_plan where id = ?", i)
			if err == sql.ErrNoRows {
				continue
			}
			if err != nil {
				return err
			}
			contractId[ent.ContractId] = true
		}

		_, err := tx.Delete("ctr_contract_collection_plan", "id in (?)", id)
		if err != nil {
			return err
		}
		_, err = tx.Delete("ctr_contract_collection", "plan_id in (?)", id)
		if err != nil {
			return err
		}
		for cid := range contractId {
			err = s.ctrSrv.UpdateCollectedAmount(tx, cid)
			if err != nil {
				return err
			}
		}
		return nil
	})
}