ImageArrJob.php 2.4 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273
  1. <?php
  2. namespace app\job;
  3. use think\Db;
  4. use think\queue\Job;
  5. use think\Queue;
  6. class ImageArrJob
  7. {
  8. public function fire(Job $job, $data)
  9. {
  10. $task_id = $data['task_id'];
  11. $images = $data['data'];
  12. foreach ($images as $value) {
  13. $value['task_id'] = $task_id;
  14. // 1. 清理 sourceDir,去掉末尾重复的 Preview
  15. $sourceDir = rtrim($value["sourceDir"], '/\\');
  16. $dirParts = explode('/', str_replace('\\', '/', $sourceDir));
  17. if (end($dirParts) === 'Preview') {
  18. array_pop($dirParts); // 删除末尾 Preview
  19. }
  20. $cleanedSourceDir = implode('/', $dirParts);
  21. // 2. 获取纯文件名
  22. $fileName = basename($value['file_name']);
  23. // 3. 拼接最终路径
  24. $fullPath = $cleanedSourceDir . '/' . $fileName;
  25. // 4. 写入日志记录
  26. $log_id = Db::name('image_task_log')->insertGetId([
  27. 'task_id' => $task_id,
  28. 'file_name' => $fullPath,
  29. 'type' => $value['type'] ?? '',
  30. 'status' => 0,
  31. 'log' => '队列中',
  32. 'create_time' => date('Y-m-d H:i:s')
  33. ]);
  34. $value['log_id'] = $log_id;
  35. // 5. 推送任务到对应队列
  36. switch (trim($value['type'])) {
  37. case '图生文':
  38. Queue::push('app\job\ImageJob', $value, 'imgtotxt');
  39. break;
  40. case '文生文':
  41. Queue::push('app\job\TextToTextJob', $value, 'txttotxt');
  42. break;
  43. case '文生图':
  44. Queue::push('app\job\TextToImageJob', $value, 'txttoimg');
  45. break;
  46. default:
  47. // 类型未知时不推送,并可写入错误日志
  48. \think\Log::warning("未识别的任务类型:" . json_encode($value, JSON_UNESCAPED_UNICODE));
  49. break;
  50. }
  51. }
  52. // 6. 更新任务状态为已启动
  53. Db::name('queue_logs')->where('id', $task_id)->update(['status' => '已启动队列']);
  54. echo date('Y-m-d H:i:s') . " 队列已启动\n";
  55. // 7. 删除当前队列任务
  56. $job->delete();
  57. }
  58. public function failed($data)
  59. {
  60. \think\Log::error("ImageArrJob 任务失败:" . json_encode($data, JSON_UNESCAPED_UNICODE));
  61. }
  62. }