urgent_task_conf.go 4.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163
  1. package task
  2. import (
  3. "github.com/gin-gonic/gin"
  4. "go.uber.org/zap"
  5. "log-server/global"
  6. "log-server/model/common/response"
  7. "log-server/model/task"
  8. "log-server/model/task/request"
  9. "log-server/utils"
  10. )
  11. type UrgentTaskApi struct {
  12. }
  13. //查询空闲租机
  14. func (a *UrgentTaskApi) GetUnusedPc (c *gin.Context) {
  15. if pcList,err := urgentTaskService.GetUnusedPc(); err != nil {
  16. global.GVA_LOG.Error("创建失败!", zap.Error(err))
  17. response.FailWithMessage(err.Error(), c)
  18. } else {
  19. response.OkWithDetailed(pcList, "获取成功", c)
  20. }
  21. }
  22. //创建紧急任务
  23. func (a *UrgentTaskApi) CreateUrgentTask (c *gin.Context) {
  24. var task1 task.CreateAndUpdateUrgentTaskConf
  25. _ = c.ShouldBindJSON(&task1)
  26. var pcCodeString string
  27. if task1.TaskId == 0 {
  28. response.FailWithMessage("请传入任务id", c)
  29. return
  30. }
  31. if len(task1.PcCode) == 0 {
  32. response.FailWithMessage("请传入租机列表", c)
  33. return
  34. }
  35. //处理pc_code数组
  36. for k,pcCode := range task1.PcCode{
  37. if k != len(task1.PcCode) - 1 {
  38. pcCodeString = pcCodeString + pcCode + ","
  39. } else {
  40. pcCodeString = pcCodeString + pcCode
  41. }
  42. }
  43. //创建实体
  44. entity := task.UrgentTaskConf {
  45. Id: task1.Id,
  46. TaskId: task1.TaskId,
  47. PcCode: pcCodeString,
  48. }
  49. if err := urgentTaskService.CreateUrgentTask(entity); err != nil {
  50. global.GVA_LOG.Error("创建失败!", zap.Error(err))
  51. response.FailWithMessage(err.Error(), c)
  52. } else {
  53. response.OkWithMessage("创建成功", c)
  54. }
  55. }
  56. //删除单个紧急任务
  57. func (a *UrgentTaskApi) DeleteUrgentTask(c *gin.Context) {
  58. var task1 task.UrgentTaskConf
  59. _ = c.ShouldBindJSON(&task1)
  60. if err := utils.Verify(task1, utils.UrgentTaskVerify); err != nil {
  61. response.FailWithMessage(err.Error(), c)
  62. return
  63. }
  64. if err := urgentTaskService.DeleteUrgentTask(task1); err != nil {
  65. global.GVA_LOG.Error("删除失败!", zap.Error(err))
  66. response.FailWithMessage("删除失败", c)
  67. } else {
  68. response.OkWithMessage("删除成功", c)
  69. }
  70. }
  71. //批量删除紧急任务
  72. func (a *UrgentTaskApi) DeleteUrgentTaskByIds (c *gin.Context) {
  73. var ids request.IdsReq
  74. _ = c.ShouldBindJSON(&ids)
  75. if err := urgentTaskService.DeleteUrgentTaskByIds(ids.Ids); err != nil {
  76. global.GVA_LOG.Error("删除失败!", zap.Error(err))
  77. response.FailWithMessage("删除失败", c)
  78. } else {
  79. response.OkWithMessage("删除成功", c)
  80. }
  81. }
  82. //更新紧急任务
  83. func (a *UrgentTaskApi) UpdateUrgentTask (c *gin.Context) {
  84. var task1 task.CreateAndUpdateUrgentTaskConf
  85. _ = c.ShouldBindJSON(&task1)
  86. var pcCodeString string
  87. if task1.TaskId == 0 {
  88. response.FailWithMessage("请传入任务id", c)
  89. return
  90. }
  91. if len(task1.PcCode) == 0 {
  92. response.FailWithMessage("请传入租机列表", c)
  93. return
  94. }
  95. //处理pc_code数组
  96. for k,pcCode := range task1.PcCode{
  97. if k != len(task1.PcCode) - 1 {
  98. pcCodeString = pcCodeString + pcCode + ","
  99. } else {
  100. pcCodeString = pcCodeString + pcCode
  101. }
  102. }
  103. //创建实体
  104. entity := task.UrgentTaskConf {
  105. Id: task1.Id,
  106. TaskId: task1.TaskId,
  107. PcCode: pcCodeString,
  108. CreateDate: task1.CreateDate[:10],
  109. }
  110. if err := urgentTaskService.UpdateUrgentTask(entity); err != nil{
  111. global.GVA_LOG.Error("更新失败!", zap.Error(err))
  112. response.FailWithMessage("更新失败", c)
  113. } else {
  114. response.OkWithMessage("更新成功", c)
  115. }
  116. }
  117. //id获取紧急任务
  118. func (a *UrgentTaskApi) GetUrgentTaskById(c *gin.Context){
  119. var id request.GetById
  120. _ = c.ShouldBindJSON(&id)
  121. if err := utils.Verify(id, utils.IdVerify); err != nil {
  122. response.FailWithMessage(err.Error(), c)
  123. return
  124. }
  125. if task1, err := urgentTaskService.GetUrgentTaskById(id.ID); err != nil{
  126. global.GVA_LOG.Error("获取失败!", zap.Error(err))
  127. response.FailWithMessage("获取失败", c)
  128. } else {
  129. response.OkWithDetailed(task1, "获取成功", c)
  130. }
  131. }
  132. //获取紧急任务列表
  133. func (a *UrgentTaskApi) GetUrgentTaskList(c *gin.Context) {
  134. var paramsInfo request.SearchUrgentTaskParams
  135. _ = c.ShouldBindJSON(&paramsInfo)
  136. if err := utils.Verify(paramsInfo.PageInfo, utils.PageInfoVerify); err != nil {
  137. response.FailWithMessage(err.Error(), c)
  138. return
  139. }
  140. list, total, err := urgentTaskService.GetUrgentTaskList(paramsInfo.UrgentTaskConfRequest, paramsInfo.PageInfo, paramsInfo.OrderKey, paramsInfo.Desc)
  141. if err != nil {
  142. global.GVA_LOG.Error("获取失败!", zap.Error(err))
  143. response.FailWithMessage("获取失败", c)
  144. } else {
  145. response.OkWithDetailed(response.PageResult{
  146. List: list,
  147. Total: total,
  148. Page: paramsInfo.Page,
  149. PageSize: paramsInfo.PageSize,
  150. }, "获取成功", c)
  151. }
  152. }