ImageToSingleJob.php 6.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176
  1. <?php
  2. namespace app\job;
  3. use app\service\AIGatewayService;
  4. use think\Db;
  5. use think\queue\Job;
  6. use think\Queue;
  7. class ImageToSingleJob{
  8. public function fire(Job $job, $data)
  9. {
  10. $logId = $data['log_id'] ?? null;
  11. try {
  12. if (!isset($data['type']) || $data['type'] !== '高清放大') {
  13. $job->delete();
  14. return;
  15. }
  16. $startTime = date('Y-m-d H:i:s');
  17. echo "━━━━━━━━━━ ▶ 高清放大任务开始处理━━━━━━━━━━\n";
  18. echo "处理时间:{$startTime}\n";
  19. // 更新日志状态:处理中
  20. if ($logId) {
  21. Db::name('image_task_log')->where('id', $logId)->update([
  22. 'status' => 1,
  23. 'log' => '高清放大处理中',
  24. 'update_time' => $startTime
  25. ]);
  26. }
  27. //拼接原图文件路径 + 图片名称
  28. $old_image_url = rtrim($data['sourceDir'], '/') . '/' . ltrim($data['file_name'], '/');
  29. $list = Db::name("text_to_image")
  30. ->where('old_image_url', $old_image_url)
  31. ->where('img_name', '<>', '')
  32. ->where('status', 1)
  33. ->select();
  34. if (!empty($list)) {
  35. $total = count($list);
  36. echo "📊 共需处理:{$total} 条记录\n\n";
  37. foreach ($list as $index => $row) {
  38. $currentIndex = $index + 1;
  39. $begin = date('Y-m-d H:i:s');
  40. echo "处理时间:{$begin}\n";
  41. echo "👉 正在处理第 {$currentIndex} 条,ID: {$row['id']}\n";
  42. // 调用生成图像方法
  43. $result = $this->ImageToImage(
  44. $data["file_name"],
  45. $data["outputDir"],
  46. $row["new_image_url"],
  47. $row["img_name"],
  48. 679,
  49. 862
  50. );
  51. $resultText = ($result === true || $result === 1 || $result === '成功') ? '成功' : '失败或无返回';
  52. echo "✅ 处理结果:{$resultText}\n";
  53. $end = date('Y-m-d H:i:s');
  54. echo "完成时间:{$end}\n";
  55. echo "Processed: " . static::class . "\n";
  56. echo "高清放大已处理完成\n\n";
  57. }
  58. // 更新日志状态:成功
  59. if ($logId) {
  60. Db::name('image_task_log')->where('id', $logId)->update([
  61. 'status' => 2,
  62. 'log' => '高清放大处理成功',
  63. 'update_time' => date('Y-m-d H:i:s')
  64. ]);
  65. }
  66. echo date('Y-m-d H:i:s') . " 高清放大任务全部完成\n";
  67. } else {
  68. echo "未找到可处理的数据,跳过执行\n";
  69. if ($logId) {
  70. Db::name('image_task_log')->where('id', $logId)->update([
  71. 'status' => 2,
  72. 'log' => '无数据可处理,已跳过'.$old_image_url,
  73. 'update_time' => date('Y-m-d H:i:s')
  74. ]);
  75. }
  76. }
  77. // 如果还有链式任务,继续推送
  78. if (!empty($data['chain_next'])) {
  79. $nextType = array_shift($data['chain_next']);
  80. $data['type'] = $nextType;
  81. Queue::push('app\job\ImageArrJob', [
  82. 'task_id' => $data['task_id'],
  83. 'data' => [$data]
  84. ], 'arrimage');
  85. }
  86. $job->delete();
  87. } catch (\Exception $e) {
  88. //异常处理,记录失败日志
  89. echo "错误信息: " . $e->getMessage() . "\n";
  90. echo "文件: " . $e->getFile() . "\n";
  91. echo "行号: " . $e->getLine() . "\n";
  92. // 删除当前任务
  93. $job->delete();
  94. }
  95. }
  96. /**
  97. * 失败回调(可用于后续通知或重试机制)
  98. */
  99. public function failed($data)
  100. {
  101. echo "ImageJob failed: " . json_encode($data);
  102. }
  103. public function ImageToImage($fileName, $outputDirRaw, $new_image_url, $width, $height)
  104. {
  105. $rootPath = str_replace('\\', '/', ROOT_PATH);
  106. $outputDir = rtrim($rootPath . 'public/' . $outputDirRaw, '/') . '/';
  107. $dateDir = date('Y-m-d') . '/';
  108. $fullBaseDir = $outputDir . $dateDir;
  109. foreach ([$fullBaseDir, $fullBaseDir . 'high_definition/', $fullBaseDir . "{$width}x{$height}/"] as $dir) {
  110. if (!is_dir($dir)) {
  111. mkdir($dir, 0755, true);
  112. }
  113. }
  114. $record = Db::name('text_to_image')
  115. ->where('old_image_url', 'like', "%{$fileName}")
  116. ->order('id desc')
  117. ->find();
  118. if (!$record) {
  119. return json(['code' => 1, 'msg' => '没有找到匹配的图像记录']);
  120. }
  121. $ai = new AIGatewayService();
  122. $res = $ai->imgtogqGptApi($new_image_url);
  123. if (!isset($res['code']) || $res['code'] !== 0) {
  124. return json(['code' => 1, 'msg' => $res['msg'] ?? '图像高清放大生成失败']);
  125. }
  126. $base64Image = $res['data']['base64_image'];
  127. $originalBaseName = pathinfo($new_image_url, PATHINFO_FILENAME);
  128. $finalFileName = $originalBaseName . '.png';
  129. $targetDir = $fullBaseDir . 'high_definition/';
  130. $savePath = $targetDir . $finalFileName;
  131. if (!file_put_contents($savePath, base64_decode($base64Image))) {
  132. return json(['code' => 1, 'msg' => '保存图片失败']);
  133. }
  134. // Db::name('text_to_image')->where('id', $record['id'])->update([
  135. // 'imgtoimg_url' => $outputDirRaw . '/' . $dateDir . 'high_definition/' . $finalFileName,
  136. // 'status_name' => '高清放大',
  137. // 'error_msg' => '',
  138. // 'update_time' => date('Y-m-d H:i:s')
  139. // ]);
  140. return '成功';
  141. }
  142. }