mes-sdai-server-api/application/job/ImageToImageJob.php

<?php
namespace app\job;
use app\service\AIGatewayService;
use think\Db;
use think\queue\Job;
use think\Queue;

class ImageToImageJob{

    public function fire(Job $job, $data)
    {

        $logId = $data['log_id'] ?? null;

        try {
            // 任务类型校验（必须是图生图）
            if (!isset($data['type']) || $data['type'] !== '图生图') {
                $job->delete();
                return;
            }

            $startTime = date('Y-m-d H:i:s');
            echo "━━━━━━━━━━ ▶ 图生图任务开始处理━━━━━━━━━━\n";
            echo "处理时间：{$startTime}\n";

            // 更新日志状态：处理中
            if ($logId) {
                Db::name('image_task_log')->where('id', $logId)->update([
                    'status' => 1,
                    'log' => '图生图处理中',
                    'update_time' => $startTime
                ]);
            }

            //拼接原图文件路径 + 图片名称
            $old_image_url = rtrim($data['sourceDir'], '/') . '/' . ltrim($data['file_name'], '/');

            $list = Db::name("text_to_image")
                ->where('old_image_url', $old_image_url)
                ->where('img_name', '<>', '')
                ->where('status', 1)
                ->select();

            if (!empty($list)) {
                $total = count($list);
                echo "📊 共需处理：{$total} 条记录\n\n";

                foreach ($list as $index => $row) {
                    $currentIndex = $index + 1;
                    $begin = date('Y-m-d H:i:s');
                    echo "处理时间：{$begin}\n";
                    echo "👉 正在处理第 {$currentIndex} 条，ID: {$row['id']}\n";

                    // 调用生成图像方法
                    $result = $this->ImageToImage(
                        $data["file_name"],
                        $data["outputDir"],
                        $row["new_image_url"],
                        $row["img_name"],
                        679,
                        862
                    );

                    $resultText = ($result === true || $result === 1 || $result === '成功') ? '成功' : '失败或无返回';
                    echo "✅ 处理结果：{$resultText}\n";

                    $end = date('Y-m-d H:i:s');
                    echo "完成时间：{$end}\n";
                    echo "Processed: " . static::class . "\n";
                    echo "图生图已处理完成\n\n";
                }

                // 更新日志状态：成功
                if ($logId) {
                    Db::name('image_task_log')->where('id', $logId)->update([
                        'status' => 2,
                        'log' => '图生图处理成功',
                        'update_time' => date('Y-m-d H:i:s')
                    ]);
                }

                echo date('Y-m-d H:i:s') . " 图生图任务全部完成\n";
            } else {

                echo "未找到可处理的数据，跳过执行\n";
                if ($logId) {
                    Db::name('image_task_log')->where('id', $logId)->update([
                        'status' => 2,
                        'log' => '无数据可处理，已跳过'.$old_image_url,
                        'update_time' => date('Y-m-d H:i:s')
                    ]);
                }
            }

            // 如果还有链式任务，继续推送
            if (!empty($data['chain_next'])) {
                $nextType = array_shift($data['chain_next']);
                $data['type'] = $nextType;

                Queue::push('app\job\ImageArrJob', [
                    'task_id' => $data['task_id'],
                    'data' => [$data]
                ], 'arrimage');
            }

            $job->delete();

        }  catch (\Exception $e) {
            //异常处理，记录失败日志
            echo "错误信息: " . $e->getMessage() . "\n";
            echo "文件: " . $e->getFile() . "\n";
            echo "行号: " . $e->getLine() . "\n";

            // 删除当前任务
            $job->delete();
        }
    }

    /**
     * 失败回调（可用于后续通知或重试机制）
     */
    public function failed($data)
    {
        echo "ImageJob failed: " . json_encode($data);
    }

//    public function ImageToImage($fileName, $outputDirRaw, $new_image_url,$img_name, $width, $height)
//    {
//        // 统一路径分隔符
//        $rootPath = str_replace('\\', '/', ROOT_PATH);
//
//        // 输出目录，如：ROOT/public/uploads/operate/ai/dall-e/hua/
//        $outputDir = rtrim($rootPath . 'public/' . $outputDirRaw, '/') . '/';
//
//        // 当前日期目录，如：2025-06-16/
//        $dateDir = date('Y-m-d') . '/';
//
//        // 完整基本路径，如：ROOT/public/uploads/operate/ai/dall-e/hua/2025-06-16/
//        $fullBaseDir = $outputDir . $dateDir;
//
//        // 只创建 img_679x862 目录
//        $saveDir = $fullBaseDir . 'new_679x862/';
//        if (!is_dir($saveDir)) {
//            mkdir($saveDir, 0755, true);
//        }
//
//        // 从数据库中查询原图记录
//        $record = Db::name('text_to_image')
//            ->where('old_image_url', 'like', "%{$fileName}")
//            ->order('id desc')
//            ->find();
//
//        if (!$record) {
//            return json([
//                'code' => 1,
//                'msg' => '没有找到匹配的图像记录'
//            ]);
//        }
//
//        // 调用图生图 API
//        $ai = new AIGatewayService();
//        $res = $ai->imgtoimgGptApi('', $new_image_url);
//
//        // 检查返回结果
//        if (!isset($res['code']) || $res['code'] !== 0) {
//            return json([
//                'code' => 1,
//                'msg' => $res['msg'] ?? '图像生成失败'
//            ]);
//        }
//
//        // 保存图片路径
//        $img_name = mb_substr(preg_replace('/[^\x{4e00}-\x{9fa5}A-Za-z0-9_\- ]/u', '', $img_name), 0, 30);
//        $filename = $img_name . '.png';
//        $path = $saveDir . $filename;
//
//        // 解码图像 base64 数据
//        $imgData = base64_decode($res['data']['url']);
//
//        // 解析图像内容
//        try {
//            $im = \imagecreatefromstring($imgData);
//            if (!$im) {
//                file_put_contents('/tmp/corrupted.png', $imgData);
//                throw new \Exception("❌ 图像无法解析，写入 /tmp/corrupted.png");
//            }
//        } catch (\Throwable $e) {
//            file_put_contents('/tmp/corrupted.png', $imgData);
//            throw new \Exception("❌ 图像处理异常：" . $e->getMessage());
//        }
//
//        // 裁剪
//        $srcW = imagesx($im);
//        $srcH = imagesy($im);
//        $srcRatio = $srcW / $srcH;
//        $dstRatio = $width / $height;
//
//        if ($srcRatio > $dstRatio) {
//            $cropW = intval($srcH * $dstRatio);
//            $cropH = $srcH;
//            $srcX = intval(($srcW - $cropW) / 2);
//            $srcY = 0;
//        } else {
//            $cropW = $srcW;
//            $cropH = intval($srcW / $dstRatio);
//            $srcX = 0;
//            $srcY = intval(($srcH - $cropH) / 2);
//        }
//
//        $dstImg = imagecreatetruecolor($width, $height);
//        imagecopyresampled($dstImg, $im, 0, 0, $srcX, $srcY, $width, $height, $cropW, $cropH);
//
//        // 保存裁剪图
//        imagepng($dstImg, $path);
//        imagedestroy($im);
//        imagedestroy($dstImg);
//
//        // 更新数据库记录
//        Db::name('text_to_image')->where('id', $record['id'])->update([
//            'imgtoimg_url' => str_replace($rootPath . 'public/', '', $path),
//            'status_name' => '图生图',
//            'error_msg' => '',
//            'update_time' => date('Y-m-d H:i:s')
//        ]);
//
//        return '成功';
//    }


    public function ImageToImage($fileName, $outputDirRaw, $new_image_url, $width, $height)
    {
        // 统一路径分隔符
        $rootPath = str_replace('\\', '/', ROOT_PATH);

        // 输出目录，如：ROOT/public/uploads/operate/ai/dall-e/
        $outputDir = rtrim($rootPath . 'public/' . $outputDirRaw, '/') . '/';

        // 当前日期目录，如：2025-06-16/
        $dateDir = date('Y-m-d') . '/';

        // 完整基本路径，如：ROOT/public/uploads/operate/ai/dall-e/hua/2025-06-16/
        $fullBaseDir = $outputDir . $dateDir;

        // 创建输出目录，包括原图目录、目录、自定义尺寸目录
        foreach ([$fullBaseDir, $fullBaseDir . 'newimg_679x862/', $fullBaseDir . "{$width}x{$height}/"] as $dir) {
            if (!is_dir($dir)) {
                mkdir($dir, 0755, true);
            }
        }

        // 从数据库中查询原图记录
        $record = Db::name('text_to_image')
            ->where('old_image_url', 'like', "%{$fileName}")
            ->order('id desc')
            ->find();

        if (!$record) {
            return json([
                'code' => 1,
                'msg' => '没有找到匹配的图像记录'
            ]);
        }

        // 调用图生图 API
        $ai = new AIGatewayService();
        $res = $ai->imgtoimgGptApi('', $new_image_url);

        // 检查返回结果
        if (!isset($res['code']) || $res['code'] !== 0) {
            return json([
                'code' => 1,
                'msg' => $res['msg'] ?? '图像生成失败'
            ]);
        }

        // 解码图像 base64 数据
        $originalBaseName = pathinfo($new_image_url, PATHINFO_FILENAME);
        $finalFileName = $originalBaseName . '.png';

        // 保存到 子目录
        $targetDir = $fullBaseDir . 'newimg_679x862/';
        $savePath = $targetDir . $finalFileName;
        file_put_contents($savePath, base64_decode($res['data']['url']));

        Db::name('text_to_image')->where('id', $record['id'])->update([
            'imgtoimg_url' => $outputDirRaw . '/' . $dateDir . 'newimg_679x862/' . $finalFileName,
            'status_name' => '图生图',
            'error_msg' => '',
            'update_time' => date('Y-m-d H:i:s')
        ]);

        return '成功';
    }

}