sop_task_stop_logic.go 3.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106
  1. package sop_task
  2. import (
  3. "context"
  4. "github.com/suyuan32/simple-admin-common/msg/errormsg"
  5. "wechat-api/ent/messagerecords"
  6. "wechat-api/ent/sopstage"
  7. "wechat-api/ent/soptask"
  8. "wechat-api/internal/utils/dberrorhandler"
  9. "wechat-api/internal/svc"
  10. "wechat-api/internal/types"
  11. "github.com/zeromicro/go-zero/core/logx"
  12. )
  13. type SopTaskStopLogic struct {
  14. logx.Logger
  15. ctx context.Context
  16. svcCtx *svc.ServiceContext
  17. }
  18. func NewSopTaskStopLogic(ctx context.Context, svcCtx *svc.ServiceContext) *SopTaskStopLogic {
  19. return &SopTaskStopLogic{
  20. Logger: logx.WithContext(ctx),
  21. ctx: ctx,
  22. svcCtx: svcCtx}
  23. }
  24. func (l *SopTaskStopLogic) SopTaskStop(req *types.IDReq) (resp *types.BaseMsgResp, err error) {
  25. organizationId := l.ctx.Value("organizationId").(uint64)
  26. isAdmin := l.ctx.Value("isAdmin").(bool)
  27. // 开始事务
  28. tx, err := l.svcCtx.DB.Tx(context.Background())
  29. if err != nil {
  30. return nil, dberrorhandler.DefaultEntError(l.Logger, err, req)
  31. }
  32. // 根据 id 查询 task 信息,如果 Status 的值不为 1,则不允许修改
  33. task, err := tx.SopTask.Query().
  34. Where(
  35. soptask.IDEQ(req.Id), // Filter by ID
  36. soptask.OrganizationID(organizationId), // Additional filter by organizationId
  37. ).
  38. WithTaskStages().
  39. Only(l.ctx)
  40. if err != nil {
  41. return nil, dberrorhandler.DefaultEntError(l.Logger, err, req)
  42. }
  43. if task.Status != 3 {
  44. return nil, dberrorhandler.DefaultEntError(l.Logger, err, req)
  45. }
  46. if isAdmin || organizationId == 12 {
  47. err = tx.SopTask.UpdateOneID(req.Id).
  48. SetStatus(1).
  49. Exec(l.ctx)
  50. sourceInfo, err := l.svcCtx.DB.SopStage.Query().
  51. Where(sopstage.TaskID(req.Id)).
  52. WithStageNodes().
  53. All(l.ctx)
  54. if err != nil {
  55. return nil, dberrorhandler.DefaultEntError(l.Logger, err, req)
  56. }
  57. for _, stage := range sourceInfo {
  58. _, err := tx.MessageRecords.Delete().Where(messagerecords.SourceIDEQ(stage.ID)).Exec(l.ctx)
  59. if err != nil {
  60. return nil, dberrorhandler.DefaultEntError(l.Logger, err, req)
  61. }
  62. for _, node := range stage.Edges.StageNodes {
  63. _, err := tx.MessageRecords.Delete().Where(messagerecords.SourceIDEQ(node.ID)).Exec(l.ctx)
  64. if err != nil {
  65. return nil, dberrorhandler.DefaultEntError(l.Logger, err, req)
  66. }
  67. }
  68. }
  69. } else {
  70. err = tx.SopTask.UpdateOneID(req.Id).
  71. SetStatus(2).
  72. Exec(l.ctx)
  73. }
  74. if err != nil {
  75. _ = tx.Rollback()
  76. return nil, dberrorhandler.DefaultEntError(l.Logger, err, req)
  77. }
  78. // 查询task的所有sop_stages
  79. for _, stage := range task.Edges.TaskStages {
  80. // 更新最旧的 p.Number 条 status 值为 1 或 4 的 message_records 记录的 status 为 2
  81. messages, _ := tx.MessageRecords.Query().
  82. Where(messagerecords.StatusIn(1, 2, 4), messagerecords.SourceType(3), messagerecords.SourceID(stage.ID)).
  83. All(l.ctx)
  84. for _, m := range messages {
  85. _, err := tx.MessageRecords.UpdateOneID(m.ID).SetStatus(5).Save(l.ctx)
  86. if err != nil {
  87. _ = tx.Rollback()
  88. }
  89. }
  90. }
  91. err = tx.Commit()
  92. if err != nil {
  93. return nil, dberrorhandler.DefaultEntError(l.Logger, err, req)
  94. }
  95. return &types.BaseMsgResp{Msg: errormsg.UpdateSuccess}, nil
  96. }