project_service.go 8.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335
  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, manager *models.CmManager) error
  34. GetListBs(pageData viewmodels.ProjectPage, staffName string) (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, staffName string) (data []viewmodels.Project, total int64) {
  131. if pageData.Size > 100 {
  132. pageData.Size = 100
  133. }
  134. datalist := make([]viewmodels.Project, 0)
  135. // 1.获得管理员信息
  136. managerDao := dao.NewManagerDao(datasource.InstanceDbMaster())
  137. managerData := managerDao.GetUserName(staffName)
  138. if managerData.Id == 0 {
  139. return datalist, 0
  140. }
  141. // 未启用--前端做完后启用
  142. // fmt.Println(staffName)
  143. // staffName = ""
  144. //待启用
  145. if managerData.IsAdmin == 1 {
  146. staffName = ""
  147. }
  148. //待启用
  149. projectData, count := s.dao.GetPageBsByStaffName(pageData, staffName)
  150. for _, data := range projectData {
  151. projectVM := makeProjectVM(&data)
  152. datalist = append(datalist, projectVM)
  153. }
  154. return datalist, count
  155. }
  156. // 获得项目名称
  157. func (s *projectService) GetName(code string) []viewmodels.Project {
  158. datalist := make([]viewmodels.Project, 0)
  159. if code != "" {
  160. projectData := s.dao.GetListByCode(code)
  161. for _, data := range projectData {
  162. projectVM := makeProjectNameVM(&data)
  163. datalist = append(datalist, projectVM)
  164. }
  165. return datalist
  166. }
  167. return datalist
  168. }
  169. // 保存项目信息
  170. func (s *projectService) Save(id int, projectVM viewmodels.Project) error {
  171. project := models.CmProject{}
  172. project.Id = id
  173. project.Name = projectVM.Name
  174. err := s.dao.Update(&project, []string{"Name"})
  175. return err
  176. }
  177. // 新增项目
  178. func (s *projectService) Add(projectVM viewmodels.Project) error {
  179. project := &models.CmProject{}
  180. project.Name = projectVM.Name
  181. project.Code = projectVM.Code
  182. project.CategoryId = projectVM.CategoryId
  183. project.Category = projectVM.Category
  184. project.StaffId = projectVM.StaffId
  185. project.StaffName = projectVM.StaffName
  186. project.CreateTime = time.Now()
  187. project.CreateName = projectVM.CreateName
  188. project.CreateCategory = projectVM.CreateCategory
  189. project.Remark = projectVM.Remark
  190. err := s.dao.Add(project)
  191. return err
  192. }
  193. // 更新项目-后台
  194. func (s *projectService) SaveBs(projectVM viewmodels.Project, id int, manager *models.CmManager) error {
  195. // 1.获得项目信息
  196. projectData, _ := s.dao.FindById(id)
  197. // 管理员和自己可以更新
  198. if manager.IsAdmin != 1 {
  199. if manager.Username != projectData.StaffName {
  200. return errors.New("无权编辑")
  201. }
  202. }
  203. project := &models.CmProject{}
  204. project.Id = id
  205. project.Name = projectVM.Name
  206. project.InsideCategoryid = projectVM.InsideCategoryid
  207. project.InsideCategory = projectVM.InsideCategory
  208. project.CategoryId = projectVM.CategoryId
  209. project.Category = projectVM.Category
  210. project.StaffId = projectVM.StaffId
  211. project.StaffName = projectVM.StaffName
  212. project.Remark = projectVM.Remark
  213. project.Status = projectVM.Status
  214. err := s.dao.Update(project, []string{"Name", "InsideCategoryid", "InsideCategory", "CategoryId", "Category", "StaffId", "StaffName", "Remark", "Status"})
  215. return err
  216. }
  217. // 构造视图层models
  218. func makeProjectVM(projectCM *models.CmProject) viewmodels.Project {
  219. projectVM := viewmodels.Project{}
  220. id, _ := comm.AesEncrypt(strconv.Itoa(projectCM.Id), conf.SignSecret)
  221. userId, _ := comm.AesEncrypt(strconv.Itoa(projectCM.UserId), conf.SignSecret)
  222. staffId, _ := comm.AesEncrypt(projectCM.StaffId, conf.SignSecret)
  223. projectVM.Id = id
  224. projectVM.Code = projectCM.Code
  225. projectVM.Name = projectCM.Name
  226. projectVM.CategoryId = projectCM.CategoryId
  227. projectVM.Category = projectCM.Category
  228. projectVM.StaffIdEncrypted = staffId
  229. projectVM.StaffId = projectCM.StaffId
  230. projectVM.StaffName = projectCM.StaffName
  231. projectVM.StaffPhone = projectCM.StaffPhone
  232. projectVM.StaffQq = projectCM.StaffQq
  233. projectVM.StaffTelephone = projectCM.StaffTelephone
  234. projectVM.UserId = userId
  235. projectVM.UserAccount = projectCM.UserAccount
  236. projectVM.CreateName = projectCM.CreateName
  237. projectVM.Status = projectCM.Status
  238. projectVM.InsideCategory = projectCM.InsideCategory
  239. projectVM.InsideCategoryid = projectCM.InsideCategoryid
  240. if !projectCM.CreateTime.IsZero() {
  241. projectVM.CreateTime = projectCM.CreateTime.Format(conf.SysTimeform)
  242. }
  243. return projectVM
  244. }
  245. func makeProjectNameVM(projectCM *models.CmProject) viewmodels.Project {
  246. projectVM := viewmodels.Project{}
  247. projectVM.Code = projectCM.Code
  248. projectVM.Name = projectCM.Name
  249. return projectVM
  250. }
  251. // 获得单个项目信息-ID 编号
  252. // func (s *projectService) Get(projectVM *viewmodels.Project) {
  253. // projectCM := models.CmProject{}
  254. // // 项目ID或编号
  255. // if projectVM.Id == "" && projectVM.Code == "" {
  256. // projectVM.Id = "0"
  257. // return
  258. // //return errors.New("项目ID或者项目编号不存在")
  259. // }
  260. // if projectVM.Id != "" {
  261. // projectId, err := comm.AesDecrypt(projectVM.Id, conf.SignSecret)
  262. // if err != nil {
  263. // projectVM.Id = "0"
  264. // return
  265. // }
  266. // projectIdInt, err := strconv.Atoi(projectId)
  267. // if err != nil {
  268. // projectVM.Id = "0"
  269. // return
  270. // }
  271. // if projectIdInt != 0 {
  272. // projectCM.Id = projectIdInt
  273. // }
  274. // } else if projectVM.Code != "" {
  275. // projectCM.Code = projectVM.Code
  276. // } else {
  277. // projectVM.Id = "0"
  278. // return
  279. // }
  280. // s.dao.Get(&projectCM)
  281. // if projectCM.Id != 0 {
  282. // // 项目信息补充完整
  283. // projectB := makeProjectVM(&projectCM)
  284. // projectVM = &projectB
  285. // } else {
  286. // projectVM.Id = "0"
  287. // return
  288. // }
  289. // }