contract_service.go 7.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266
  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, ext string) (int, 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. uploadDao *dao.UploadDao
  49. }
  50. //创建项目用户service
  51. func NewContractService() ContractService {
  52. return &contractService{
  53. treeContractDao: dao.NewTreeContractDao(datasource.InstanceDbMaster()),
  54. contractDao: dao.NewContractDao(datasource.InstanceDbMaster()),
  55. contractReturnDao: dao.NewContractReturnDao(datasource.InstanceDbMaster()),
  56. contractPaidDao: dao.NewContractPaidDao(datasource.InstanceDbMaster()),
  57. treeDao: dao.NewTreeDao(datasource.InstanceDbMaster()),
  58. annexDao: dao.NewAnnexDao(datasource.InstanceDbMaster()),
  59. uploadDao: dao.NewUploadDao(datasource.InstanceDbMaster()),
  60. }
  61. }
  62. // 升级降级规则验证
  63. func (s *contractService) ValidRuleDepth(ctx iris.Context) (*viewmodels.TreeSectionContract, error) {
  64. treeSectionVaild := &viewmodels.TreeSectionContract{}
  65. err := ctx.ReadJSON(treeSectionVaild)
  66. if err != nil {
  67. log.Println("folder-ValidRule-ReadForm转换异常, error=", err)
  68. return treeSectionVaild, err
  69. }
  70. err = treeSectionVaild.ValidateDepth()
  71. if err != nil {
  72. log.Println("参数验证错误, error=", err)
  73. return treeSectionVaild, err
  74. }
  75. return treeSectionVaild, nil
  76. }
  77. // 模板规则验证
  78. func (s *contractService) ValidRuleTemplate(ctx iris.Context) (*viewmodels.TreeSectionContract, error) {
  79. treeSectionVaild := &viewmodels.TreeSectionContract{}
  80. err := ctx.ReadJSON(treeSectionVaild)
  81. if err != nil {
  82. log.Println("folder-ValidRule-ReadForm转换异常, error=", err)
  83. return treeSectionVaild, err
  84. }
  85. err = treeSectionVaild.ValidateTemplate()
  86. if err != nil {
  87. log.Println("参数验证错误, error=", err)
  88. return treeSectionVaild, err
  89. }
  90. return treeSectionVaild, nil
  91. }
  92. // 模板规则新增项目节
  93. func (s *contractService) ValidRuleSectionAdd(ctx iris.Context) (*viewmodels.TreeSectionContract, error) {
  94. treeSectionVaild := &viewmodels.TreeSectionContract{}
  95. err := ctx.ReadJSON(treeSectionVaild)
  96. if err != nil {
  97. log.Println("folder-ValidRule-ReadForm转换异常, error=", err)
  98. return treeSectionVaild, err
  99. }
  100. err = treeSectionVaild.ValidateSectionAdd()
  101. if err != nil {
  102. log.Println("参数验证错误, error=", err)
  103. return treeSectionVaild, err
  104. }
  105. return treeSectionVaild, nil
  106. }
  107. // 模板规则新增项目节
  108. func (s *contractService) ValidRuleSectionDelete(ctx iris.Context) (*viewmodels.TreeSectionContract, error) {
  109. treeSectionVaild := &viewmodels.TreeSectionContract{}
  110. err := ctx.ReadForm(treeSectionVaild)
  111. if err != nil {
  112. log.Println("folder-ValidRule-ReadForm转换异常, error=", err)
  113. return treeSectionVaild, err
  114. }
  115. err = treeSectionVaild.ValidateSectionDelete()
  116. if err != nil {
  117. log.Println("参数验证错误, error=", err)
  118. return treeSectionVaild, err
  119. }
  120. return treeSectionVaild, nil
  121. }
  122. func (s *contractService) ValidRuleSectionNot(ctx iris.Context) (*viewmodels.TreeSectionContract, error) {
  123. treeSectionVaild := &viewmodels.TreeSectionContract{}
  124. err := ctx.ReadForm(treeSectionVaild)
  125. if err != nil {
  126. log.Println("folder-ValidRule-ReadForm转换异常, error=", err)
  127. return treeSectionVaild, err
  128. }
  129. err = treeSectionVaild.ValidateSectionNot()
  130. if err != nil {
  131. log.Println("参数验证错误, error=", err)
  132. return treeSectionVaild, err
  133. }
  134. return treeSectionVaild, nil
  135. }
  136. func (s *contractService) ValidRuleGet(ctx iris.Context) (*viewmodels.TreeSectionContract, error) {
  137. treeSectionVaild := &viewmodels.TreeSectionContract{}
  138. err := ctx.ReadForm(treeSectionVaild)
  139. if err != nil {
  140. log.Println("folder-ValidRule-ReadForm转换异常, error=", err)
  141. return treeSectionVaild, err
  142. }
  143. err = treeSectionVaild.ValidateSectionDelete()
  144. if err != nil {
  145. log.Println("参数验证错误, error=", err)
  146. return treeSectionVaild, err
  147. }
  148. return treeSectionVaild, nil
  149. }
  150. // 验证序号相关
  151. func (s *contractService) ValidRuleSerial(ctx iris.Context) (*viewmodels.TreeSectionContract, error) {
  152. treeSectionVaild := &viewmodels.TreeSectionContract{}
  153. err := ctx.ReadJSON(treeSectionVaild)
  154. if err != nil {
  155. log.Println("folder-ValidRule-ReadForm转换异常, error=", err)
  156. return treeSectionVaild, err
  157. }
  158. err = treeSectionVaild.ValidateSectionSerial()
  159. if err != nil {
  160. log.Println("参数验证错误, error=", err)
  161. return treeSectionVaild, err
  162. }
  163. return treeSectionVaild, nil
  164. }
  165. // 新增合同参数验证
  166. func (s *contractService) ValidRuleContractAdd(ctx iris.Context) (*viewmodels.Contracts, error) {
  167. // 创建一个存放前端传过来参数
  168. contractsVaild := &viewmodels.Contracts{}
  169. // 存放raw的值,放入到contractsVaild
  170. err := ctx.ReadJSON(contractsVaild)
  171. if err != nil {
  172. log.Println("folder-ValidRule-ReadForm转换异常, error=", err)
  173. return contractsVaild, err
  174. }
  175. // 验证合同传参
  176. err = contractsVaild.ValidateAdd()
  177. if err != nil {
  178. log.Println("参数验证错误, error=", err)
  179. return contractsVaild, err
  180. }
  181. // xss
  182. contractsVaild.Code = html.EscapeString(contractsVaild.Code)
  183. contractsVaild.Name = html.EscapeString(contractsVaild.Name)
  184. contractsVaild.Price = html.EscapeString(contractsVaild.Price)
  185. return contractsVaild, nil
  186. }
  187. // 新增合同
  188. func (s *contractService) Add(contractData *viewmodels.Contracts, treeId int) error {
  189. // 1. 项目节存在
  190. contracts := s.treeContractDao.Get(treeId)
  191. if contracts.Id == 0 {
  192. return errors.New("未找到项目节")
  193. }
  194. // 2.项目节是没有合同
  195. if contracts.ContractId != 0 {
  196. return errors.New("该项目节上已经存在合同")
  197. }
  198. // 3.新增合同 --合计标段上的金额
  199. contractsCm := &models.CmContracts{}
  200. contractsCm.Code = contractData.Code
  201. contractsCm.Name = contractData.Name
  202. contractsCm.ContractsType = 1
  203. contractsCm.Price = contractData.Price
  204. contractsCm.Returned = "0"
  205. contractsCm.Paid = "0"
  206. contractsCm.TreeId = treeId
  207. contractsCm.Status = 0
  208. contractsCm.CreateTime = time.Now()
  209. contractsCm.UpdateTime = time.Now()
  210. err := s.contractDao.Add(contractsCm)
  211. if err != nil {
  212. return err
  213. }
  214. return nil
  215. }
  216. // 保存上传文件
  217. func (s *contractService) SaveUpload(Location string, Filename string, ext string) (int, error) {
  218. // 文件保存
  219. uploadCm := &models.CmUpload{}
  220. uploadCm.Name = Filename
  221. uploadCm.Path = Location
  222. uploadCm.Ext = ext
  223. _, err := s.uploadDao.Add(uploadCm)
  224. if err != nil {
  225. return 0, err
  226. }
  227. return uploadCm.Id, nil
  228. }