contract_return_dao.go 4.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169
  1. /*
  2. * @description: 回款相关数据操作
  3. * @Author: CP
  4. * @Date: 2020-11-30 11:19:14
  5. * @FilePath: \construction_management\dao\contract_return_dao.go
  6. */
  7. package dao
  8. import (
  9. "errors"
  10. "fmt"
  11. "strconv"
  12. "github.com/go-xorm/xorm"
  13. "go.mod/models"
  14. )
  15. //数据库操作引擎
  16. type ContractReturnDao struct {
  17. engine *xorm.Engine
  18. }
  19. //获得一个DAO对象
  20. func NewContractReturnDao(engine *xorm.Engine) *ContractReturnDao {
  21. return &ContractReturnDao{
  22. engine: engine,
  23. }
  24. }
  25. // 获得回款
  26. func (d *ContractReturnDao) Get(id int, contractsId int, bidsectionId int) *models.CmContractsReturn {
  27. data := &models.CmContractsReturn{Id: id, ContractsId: contractsId, BidsectionId: bidsectionId}
  28. ok, err := d.engine.Get(data)
  29. if ok && err == nil {
  30. return data
  31. } else {
  32. data.Id = 0
  33. return data
  34. }
  35. }
  36. func (d *ContractReturnDao) GetAll(projectId int, bidsectionId int, contractsId int) []models.CmContractsReturn {
  37. datalist := make([]models.CmContractsReturn, 0)
  38. err := d.engine.
  39. Where("project_id = ? and bidsection_id =? and contracts_id=?", projectId, bidsectionId, contractsId).
  40. Desc("id").
  41. Find(&datalist)
  42. if err != nil {
  43. return datalist
  44. } else {
  45. return datalist
  46. }
  47. }
  48. //回款分页
  49. func (d *ContractReturnDao) GetPage(projectId int, bidsectionId int, contractsId int, page int) []models.CmContractsReturn {
  50. datalist := make([]models.CmContractsReturn, 0)
  51. size := 8
  52. if page == 1 {
  53. page = 0
  54. } else {
  55. page--
  56. page = page * size
  57. }
  58. _ = d.engine.Where("project_id = ? and bidsection_id =? and contracts_id=?", projectId, bidsectionId, contractsId).
  59. Limit(size, page).
  60. Find(&datalist)
  61. return datalist
  62. }
  63. // 创建回款
  64. func (d *ContractReturnDao) Add(contractsReturn *models.CmContractsReturn) error {
  65. _, err := d.engine.Insert(contractsReturn)
  66. return err
  67. }
  68. // 更新回款
  69. func (d *ContractReturnDao) Update(contractsReturn *models.CmContractsReturn, contracts_id int, bidsectionId int) error {
  70. is, err := d.engine.Where("id = ? and contracts_id = ? and bidsection_id=? ", contractsReturn.Id, contracts_id, bidsectionId).Update(contractsReturn)
  71. if is == 0 {
  72. return errors.New("未找到回款信息")
  73. }
  74. return err
  75. }
  76. // 删除回款
  77. func (d *ContractReturnDao) Delete(id int, contractsId int, bidsectionId int, projectId int) error {
  78. data := &models.CmContractsReturn{}
  79. is, err := d.engine.Where("id = ? and contracts_id = ? and project_id =? and bidsection_id = ? ", id, contractsId, projectId, bidsectionId).Delete(data)
  80. if is == 0 {
  81. return errors.New("未找到回款")
  82. }
  83. return err
  84. }
  85. // 更新回款总金额
  86. func (d *ContractReturnDao) UpdateTotalPrice(projectId int, bidsectionId int, contractsId int) error {
  87. session := d.engine.NewSession()
  88. defer session.Close()
  89. err := session.Begin()
  90. if err != nil {
  91. return errors.New("session出错-db")
  92. }
  93. // 0.获得合同金额
  94. contractsDetail := &models.CmContracts{}
  95. _, err = d.engine.Where(" id = ? ", contractsId).Get(contractsDetail)
  96. if err != nil {
  97. return errors.New("未找到合同")
  98. }
  99. // 1.获得合同回款总金额
  100. datalist := make([]models.CmContractsReturn, 0)
  101. err = d.engine.Where(" project_id =? and bidsection_id = ? ", projectId, bidsectionId).Find(&datalist)
  102. if err != nil {
  103. session.Rollback()
  104. return errors.New("编辑合同出错-项目节更新失败")
  105. }
  106. priceTotal := 0.00
  107. contractsPrice := 0.00
  108. for _, item := range datalist {
  109. price, _ := strconv.ParseFloat(item.Price, 64)
  110. priceTotal = priceTotal + price
  111. if item.ContractsId == contractsId {
  112. contractsPrice = contractsPrice + price
  113. }
  114. }
  115. priceTotal, _ = strconv.ParseFloat(fmt.Sprintf("%.2f", priceTotal), 64)
  116. contractsPrice, _ = strconv.ParseFloat(fmt.Sprintf("%.2f", contractsPrice), 64)
  117. // 1-1合同状态的判定
  118. contractsDetailPrice, _ := strconv.ParseFloat(contractsDetail.Price, 64)
  119. // 总回款大于等于合同金额 待关闭
  120. contractStatus := 0
  121. if contractsPrice >= contractsDetailPrice {
  122. contractStatus = 1
  123. }
  124. // 2.更新合同表 合同下回款总金额
  125. _, err = session.Exec("UPDATE cm_contracts SET `returned` = ? , status = ? where id = ? ", contractsPrice, contractStatus, contractsId)
  126. if err != nil {
  127. session.Rollback()
  128. return errors.New("金额更新失败")
  129. }
  130. // 3.更新项目节 合同下回款总金额
  131. _, err = session.Exec("UPDATE cm_tree_contracts SET `contract_returned` = ? , contract_status = ? where project_id = ? and bidsection_id=? and contract_id=? ",
  132. contractsPrice, contractStatus, projectId, bidsectionId, contractsId)
  133. if err != nil {
  134. session.Rollback()
  135. return errors.New("金额更新失败")
  136. }
  137. // 4.更新标段树 整个标段下回款总金额
  138. _, err = session.Exec("UPDATE cm_tree SET `contracts_returned` = ? where project_id = ? and bidsection_id=? ", priceTotal, projectId, bidsectionId)
  139. if err != nil {
  140. session.Rollback()
  141. return errors.New("金额更新失败")
  142. }
  143. err = session.Commit()
  144. if err != nil {
  145. session.Rollback()
  146. return errors.New("session出错-db")
  147. }
  148. return nil
  149. }