ProcessVolcImageGenerationJob.php 5.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156
  1. <?php
  2. namespace App\Jobs;
  3. use App\Models\MpGeneratePicTask;
  4. use App\Services\AIGeneration\AIImageGenerationService;
  5. use Illuminate\Bus\Queueable;
  6. use Illuminate\Contracts\Queue\ShouldQueue;
  7. use Illuminate\Foundation\Bus\Dispatchable;
  8. use Illuminate\Queue\InteractsWithQueue;
  9. use Illuminate\Queue\SerializesModels;
  10. class ProcessVolcImageGenerationJob implements ShouldQueue
  11. {
  12. use Dispatchable, InteractsWithQueue, Queueable, SerializesModels;
  13. /**
  14. * 任务最大尝试次数
  15. *
  16. * @var int
  17. */
  18. public $tries = 3;
  19. /**
  20. * 任务超时时间(秒)
  21. *
  22. * 必须大于上游请求超时(AIImageGenerationService::REQUEST_TIMEOUT_SECONDS = 480s),
  23. * 让请求先超时、由业务层记录错误并重新排队;supervisor 的 stopwaitsecs 需不小于该值。
  24. *
  25. * @var int
  26. */
  27. public $timeout = 600;
  28. /**
  29. * 任务ID
  30. *
  31. * @var int
  32. */
  33. protected $taskId;
  34. /**
  35. * 创建一个新的任务实例
  36. *
  37. * @param int $taskId
  38. * @return void
  39. */
  40. public function __construct(int $taskId)
  41. {
  42. $this->taskId = $taskId;
  43. $this->onQueue('{GenerateVolcPics}');
  44. }
  45. /**
  46. * 执行任务
  47. *
  48. * @param AIImageGenerationService $service
  49. * @return void
  50. */
  51. public function handle(AIImageGenerationService $service)
  52. {
  53. // 本次 job 是否已正常结束;异常时不释放派发锁,交给队列重试或 failed() 释放
  54. $finished = false;
  55. try {
  56. // 查找任务
  57. $task = MpGeneratePicTask::find($this->taskId);
  58. if (!$task) {
  59. dLog('generate')->error('队列任务:找不到图片生成任务', ['task_id' => $this->taskId]);
  60. logDB('generate', 'error', '队列任务:找不到图片生成任务', ['task_id' => $this->taskId]);
  61. $finished = true;
  62. return;
  63. }
  64. // 检查任务状态,只处理pending状态的任务
  65. if ($task->status !== MpGeneratePicTask::STATUS_PENDING) {
  66. dLog('generate')->info('队列任务:任务状态不是pending,跳过处理', [
  67. 'task_id' => $this->taskId,
  68. 'status' => $task->status
  69. ]);
  70. $finished = true;
  71. return;
  72. }
  73. dLog('generate')->info('队列任务:开始处理火山API图片生成任务', ['task_id' => $this->taskId]);
  74. // 调用服务方法提交任务到火山API
  75. $service->submitTaskToVolcApi($task);
  76. dLog('generate')->info('队列任务:火山API图片生成任务处理完成', ['task_id' => $this->taskId]);
  77. $finished = true;
  78. } catch (\Exception $e) {
  79. dLog('generate')->error('队列任务:处理火山API图片生成任务失败', [
  80. 'task_id' => $this->taskId,
  81. 'error' => $e->getMessage()
  82. ]);
  83. logDB('generate', 'error', '队列任务:处理火山API图片生成任务失败', [
  84. 'task_id' => $this->taskId,
  85. 'error' => $e->getMessage(),
  86. 'trace' => $e->getTraceAsString()
  87. ]);
  88. // 重新抛出异常,让队列系统处理重试;此处不释放派发锁,
  89. // 避免重试期间任务被重复派发,最终失败时由 failed() 释放
  90. throw $e;
  91. } finally {
  92. if ($finished) {
  93. AIImageGenerationService::releaseTaskDispatchLock(
  94. AIImageGenerationService::DISPATCH_GROUP_VOLC,
  95. $this->taskId
  96. );
  97. }
  98. }
  99. }
  100. /**
  101. * 任务失败时的处理
  102. *
  103. * @param \Throwable $exception
  104. * @return void
  105. */
  106. public function failed(\Throwable $exception)
  107. {
  108. dLog('generate')->error('队列任务:火山API图片生成任务最终失败', [
  109. 'task_id' => $this->taskId,
  110. 'error' => $exception->getMessage()
  111. ]);
  112. logDB('generate', 'error', '队列任务:火山API图片生成任务最终失败', [
  113. 'task_id' => $this->taskId,
  114. 'error' => $exception->getMessage(),
  115. 'attempts' => $this->attempts()
  116. ]);
  117. // 更新任务状态为失败
  118. try {
  119. $task = MpGeneratePicTask::find($this->taskId);
  120. if ($task && $task->status === MpGeneratePicTask::STATUS_PENDING) {
  121. $task->updateStatus(MpGeneratePicTask::STATUS_FAILED, [
  122. 'error_message' => '队列任务处理失败: ' . $exception->getMessage()
  123. ]);
  124. }
  125. } catch (\Exception $e) {
  126. dLog('generate')->error('队列任务:更新任务状态失败', [
  127. 'task_id' => $this->taskId,
  128. 'error' => $e->getMessage()
  129. ]);
  130. }
  131. // 最终失败:释放派发锁,使任务在下一次扫描时可以重新派发
  132. AIImageGenerationService::releaseTaskDispatchLock(
  133. AIImageGenerationService::DISPATCH_GROUP_VOLC,
  134. $this->taskId
  135. );
  136. }
  137. }