project_service.go 7.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280
  1. /*
  2. * @description: 项目相关数据处理
  3. * @Author: CP
  4. * @Date: 2020-09-04 14:42:27
  5. * @FilePath: \construction_management\services\project_service.go
  6. */
  7. package services
  8. import (
  9. "errors"
  10. "html"
  11. "log"
  12. "strconv"
  13. "time"
  14. "github.com/kataras/iris/v12"
  15. "go.mod/comm"
  16. "go.mod/conf"
  17. "go.mod/dao"
  18. "go.mod/datasource"
  19. "go.mod/models"
  20. "go.mod/web/viewmodels"
  21. )
  22. //定义项目Service接口
  23. type ProjectService interface {
  24. //ValidManager(code string, account string, password string) error
  25. ValidRule(iris.Context) (viewmodels.Project, error)
  26. ValidRulePage(iris.Context) (viewmodels.ProjectPage, error)
  27. ValidRuleId(ctx iris.Context) (viewmodels.Project, error)
  28. Get(projectId int) (*viewmodels.Project, error)
  29. GetList(page int, size int) []viewmodels.Project
  30. GetName(code string) []viewmodels.Project
  31. Save(projectId int, projectVM viewmodels.Project) error
  32. Add(projectVM viewmodels.Project) error
  33. SaveBs(projectVM viewmodels.Project, id int) error
  34. }
  35. //返回service操作类
  36. type projectService struct {
  37. dao *dao.ProjectDao
  38. validSave string
  39. }
  40. //创建项目service
  41. func NewProjectService() ProjectService {
  42. return &projectService{
  43. dao: dao.NewProjectDao(datasource.InstanceDbMaster()),
  44. validSave: "/api/projectSetting/project/save",
  45. }
  46. }
  47. func (s *projectService) ValidRulePage(ctx iris.Context) (viewmodels.ProjectPage, error) {
  48. projectVaild := viewmodels.ProjectPage{}
  49. err := ctx.ReadForm(&projectVaild)
  50. if err != nil {
  51. log.Println("folder-ValidRule-ReadForm转换异常, error=", err)
  52. return projectVaild, err
  53. }
  54. err = projectVaild.ValidatePage()
  55. if err != nil {
  56. log.Println("参数验证错误, error=", err)
  57. return projectVaild, err
  58. }
  59. return projectVaild, nil
  60. }
  61. // 验证项目
  62. func (s *projectService) ValidRuleId(ctx iris.Context) (viewmodels.Project, error) {
  63. projectVaild := viewmodels.Project{}
  64. err := ctx.ReadForm(&projectVaild)
  65. if err != nil {
  66. log.Println("ReadForm转换异常, error=", err)
  67. return projectVaild, err
  68. }
  69. err = projectVaild.ValidateId()
  70. if err != nil {
  71. log.Println("验证项目, error=", err)
  72. return projectVaild, err
  73. }
  74. return projectVaild, nil
  75. }
  76. // 验证项目项的内容
  77. func (s *projectService) ValidRule(ctx iris.Context) (viewmodels.Project, error) {
  78. projectVaild := viewmodels.Project{}
  79. var err error
  80. if ctx.Method() == "POST" {
  81. err = ctx.ReadJSON(&projectVaild)
  82. if err != nil {
  83. log.Println("project-ValidRule-ReadForm转换异常, error=", err)
  84. return projectVaild, err
  85. }
  86. } else {
  87. err = ctx.ReadForm(&projectVaild)
  88. if err != nil {
  89. log.Println("project-ValidRule-ReadForm转换异常, error=", err)
  90. return projectVaild, err
  91. }
  92. }
  93. if ctx.Path() == s.validSave {
  94. err = projectVaild.ValidateName()
  95. } else {
  96. err = projectVaild.Validate()
  97. }
  98. if err != nil {
  99. log.Println("项目信息验证, error=", err)
  100. return projectVaild, err
  101. }
  102. projectVaild.Code = html.EscapeString(projectVaild.Code)
  103. projectVaild.Name = html.EscapeString(projectVaild.Name)
  104. return projectVaild, nil
  105. }
  106. // ID获得项目信息-TODO
  107. func (s *projectService) Get(projectId int) (*viewmodels.Project, error) {
  108. data, err := s.dao.FindById(projectId)
  109. if err != nil {
  110. return nil, errors.New("获得项目详情出错")
  111. }
  112. project := makeProjectVM(data)
  113. return &project, nil
  114. }
  115. // 获得一组项目信息 -项目编号
  116. func (s *projectService) GetList(page int, size int) []viewmodels.Project {
  117. if size > 100 {
  118. size = 100
  119. }
  120. datalist := make([]viewmodels.Project, 0)
  121. projectData := s.dao.GetPage(page, size)
  122. for _, data := range projectData {
  123. projectVM := makeProjectVM(&data)
  124. datalist = append(datalist, projectVM)
  125. }
  126. return datalist
  127. }
  128. // 获得项目名称
  129. func (s *projectService) GetName(code string) []viewmodels.Project {
  130. datalist := make([]viewmodels.Project, 0)
  131. if code != "" {
  132. projectData := s.dao.GetListByCode(code)
  133. for _, data := range projectData {
  134. projectVM := makeProjectNameVM(&data)
  135. datalist = append(datalist, projectVM)
  136. }
  137. return datalist
  138. }
  139. return datalist
  140. }
  141. // 保存项目信息
  142. func (s *projectService) Save(id int, projectVM viewmodels.Project) error {
  143. project := models.CmProject{}
  144. project.Id = id
  145. project.Name = projectVM.Name
  146. err := s.dao.Update(&project, []string{"Name"})
  147. return err
  148. }
  149. // 新增项目
  150. func (s *projectService) Add(projectVM viewmodels.Project) error {
  151. project := &models.CmProject{}
  152. project.Name = projectVM.Name
  153. project.Code = projectVM.Code
  154. project.CategoryId = projectVM.CategoryId
  155. project.Category = projectVM.Category
  156. project.StaffId = projectVM.StaffId
  157. project.StaffName = projectVM.StaffName
  158. project.CreateTime = time.Now()
  159. project.CreateName = projectVM.CreateName
  160. project.CreateCategory = projectVM.CreateCategory
  161. project.Remark = projectVM.Remark
  162. err := s.dao.Add(project)
  163. return err
  164. }
  165. // 更新项目-后台
  166. func (s *projectService) SaveBs(projectVM viewmodels.Project, id int) error {
  167. project := &models.CmProject{}
  168. project.Id = id
  169. project.Name = projectVM.Name
  170. project.InsideCategoryid = projectVM.InsideCategoryId
  171. project.InsideCategory = projectVM.InsideCategory
  172. project.CategoryId = projectVM.CategoryId
  173. project.Category = projectVM.Category
  174. project.StaffId = projectVM.StaffId
  175. project.StaffName = projectVM.StaffName
  176. project.Remark = projectVM.Remark
  177. err := s.dao.Update(project, []string{"Name", "InsideCategoryid", "InsideCategory", "CategoryId", "Category", "StaffId", "StaffName", "Remark"})
  178. return err
  179. }
  180. // 构造视图层models
  181. func makeProjectVM(projectCM *models.CmProject) viewmodels.Project {
  182. projectVM := viewmodels.Project{}
  183. id, _ := comm.AesEncrypt(strconv.Itoa(projectCM.Id), conf.SignSecret)
  184. userId, _ := comm.AesEncrypt(strconv.Itoa(projectCM.UserId), conf.SignSecret)
  185. projectVM.Id = id
  186. projectVM.Code = projectCM.Code
  187. projectVM.Name = projectCM.Name
  188. projectVM.Category = projectCM.Category
  189. projectVM.StaffName = projectCM.StaffName
  190. projectVM.UserId = userId
  191. projectVM.UserAccount = projectCM.UserAccount
  192. if !projectCM.CreateTime.IsZero() {
  193. projectVM.CreateTime = projectCM.CreateTime.Format(conf.SysTimeform)
  194. }
  195. return projectVM
  196. }
  197. func makeProjectNameVM(projectCM *models.CmProject) viewmodels.Project {
  198. projectVM := viewmodels.Project{}
  199. projectVM.Code = projectCM.Code
  200. projectVM.Name = projectCM.Name
  201. return projectVM
  202. }
  203. // 获得单个项目信息-ID 编号
  204. // func (s *projectService) Get(projectVM *viewmodels.Project) {
  205. // projectCM := models.CmProject{}
  206. // // 项目ID或编号
  207. // if projectVM.Id == "" && projectVM.Code == "" {
  208. // projectVM.Id = "0"
  209. // return
  210. // //return errors.New("项目ID或者项目编号不存在")
  211. // }
  212. // if projectVM.Id != "" {
  213. // projectId, err := comm.AesDecrypt(projectVM.Id, conf.SignSecret)
  214. // if err != nil {
  215. // projectVM.Id = "0"
  216. // return
  217. // }
  218. // projectIdInt, err := strconv.Atoi(projectId)
  219. // if err != nil {
  220. // projectVM.Id = "0"
  221. // return
  222. // }
  223. // if projectIdInt != 0 {
  224. // projectCM.Id = projectIdInt
  225. // }
  226. // } else if projectVM.Code != "" {
  227. // projectCM.Code = projectVM.Code
  228. // } else {
  229. // projectVM.Id = "0"
  230. // return
  231. // }
  232. // s.dao.Get(&projectCM)
  233. // if projectCM.Id != 0 {
  234. // // 项目信息补充完整
  235. // projectB := makeProjectVM(&projectCM)
  236. // projectVM = &projectB
  237. // } else {
  238. // projectVM.Id = "0"
  239. // return
  240. // }
  241. // }