ops_event_task.go 53 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601
  1. package opsdev
  2. import (
  3. "context"
  4. "fmt"
  5. "math"
  6. "strings"
  7. "dashoo.cn/opms_libary/myerrors"
  8. opsdevdao "dashoo.cn/opms_parent/app/dao/opsdev"
  9. opsdevmodel "dashoo.cn/opms_parent/app/model/opsdev"
  10. "dashoo.cn/opms_parent/app/service"
  11. "github.com/gogf/gf/database/gdb"
  12. "github.com/gogf/gf/frame/g"
  13. "github.com/gogf/gf/os/gtime"
  14. "github.com/gogf/gf/util/gconv"
  15. )
  16. // OpsEventTaskService 任务业务逻辑实现类
  17. type OpsEventTaskService struct {
  18. *service.ContextService
  19. TaskDao *opsdevdao.OpsEventTaskDao
  20. RecordDao *opsdevdao.OpsEventTaskRecordDao
  21. AttachmentDao *opsdevdao.OpsEventTaskAttachmentDao
  22. ProjectDao *opsdevdao.OpsDeliveryProjectDao
  23. ReleaseDao *opsdevdao.OpsEventTaskReleaseDao
  24. WorkHourDao *opsdevdao.OpsEventTaskWorkHourDao
  25. EventDao *opsdevdao.OpsDeliveryProjectEventDao
  26. EventRecordDao *opsdevdao.OpsDeliveryProjectEventRecordDao
  27. }
  28. // NewOpsEventTaskService 初始化service
  29. func NewOpsEventTaskService(ctx context.Context) (svc *OpsEventTaskService, err error) {
  30. svc = new(OpsEventTaskService)
  31. if svc.ContextService, err = svc.Init(ctx); err != nil {
  32. return nil, err
  33. }
  34. svc.TaskDao = opsdevdao.NewOpsEventTaskDao(svc.Tenant)
  35. svc.RecordDao = opsdevdao.NewOpsEventTaskRecordDao(svc.Tenant)
  36. svc.AttachmentDao = opsdevdao.NewOpsEventTaskAttachmentDao(svc.Tenant)
  37. svc.ProjectDao = opsdevdao.NewOpsDeliveryProjectDao(svc.Tenant)
  38. svc.ReleaseDao = opsdevdao.NewOpsEventTaskReleaseDao(svc.Tenant)
  39. svc.WorkHourDao = opsdevdao.NewOpsEventTaskWorkHourDao(svc.Tenant)
  40. svc.EventDao = opsdevdao.NewOpsDeliveryProjectEventDao(svc.Tenant)
  41. svc.EventRecordDao = opsdevdao.NewOpsDeliveryProjectEventRecordDao(svc.Tenant)
  42. return svc, nil
  43. }
  44. // GetList 分页查询任务列表
  45. func (s *OpsEventTaskService) GetList(req *opsdevmodel.OpsEventTaskSearchReq) (total int, list []*opsdevmodel.OpsEventTaskRsp, err error) {
  46. db := s.TaskDao.FieldsEx(s.TaskDao.Columns.DeletedTime)
  47. // 项目ID筛选(单选)
  48. if req.ProjectId > 0 {
  49. db = db.Where(s.TaskDao.Columns.ProjectId, req.ProjectId)
  50. }
  51. // 项目ID列表筛选(多选,优先于单选)
  52. if len(req.ProjectIds) > 0 {
  53. db = db.Where(s.TaskDao.Columns.ProjectId+" in (?)", req.ProjectIds)
  54. }
  55. // 关联事件ID筛选
  56. if req.EventId > 0 {
  57. db = db.Where(s.TaskDao.Columns.EventId, req.EventId)
  58. }
  59. // 任务标题模糊查询
  60. if req.TaskTitle != "" {
  61. db = db.Where(s.TaskDao.Columns.TaskTitle+" like ?", "%"+req.TaskTitle+"%")
  62. }
  63. // 任务类型筛选(多选)
  64. if len(req.TaskType) > 0 {
  65. db = db.Where(s.TaskDao.Columns.TaskType+" in (?)", req.TaskType)
  66. }
  67. // 任务状态筛选(多选)
  68. if len(req.TaskStatus) > 0 {
  69. db = db.Where(s.TaskDao.Columns.TaskStatus+" in (?)", req.TaskStatus)
  70. }
  71. // 优先级筛选(多选)
  72. if len(req.Priority) > 0 {
  73. db = db.Where(s.TaskDao.Columns.Priority+" in (?)", req.Priority)
  74. }
  75. // 执行人筛选(多选)
  76. if len(req.OpsUserName) > 0 {
  77. db = db.Where(s.TaskDao.Columns.OpsUserName+" in (?)", req.OpsUserName)
  78. }
  79. // 发布版本为空筛选(用于发版任务选择未发版任务)
  80. if req.ReleaseVersionEmpty {
  81. db = db.Where(s.TaskDao.Columns.ReleaseVersion + " is null")
  82. }
  83. // 排期状态筛选(plan_start_time/plan_end_time 为 datetime 类型,需用 IS NULL)
  84. if req.ScheduleStatus == "scheduled" {
  85. db = db.Where(s.TaskDao.Columns.PlanStartTime + " is not null").Where(s.TaskDao.Columns.PlanEndTime + " is not null")
  86. } else if req.ScheduleStatus == "unscheduled" {
  87. db = db.Where(s.TaskDao.Columns.PlanStartTime + " is null or " + s.TaskDao.Columns.PlanEndTime + " is null")
  88. }
  89. // 计划结束日期范围筛选
  90. if req.PlanEndDateStart != "" {
  91. db = db.Where(s.TaskDao.Columns.PlanEndTime+" >= ?", req.PlanEndDateStart+" 00:00:00")
  92. }
  93. if req.PlanEndDateEnd != "" {
  94. db = db.Where(s.TaskDao.Columns.PlanEndTime+" <= ?", req.PlanEndDateEnd+" 23:59:59")
  95. }
  96. // 创建日期范围筛选
  97. if req.CreatedTimeStart != "" {
  98. db = db.Where(s.TaskDao.Columns.CreatedTime+" >= ?", req.CreatedTimeStart+" 00:00:00")
  99. }
  100. if req.CreatedTimeEnd != "" {
  101. db = db.Where(s.TaskDao.Columns.CreatedTime+" <= ?", req.CreatedTimeEnd+" 23:59:59")
  102. }
  103. // 完成日期范围筛选
  104. if req.CompleteTimeStart != "" {
  105. db = db.Where(s.TaskDao.Columns.CompleteTime+" >= ?", req.CompleteTimeStart+" 00:00:00")
  106. }
  107. if req.CompleteTimeEnd != "" {
  108. db = db.Where(s.TaskDao.Columns.CompleteTime+" <= ?", req.CompleteTimeEnd+" 23:59:59")
  109. }
  110. // 统计总数
  111. total, err = db.Count()
  112. if err != nil {
  113. g.Log().Error(err)
  114. return 0, nil, myerrors.DbError("获取任务总数失败")
  115. }
  116. // 分页查询
  117. pageNum, pageSize := req.GetPage()
  118. var entityList []*opsdevmodel.OpsEventTask
  119. // 处理排序
  120. if len(req.SortFields) > 0 {
  121. orderClauses := []string{}
  122. for _, sort := range req.SortFields {
  123. // 将前端字段名转换为数据库列名
  124. colName := s.getSortColumnName(sort.Field)
  125. if colName != "" {
  126. orderClauses = append(orderClauses, colName+" "+strings.ToUpper(sort.Order))
  127. }
  128. }
  129. if len(orderClauses) > 0 {
  130. db = db.Order(strings.Join(orderClauses, ", "))
  131. } else {
  132. db = db.Order(s.TaskDao.Columns.CreatedTime + " desc")
  133. }
  134. } else {
  135. db = db.Order(s.TaskDao.Columns.CreatedTime + " desc")
  136. }
  137. err = db.Page(pageNum, pageSize).Scan(&entityList)
  138. if err != nil {
  139. g.Log().Error(err)
  140. return 0, nil, myerrors.DbError("查询任务列表失败")
  141. }
  142. // 转换为响应结构体
  143. if err = gconv.Structs(entityList, &list); err != nil {
  144. g.Log().Error(err)
  145. return 0, nil, myerrors.DbError("数据转换失败")
  146. }
  147. return
  148. }
  149. // getSortColumnName 将前端排序字段名转换为数据库列名
  150. func (s *OpsEventTaskService) getSortColumnName(field string) string {
  151. // 前端字段名到数据库列名的映射
  152. fieldMap := map[string]string{
  153. "taskTitle": s.TaskDao.Columns.TaskTitle,
  154. "functionName": s.TaskDao.Columns.FunctionName,
  155. "taskType": s.TaskDao.Columns.TaskType,
  156. "taskStatus": s.TaskDao.Columns.TaskStatus,
  157. "priority": s.TaskDao.Columns.Priority,
  158. "opsUserName": s.TaskDao.Columns.OpsUserName,
  159. "planStartTime": s.TaskDao.Columns.PlanStartTime,
  160. "planEndTime": s.TaskDao.Columns.PlanEndTime,
  161. "completeTime": s.TaskDao.Columns.CompleteTime,
  162. "releaseVersion": s.TaskDao.Columns.ReleaseVersion,
  163. "createdTime": s.TaskDao.Columns.CreatedTime,
  164. }
  165. if colName, ok := fieldMap[field]; ok {
  166. return colName
  167. }
  168. return ""
  169. }
  170. const maxTaskNoRetries = 3
  171. // isDuplicateEntryError 判断是否为数据库唯一键冲突
  172. func isDuplicateEntryError(err error) bool {
  173. if err == nil {
  174. return false
  175. }
  176. return strings.Contains(strings.ToLower(err.Error()), "duplicate entry")
  177. }
  178. // Create 新增任务,包含事务控制和过程记录
  179. func (s *OpsEventTaskService) Create(req *opsdevmodel.OpsEventTaskAddReq) (err error) {
  180. // 判断任务状态:如果执行人、计划开始时间、计划结束时间都填写了,则状态为处理中,否则为待处理
  181. taskStatus := opsdevmodel.TaskStatusTodo
  182. if req.OpsUserId > 0 && req.PlanStartTime != "" && req.PlanEndTime != "" {
  183. taskStatus = opsdevmodel.TaskStatusProcessing
  184. }
  185. for i := 0; i < maxTaskNoRetries; i++ {
  186. err = s.doCreate(req, taskStatus)
  187. if err == nil {
  188. return nil
  189. }
  190. if !isDuplicateEntryError(err) {
  191. return err
  192. }
  193. }
  194. return myerrors.DbError("任务编号生成冲突,请稍后重试")
  195. }
  196. // doCreate 执行新增任务(可在任务编号冲突时重试)
  197. func (s *OpsEventTaskService) doCreate(req *opsdevmodel.OpsEventTaskAddReq, taskStatus string) error {
  198. // 生成任务编号
  199. taskNo := s.generateTaskNo()
  200. // 构造数据
  201. data := g.Map{
  202. s.TaskDao.Columns.TaskNo: taskNo,
  203. s.TaskDao.Columns.ProjectId: req.ProjectId,
  204. s.TaskDao.Columns.ProjectName: s.getProjectName(req.ProjectId),
  205. s.TaskDao.Columns.TaskTitle: req.TaskTitle,
  206. s.TaskDao.Columns.TaskDesc: req.TaskDesc,
  207. s.TaskDao.Columns.FunctionName: req.FunctionName,
  208. s.TaskDao.Columns.TaskType: req.TaskType,
  209. s.TaskDao.Columns.TaskStatus: taskStatus,
  210. s.TaskDao.Columns.Priority: req.Priority,
  211. s.TaskDao.Columns.OpsUserId: req.OpsUserId,
  212. s.TaskDao.Columns.OpsUserName: req.OpsUserName,
  213. s.TaskDao.Columns.EstimateWorkHour: req.EstimateWorkHour,
  214. s.TaskDao.Columns.DefectType: req.DefectType,
  215. s.TaskDao.Columns.ReleaseVersion: req.ReleaseVersion,
  216. s.TaskDao.Columns.Remark: req.Remark,
  217. s.TaskDao.Columns.EventId: req.EventId,
  218. s.TaskDao.Columns.EventType: req.EventType,
  219. s.TaskDao.Columns.TaskParentId: req.TaskParentId,
  220. s.TaskDao.Columns.Attribute2: req.Attribute2,
  221. }
  222. if req.PlanStartTime != "" {
  223. data[s.TaskDao.Columns.PlanStartTime] = req.PlanStartTime
  224. }
  225. if req.PlanEndTime != "" {
  226. data[s.TaskDao.Columns.PlanEndTime] = req.PlanEndTime
  227. }
  228. // 补齐审计字段
  229. service.SetCreatedInfo(data, s.GetCxtUserId(), s.GetCxtUserName())
  230. // 使用事务控制
  231. return s.TaskDao.Transaction(context.TODO(), func(ctx context.Context, tx *gdb.TX) error {
  232. // 1. 创建任务记录
  233. result, err := s.TaskDao.TX(tx).Data(data).Insert()
  234. if err != nil {
  235. g.Log().Error(err)
  236. if isDuplicateEntryError(err) {
  237. return err
  238. }
  239. return myerrors.DbError("新增任务失败")
  240. }
  241. // 获取新创建的任务ID
  242. taskId, err := result.LastInsertId()
  243. if err != nil {
  244. g.Log().Error(err)
  245. return myerrors.DbError("获取任务ID失败")
  246. }
  247. // 2. 创建任务过程记录
  248. recordData := g.Map{
  249. s.RecordDao.Columns.TaskId: taskId,
  250. s.RecordDao.Columns.HandleUserId: s.GetCxtUserId(),
  251. s.RecordDao.Columns.HandleUserName: s.GetCxtUserName(),
  252. s.RecordDao.Columns.HandleContent: "创建任务<br/>任务标题: " + req.TaskTitle,
  253. }
  254. service.SetCreatedInfo(recordData, s.GetCxtUserId(), s.GetCxtUserName())
  255. recordResult, err := s.RecordDao.TX(tx).Data(recordData).Insert()
  256. if err != nil {
  257. g.Log().Error(err)
  258. return myerrors.DbError("新增任务过程记录失败")
  259. }
  260. // 获取过程记录ID
  261. recordId, err := recordResult.LastInsertId()
  262. if err != nil {
  263. g.Log().Error(err)
  264. return myerrors.DbError("获取过程记录ID失败")
  265. }
  266. // 3. 保存附件到任务附件表,关联过程记录
  267. if len(req.Attachments) > 0 {
  268. for _, att := range req.Attachments {
  269. attData := g.Map{
  270. s.AttachmentDao.Columns.TaskId: taskId,
  271. s.AttachmentDao.Columns.TaskRecordId: recordId,
  272. s.AttachmentDao.Columns.FileName: att.FileName,
  273. s.AttachmentDao.Columns.FileUrl: att.FileUrl,
  274. s.AttachmentDao.Columns.FileType: att.FileType,
  275. }
  276. service.SetCreatedInfo(attData, s.GetCxtUserId(), s.GetCxtUserName())
  277. _, err = s.AttachmentDao.TX(tx).Data(attData).Insert()
  278. if err != nil {
  279. g.Log().Error(err)
  280. return myerrors.DbError("保存附件信息失败")
  281. }
  282. }
  283. }
  284. return nil
  285. })
  286. }
  287. // UpdateById 根据ID更新任务
  288. func (s *OpsEventTaskService) UpdateById(req *opsdevmodel.OpsEventTaskUpdateReq) error {
  289. // 校验数据是否存在
  290. var entity opsdevmodel.OpsEventTask
  291. err := s.TaskDao.FieldsEx(s.TaskDao.Columns.DeletedTime).WherePri(s.TaskDao.Columns.Id, req.Id).Scan(&entity)
  292. if err != nil {
  293. g.Log().Error(err)
  294. return myerrors.DbError("查询任务数据失败")
  295. }
  296. if entity.Id <= 0 {
  297. return myerrors.TipsError("任务数据不存在")
  298. }
  299. // 已完成(30)或已作废(90)状态的任务不允许编辑
  300. if !s.canEdit(entity.TaskStatus) {
  301. return myerrors.TipsError("已完成或已作废状态的任务不允许编辑")
  302. }
  303. // 构造更新数据 - 只更新传入的非空字段
  304. data := g.Map{}
  305. if req.TaskTitle != "" {
  306. data[s.TaskDao.Columns.TaskTitle] = req.TaskTitle
  307. }
  308. if req.TaskDesc != "" {
  309. data[s.TaskDao.Columns.TaskDesc] = req.TaskDesc
  310. }
  311. if req.FunctionName != "" {
  312. data[s.TaskDao.Columns.FunctionName] = req.FunctionName
  313. }
  314. if req.TaskType != "" {
  315. data[s.TaskDao.Columns.TaskType] = req.TaskType
  316. }
  317. if req.Priority != "" {
  318. data[s.TaskDao.Columns.Priority] = req.Priority
  319. }
  320. if req.OpsUserId > 0 {
  321. data[s.TaskDao.Columns.OpsUserId] = req.OpsUserId
  322. }
  323. if req.OpsUserName != "" {
  324. data[s.TaskDao.Columns.OpsUserName] = req.OpsUserName
  325. }
  326. if req.PlanStartTime != "" {
  327. data[s.TaskDao.Columns.PlanStartTime] = req.PlanStartTime
  328. }
  329. if req.PlanEndTime != "" {
  330. data[s.TaskDao.Columns.PlanEndTime] = req.PlanEndTime
  331. }
  332. if req.EstimateWorkHour > 0 {
  333. data[s.TaskDao.Columns.EstimateWorkHour] = req.EstimateWorkHour
  334. }
  335. if req.DefectType != "" {
  336. data[s.TaskDao.Columns.DefectType] = req.DefectType
  337. }
  338. if req.ReleaseVersion != "" {
  339. data[s.TaskDao.Columns.ReleaseVersion] = req.ReleaseVersion
  340. }
  341. if req.Remark != "" {
  342. data[s.TaskDao.Columns.Remark] = req.Remark
  343. }
  344. if req.Attribute2 != "" {
  345. data[s.TaskDao.Columns.Attribute2] = req.Attribute2
  346. }
  347. // 补齐审计字段
  348. service.SetUpdatedInfo(data, s.GetCxtUserId(), s.GetCxtUserName())
  349. // 更新操作,排除不可改字段
  350. _, err = s.TaskDao.FieldsEx(service.UpdateFieldEx...).Data(data).WherePri(s.TaskDao.Columns.Id, req.Id).Update()
  351. if err != nil {
  352. g.Log().Error(err)
  353. return myerrors.DbError("更新任务失败")
  354. }
  355. return nil
  356. }
  357. // DeleteByIds 根据ID批量删除
  358. func (s *OpsEventTaskService) DeleteByIds(ids []int64) error {
  359. if len(ids) == 0 {
  360. return myerrors.TipsError("请选择需要删除的任务")
  361. }
  362. return s.TaskDao.Transaction(context.TODO(), func(ctx context.Context, tx *gdb.TX) error {
  363. // 1. 删除任务
  364. _, err := s.TaskDao.TX(tx).WhereIn(s.TaskDao.Columns.Id, ids).Delete()
  365. if err != nil {
  366. g.Log().Error(err)
  367. return myerrors.DbError("删除任务失败")
  368. }
  369. // 2. 删除关联的过程记录
  370. _, err = s.RecordDao.TX(tx).WhereIn(s.RecordDao.Columns.TaskId, ids).Delete()
  371. if err != nil {
  372. g.Log().Error(err)
  373. return myerrors.DbError("删除任务过程记录失败")
  374. }
  375. // 3. 删除关联的附件
  376. _, err = s.AttachmentDao.TX(tx).WhereIn(s.AttachmentDao.Columns.TaskId, ids).Delete()
  377. if err != nil {
  378. g.Log().Error(err)
  379. return myerrors.DbError("删除任务附件失败")
  380. }
  381. return nil
  382. })
  383. }
  384. // GetById 根据ID查询单条(关联项目信息)
  385. func (s *OpsEventTaskService) GetById(id int) (*opsdevmodel.OpsEventTaskRsp, error) {
  386. var entity opsdevmodel.OpsEventTask
  387. err := s.TaskDao.FieldsEx(s.TaskDao.Columns.DeletedTime).WherePri(s.TaskDao.Columns.Id, id).Scan(&entity)
  388. if err != nil {
  389. g.Log().Error(err)
  390. return nil, myerrors.DbError("查询任务数据失败")
  391. }
  392. if entity.Id <= 0 {
  393. return nil, myerrors.TipsError("任务数据不存在")
  394. }
  395. var rsp opsdevmodel.OpsEventTaskRsp
  396. if err := gconv.Struct(entity, &rsp); err != nil {
  397. g.Log().Error(err)
  398. return nil, myerrors.DbError("数据转换失败")
  399. }
  400. return &rsp, nil
  401. }
  402. // Schedule 任务排期
  403. func (s *OpsEventTaskService) Schedule(req *opsdevmodel.OpsEventTaskScheduleReq) error {
  404. // 校验数据是否存在
  405. var entity opsdevmodel.OpsEventTask
  406. err := s.TaskDao.FieldsEx(s.TaskDao.Columns.DeletedTime).WherePri(s.TaskDao.Columns.Id, req.Id).Scan(&entity)
  407. if err != nil {
  408. g.Log().Error(err)
  409. return myerrors.DbError("查询任务数据失败")
  410. }
  411. if entity.Id <= 0 {
  412. return myerrors.TipsError("任务数据不存在")
  413. }
  414. // 只有待处理状态可以排期
  415. if !s.canSchedule(entity.TaskStatus) {
  416. return myerrors.TipsError("只有待处理状态的任务可以排期")
  417. }
  418. data := g.Map{
  419. s.TaskDao.Columns.OpsUserId: req.OpsUserId,
  420. s.TaskDao.Columns.OpsUserName: req.OpsUserName,
  421. s.TaskDao.Columns.PlanStartTime: req.PlanStartTime,
  422. s.TaskDao.Columns.PlanEndTime: req.PlanEndTime,
  423. s.TaskDao.Columns.EstimateWorkHour: req.EstimateWorkHour,
  424. s.TaskDao.Columns.TaskStatus: opsdevmodel.TaskStatusProcessing,
  425. }
  426. // 补齐审计字段
  427. service.SetUpdatedInfo(data, s.GetCxtUserId(), s.GetCxtUserName())
  428. // 使用事务
  429. return s.TaskDao.Transaction(context.TODO(), func(ctx context.Context, tx *gdb.TX) error {
  430. // 1. 更新任务
  431. _, err := s.TaskDao.TX(tx).FieldsEx(service.UpdateFieldEx...).Data(data).WherePri(s.TaskDao.Columns.Id, req.Id).Update()
  432. if err != nil {
  433. g.Log().Error(err)
  434. return myerrors.DbError("任务排期失败")
  435. }
  436. // 2. 创建过程记录
  437. handleContent := "任务排期<br/>执行人: " + req.OpsUserName +
  438. "<br/>计划开始: " + req.PlanStartTime +
  439. "<br/>计划结束: " + req.PlanEndTime +
  440. "<br/>预估工时: " + gconv.String(req.EstimateWorkHour) + "小时"
  441. recordData := g.Map{
  442. s.RecordDao.Columns.TaskId: req.Id,
  443. s.RecordDao.Columns.HandleUserId: s.GetCxtUserId(),
  444. s.RecordDao.Columns.HandleUserName: s.GetCxtUserName(),
  445. s.RecordDao.Columns.HandleContent: handleContent,
  446. }
  447. service.SetCreatedInfo(recordData, s.GetCxtUserId(), s.GetCxtUserName())
  448. _, err = s.RecordDao.TX(tx).Data(recordData).Insert()
  449. if err != nil {
  450. g.Log().Error(err)
  451. return myerrors.DbError("新增过程记录失败")
  452. }
  453. return nil
  454. })
  455. }
  456. // Start 开始任务
  457. func (s *OpsEventTaskService) Start(req *opsdevmodel.OpsEventTaskStartReq) error {
  458. // 校验数据是否存在
  459. var entity opsdevmodel.OpsEventTask
  460. err := s.TaskDao.FieldsEx(s.TaskDao.Columns.DeletedTime).WherePri(s.TaskDao.Columns.Id, req.Id).Scan(&entity)
  461. if err != nil {
  462. g.Log().Error(err)
  463. return myerrors.DbError("查询任务数据失败")
  464. }
  465. if entity.Id <= 0 {
  466. return myerrors.TipsError("任务数据不存在")
  467. }
  468. // 只有待处理(10)或暂停(25)状态可以开始
  469. if !s.canStart(entity.TaskStatus) {
  470. return myerrors.TipsError("只有待处理或暂停状态的任务可以开始")
  471. }
  472. // 构造更新数据
  473. data := g.Map{
  474. s.TaskDao.Columns.TaskStatus: opsdevmodel.TaskStatusProcessing, // 处理中
  475. }
  476. // 补齐审计字段
  477. service.SetUpdatedInfo(data, s.GetCxtUserId(), s.GetCxtUserName())
  478. // 使用事务
  479. return s.TaskDao.Transaction(context.TODO(), func(ctx context.Context, tx *gdb.TX) error {
  480. // 1. 更新任务状态
  481. _, err := s.TaskDao.TX(tx).FieldsEx(service.UpdateFieldEx...).Data(data).WherePri(s.TaskDao.Columns.Id, req.Id).Update()
  482. if err != nil {
  483. g.Log().Error(err)
  484. return myerrors.DbError("开始任务失败")
  485. }
  486. // 2. 创建过程记录
  487. recordData := g.Map{
  488. s.RecordDao.Columns.TaskId: req.Id,
  489. s.RecordDao.Columns.HandleUserId: s.GetCxtUserId(),
  490. s.RecordDao.Columns.HandleUserName: s.GetCxtUserName(),
  491. s.RecordDao.Columns.HandleContent: "开始处理任务<br/>任务标题: " + entity.TaskTitle,
  492. }
  493. service.SetCreatedInfo(recordData, s.GetCxtUserId(), s.GetCxtUserName())
  494. _, err = s.RecordDao.TX(tx).Data(recordData).Insert()
  495. if err != nil {
  496. g.Log().Error(err)
  497. return myerrors.DbError("新增过程记录失败")
  498. }
  499. return nil
  500. })
  501. }
  502. // Complete 完成任务
  503. func (s *OpsEventTaskService) Complete(req *opsdevmodel.OpsEventTaskCompleteReq) (err error) {
  504. g.Log().Infof("Handler received req: %+v", req)
  505. // 校验数据是否存在
  506. var entity opsdevmodel.OpsEventTask
  507. err = s.TaskDao.FieldsEx(s.TaskDao.Columns.DeletedTime).WherePri(s.TaskDao.Columns.Id, req.Id).Scan(&entity)
  508. if err != nil {
  509. g.Log().Error(err)
  510. return myerrors.DbError("查询任务数据失败")
  511. }
  512. if entity.Id <= 0 {
  513. return myerrors.TipsError("任务数据不存在")
  514. }
  515. // 只有处理中(20)状态可以完成
  516. if !s.canComplete(entity.TaskStatus) {
  517. return myerrors.TipsError("只有处理中的任务可以完成")
  518. }
  519. for i := 0; i < maxTaskNoRetries; i++ {
  520. err = s.doComplete(req, &entity)
  521. if err == nil {
  522. return nil
  523. }
  524. if !isDuplicateEntryError(err) {
  525. return err
  526. }
  527. }
  528. return myerrors.DbError("任务编号生成冲突,请稍后重试")
  529. }
  530. // doComplete 执行完成任务(可在下游任务编号冲突时重试)
  531. func (s *OpsEventTaskService) doComplete(req *opsdevmodel.OpsEventTaskCompleteReq, entity *opsdevmodel.OpsEventTask) error {
  532. // 解析完成日期
  533. completeTime := gtime.Now()
  534. if req.CompleteDate != "" {
  535. if parsed := gtime.NewFromStrFormat(req.CompleteDate, "Y-m-d"); parsed != nil {
  536. completeTime = parsed
  537. }
  538. }
  539. // 构造更新数据
  540. data := g.Map{
  541. s.TaskDao.Columns.TaskStatus: opsdevmodel.TaskStatusCompleted, // 已完成
  542. s.TaskDao.Columns.CompleteTime: completeTime,
  543. s.TaskDao.Columns.ActualWorkHour: req.ActualWorkHour,
  544. s.TaskDao.Columns.Remark: req.Remark,
  545. }
  546. // 补齐审计字段
  547. service.SetUpdatedInfo(data, s.GetCxtUserId(), s.GetCxtUserName())
  548. // 预生成下游任务编号(避免事务内查询重复)
  549. var testTaskNo string
  550. if entity.TaskType == opsdevmodel.TaskTypeFeatureDev {
  551. testTaskNo = s.generateTaskNo()
  552. }
  553. if entity.TaskType == opsdevmodel.TaskTypeBug {
  554. testTaskNo = s.generateTaskNo()
  555. }
  556. // 使用事务
  557. return s.TaskDao.Transaction(context.TODO(), func(ctx context.Context, tx *gdb.TX) error {
  558. // 1. 更新任务状态
  559. _, err := s.TaskDao.TX(tx).FieldsEx(service.UpdateFieldEx...).Data(data).WherePri(s.TaskDao.Columns.Id, req.Id).Update()
  560. if err != nil {
  561. g.Log().Error(err)
  562. return myerrors.DbError("完成任务失败")
  563. }
  564. // 2. 检查实际工时差异,自动登记工时差额
  565. delta := req.ActualWorkHour - entity.ActualWorkHour
  566. if delta != 0 {
  567. workHourData := g.Map{
  568. s.WorkHourDao.Columns.TaskId: req.Id,
  569. s.WorkHourDao.Columns.OpsUserId: s.GetCxtUserId(),
  570. s.WorkHourDao.Columns.OpsUserName: s.GetCxtUserName(),
  571. s.WorkHourDao.Columns.ActualWorkDate: completeTime.Format("Y-m-d"),
  572. s.WorkHourDao.Columns.ActualWorkHour: delta,
  573. s.WorkHourDao.Columns.Remark: "完成任务自动登记工时差额",
  574. }
  575. service.SetCreatedInfo(workHourData, s.GetCxtUserId(), s.GetCxtUserName())
  576. _, err = s.WorkHourDao.TX(tx).Data(workHourData).Insert()
  577. if err != nil {
  578. g.Log().Error(err)
  579. return myerrors.DbError("登记工时差额失败")
  580. }
  581. }
  582. // 3. 创建过程记录
  583. var handleContent string
  584. if entity.TaskType == opsdevmodel.TaskTypeFeatureTest && req.TestResult != "" {
  585. testResultText := "通过"
  586. if req.TestResult == "fail" {
  587. testResultText = "不通过"
  588. }
  589. handleContent = fmt.Sprintf("测试结果:%s <br/>测试时间:%s", testResultText, gtime.Now().Format("Y-m-d"))
  590. } else if entity.TaskType == opsdevmodel.TaskTypeSystemReleaseEvt && req.IsReleaseComplete {
  591. // 系统发版完成记录
  592. handleContent = "发版完成<br/>任务标题: " + entity.TaskTitle + "<br/>发版工时: " + gconv.String(req.ActualWorkHour) + "小时"
  593. if req.Remark != "" {
  594. handleContent += "<br/>发版说明: " + req.Remark
  595. }
  596. if len(req.DevTaskIds) > 0 {
  597. handleContent += "<br/>关联研发任务数: " + gconv.String(len(req.DevTaskIds)) + "个"
  598. }
  599. } else {
  600. // 普通任务完成记录
  601. handleContent = "完成任务<br/>任务标题: " + entity.TaskTitle + "<br/>实际工作量: " + gconv.String(req.ActualWorkHour) + "小时"
  602. }
  603. recordData := g.Map{
  604. s.RecordDao.Columns.TaskId: req.Id,
  605. s.RecordDao.Columns.HandleUserId: s.GetCxtUserId(),
  606. s.RecordDao.Columns.HandleUserName: s.GetCxtUserName(),
  607. s.RecordDao.Columns.HandleContent: handleContent,
  608. }
  609. service.SetCreatedInfo(recordData, s.GetCxtUserId(), s.GetCxtUserName())
  610. result, err := s.RecordDao.TX(tx).Data(recordData).Insert()
  611. if err != nil {
  612. g.Log().Error(err)
  613. return myerrors.DbError("新增过程记录失败")
  614. }
  615. recordId, _ := result.LastInsertId()
  616. // 4. 保存附件
  617. if len(req.Attachments) > 0 {
  618. for _, att := range req.Attachments {
  619. attData := g.Map{
  620. s.AttachmentDao.Columns.TaskId: req.Id,
  621. s.AttachmentDao.Columns.TaskRecordId: recordId,
  622. s.AttachmentDao.Columns.FileName: att.FileName,
  623. s.AttachmentDao.Columns.FileUrl: att.FileUrl,
  624. s.AttachmentDao.Columns.FileType: att.FileType,
  625. }
  626. service.SetCreatedInfo(attData, s.GetCxtUserId(), s.GetCxtUserName())
  627. _, err = s.AttachmentDao.TX(tx).Data(attData).Insert()
  628. if err != nil {
  629. g.Log().Error(err)
  630. return myerrors.DbError("保存附件信息失败")
  631. }
  632. }
  633. }
  634. // 5. 根据任务类型自动创建下游任务
  635. // 5.1 功能开发完成时,自动创建功能测试任务(原4.2)
  636. if entity.TaskType == opsdevmodel.TaskTypeFeatureDev {
  637. testTaskData := g.Map{
  638. s.TaskDao.Columns.TaskNo: testTaskNo,
  639. s.TaskDao.Columns.ProjectId: entity.ProjectId,
  640. s.TaskDao.Columns.ProjectName: entity.ProjectName,
  641. s.TaskDao.Columns.EventId: entity.EventId,
  642. s.TaskDao.Columns.EventType: entity.EventType,
  643. s.TaskDao.Columns.TaskTitle: entity.TaskTitle,
  644. s.TaskDao.Columns.TaskDesc: entity.TaskDesc,
  645. s.TaskDao.Columns.FunctionName: entity.FunctionName,
  646. s.TaskDao.Columns.TaskType: opsdevmodel.TaskTypeFeatureTest, // 功能测试
  647. s.TaskDao.Columns.TaskStatus: opsdevmodel.TaskStatusTodo, // 待处理
  648. s.TaskDao.Columns.Priority: entity.Priority,
  649. s.TaskDao.Columns.TaskParentId: req.Id,
  650. }
  651. service.SetCreatedInfo(testTaskData, s.GetCxtUserId(), s.GetCxtUserName())
  652. testResult, err := s.TaskDao.TX(tx).Data(testTaskData).Insert()
  653. if err != nil {
  654. g.Log().Error(err)
  655. if isDuplicateEntryError(err) {
  656. return err
  657. }
  658. return myerrors.DbError("自动创建功能测试任务失败")
  659. }
  660. testTaskId, _ := testResult.LastInsertId()
  661. // 创建功能测试任务的过程记录
  662. testRecordData := g.Map{
  663. s.RecordDao.Columns.TaskId: testTaskId,
  664. s.RecordDao.Columns.HandleUserId: s.GetCxtUserId(),
  665. s.RecordDao.Columns.HandleUserName: s.GetCxtUserName(),
  666. s.RecordDao.Columns.HandleContent: "创建功能测试任务<br/>说明: 由功能开发任务自动创建",
  667. }
  668. service.SetCreatedInfo(testRecordData, s.GetCxtUserId(), s.GetCxtUserName())
  669. _, err = s.RecordDao.TX(tx).Data(testRecordData).Insert()
  670. if err != nil {
  671. g.Log().Error(err)
  672. return myerrors.DbError("新增功能测试任务过程记录失败")
  673. }
  674. }
  675. // 5.2b BUG任务完成时,自动创建功能测试任务
  676. if entity.TaskType == opsdevmodel.TaskTypeBug {
  677. testTaskData := g.Map{
  678. s.TaskDao.Columns.TaskNo: testTaskNo,
  679. s.TaskDao.Columns.ProjectId: entity.ProjectId,
  680. s.TaskDao.Columns.ProjectName: entity.ProjectName,
  681. s.TaskDao.Columns.EventId: entity.EventId,
  682. s.TaskDao.Columns.EventType: entity.EventType,
  683. s.TaskDao.Columns.TaskTitle: entity.TaskTitle,
  684. s.TaskDao.Columns.TaskDesc: entity.TaskDesc,
  685. s.TaskDao.Columns.FunctionName: entity.FunctionName,
  686. s.TaskDao.Columns.TaskType: opsdevmodel.TaskTypeFeatureTest, // 功能测试
  687. s.TaskDao.Columns.TaskStatus: opsdevmodel.TaskStatusTodo, // 待处理
  688. s.TaskDao.Columns.Priority: entity.Priority,
  689. s.TaskDao.Columns.TaskParentId: req.Id,
  690. }
  691. if entity.TaskParentId > 0 {
  692. var parentTask opsdevmodel.OpsEventTask
  693. err := s.TaskDao.TX(tx).FieldsEx(s.TaskDao.Columns.DeletedTime).
  694. WherePri(s.TaskDao.Columns.Id, entity.TaskParentId).
  695. Scan(&parentTask)
  696. if err == nil && parentTask.Id > 0 {
  697. testTaskData[s.TaskDao.Columns.OpsUserId] = parentTask.OpsUserId
  698. testTaskData[s.TaskDao.Columns.OpsUserName] = parentTask.OpsUserName
  699. }
  700. }
  701. service.SetCreatedInfo(testTaskData, s.GetCxtUserId(), s.GetCxtUserName())
  702. testResult, err := s.TaskDao.TX(tx).Data(testTaskData).Insert()
  703. if err != nil {
  704. g.Log().Error(err)
  705. if isDuplicateEntryError(err) {
  706. return err
  707. }
  708. return myerrors.DbError("自动创建功能测试任务失败")
  709. }
  710. testTaskId, _ := testResult.LastInsertId()
  711. // 创建功能测试任务的过程记录
  712. testRecordData := g.Map{
  713. s.RecordDao.Columns.TaskId: testTaskId,
  714. s.RecordDao.Columns.HandleUserId: s.GetCxtUserId(),
  715. s.RecordDao.Columns.HandleUserName: s.GetCxtUserName(),
  716. s.RecordDao.Columns.HandleContent: "创建功能测试任务<br/>说明: 由BUG任务自动创建",
  717. }
  718. service.SetCreatedInfo(testRecordData, s.GetCxtUserId(), s.GetCxtUserName())
  719. _, err = s.RecordDao.TX(tx).Data(testRecordData).Insert()
  720. if err != nil {
  721. g.Log().Error(err)
  722. return myerrors.DbError("新增功能测试任务过程记录失败")
  723. }
  724. }
  725. // 5.3 系统发版完成时,保存关联的研发任务到ops_event_task_release表
  726. if entity.TaskType == opsdevmodel.TaskTypeSystemReleaseEvt && req.IsReleaseComplete && len(req.DevTaskIds) > 0 {
  727. for _, devTaskId := range req.DevTaskIds {
  728. releaseData := g.Map{
  729. s.ReleaseDao.Columns.ReleaseTaskId: req.Id,
  730. s.ReleaseDao.Columns.DevTaskId: devTaskId,
  731. s.ReleaseDao.Columns.ProjectId: entity.ProjectId,
  732. }
  733. service.SetCreatedInfo(releaseData, s.GetCxtUserId(), s.GetCxtUserName())
  734. _, err := s.ReleaseDao.TX(tx).Data(releaseData).Insert()
  735. if err != nil {
  736. g.Log().Error(err)
  737. return myerrors.DbError("保存发版关联任务失败")
  738. }
  739. }
  740. }
  741. // 6. 任务完成时自动更新关联的交付事件状态为完成并生成过程记录
  742. if entity.EventId > 0 && entity.EventType == opsdevmodel.EventTypeDelivery {
  743. // 触发事件自动完成的任务类型:10(需求评审)、30(功能测试-通过)、38(系统发版)、40(系统发版/硬件发货)、41(硬件安装)
  744. if s.shouldAutoCompleteEvent(entity.TaskType, req.TestResult) {
  745. eventData := g.Map{
  746. s.EventDao.Columns.DeliveryEventStatus: opsdevmodel.DeliveryEventStatusClosed,
  747. s.EventDao.Columns.CompleteTime: gtime.Now(),
  748. }
  749. service.SetUpdatedInfo(eventData, s.GetCxtUserId(), s.GetCxtUserName())
  750. _, err := s.EventDao.TX(tx).FieldsEx(service.UpdateFieldEx...).
  751. Data(eventData).
  752. WherePri(s.EventDao.Columns.Id, entity.EventId).
  753. Update()
  754. if err != nil {
  755. g.Log().Error(err)
  756. return myerrors.DbError("自动更新事件状态失败")
  757. }
  758. eventRecordData := g.Map{
  759. s.EventRecordDao.Columns.DeliveryEventId: entity.EventId,
  760. s.EventRecordDao.Columns.HandleUserId: s.GetCxtUserId(),
  761. s.EventRecordDao.Columns.HandleUserName: s.GetCxtUserName(),
  762. s.EventRecordDao.Columns.HandleContent: "研发任务已完成,自动关闭事件<br/>任务编号: " + entity.TaskNo + "<br/>任务标题: " + entity.TaskTitle,
  763. }
  764. service.SetCreatedInfo(eventRecordData, s.GetCxtUserId(), s.GetCxtUserName())
  765. _, err = s.EventRecordDao.TX(tx).Data(eventRecordData).Insert()
  766. if err != nil {
  767. g.Log().Error(err)
  768. return myerrors.DbError("新增事件过程记录失败")
  769. }
  770. }
  771. }
  772. return nil
  773. })
  774. }
  775. // Pause 暂停任务
  776. func (s *OpsEventTaskService) Pause(req *opsdevmodel.OpsEventTaskPauseReq) error {
  777. // 校验数据是否存在
  778. var entity opsdevmodel.OpsEventTask
  779. err := s.TaskDao.FieldsEx(s.TaskDao.Columns.DeletedTime).WherePri(s.TaskDao.Columns.Id, req.Id).Scan(&entity)
  780. if err != nil {
  781. g.Log().Error(err)
  782. return myerrors.DbError("查询任务数据失败")
  783. }
  784. if entity.Id <= 0 {
  785. return myerrors.TipsError("任务数据不存在")
  786. }
  787. // 只有处理中(20)状态可以暂停
  788. if !s.canPause(entity.TaskStatus) {
  789. return myerrors.TipsError("只有处理中的任务可以暂停")
  790. }
  791. // 构造更新数据
  792. data := g.Map{
  793. s.TaskDao.Columns.TaskStatus: opsdevmodel.TaskStatusPaused, // 暂停
  794. }
  795. // 补齐审计字段
  796. service.SetUpdatedInfo(data, s.GetCxtUserId(), s.GetCxtUserName())
  797. // 使用事务
  798. return s.TaskDao.Transaction(context.TODO(), func(ctx context.Context, tx *gdb.TX) error {
  799. // 1. 更新任务状态
  800. _, err := s.TaskDao.TX(tx).FieldsEx(service.UpdateFieldEx...).Data(data).WherePri(s.TaskDao.Columns.Id, req.Id).Update()
  801. if err != nil {
  802. g.Log().Error(err)
  803. return myerrors.DbError("暂停任务失败")
  804. }
  805. // 2. 创建过程记录
  806. recordData := g.Map{
  807. s.RecordDao.Columns.TaskId: req.Id,
  808. s.RecordDao.Columns.HandleUserId: s.GetCxtUserId(),
  809. s.RecordDao.Columns.HandleUserName: s.GetCxtUserName(),
  810. s.RecordDao.Columns.HandleContent: "暂停任务<br/>暂停原因: " + req.Remark,
  811. }
  812. service.SetCreatedInfo(recordData, s.GetCxtUserId(), s.GetCxtUserName())
  813. _, err = s.RecordDao.TX(tx).Data(recordData).Insert()
  814. if err != nil {
  815. g.Log().Error(err)
  816. return myerrors.DbError("新增过程记录失败")
  817. }
  818. return nil
  819. })
  820. }
  821. // Block 阻塞任务
  822. func (s *OpsEventTaskService) Block(req *opsdevmodel.OpsEventTaskBlockReq) error {
  823. // 校验数据是否存在
  824. var entity opsdevmodel.OpsEventTask
  825. err := s.TaskDao.FieldsEx(s.TaskDao.Columns.DeletedTime).WherePri(s.TaskDao.Columns.Id, req.Id).Scan(&entity)
  826. if err != nil {
  827. g.Log().Error(err)
  828. return myerrors.DbError("查询任务数据失败")
  829. }
  830. if entity.Id <= 0 {
  831. return myerrors.TipsError("任务数据不存在")
  832. }
  833. // 只有处理中(20)或暂停(25)状态可以阻塞
  834. if !s.canBlock(entity.TaskStatus) {
  835. return myerrors.TipsError("只有处理中或暂停状态的任务可以阻塞")
  836. }
  837. // 构造更新数据
  838. data := g.Map{
  839. s.TaskDao.Columns.TaskStatus: opsdevmodel.TaskStatusBlocked, // 阻塞
  840. }
  841. // 补齐审计字段
  842. service.SetUpdatedInfo(data, s.GetCxtUserId(), s.GetCxtUserName())
  843. // 使用事务
  844. return s.TaskDao.Transaction(context.TODO(), func(ctx context.Context, tx *gdb.TX) error {
  845. // 1. 更新任务状态
  846. _, err := s.TaskDao.TX(tx).FieldsEx(service.UpdateFieldEx...).Data(data).WherePri(s.TaskDao.Columns.Id, req.Id).Update()
  847. if err != nil {
  848. g.Log().Error(err)
  849. return myerrors.DbError("阻塞任务失败")
  850. }
  851. // 2. 创建过程记录
  852. recordData := g.Map{
  853. s.RecordDao.Columns.TaskId: req.Id,
  854. s.RecordDao.Columns.HandleUserId: s.GetCxtUserId(),
  855. s.RecordDao.Columns.HandleUserName: s.GetCxtUserName(),
  856. s.RecordDao.Columns.HandleContent: "阻塞任务<br/>阻塞原因: " + req.Remark,
  857. }
  858. service.SetCreatedInfo(recordData, s.GetCxtUserId(), s.GetCxtUserName())
  859. _, err = s.RecordDao.TX(tx).Data(recordData).Insert()
  860. if err != nil {
  861. g.Log().Error(err)
  862. return myerrors.DbError("新增过程记录失败")
  863. }
  864. return nil
  865. })
  866. }
  867. // Cancel 作废任务
  868. func (s *OpsEventTaskService) Cancel(req *opsdevmodel.OpsEventTaskCancelReq) error {
  869. // 校验数据是否存在
  870. var entity opsdevmodel.OpsEventTask
  871. err := s.TaskDao.FieldsEx(s.TaskDao.Columns.DeletedTime).WherePri(s.TaskDao.Columns.Id, req.Id).Scan(&entity)
  872. if err != nil {
  873. g.Log().Error(err)
  874. return myerrors.DbError("查询任务数据失败")
  875. }
  876. if entity.Id <= 0 {
  877. return myerrors.TipsError("任务数据不存在")
  878. }
  879. // 已完成(30)的任务不能作废
  880. if !s.canCancel(entity.TaskStatus) {
  881. return myerrors.TipsError("已完成的任务不能作废")
  882. }
  883. // 构造更新数据
  884. data := g.Map{
  885. s.TaskDao.Columns.TaskStatus: opsdevmodel.TaskStatusCancelled, // 作废
  886. }
  887. // 补齐审计字段
  888. service.SetUpdatedInfo(data, s.GetCxtUserId(), s.GetCxtUserName())
  889. // 使用事务
  890. return s.TaskDao.Transaction(context.TODO(), func(ctx context.Context, tx *gdb.TX) error {
  891. // 1. 更新任务状态
  892. _, err := s.TaskDao.TX(tx).FieldsEx(service.UpdateFieldEx...).Data(data).WherePri(s.TaskDao.Columns.Id, req.Id).Update()
  893. if err != nil {
  894. g.Log().Error(err)
  895. return myerrors.DbError("作废任务失败")
  896. }
  897. // 2. 创建过程记录
  898. recordData := g.Map{
  899. s.RecordDao.Columns.TaskId: req.Id,
  900. s.RecordDao.Columns.HandleUserId: s.GetCxtUserId(),
  901. s.RecordDao.Columns.HandleUserName: s.GetCxtUserName(),
  902. s.RecordDao.Columns.HandleContent: "作废任务<br/>作废原因: " + req.Remark,
  903. }
  904. service.SetCreatedInfo(recordData, s.GetCxtUserId(), s.GetCxtUserName())
  905. _, err = s.RecordDao.TX(tx).Data(recordData).Insert()
  906. if err != nil {
  907. g.Log().Error(err)
  908. return myerrors.DbError("新增过程记录失败")
  909. }
  910. return nil
  911. })
  912. }
  913. // GetRecords 获取任务过程记录列表(包含附件)
  914. func (s *OpsEventTaskService) GetRecords(req *opsdevmodel.OpsEventTaskRecordSearchReq) ([]*opsdevmodel.OpsEventTaskRecordWithAttachments, error) {
  915. var records []*opsdevmodel.OpsEventTaskRecord
  916. err := s.RecordDao.Where(s.RecordDao.Columns.TaskId, req.TaskId).
  917. Order(s.RecordDao.Columns.CreatedTime + " desc").
  918. Scan(&records)
  919. if err != nil {
  920. g.Log().Error(err)
  921. return nil, myerrors.DbError("查询过程记录失败")
  922. }
  923. result := make([]*opsdevmodel.OpsEventTaskRecordWithAttachments, 0, len(records))
  924. for _, record := range records {
  925. recordRsp := &opsdevmodel.OpsEventTaskRecordWithAttachments{
  926. OpsEventTaskRecord: *record,
  927. Attachments: []*opsdevmodel.OpsEventTaskAttachment{},
  928. }
  929. // 查询该记录关联的附件
  930. if record.Id > 0 {
  931. var attachments []*opsdevmodel.OpsEventTaskAttachment
  932. err := s.AttachmentDao.Where(s.AttachmentDao.Columns.TaskRecordId, record.Id).
  933. Scan(&attachments)
  934. if err != nil {
  935. g.Log().Error(err)
  936. } else {
  937. recordRsp.Attachments = attachments
  938. }
  939. }
  940. result = append(result, recordRsp)
  941. }
  942. return result, nil
  943. }
  944. // generateTaskNo 生成任务编号
  945. func (s *OpsEventTaskService) generateTaskNo() string {
  946. // 格式: TSK + 年月日 + 4位序列号
  947. now := gtime.Now()
  948. prefix := "TSK" + now.Format("Ymd")
  949. // 使用数据库序列生成唯一序号(按天重置)
  950. seqVal, err := s.TaskDao.DB.GetValue("SELECT next_day_reset_val('task_no_seq')")
  951. if err != nil {
  952. // 如果序列不存在或出错,使用备用方案:查询当天最大序号+1
  953. var maxNoResult struct {
  954. TaskNo string
  955. }
  956. err = s.TaskDao.Where(s.TaskDao.Columns.TaskNo+" like ?", prefix+"%").Order(s.TaskDao.Columns.TaskNo + " desc").Fields(s.TaskDao.Columns.TaskNo).Scan(&maxNoResult)
  957. if err != nil || maxNoResult.TaskNo == "" {
  958. return prefix + "0001"
  959. }
  960. maxNoStr := maxNoResult.TaskNo
  961. if len(maxNoStr) >= len(prefix)+4 {
  962. seq := maxNoStr[len(prefix):]
  963. seqNum := gconv.Int(seq)
  964. seqNum++
  965. return prefix + fmt.Sprintf("%04d", seqNum)
  966. }
  967. return prefix + "0001"
  968. }
  969. return prefix + fmt.Sprintf("%04d", seqVal.Int())
  970. }
  971. // 状态校验辅助方法
  972. func (s *OpsEventTaskService) canEdit(status string) bool {
  973. return status != opsdevmodel.TaskStatusCompleted && status != opsdevmodel.TaskStatusCancelled
  974. }
  975. func (s *OpsEventTaskService) canSchedule(status string) bool {
  976. return status == opsdevmodel.TaskStatusTodo
  977. }
  978. func (s *OpsEventTaskService) canStart(status string) bool {
  979. return status == opsdevmodel.TaskStatusTodo || status == opsdevmodel.TaskStatusPaused || status == opsdevmodel.TaskStatusBlocked
  980. }
  981. func (s *OpsEventTaskService) canComplete(status string) bool {
  982. return status == opsdevmodel.TaskStatusProcessing
  983. }
  984. func (s *OpsEventTaskService) canPause(status string) bool {
  985. return status == opsdevmodel.TaskStatusProcessing
  986. }
  987. func (s *OpsEventTaskService) canBlock(status string) bool {
  988. return status == opsdevmodel.TaskStatusProcessing || status == opsdevmodel.TaskStatusPaused
  989. }
  990. // shouldAutoCompleteEvent 判断当前任务完成时是否应自动完成关联的交付事件
  991. func (s *OpsEventTaskService) shouldAutoCompleteEvent(taskType, testResult string) bool {
  992. switch taskType {
  993. case "10": // 需求评审 — 完成即触发
  994. return true
  995. case "30": // 功能测试 — 仅通过时触发
  996. return testResult == "pass"
  997. case "38": // 系统发版(事件关联) — 完成即触发
  998. return true
  999. default:
  1000. return false
  1001. }
  1002. }
  1003. func (s *OpsEventTaskService) canCancel(status string) bool {
  1004. return status != opsdevmodel.TaskStatusCompleted
  1005. }
  1006. // getProjectName 根据项目ID获取项目名称
  1007. func (s *OpsEventTaskService) getProjectName(projectId int) string {
  1008. if projectId <= 0 {
  1009. return ""
  1010. }
  1011. var project opsdevmodel.OpsDeliveryProject
  1012. err := s.ProjectDao.FieldsEx(s.ProjectDao.Columns.DeletedTime).
  1013. WherePri(s.ProjectDao.Columns.Id, projectId).
  1014. Scan(&project)
  1015. if err != nil {
  1016. g.Log().Error(err)
  1017. return ""
  1018. }
  1019. return project.ProjectName
  1020. }
  1021. // GetAttachments 获取任务附件列表
  1022. func (s *OpsEventTaskService) GetAttachments(taskId int) ([]*opsdevmodel.OpsEventTaskAttachment, error) {
  1023. var attachments []*opsdevmodel.OpsEventTaskAttachment
  1024. err := s.AttachmentDao.Where(s.AttachmentDao.Columns.TaskId, taskId).
  1025. Order(s.AttachmentDao.Columns.CreatedTime + " desc").
  1026. Scan(&attachments)
  1027. if err != nil {
  1028. g.Log().Error(err)
  1029. return nil, myerrors.DbError("查询附件失败")
  1030. }
  1031. return attachments, nil
  1032. }
  1033. // GetTaskReleaseList 根据发版任务ID查询关联的开发任务列表
  1034. func (s *OpsEventTaskService) GetTaskReleaseList(req *opsdevmodel.OpsEventTaskReleaseListReq) ([]*opsdevmodel.OpsEventTaskReleaseRsp, error) {
  1035. // 1. 查询关联表获取关联的任务ID列表
  1036. var releaseList []*opsdevmodel.OpsEventTaskRelease
  1037. err := s.ReleaseDao.Where(s.ReleaseDao.Columns.ReleaseTaskId, req.ReleaseTaskId).
  1038. Scan(&releaseList)
  1039. if err != nil {
  1040. g.Log().Error(err)
  1041. return nil, myerrors.DbError("查询发布版本关联记录失败")
  1042. }
  1043. if len(releaseList) == 0 {
  1044. return []*opsdevmodel.OpsEventTaskReleaseRsp{}, nil
  1045. }
  1046. // 2. 提取开发任务ID列表
  1047. devTaskIds := make([]int, 0, len(releaseList))
  1048. for _, r := range releaseList {
  1049. devTaskIds = append(devTaskIds, r.DevTaskId)
  1050. }
  1051. // 3. 查询任务详情
  1052. var taskList []*opsdevmodel.OpsEventTask
  1053. err = s.TaskDao.WhereIn(s.TaskDao.Columns.Id, devTaskIds).
  1054. Scan(&taskList)
  1055. if err != nil {
  1056. g.Log().Error(err)
  1057. return nil, myerrors.DbError("查询关联任务详情失败")
  1058. }
  1059. // 4. 构建响应数据
  1060. result := make([]*opsdevmodel.OpsEventTaskReleaseRsp, 0, len(taskList))
  1061. for _, task := range taskList {
  1062. result = append(result, &opsdevmodel.OpsEventTaskReleaseRsp{
  1063. Id: task.Id,
  1064. DevTaskId: task.Id,
  1065. TaskNo: task.TaskNo,
  1066. TaskTitle: task.TaskTitle,
  1067. TaskType: task.TaskType,
  1068. TaskStatus: task.TaskStatus,
  1069. OpsUserName: task.OpsUserName,
  1070. ProjectId: task.ProjectId,
  1071. CreatedTime: task.CreatedTime.Format("Y-m-d H:i:s"),
  1072. })
  1073. }
  1074. return result, nil
  1075. }
  1076. // AddRecord 添加任务过程记录
  1077. func (s *OpsEventTaskService) AddRecord(req *opsdevmodel.OpsEventTaskRecordAddReq) error {
  1078. // 构造记录数据
  1079. recordData := g.Map{
  1080. s.RecordDao.Columns.TaskId: req.TaskId,
  1081. s.RecordDao.Columns.HandleUserId: s.GetCxtUserId(),
  1082. s.RecordDao.Columns.HandleUserName: s.GetCxtUserName(),
  1083. s.RecordDao.Columns.HandleContent: req.HandleContent,
  1084. }
  1085. // 补齐审计字段
  1086. service.SetCreatedInfo(recordData, s.GetCxtUserId(), s.GetCxtUserName())
  1087. // 使用事务控制
  1088. return s.RecordDao.Transaction(context.TODO(), func(ctx context.Context, tx *gdb.TX) error {
  1089. // 1. 创建过程记录
  1090. result, err := s.RecordDao.TX(tx).Data(recordData).Insert()
  1091. if err != nil {
  1092. g.Log().Error(err)
  1093. return myerrors.DbError("添加过程记录失败")
  1094. }
  1095. // 获取新创建的记录ID
  1096. recordId, err := result.LastInsertId()
  1097. if err != nil {
  1098. g.Log().Error(err)
  1099. return myerrors.DbError("获取记录ID失败")
  1100. }
  1101. // 2. 如果有附件,保存附件
  1102. if len(req.Attachments) > 0 {
  1103. for _, att := range req.Attachments {
  1104. attData := g.Map{
  1105. s.AttachmentDao.Columns.TaskId: req.TaskId,
  1106. s.AttachmentDao.Columns.TaskRecordId: int(recordId),
  1107. s.AttachmentDao.Columns.FileName: att.FileName,
  1108. s.AttachmentDao.Columns.FileUrl: att.FileUrl,
  1109. s.AttachmentDao.Columns.FileType: att.FileType,
  1110. }
  1111. service.SetCreatedInfo(attData, s.GetCxtUserId(), s.GetCxtUserName())
  1112. _, err = s.AttachmentDao.TX(tx).Data(attData).Insert()
  1113. if err != nil {
  1114. g.Log().Error(err)
  1115. return myerrors.DbError("保存附件失败")
  1116. }
  1117. }
  1118. }
  1119. return nil
  1120. })
  1121. }
  1122. // AddWorkHour 添加工时登记(累加actual_work_hour并记录过程)
  1123. func (s *OpsEventTaskService) AddWorkHour(req *opsdevmodel.OpsEventTaskWorkHourAddReq) error {
  1124. var entity opsdevmodel.OpsEventTask
  1125. err := s.TaskDao.FieldsEx(s.TaskDao.Columns.DeletedTime).WherePri(s.TaskDao.Columns.Id, req.TaskId).Scan(&entity)
  1126. if err != nil {
  1127. g.Log().Error(err)
  1128. return myerrors.DbError("查询任务数据失败")
  1129. }
  1130. if entity.Id <= 0 {
  1131. return myerrors.TipsError("任务数据不存在")
  1132. }
  1133. if !s.canAddWorkHour(entity.TaskStatus) {
  1134. return myerrors.TipsError("只有处理中的任务可以登记工时")
  1135. }
  1136. return s.TaskDao.Transaction(context.TODO(), func(ctx context.Context, tx *gdb.TX) error {
  1137. workHourData := g.Map{
  1138. s.WorkHourDao.Columns.TaskId: req.TaskId,
  1139. s.WorkHourDao.Columns.OpsUserId: s.GetCxtUserId(),
  1140. s.WorkHourDao.Columns.OpsUserName: s.GetCxtUserName(),
  1141. s.WorkHourDao.Columns.ActualWorkDate: req.WorkDate,
  1142. s.WorkHourDao.Columns.ActualWorkHour: req.ActualHour,
  1143. s.WorkHourDao.Columns.Remark: req.Remark,
  1144. }
  1145. service.SetCreatedInfo(workHourData, s.GetCxtUserId(), s.GetCxtUserName())
  1146. _, err := s.WorkHourDao.TX(tx).Data(workHourData).Insert()
  1147. if err != nil {
  1148. g.Log().Error(err)
  1149. return myerrors.DbError("工时登记失败")
  1150. }
  1151. newActualWorkHour := entity.ActualWorkHour + req.ActualHour
  1152. updateData := g.Map{
  1153. s.TaskDao.Columns.ActualWorkHour: newActualWorkHour,
  1154. }
  1155. service.SetUpdatedInfo(updateData, s.GetCxtUserId(), s.GetCxtUserName())
  1156. _, err = s.TaskDao.TX(tx).FieldsEx(service.UpdateFieldEx...).Data(updateData).WherePri(s.TaskDao.Columns.Id, req.TaskId).Update()
  1157. if err != nil {
  1158. g.Log().Error(err)
  1159. return myerrors.DbError("更新实际工时失败")
  1160. }
  1161. handleContent := fmt.Sprintf("工时登记<br/>工作日期: %s<br/>实际工时: %s小时<br/>工作进展: %s",
  1162. req.WorkDate, gconv.String(req.ActualHour), req.Remark)
  1163. recordData := g.Map{
  1164. s.RecordDao.Columns.TaskId: req.TaskId,
  1165. s.RecordDao.Columns.HandleUserId: s.GetCxtUserId(),
  1166. s.RecordDao.Columns.HandleUserName: s.GetCxtUserName(),
  1167. s.RecordDao.Columns.HandleContent: handleContent,
  1168. }
  1169. service.SetCreatedInfo(recordData, s.GetCxtUserId(), s.GetCxtUserName())
  1170. _, err = s.RecordDao.TX(tx).Data(recordData).Insert()
  1171. if err != nil {
  1172. g.Log().Error(err)
  1173. return myerrors.DbError("新增过程记录失败")
  1174. }
  1175. return nil
  1176. })
  1177. }
  1178. // GetWorkHourList 获取工时登记列表
  1179. func (s *OpsEventTaskService) GetWorkHourList(req *opsdevmodel.OpsEventTaskWorkHourListReq) ([]*opsdevmodel.OpsEventTaskWorkHourRsp, error) {
  1180. var entities []*opsdevmodel.OpsEventTaskWorkHour
  1181. err := s.WorkHourDao.FieldsEx(s.WorkHourDao.Columns.DeletedTime).
  1182. Where(s.WorkHourDao.Columns.TaskId, req.TaskId).
  1183. Order(s.WorkHourDao.Columns.CreatedTime + " desc").
  1184. Scan(&entities)
  1185. if err != nil {
  1186. g.Log().Error(err)
  1187. return nil, myerrors.DbError("查询工时登记列表失败")
  1188. }
  1189. list := make([]*opsdevmodel.OpsEventTaskWorkHourRsp, 0, len(entities))
  1190. for _, entity := range entities {
  1191. workDate := ""
  1192. if entity.ActualWorkDate != nil {
  1193. workDate = entity.ActualWorkDate.Format("Y-m-d")
  1194. }
  1195. createdTime := ""
  1196. if entity.CreatedTime != nil {
  1197. createdTime = entity.CreatedTime.Format("Y-m-d H:i:s")
  1198. }
  1199. list = append(list, &opsdevmodel.OpsEventTaskWorkHourRsp{
  1200. Id: entity.Id,
  1201. TaskId: entity.TaskId,
  1202. WorkDate: workDate,
  1203. ActualHour: entity.ActualWorkHour,
  1204. Remark: entity.Remark,
  1205. CreatedName: entity.CreatedName,
  1206. CreatedTime: createdTime,
  1207. })
  1208. }
  1209. return list, nil
  1210. }
  1211. // GetDashboardData 获取工作台看板数据(周视图)
  1212. func (s *OpsEventTaskService) GetDashboardData(startDate, endDate string) (*opsdevmodel.OpsEventTaskWorkHourDashboardRsp, error) {
  1213. userId := s.GetCxtUserId()
  1214. db := g.DB(s.Tenant)
  1215. type taskRow struct {
  1216. Id int `json:"id"`
  1217. TaskNo string `json:"taskNo"`
  1218. TaskTitle string `json:"taskTitle"`
  1219. TaskType string `json:"taskType"`
  1220. TaskStatus string `json:"taskStatus"`
  1221. Priority string `json:"priority"`
  1222. ProjectName string `json:"projectName"`
  1223. ActualWorkHour float64 `json:"actualWorkHour"`
  1224. EstimateWorkHour float64 `json:"estimateWorkHour"`
  1225. PlanStartDate string `json:"planStartDate"`
  1226. PlanEndTime string `json:"planEndTime"`
  1227. }
  1228. taskSQL := `
  1229. SELECT id, task_no, task_title, task_type, task_status,
  1230. priority, project_name, actual_work_hour, estimate_work_hour,
  1231. DATE(plan_start_time) AS plan_start_date,
  1232. IFNULL(DATE(plan_end_time), '') AS plan_end_time
  1233. FROM ops_event_task
  1234. WHERE ops_user_id = ?
  1235. AND plan_start_time >= ? AND plan_start_time <= ?
  1236. AND deleted_time IS NULL
  1237. ORDER BY plan_start_time, priority
  1238. `
  1239. var taskRows []taskRow
  1240. err := db.GetScan(&taskRows, taskSQL, userId, startDate, endDate+" 23:59:59")
  1241. if err != nil {
  1242. g.Log().Error(err)
  1243. return nil, myerrors.DbError("查询任务数据失败")
  1244. }
  1245. type hourRow struct {
  1246. WorkDate string `json:"workDate"`
  1247. TotalHour float64 `json:"totalHour"`
  1248. }
  1249. hourSQL := `
  1250. SELECT DATE(actual_work_date) AS work_date, SUM(actual_work_hour) AS total_hour
  1251. FROM ops_event_task_work_hour
  1252. WHERE ops_user_id = ?
  1253. AND actual_work_date >= ? AND actual_work_date <= ?
  1254. AND deleted_time IS NULL
  1255. GROUP BY DATE(actual_work_date)
  1256. `
  1257. var hourRows []hourRow
  1258. err = db.GetScan(&hourRows, hourSQL, userId, startDate, endDate+" 23:59:59")
  1259. if err != nil {
  1260. g.Log().Error(err)
  1261. return nil, myerrors.DbError("查询每日工时失败")
  1262. }
  1263. hourMap := make(map[string]float64, 7)
  1264. for i := range hourRows {
  1265. hourMap[hourRows[i].WorkDate] = hourRows[i].TotalHour
  1266. }
  1267. taskMap := make(map[string][]*opsdevmodel.DashboardTaskRsp, 7)
  1268. for i := range taskRows {
  1269. row := &taskRows[i]
  1270. date := row.PlanStartDate
  1271. taskMap[date] = append(taskMap[date], &opsdevmodel.DashboardTaskRsp{
  1272. Id: row.Id,
  1273. TaskNo: row.TaskNo,
  1274. TaskTitle: row.TaskTitle,
  1275. TaskType: row.TaskType,
  1276. TaskStatus: row.TaskStatus,
  1277. Priority: row.Priority,
  1278. ProjectName: row.ProjectName,
  1279. ActualWorkHour: row.ActualWorkHour,
  1280. EstimateWorkHour: row.EstimateWorkHour,
  1281. PlanEndTime: row.PlanEndTime,
  1282. })
  1283. }
  1284. overdueCount, err := db.GetValue(
  1285. "SELECT COUNT(1) FROM ops_event_task WHERE plan_end_time < NOW() AND ops_user_id = ? AND task_status NOT IN (?,?) AND deleted_time IS NULL",
  1286. userId,
  1287. opsdevmodel.TaskStatusCompleted, opsdevmodel.TaskStatusCancelled,
  1288. )
  1289. if err != nil {
  1290. g.Log().Error(err)
  1291. return nil, myerrors.DbError("查询超期任务数失败")
  1292. }
  1293. var weekTotal float64
  1294. days := make([]*opsdevmodel.DashboardDayRsp, 0, 7)
  1295. for i := 0; i < 7; i++ {
  1296. date := startDate
  1297. if i > 0 {
  1298. d := gtime.New(startDate).AddDate(0, 0, i)
  1299. date = d.Format("Y-m-d")
  1300. }
  1301. targetHours := 8.0
  1302. if i >= 5 {
  1303. targetHours = 0
  1304. }
  1305. totalHours := hourMap[date]
  1306. weekTotal += totalHours
  1307. dayTasks := taskMap[date]
  1308. if dayTasks == nil {
  1309. dayTasks = []*opsdevmodel.DashboardTaskRsp{}
  1310. }
  1311. days = append(days, &opsdevmodel.DashboardDayRsp{
  1312. Date: date,
  1313. TotalHours: totalHours,
  1314. TargetHours: targetHours,
  1315. Tasks: dayTasks,
  1316. })
  1317. }
  1318. weekTargetHours := 40.0
  1319. return &opsdevmodel.OpsEventTaskWorkHourDashboardRsp{
  1320. WeekTargetHours: weekTargetHours,
  1321. WeekTotalHours: weekTotal,
  1322. OverdueCount: overdueCount.Int(),
  1323. Days: days,
  1324. }, nil
  1325. }
  1326. // canAddWorkHour 检查是否可以登记工时(仅处理中状态)
  1327. func (s *OpsEventTaskService) canAddWorkHour(status string) bool {
  1328. return status == opsdevmodel.TaskStatusProcessing
  1329. }
  1330. // GetScheduleStats 获取人员排期统计(按周、按天、按人汇总任务数和预估工时)
  1331. func (s *OpsEventTaskService) GetScheduleStats(req *opsdevmodel.OpsEventTaskScheduleStatReq) (*opsdevmodel.OpsEventTaskScheduleStatRsp, error) {
  1332. type dayRow struct {
  1333. OpsUserId int `json:"opsUserId"`
  1334. OpsUserName string `json:"opsUserName"`
  1335. PlanDate string `json:"planDate"`
  1336. TaskCount int `json:"taskCount"`
  1337. EstimateWorkHour float64 `json:"estimateWorkHour"`
  1338. }
  1339. sql := `
  1340. SELECT
  1341. ops_user_id,
  1342. ops_user_name,
  1343. DATE(plan_start_time) AS plan_date,
  1344. COUNT(*) AS task_count,
  1345. SUM(COALESCE(estimate_work_hour, 0)) AS estimate_work_hour
  1346. FROM ops_event_task
  1347. WHERE deleted_time IS NULL
  1348. AND task_status != ?
  1349. AND plan_start_time >= ?
  1350. AND plan_start_time < ?
  1351. `
  1352. args := []interface{}{opsdevmodel.TaskStatusCancelled, req.WeekStart, req.WeekEnd + " 23:59:59"}
  1353. if req.ProjectId > 0 {
  1354. sql += " AND project_id = ?"
  1355. args = append(args, req.ProjectId)
  1356. }
  1357. sql += `
  1358. GROUP BY ops_user_id, ops_user_name, DATE(plan_start_time)
  1359. ORDER BY ops_user_name, plan_date
  1360. `
  1361. var rows []dayRow
  1362. err := g.DB(s.Tenant).GetScan(&rows, sql, args...)
  1363. if err != nil {
  1364. g.Log().Error(err)
  1365. return nil, myerrors.DbError("查询排期统计数据失败")
  1366. }
  1367. // 按用户聚合,补全 7 天数据
  1368. userMap := make(map[int]*opsdevmodel.OpsEventTaskUserScheduleStat)
  1369. for _, row := range rows {
  1370. user, ok := userMap[row.OpsUserId]
  1371. if !ok {
  1372. user = &opsdevmodel.OpsEventTaskUserScheduleStat{
  1373. OpsUserId: row.OpsUserId,
  1374. OpsUserName: row.OpsUserName,
  1375. DayStats: make([]*opsdevmodel.OpsEventTaskUserDayStat, 7),
  1376. }
  1377. // 初始化 7 天空数据
  1378. weekStart := gtime.New(req.WeekStart)
  1379. for i := 0; i < 7; i++ {
  1380. day := weekStart.AddDate(0, 0, i)
  1381. user.DayStats[i] = &opsdevmodel.OpsEventTaskUserDayStat{
  1382. Date: day.Format("Y-m-d"),
  1383. TaskCount: 0,
  1384. EstimateWorkHour: 0,
  1385. }
  1386. }
  1387. userMap[row.OpsUserId] = user
  1388. }
  1389. // 查找对应 day index
  1390. for _, ds := range user.DayStats {
  1391. if ds.Date == row.PlanDate {
  1392. ds.TaskCount = row.TaskCount
  1393. ds.EstimateWorkHour = row.EstimateWorkHour
  1394. break
  1395. }
  1396. }
  1397. }
  1398. // 计算周合计
  1399. list := make([]*opsdevmodel.OpsEventTaskUserScheduleStat, 0, len(userMap))
  1400. for _, user := range userMap {
  1401. weekTotal := &opsdevmodel.OpsEventTaskUserDayStat{Date: "合计"}
  1402. for _, ds := range user.DayStats {
  1403. weekTotal.TaskCount += ds.TaskCount
  1404. weekTotal.EstimateWorkHour += ds.EstimateWorkHour
  1405. }
  1406. weekTotal.EstimateWorkHour = math.Round(weekTotal.EstimateWorkHour*10) / 10
  1407. user.WeekTotal = weekTotal
  1408. list = append(list, user)
  1409. }
  1410. return &opsdevmodel.OpsEventTaskScheduleStatRsp{List: list}, nil
  1411. }