| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273 |
- <?php
- namespace app\job;
- use think\Db;
- use think\queue\Job;
- use think\Queue;
- class ImageArrJob
- {
- public function fire(Job $job, $data)
- {
- $task_id = $data['task_id'];
- $images = $data['data'];
- foreach ($images as $value) {
- $value['task_id'] = $task_id;
- // 1. 清理 sourceDir,去掉末尾重复的 Preview
- $sourceDir = rtrim($value["sourceDir"], '/\\');
- $dirParts = explode('/', str_replace('\\', '/', $sourceDir));
- if (end($dirParts) === 'Preview') {
- array_pop($dirParts); // 删除末尾 Preview
- }
- $cleanedSourceDir = implode('/', $dirParts);
- // 2. 获取纯文件名
- $fileName = basename($value['file_name']);
- // 3. 拼接最终路径
- $fullPath = $cleanedSourceDir . '/' . $fileName;
- // 4. 写入日志记录
- $log_id = Db::name('image_task_log')->insertGetId([
- 'task_id' => $task_id,
- 'file_name' => $fullPath,
- 'type' => $value['type'] ?? '',
- 'status' => 0,
- 'log' => '队列中',
- 'create_time' => date('Y-m-d H:i:s')
- ]);
- $value['log_id'] = $log_id;
- // 5. 推送任务到对应队列
- switch (trim($value['type'])) {
- case '图生文':
- Queue::push('app\job\ImageJob', $value, 'imgtotxt');
- break;
- case '文生文':
- Queue::push('app\job\TextToTextJob', $value, 'txttotxt');
- break;
- case '文生图':
- Queue::push('app\job\TextToImageJob', $value, 'txttoimg');
- break;
- default:
- // 类型未知时不推送,并可写入错误日志
- \think\Log::warning("未识别的任务类型:" . json_encode($value, JSON_UNESCAPED_UNICODE));
- break;
- }
- }
- // 6. 更新任务状态为已启动
- Db::name('queue_logs')->where('id', $task_id)->update(['status' => '已启动队列']);
- echo date('Y-m-d H:i:s') . " 队列已启动\n";
- // 7. 删除当前队列任务
- $job->delete();
- }
- public function failed($data)
- {
- \think\Log::error("ImageArrJob 任务失败:" . json_encode($data, JSON_UNESCAPED_UNICODE));
- }
- }
|