|
|
@@ -0,0 +1,295 @@
|
|
|
+<?php
|
|
|
+
|
|
|
+namespace App\Services;
|
|
|
+
|
|
|
+use App\Models\MpGeneratePicTask;
|
|
|
+use GuzzleHttp\Client;
|
|
|
+use GuzzleHttp\Exception\GuzzleException;
|
|
|
+use Illuminate\Support\Facades\Log;
|
|
|
+
|
|
|
+class AiImageGenerationService
|
|
|
+{
|
|
|
+ private $volcEngineService;
|
|
|
+ private $httpClient;
|
|
|
+
|
|
|
+ public function __construct(VolcEngineService $volcEngineService)
|
|
|
+ {
|
|
|
+ $this->volcEngineService = $volcEngineService;
|
|
|
+ $this->httpClient = new Client([
|
|
|
+ 'timeout' => 300,
|
|
|
+ ]);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 异步创建图片生成任务
|
|
|
+ *
|
|
|
+ * @param array $params
|
|
|
+ * @return MpGeneratePicTask
|
|
|
+ */
|
|
|
+ public function createImageGenerationTask(array $params): MpGeneratePicTask
|
|
|
+ {
|
|
|
+ // 生成唯一的任务ID
|
|
|
+ $taskId = 'img_gen_' . time() . '_' . uniqid();
|
|
|
+
|
|
|
+ // 创建任务记录
|
|
|
+ $task = MpGeneratePicTask::create([
|
|
|
+ 'task_id' => $taskId,
|
|
|
+ 'prompt' => $params['prompt'] ?? '',
|
|
|
+ 'width' => $params['width'] ?? 2048,
|
|
|
+ 'height' => $params['height'] ?? 2048,
|
|
|
+ 'image_num' => $params['image_num'] ?? 1,
|
|
|
+ 'scale' => $params['scale'] ?? 50,
|
|
|
+ 'ref_img_url' => $params['ref_img_url'] ?? null,
|
|
|
+ 'mask_img_url' => $params['mask_img_url'] ?? null,
|
|
|
+ 'extra_params' => $params['extra_params'] ?? null,
|
|
|
+ 'status' => MpGeneratePicTask::STATUS_PENDING,
|
|
|
+ ]);
|
|
|
+
|
|
|
+ // 发送异步请求到即梦AI API创建任务
|
|
|
+ $this->submitTaskToApi($task);
|
|
|
+
|
|
|
+ return $task;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 提交任务到即梦AI API
|
|
|
+ *
|
|
|
+ * @param MpGeneratePicTask $task
|
|
|
+ * @return void
|
|
|
+ */
|
|
|
+ private function submitTaskToApi(MpGeneratePicTask $task): void
|
|
|
+ {
|
|
|
+ try {
|
|
|
+ // 验证环境变量配置
|
|
|
+ $accessKey = env('VOLC_AK');
|
|
|
+ $secretKey = env('VOLC_SK');
|
|
|
+
|
|
|
+ if (empty($accessKey) || empty($secretKey)) {
|
|
|
+ $task->updateStatus(MpGeneratePicTask::STATUS_FAILED, [
|
|
|
+ 'error_message' => '火山引擎访问密钥未配置,请检查环境变量VOLC_AK和VOLC_SK'
|
|
|
+ ]);
|
|
|
+ return;
|
|
|
+ }
|
|
|
+
|
|
|
+ //构建即梦AI 4.0 API请求参数
|
|
|
+ $apiParams = [
|
|
|
+ 'req_key' => 'jimeng_t2i_v40',
|
|
|
+ 'prompt' => $task->prompt,
|
|
|
+ 'scale' => $task->scale / 100, //为0-1范围
|
|
|
+ ];
|
|
|
+
|
|
|
+ // 添加可选参数
|
|
|
+ if ($task->image_num > 1) {
|
|
|
+ $apiParams['force_single'] = false;
|
|
|
+ } else {
|
|
|
+ $apiParams['force_single'] = true;
|
|
|
+ }
|
|
|
+
|
|
|
+ // 参考图片
|
|
|
+ if ($task->ref_img_url) {
|
|
|
+ $apiParams['image_urls'] = [$task->ref_img_url];
|
|
|
+ }
|
|
|
+
|
|
|
+ if ($task->mask_img_url) {
|
|
|
+ $apiParams['image_urls'][] = $task->mask_img_url;
|
|
|
+ }
|
|
|
+
|
|
|
+ // 尺寸参数
|
|
|
+ $area = $task->width * $task->height; //计算面积
|
|
|
+ if ($area >= 1024*1024 && $area <= 4096*4096) {
|
|
|
+ $apiParams['size'] = $area;
|
|
|
+ }
|
|
|
+
|
|
|
+ //调用即梦AI 4.0 API提交任务
|
|
|
+ $response = $this->volcEngineService->request(
|
|
|
+ 'POST',
|
|
|
+ 'visual.volcengineapi.com', //即梦AI API域名
|
|
|
+ '/', // 使用根路径
|
|
|
+ [], // Query参数
|
|
|
+ json_encode($apiParams),
|
|
|
+ $accessKey,
|
|
|
+ $secretKey,
|
|
|
+ 'cv', // 服务标识
|
|
|
+ 'cn-north-1', //区域
|
|
|
+ 'CVSync2AsyncSubmitTask',
|
|
|
+ '2022-08-31',
|
|
|
+ 'application/json'
|
|
|
+ );
|
|
|
+
|
|
|
+ $responseData = json_decode($response['body'], true);
|
|
|
+ dLog('generate')->info('即梦AI 4.0 API提交任务响应: ', $responseData);
|
|
|
+
|
|
|
+ if ($responseData['code'] !== 10000) {
|
|
|
+ // API返回错误
|
|
|
+ $task->updateStatus(MpGeneratePicTask::STATUS_FAILED, [
|
|
|
+ 'error_message' => $responseData['message'] ?? 'API Error',
|
|
|
+ 'extra_params' => $responseData['data'] ?? []
|
|
|
+ ]);
|
|
|
+ } else {
|
|
|
+ // 任务提交成功,更新状态为处理中
|
|
|
+ $updateData = [];
|
|
|
+ //存储API返回的任务ID
|
|
|
+ if (isset($responseData['data']['task_id'])) {
|
|
|
+ $updateData['task_id'] = $responseData['data']['task_id'];
|
|
|
+ $updateData['extra_params'] = $responseData['data'] ?? [];
|
|
|
+ }
|
|
|
+
|
|
|
+ $task->updateStatus(MpGeneratePicTask::STATUS_PROCESSING, $updateData);
|
|
|
+ }
|
|
|
+ } catch (\Exception $e) {
|
|
|
+ // 记录错误
|
|
|
+ $task->updateStatus(MpGeneratePicTask::STATUS_FAILED, [
|
|
|
+ 'error_message' => 'API请求失败: ' . $e->getMessage()
|
|
|
+ ]);
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ public function getTaskList($data)
|
|
|
+ {
|
|
|
+ $task_id = getProp($data, 'task_id');
|
|
|
+ $status = getProp($data, 'status');
|
|
|
+
|
|
|
+ $query = MpGeneratePicTask::select('*');
|
|
|
+ if ($task_id) {
|
|
|
+ $query->where('task_id', $task_id);
|
|
|
+ }
|
|
|
+ if ($status) {
|
|
|
+ $query->where('status', $status);
|
|
|
+ }
|
|
|
+
|
|
|
+ return $query->orderBy('created_at', 'desc')->paginate();
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 查询任务状态
|
|
|
+ *
|
|
|
+ * @param MpGeneratePicTask $task
|
|
|
+ * @return array
|
|
|
+ */
|
|
|
+ public function queryTaskStatus(MpGeneratePicTask $task): array
|
|
|
+ {
|
|
|
+ try {
|
|
|
+ // 获取API任务ID
|
|
|
+ $apiTaskId = $task->task_id ?? null;
|
|
|
+
|
|
|
+ if (!$apiTaskId) {
|
|
|
+ return [
|
|
|
+ 'status' => 'failed',
|
|
|
+ 'error_message' => 'API任务ID不存在',
|
|
|
+ 'result_url' => null
|
|
|
+ ];
|
|
|
+ }
|
|
|
+
|
|
|
+ //构建查询参数
|
|
|
+ $apiParams = [
|
|
|
+ 'req_key' => 'jimeng_t2i_v40',
|
|
|
+ 'task_id' => $apiTaskId,
|
|
|
+ 'req_json' => '{"return_url":true}' // 返回URL格式
|
|
|
+ ];
|
|
|
+
|
|
|
+ //调用即梦AI 4.0 API查询任务状态
|
|
|
+ $response = $this->volcEngineService->request(
|
|
|
+ 'POST',
|
|
|
+ 'visual.volcengineapi.com',
|
|
|
+ '/', // 使用根路径
|
|
|
+ [], // Query参数
|
|
|
+ json_encode($apiParams),
|
|
|
+ env('VOLC_AK'),
|
|
|
+ env('VOLC_SK'),
|
|
|
+ 'cv',
|
|
|
+ 'cn-north-1',
|
|
|
+ 'CVSync2AsyncGetResult',
|
|
|
+ '2022-08-31',
|
|
|
+ 'application/json'
|
|
|
+ );
|
|
|
+
|
|
|
+ $responseData = json_decode($response['body'], true);
|
|
|
+ dLog('generate')->info('即梦AI 4.0 API查询任务状态响应: ', $responseData);
|
|
|
+
|
|
|
+ if ($responseData['code'] !== 10000) {
|
|
|
+ // API返回错误
|
|
|
+ return [
|
|
|
+ 'status' => 'failed',
|
|
|
+ 'error_message' => $responseData['message'] ?? 'API Error',
|
|
|
+ 'result_url' => null
|
|
|
+ ];
|
|
|
+ }
|
|
|
+
|
|
|
+ // 解析API响应
|
|
|
+ $result = $responseData['data'] ?? [];
|
|
|
+ $taskStatus = $result['status'] ?? 'failed';
|
|
|
+
|
|
|
+ $returnData = [
|
|
|
+ 'status' => $taskStatus,
|
|
|
+ 'error_message' => null,
|
|
|
+ 'result_url' => null
|
|
|
+ ];
|
|
|
+
|
|
|
+ // 如果任务成功,获取结果URL
|
|
|
+ if ($taskStatus === 'done' && isset($result['image_urls']) && count($result['image_urls']) > 0) {
|
|
|
+ $result_urls = [];
|
|
|
+ foreach ($result['image_urls'] as $url) {
|
|
|
+ $pic_name = 'ai_generation_' . time() . '_' . uniqid();
|
|
|
+ $pic_ext = getExtFromContent($url);
|
|
|
+ $pic_name = $pic_name . '.' . $pic_ext;
|
|
|
+
|
|
|
+ // 将图片另存到tos
|
|
|
+ // $url = uploadStreamByTos('img_generation', file_get_contents($url), $pic_name);
|
|
|
+ $result_urls[] = $url;
|
|
|
+ }
|
|
|
+ $returnData['result_url'] = $result_urls;
|
|
|
+ $returnData['status'] = 'success';
|
|
|
+ } elseif (in_array($taskStatus, ['not_found', 'expired'])) {
|
|
|
+ $returnData['status'] = 'failed';
|
|
|
+ $returnData['error_message'] = '任务未找到或已过期';
|
|
|
+ } elseif ($taskStatus === 'failed') {
|
|
|
+ $returnData['status'] = 'failed';
|
|
|
+ $returnData['error_message'] = '任务执行失败';
|
|
|
+ }
|
|
|
+ // in_queue 或 generating状态保持原status
|
|
|
+
|
|
|
+ return $returnData;
|
|
|
+ } catch (\Exception $e) {
|
|
|
+ return [
|
|
|
+ 'status' => 'failed',
|
|
|
+ 'error_message' => $e->getMessage(),
|
|
|
+ 'result_url' => null
|
|
|
+ ];
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 更新所有待处理任务的状态
|
|
|
+ *
|
|
|
+ * @return void
|
|
|
+ */
|
|
|
+ public function updatePendingTasks(): void
|
|
|
+ {
|
|
|
+ // 获取所有处理中的任务
|
|
|
+ $tasks = MpGeneratePicTask::where('status', MpGeneratePicTask::STATUS_PROCESSING)
|
|
|
+ ->get();
|
|
|
+
|
|
|
+ foreach ($tasks as $task) {
|
|
|
+ $statusInfo = $this->queryTaskStatus($task);
|
|
|
+
|
|
|
+ if ($statusInfo['status'] === 'success') {
|
|
|
+ $task->updateStatus(MpGeneratePicTask::STATUS_SUCCESS, [
|
|
|
+ 'result_url' => $statusInfo['result_url']
|
|
|
+ ]);
|
|
|
+ } elseif ($statusInfo['status'] === 'failed') {
|
|
|
+ $task->updateStatus(MpGeneratePicTask::STATUS_FAILED, [
|
|
|
+ 'error_message' => $statusInfo['error_message']
|
|
|
+ ]);
|
|
|
+ }
|
|
|
+ // 如果仍然是处理中状态,不做任何操作
|
|
|
+
|
|
|
+ //处理:如果任务处理超过12小时,标记为失败
|
|
|
+ $processingTime = now()->diffInHours($task->created_at);
|
|
|
+ if ($processingTime > 12) {
|
|
|
+ $task->updateStatus(MpGeneratePicTask::STATUS_FAILED, [
|
|
|
+ 'error_message' => '任务处理超时(超过12小时)'
|
|
|
+ ]);
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+}
|