ProcessGptImage2GenerationJob.php 4.9 KB

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