contract_service.go 7.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260
  1. /*
  2. * @description: 合同数据相关操作
  3. * @Author: CP
  4. * @Date: 2020-10-27 11:28:24
  5. * @FilePath: \design_quantity\services\contract_service.go
  6. */
  7. package services
  8. import (
  9. "errors"
  10. "html"
  11. "log"
  12. "time"
  13. "github.com/kataras/iris/v12"
  14. "go.mod/dao"
  15. "go.mod/datasource"
  16. "go.mod/models"
  17. "go.mod/web/viewmodels"
  18. )
  19. //定义项目用户Service接口
  20. type ContractService interface {
  21. ValidRuleDepth(ctx iris.Context) (*viewmodels.TreeSectionContract, error)
  22. ValidRuleTemplate(ctx iris.Context) (*viewmodels.TreeSectionContract, error)
  23. ValidRuleSectionAdd(ctx iris.Context) (*viewmodels.TreeSectionContract, error)
  24. ValidRuleSectionDelete(ctx iris.Context) (*viewmodels.TreeSectionContract, error)
  25. ValidRuleSectionNot(ctx iris.Context) (*viewmodels.TreeSectionContract, error)
  26. ValidRuleGet(ctx iris.Context) (*viewmodels.TreeSectionContract, error)
  27. ValidRuleSerial(ctx iris.Context) (*viewmodels.TreeSectionContract, error)
  28. // 项目节
  29. GetSecionTree() *viewmodels.TreeSectionContract
  30. SetSection() error
  31. GetSection(id int) *viewmodels.TreeSectionDetail
  32. SectionAdd(sectionData *viewmodels.TreeSectionContract) (*models.CmTreeContracts, error)
  33. SectionSave(sectionData *viewmodels.TreeSectionContract) error
  34. UpdateSerial(sectionData *viewmodels.TreeSectionContract) error
  35. SectionDelete(treeId int) error
  36. MoveDepth(sectionData *viewmodels.TreeSectionContract) error
  37. MoveSerial(sectionData *viewmodels.TreeSectionContract) error
  38. SaveUpload(Location string, Filename string, id int, ext string) error
  39. }
  40. //返回service操作类
  41. type contractService struct {
  42. treeContractDao *dao.TreeContractDao
  43. contractDao *dao.ContractDao
  44. contractReturnDao *dao.ContractReturnDao
  45. contractPaidDao *dao.ContractPaidDao
  46. treeDao *dao.TreeDao
  47. annexDao *dao.AnnexDao
  48. }
  49. //创建项目用户service
  50. func NewContractService() ContractService {
  51. return &contractService{
  52. treeContractDao: dao.NewTreeContractDao(datasource.InstanceDbMaster()),
  53. contractDao: dao.NewContractDao(datasource.InstanceDbMaster()),
  54. contractReturnDao: dao.NewContractReturnDao(datasource.InstanceDbMaster()),
  55. contractPaidDao: dao.NewContractPaidDao(datasource.InstanceDbMaster()),
  56. treeDao: dao.NewTreeDao(datasource.InstanceDbMaster()),
  57. annexDao: dao.NewAnnexDao(datasource.InstanceDbMaster()),
  58. }
  59. }
  60. // 升级降级规则验证
  61. func (s *contractService) ValidRuleDepth(ctx iris.Context) (*viewmodels.TreeSectionContract, error) {
  62. treeSectionVaild := &viewmodels.TreeSectionContract{}
  63. err := ctx.ReadJSON(treeSectionVaild)
  64. if err != nil {
  65. log.Println("folder-ValidRule-ReadForm转换异常, error=", err)
  66. return treeSectionVaild, err
  67. }
  68. err = treeSectionVaild.ValidateDepth()
  69. if err != nil {
  70. log.Println("参数验证错误, error=", err)
  71. return treeSectionVaild, err
  72. }
  73. return treeSectionVaild, nil
  74. }
  75. // 模板规则验证
  76. func (s *contractService) ValidRuleTemplate(ctx iris.Context) (*viewmodels.TreeSectionContract, error) {
  77. treeSectionVaild := &viewmodels.TreeSectionContract{}
  78. err := ctx.ReadJSON(treeSectionVaild)
  79. if err != nil {
  80. log.Println("folder-ValidRule-ReadForm转换异常, error=", err)
  81. return treeSectionVaild, err
  82. }
  83. err = treeSectionVaild.ValidateTemplate()
  84. if err != nil {
  85. log.Println("参数验证错误, error=", err)
  86. return treeSectionVaild, err
  87. }
  88. return treeSectionVaild, nil
  89. }
  90. // 模板规则新增项目节
  91. func (s *contractService) ValidRuleSectionAdd(ctx iris.Context) (*viewmodels.TreeSectionContract, error) {
  92. treeSectionVaild := &viewmodels.TreeSectionContract{}
  93. err := ctx.ReadJSON(treeSectionVaild)
  94. if err != nil {
  95. log.Println("folder-ValidRule-ReadForm转换异常, error=", err)
  96. return treeSectionVaild, err
  97. }
  98. err = treeSectionVaild.ValidateSectionAdd()
  99. if err != nil {
  100. log.Println("参数验证错误, error=", err)
  101. return treeSectionVaild, err
  102. }
  103. return treeSectionVaild, nil
  104. }
  105. // 模板规则新增项目节
  106. func (s *contractService) ValidRuleSectionDelete(ctx iris.Context) (*viewmodels.TreeSectionContract, error) {
  107. treeSectionVaild := &viewmodels.TreeSectionContract{}
  108. err := ctx.ReadForm(treeSectionVaild)
  109. if err != nil {
  110. log.Println("folder-ValidRule-ReadForm转换异常, error=", err)
  111. return treeSectionVaild, err
  112. }
  113. err = treeSectionVaild.ValidateSectionDelete()
  114. if err != nil {
  115. log.Println("参数验证错误, error=", err)
  116. return treeSectionVaild, err
  117. }
  118. return treeSectionVaild, nil
  119. }
  120. func (s *contractService) ValidRuleSectionNot(ctx iris.Context) (*viewmodels.TreeSectionContract, error) {
  121. treeSectionVaild := &viewmodels.TreeSectionContract{}
  122. err := ctx.ReadForm(treeSectionVaild)
  123. if err != nil {
  124. log.Println("folder-ValidRule-ReadForm转换异常, error=", err)
  125. return treeSectionVaild, err
  126. }
  127. err = treeSectionVaild.ValidateSectionNot()
  128. if err != nil {
  129. log.Println("参数验证错误, error=", err)
  130. return treeSectionVaild, err
  131. }
  132. return treeSectionVaild, nil
  133. }
  134. func (s *contractService) ValidRuleGet(ctx iris.Context) (*viewmodels.TreeSectionContract, error) {
  135. treeSectionVaild := &viewmodels.TreeSectionContract{}
  136. err := ctx.ReadForm(treeSectionVaild)
  137. if err != nil {
  138. log.Println("folder-ValidRule-ReadForm转换异常, error=", err)
  139. return treeSectionVaild, err
  140. }
  141. err = treeSectionVaild.ValidateSectionDelete()
  142. if err != nil {
  143. log.Println("参数验证错误, error=", err)
  144. return treeSectionVaild, err
  145. }
  146. return treeSectionVaild, nil
  147. }
  148. // 验证序号相关
  149. func (s *contractService) ValidRuleSerial(ctx iris.Context) (*viewmodels.TreeSectionContract, error) {
  150. treeSectionVaild := &viewmodels.TreeSectionContract{}
  151. err := ctx.ReadJSON(treeSectionVaild)
  152. if err != nil {
  153. log.Println("folder-ValidRule-ReadForm转换异常, error=", err)
  154. return treeSectionVaild, err
  155. }
  156. err = treeSectionVaild.ValidateSectionSerial()
  157. if err != nil {
  158. log.Println("参数验证错误, error=", err)
  159. return treeSectionVaild, err
  160. }
  161. return treeSectionVaild, nil
  162. }
  163. // 新增合同参数验证
  164. func (s *contractService) ValidRuleContractAdd(ctx iris.Context) (*viewmodels.Contracts, error) {
  165. // 创建一个存放前端传过来参数
  166. contractsVaild := &viewmodels.Contracts{}
  167. // 存放raw的值,放入到contractsVaild
  168. err := ctx.ReadJSON(contractsVaild)
  169. if err != nil {
  170. log.Println("folder-ValidRule-ReadForm转换异常, error=", err)
  171. return contractsVaild, err
  172. }
  173. // 验证合同传参
  174. err = contractsVaild.ValidateAdd()
  175. if err != nil {
  176. log.Println("参数验证错误, error=", err)
  177. return contractsVaild, err
  178. }
  179. // xss
  180. contractsVaild.Code = html.EscapeString(contractsVaild.Code)
  181. contractsVaild.Name = html.EscapeString(contractsVaild.Name)
  182. contractsVaild.Price = html.EscapeString(contractsVaild.Price)
  183. return contractsVaild, nil
  184. }
  185. // 新增合同
  186. func (s *contractService) Add(contractData *viewmodels.Contracts, treeId int) error {
  187. // 1. 项目节存在
  188. contracts := s.treeContractDao.Get(treeId)
  189. if contracts.Id == 0 {
  190. return errors.New("未找到项目节")
  191. }
  192. // 2.项目节是没有合同
  193. if contracts.ContractId != 0 {
  194. return errors.New("该项目节上已经存在合同")
  195. }
  196. // 3.新增合同 --合计标段上的金额
  197. contractsCm := &models.CmContracts{}
  198. contractsCm.Code = contractData.Code
  199. contractsCm.Name = contractData.Name
  200. contractsCm.ContractsType = 1
  201. contractsCm.Price = contractData.Price
  202. contractsCm.Returned = "0"
  203. contractsCm.Paid = "0"
  204. contractsCm.TreeId = treeId
  205. contractsCm.Status = 0
  206. contractsCm.CreateTime = time.Now()
  207. contractsCm.UpdateTime = time.Now()
  208. err := s.contractDao.Add(contractsCm)
  209. if err != nil {
  210. return err
  211. }
  212. return nil
  213. }
  214. // 保存上传文件
  215. func (s *contractService) SaveUpload(Location string, Filename string, id int, ext string) error {
  216. // 文件保存
  217. // 绑定项目节
  218. contractsCm := &models.CmTreeContracts{}
  219. contractsCm.Id = id
  220. // contractsCm.ContractId=
  221. contractsCm.ContractName = ext
  222. s.treeContractDao.Save(contractsCm, []string{"contract_id", "contract_name"})
  223. return nil
  224. }