project_service.go 7.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300
  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. GetListBs(pageData viewmodels.ProjectPage) (data []viewmodels.Project, total int64)
  35. }
  36. //返回service操作类
  37. type projectService struct {
  38. dao *dao.ProjectDao
  39. validSave string
  40. }
  41. //创建项目service
  42. func NewProjectService() ProjectService {
  43. return &projectService{
  44. dao: dao.NewProjectDao(datasource.InstanceDbMaster()),
  45. validSave: "/api/projectSetting/project/save",
  46. }
  47. }
  48. func (s *projectService) ValidRulePage(ctx iris.Context) (viewmodels.ProjectPage, error) {
  49. projectVaild := viewmodels.ProjectPage{}
  50. err := ctx.ReadJSON(&projectVaild)
  51. if err != nil {
  52. log.Println("folder-ValidRule-ReadForm转换异常, error=", err)
  53. return projectVaild, err
  54. }
  55. err = projectVaild.ValidatePage()
  56. if err != nil {
  57. log.Println("参数验证错误, error=", err)
  58. return projectVaild, err
  59. }
  60. return projectVaild, nil
  61. }
  62. // 验证项目
  63. func (s *projectService) ValidRuleId(ctx iris.Context) (viewmodels.Project, error) {
  64. projectVaild := viewmodels.Project{}
  65. err := ctx.ReadForm(&projectVaild)
  66. if err != nil {
  67. log.Println("ReadForm转换异常, error=", err)
  68. return projectVaild, err
  69. }
  70. err = projectVaild.ValidateId()
  71. if err != nil {
  72. log.Println("验证项目, error=", err)
  73. return projectVaild, err
  74. }
  75. return projectVaild, nil
  76. }
  77. // 验证项目项的内容
  78. func (s *projectService) ValidRule(ctx iris.Context) (viewmodels.Project, error) {
  79. projectVaild := viewmodels.Project{}
  80. var err error
  81. if ctx.Method() == "POST" {
  82. err = ctx.ReadJSON(&projectVaild)
  83. if err != nil {
  84. log.Println("project-ValidRule-ReadForm转换异常, error=", err)
  85. return projectVaild, err
  86. }
  87. } else {
  88. err = ctx.ReadForm(&projectVaild)
  89. if err != nil {
  90. log.Println("project-ValidRule-ReadForm转换异常, error=", err)
  91. return projectVaild, err
  92. }
  93. }
  94. if ctx.Path() == s.validSave {
  95. err = projectVaild.ValidateName()
  96. } else {
  97. err = projectVaild.Validate()
  98. }
  99. if err != nil {
  100. log.Println("项目信息验证, error=", err)
  101. return projectVaild, err
  102. }
  103. projectVaild.Code = html.EscapeString(projectVaild.Code)
  104. projectVaild.Name = html.EscapeString(projectVaild.Name)
  105. return projectVaild, nil
  106. }
  107. // ID获得项目信息-TODO
  108. func (s *projectService) Get(projectId int) (*viewmodels.Project, error) {
  109. data, err := s.dao.FindById(projectId)
  110. if err != nil {
  111. return nil, errors.New("获得项目详情出错")
  112. }
  113. project := makeProjectVM(data)
  114. return &project, nil
  115. }
  116. // 获得一组项目信息 -项目编号
  117. func (s *projectService) GetList(page int, size int) []viewmodels.Project {
  118. if size > 100 {
  119. size = 100
  120. }
  121. datalist := make([]viewmodels.Project, 0)
  122. projectData := s.dao.GetPage(page, size)
  123. for _, data := range projectData {
  124. projectVM := makeProjectVM(&data)
  125. datalist = append(datalist, projectVM)
  126. }
  127. return datalist
  128. }
  129. // 后端获得列表
  130. func (s *projectService) GetListBs(pageData viewmodels.ProjectPage) (data []viewmodels.Project, total int64) {
  131. if pageData.Size > 100 {
  132. pageData.Size = 100
  133. }
  134. datalist := make([]viewmodels.Project, 0)
  135. projectData, count := s.dao.GetPageBs(pageData)
  136. for _, data := range projectData {
  137. projectVM := makeProjectVM(&data)
  138. datalist = append(datalist, projectVM)
  139. }
  140. return datalist, count
  141. }
  142. // 获得项目名称
  143. func (s *projectService) GetName(code string) []viewmodels.Project {
  144. datalist := make([]viewmodels.Project, 0)
  145. if code != "" {
  146. projectData := s.dao.GetListByCode(code)
  147. for _, data := range projectData {
  148. projectVM := makeProjectNameVM(&data)
  149. datalist = append(datalist, projectVM)
  150. }
  151. return datalist
  152. }
  153. return datalist
  154. }
  155. // 保存项目信息
  156. func (s *projectService) Save(id int, projectVM viewmodels.Project) error {
  157. project := models.CmProject{}
  158. project.Id = id
  159. project.Name = projectVM.Name
  160. err := s.dao.Update(&project, []string{"Name"})
  161. return err
  162. }
  163. // 新增项目
  164. func (s *projectService) Add(projectVM viewmodels.Project) error {
  165. project := &models.CmProject{}
  166. project.Name = projectVM.Name
  167. project.Code = projectVM.Code
  168. project.CategoryId = projectVM.CategoryId
  169. project.Category = projectVM.Category
  170. project.StaffId = projectVM.StaffId
  171. project.StaffName = projectVM.StaffName
  172. project.CreateTime = time.Now()
  173. project.CreateName = projectVM.CreateName
  174. project.CreateCategory = projectVM.CreateCategory
  175. project.Remark = projectVM.Remark
  176. err := s.dao.Add(project)
  177. return err
  178. }
  179. // 更新项目-后台
  180. func (s *projectService) SaveBs(projectVM viewmodels.Project, id int) error {
  181. project := &models.CmProject{}
  182. project.Id = id
  183. project.Name = projectVM.Name
  184. project.InsideCategoryid = projectVM.InsideCategoryId
  185. project.InsideCategory = projectVM.InsideCategory
  186. project.CategoryId = projectVM.CategoryId
  187. project.Category = projectVM.Category
  188. project.StaffId = projectVM.StaffId
  189. project.StaffName = projectVM.StaffName
  190. project.Remark = projectVM.Remark
  191. err := s.dao.Update(project, []string{"Name", "InsideCategoryid", "InsideCategory", "CategoryId", "Category", "StaffId", "StaffName", "Remark"})
  192. return err
  193. }
  194. // 构造视图层models
  195. func makeProjectVM(projectCM *models.CmProject) viewmodels.Project {
  196. projectVM := viewmodels.Project{}
  197. id, _ := comm.AesEncrypt(strconv.Itoa(projectCM.Id), conf.SignSecret)
  198. userId, _ := comm.AesEncrypt(strconv.Itoa(projectCM.UserId), conf.SignSecret)
  199. projectVM.Id = id
  200. projectVM.Code = projectCM.Code
  201. projectVM.Name = projectCM.Name
  202. projectVM.Category = projectCM.Category
  203. projectVM.StaffName = projectCM.StaffName
  204. projectVM.StaffPhone = projectCM.StaffPhone
  205. projectVM.StaffQq = projectCM.StaffQq
  206. projectVM.StaffTelephone = projectCM.StaffTelephone
  207. projectVM.UserId = userId
  208. projectVM.UserAccount = projectCM.UserAccount
  209. projectVM.Status = projectCM.Status
  210. if !projectCM.CreateTime.IsZero() {
  211. projectVM.CreateTime = projectCM.CreateTime.Format(conf.SysTimeform)
  212. }
  213. return projectVM
  214. }
  215. func makeProjectNameVM(projectCM *models.CmProject) viewmodels.Project {
  216. projectVM := viewmodels.Project{}
  217. projectVM.Code = projectCM.Code
  218. projectVM.Name = projectCM.Name
  219. return projectVM
  220. }
  221. // 获得单个项目信息-ID 编号
  222. // func (s *projectService) Get(projectVM *viewmodels.Project) {
  223. // projectCM := models.CmProject{}
  224. // // 项目ID或编号
  225. // if projectVM.Id == "" && projectVM.Code == "" {
  226. // projectVM.Id = "0"
  227. // return
  228. // //return errors.New("项目ID或者项目编号不存在")
  229. // }
  230. // if projectVM.Id != "" {
  231. // projectId, err := comm.AesDecrypt(projectVM.Id, conf.SignSecret)
  232. // if err != nil {
  233. // projectVM.Id = "0"
  234. // return
  235. // }
  236. // projectIdInt, err := strconv.Atoi(projectId)
  237. // if err != nil {
  238. // projectVM.Id = "0"
  239. // return
  240. // }
  241. // if projectIdInt != 0 {
  242. // projectCM.Id = projectIdInt
  243. // }
  244. // } else if projectVM.Code != "" {
  245. // projectCM.Code = projectVM.Code
  246. // } else {
  247. // projectVM.Id = "0"
  248. // return
  249. // }
  250. // s.dao.Get(&projectCM)
  251. // if projectCM.Id != 0 {
  252. // // 项目信息补充完整
  253. // projectB := makeProjectVM(&projectCM)
  254. // projectVM = &projectB
  255. // } else {
  256. // projectVM.Id = "0"
  257. // return
  258. // }
  259. // }