AnimeController.php 36 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869
  1. <?php
  2. namespace App\Http\Controllers\Anime;
  3. use App\Transformer\Anime\AnimeTransformer;
  4. use App\Facade\Site;
  5. use App\Consts\ErrorConst;
  6. use App\Exceptions\ApiException;
  7. use App\Libs\ApiResponse;
  8. use App\Libs\Utils;
  9. use App\Models\MpGenerateVideoTask;
  10. use App\Services\AIGeneration\AIImageGenerationService;
  11. use App\Services\AIGeneration\AIVideoGenerationService;
  12. use App\Services\Anime\AnimeService;
  13. use Illuminate\Http\Request;
  14. use Illuminate\Routing\Controller as BaseController;
  15. use Illuminate\Support\Facades\DB;
  16. use Illuminate\Support\Facades\Redis;
  17. use Illuminate\Support\Facades\Validator;
  18. class AnimeController extends BaseController
  19. {
  20. use ApiResponse;
  21. protected $AnimeService;
  22. protected $AIImageGenerationService;
  23. protected $AIVideoGenerationService;
  24. public function __construct(
  25. AnimeService $AnimeService,
  26. AIImageGenerationService $AIImageGenerationService,
  27. AIVideoGenerationService $AIVideoGenerationService
  28. ) {
  29. $this->AnimeService = $AnimeService;
  30. $this->AIImageGenerationService = $AIImageGenerationService;
  31. $this->AIVideoGenerationService = $AIVideoGenerationService;
  32. }
  33. public function textModel() {
  34. $models = DB::table('mp_text_models')->where('is_enabled', 1)->select('model', 'name')->get()->map(function ($value) {
  35. return (array)$value;
  36. })->toArray();
  37. return $this->success($models);
  38. }
  39. public function artStyleList()
  40. {
  41. $result = [
  42. [
  43. 'art_style' => '日系动漫风格',
  44. 'pic_url' => "https://cdn-zwai.ycsd.cn/mp_audio/art_styles/日系动漫风格.jpg"
  45. ],
  46. [
  47. 'art_style' => '国漫风格',
  48. 'pic_url' => "https://cdn-zwai.ycsd.cn/mp_audio/art_styles/国漫风格.jpg"
  49. ],
  50. [
  51. 'art_style' => 'Q版卡通风格',
  52. 'pic_url' => "https://cdn-zwai.ycsd.cn/mp_audio/art_styles/Q版卡通风格.jpg"
  53. ],
  54. [
  55. 'art_style' => '简约扁平风格',
  56. 'pic_url' => "https://cdn-zwai.ycsd.cn/mp_audio/art_styles/简约扁平风格.jpg"
  57. ],
  58. [
  59. 'art_style' => '古风仙侠风格',
  60. 'pic_url' => "https://cdn-zwai.ycsd.cn/mp_audio/art_styles/古风仙侠风格.jpg"
  61. ],
  62. [
  63. 'art_style' => '武侠风格',
  64. 'pic_url' => "https://cdn-zwai.ycsd.cn/mp_audio/art_styles/武侠风格.jpg"
  65. ],
  66. [
  67. 'art_style' => '新中式水墨风格',
  68. 'pic_url' => "https://cdn-zwai.ycsd.cn/mp_audio/art_styles/新中式水墨风格.jpg"
  69. ],
  70. [
  71. 'art_style' => '写实插画风格',
  72. 'pic_url' => "https://cdn-zwai.ycsd.cn/mp_audio/art_styles/写实插画风格.jpg"
  73. ],
  74. [
  75. 'art_style' => '3D卡通风格',
  76. 'pic_url' => "https://cdn-zwai.ycsd.cn/mp_audio/art_styles/3D卡通风格.jpg"
  77. ],
  78. [
  79. 'art_style' => '条漫风格',
  80. 'pic_url' => "https://cdn-zwai.ycsd.cn/mp_audio/art_styles/条漫风格.jpg"
  81. ],
  82. [
  83. 'art_style' => '赛博朋克风格',
  84. 'pic_url' => "https://cdn-zwai.ycsd.cn/mp_audio/art_styles/赛博朋克风格.jpg"
  85. ],
  86. [
  87. 'art_style' => '暗黑悬疑风格',
  88. 'pic_url' => "https://cdn-zwai.ycsd.cn/mp_audio/art_styles/暗黑悬疑风格.jpg"
  89. ],
  90. [
  91. 'art_style' => '治愈清新风格',
  92. 'pic_url' => "https://cdn-zwai.ycsd.cn/mp_audio/art_styles/治愈清新风格.jpg"
  93. ],
  94. ];
  95. return $this->success($result);
  96. }
  97. // 批量生成主体图片
  98. public function batchSetRoleImg(Request $request) {
  99. // 忽略所有超时限制
  100. set_time_limit(0);
  101. ini_set('max_execution_time', '0');
  102. $data = $request->all();
  103. $result = $this->AnimeService->batchSetRoleImg($data);
  104. return $this->success($result);
  105. }
  106. // 批量生成场景图片
  107. public function batchSetSceneImg(Request $request) {
  108. // 忽略所有超时限制
  109. set_time_limit(0);
  110. ini_set('max_execution_time', '0');
  111. $data = $request->all();
  112. $result = $this->AnimeService->batchSetSceneImg($data);
  113. return $this->success($result);
  114. }
  115. // 编辑动漫
  116. public function editAnime(Request $request) {
  117. $data = $request->all();
  118. $result = $this->AnimeService->editAnime($data);
  119. return $this->success(['success'=>$result ? 1 : 0]);
  120. }
  121. // 修改分集主体列表
  122. public function changeEpisodeRoles(Request $request) {
  123. $data = $request->all();
  124. $result = $this->AnimeService->changeEpisodeRoles($data);
  125. return $this->success(['success'=>$result ? 1 : 0]);
  126. }
  127. // 修改分集场景列表
  128. public function changeEpisodeScenes(Request $request) {
  129. $data = $request->all();
  130. $result = $this->AnimeService->changeEpisodeScenes($data);
  131. return $this->success(['success'=>$result ? 1 : 0]);
  132. }
  133. // 编辑分镜剧本
  134. public function editSegment(Request $request) {
  135. $data = $request->all();
  136. $result = $this->AnimeService->editSegment($data);
  137. return $this->success(['success'=>$result ? 1 : 0]);
  138. }
  139. // 动漫对话列表
  140. public function chatList(Request $request) {
  141. $data = $request->all();
  142. $result = $this->AnimeService->chatList($data);
  143. return $this->success($result, [new AnimeTransformer(), 'newBuildChatList']);
  144. }
  145. // 动漫对话历史记录
  146. public function chatHistory(Request $request) {
  147. $data = $request->all();
  148. $result = $this->AnimeService->chatHistory($data);
  149. return $this->success($result);
  150. }
  151. // 动漫大纲
  152. public function animeDetail(Request $request) {
  153. $data = $request->all();
  154. $result = $this->AnimeService->animeDetail($data);
  155. return $this->success($result);
  156. }
  157. // 动漫剧集
  158. public function episodeInfo(Request $request) {
  159. $data = $request->all();
  160. $result = $this->AnimeService->episodeInfo($data);
  161. return $this->success($result);
  162. }
  163. // 复制剧集副本
  164. public function copyEpisodeVersion(Request $request) {
  165. $data = $request->all();
  166. $result = $this->AnimeService->copyEpisodeVersion($data);
  167. return $this->success(['success' => $result ? 1 : 0]);
  168. }
  169. // 剧集副本列表
  170. public function episodeVersions(Request $request) {
  171. $data = $request->all();
  172. $result = $this->AnimeService->episodeVersions($data);
  173. return $this->success($result);
  174. }
  175. // 绑定剧集副本
  176. public function bindEpisodeVersion(Request $request) {
  177. $data = $request->all();
  178. $result = $this->AnimeService->bindEpisodeVersion($data);
  179. return $this->success(['success' => $result ? 1 : 0]);
  180. }
  181. // 对话改图
  182. public function chatChangeImg(Request $request) {
  183. $data = $request->all();
  184. $result = $this->AnimeService->chatChangeImg($data);
  185. return $this->success(['img_url' => $result]);
  186. }
  187. // 一键生成分镜
  188. public function batchSetSegmentPics(Request $request) {
  189. // 忽略所有超时限制
  190. set_time_limit(0);
  191. ini_set('max_execution_time', '0');
  192. $data = $request->all();
  193. $result = $this->AnimeService->batchSetSegmentPics($data);
  194. return $this->success($result);
  195. }
  196. // 重新生成分镜
  197. public function reGenerateSegment(Request $request) {
  198. // 忽略所有超时限制
  199. set_time_limit(0);
  200. ini_set('max_execution_time', '0');
  201. $data = $request->all();
  202. $result = $this->AnimeService->reGenerateSegment($data);
  203. return $this->success(['img_url'=>$result]);
  204. }
  205. // 添加分镜
  206. public function addSegment(Request $request) {
  207. // 忽略所有超时限制
  208. set_time_limit(0);
  209. ini_set('max_execution_time', '0');
  210. $data = $request->all();
  211. $result = $this->AnimeService->addSegment($data);
  212. return $this->success($result);
  213. }
  214. // 复制分镜
  215. public function copySegment(Request $request) {
  216. $data = $request->all();
  217. $result = $this->AnimeService->copySegment($data);
  218. return $this->success($result);
  219. }
  220. // 移动分镜
  221. public function moveSegment(Request $request) {
  222. $data = $request->all();
  223. $result = $this->AnimeService->moveSegment($data);
  224. return $this->success(['success'=>$result ? 1: 0]);
  225. }
  226. // 删除分镜
  227. public function delSegment(Request $request) {
  228. $data = $request->all();
  229. $result = $this->AnimeService->delSegment($data);
  230. return $this->success(['success'=>$result ? 1: 0]);
  231. }
  232. // 分镜历史图片|视频
  233. public function segmentHistory(Request $request) {
  234. $data = $request->all();
  235. $result = $this->AnimeService->segmentHistory($data);
  236. return $this->success($result);
  237. }
  238. public function createSegmentVideoTask(Request $request) {
  239. // 忽略所有超时限制
  240. set_time_limit(0);
  241. ini_set('max_execution_time', '0');
  242. $data = $request->all();
  243. // 验证参数
  244. $validator = Validator::make($data, [
  245. 'segment_id' => 'required|string',
  246. 'tail_frame' => 'nullable|string|max:500',
  247. ], [
  248. 'segment_id.required' => '分镜ID不能为空',
  249. 'tail_frame.max' => '尾帧描述不能超过500个字符',
  250. ]);
  251. if ($validator->fails()) {
  252. Utils::throwError('1002:' . $validator->errors()->first());
  253. }
  254. // 创建视频生成任务
  255. $result = $this->AnimeService->createSegmentVideoTask($data);
  256. $taskId = $result['task_id'];
  257. // 设置 SSE 响应头
  258. return response()->stream(function () use ($taskId, $result) {
  259. // 设置 SSE 响应头
  260. echo "data: " . json_encode([
  261. 'type' => 'task_created',
  262. 'data' => $result
  263. ]) . "\n\n";
  264. ob_flush();
  265. flush();
  266. $startTime = time();
  267. $maxDuration = 300; // 5分钟超时
  268. $checkInterval = 5; // 每5秒检查一次
  269. while (time() - $startTime < $maxDuration) {
  270. try {
  271. // 查询任务状态
  272. $task = \App\Models\MpGenerateVideoTask::find($taskId);
  273. if (!$task) {
  274. echo "data: " . json_encode([
  275. 'type' => 'error',
  276. 'message' => '任务不存在'
  277. ]) . "\n\n";
  278. ob_flush();
  279. flush();
  280. break;
  281. }
  282. // 如果任务还在处理中,查询最新状态
  283. if ($task->status === 'processing') {
  284. $statusResult = $this->AIVideoGenerationService->querySeedanceTaskStatus($task);
  285. if (isset($statusResult['status'])) {
  286. // 更新任务状态
  287. $task->update([
  288. 'status' => $statusResult['status'],
  289. 'result_url' => $statusResult['result_url'] ?? null,
  290. 'last_frame_url' => $statusResult['last_frame_url'] ?? '',
  291. 'error_message' => $statusResult['error_message'] ?? null,
  292. 'completed_at' => in_array($statusResult['status'], [
  293. 'success',
  294. 'failed'
  295. ]) ? now() : null
  296. ]);
  297. // 如果任务成功,更新分镜表
  298. if ($statusResult['status'] === 'success' && isset($statusResult['result_url'])) {
  299. DB::table('mp_episode_segments')
  300. ->where('video_task_id', $taskId)
  301. ->update([
  302. 'video_url' => $statusResult['result_url'],
  303. 'video_task_status' => '已完成',
  304. 'last_frame_url' => $statusResult['last_frame_url'] ?? '',
  305. 'updated_at' => date('Y-m-d H:i:s')
  306. ]);
  307. } elseif ($statusResult['status'] === 'failed') {
  308. DB::table('mp_episode_segments')
  309. ->where('video_task_id', $taskId)
  310. ->update([
  311. 'video_task_status' => '失败',
  312. 'updated_at' => date('Y-m-d H:i:s')
  313. ]);
  314. }
  315. }
  316. }
  317. // 发送当前状态
  318. echo "data: " . json_encode([
  319. 'type' => 'status_update',
  320. 'data' => [
  321. 'task_id' => $task->id,
  322. 'status' => $task->status,
  323. 'result_url' => $task->result_url,
  324. 'error_message' => $task->error_message,
  325. 'progress' => $this->getTaskProgress($task->status),
  326. 'elapsed_time' => time() - $startTime
  327. ]
  328. ]) . "\n\n";
  329. ob_flush();
  330. flush();
  331. // 如果任务完成(成功或失败),结束连接
  332. if (in_array($task->status, [
  333. 'success',
  334. 'failed'
  335. ])) {
  336. echo "data: " . json_encode([
  337. 'type' => 'completed',
  338. 'data' => [
  339. 'task_id' => $task->id,
  340. 'status' => $task->status,
  341. 'video_url' => $task->result_url,
  342. 'last_frame_url' => $task->last_frame_url,
  343. 'error_message' => $task->error_message
  344. ]
  345. ]) . "\n\n";
  346. ob_flush();
  347. flush();
  348. break;
  349. }
  350. sleep($checkInterval);
  351. } catch (\Exception $e) {
  352. echo "data: " . json_encode([
  353. 'type' => 'error',
  354. 'message' => '查询任务状态失败: ' . $e->getMessage()
  355. ]) . "\n\n";
  356. ob_flush();
  357. flush();
  358. sleep($checkInterval);
  359. }
  360. }
  361. // 超时处理
  362. if (time() - $startTime >= $maxDuration) {
  363. echo "data: " . json_encode([
  364. 'type' => 'timeout',
  365. 'message' => '任务执行超时,请稍后查询任务状态',
  366. 'data' => [
  367. 'task_id' => $taskId
  368. ]
  369. ]) . "\n\n";
  370. ob_flush();
  371. flush();
  372. }
  373. }, 200, [
  374. 'Content-Type' => 'text/event-stream',
  375. 'Cache-Control' => 'no-cache',
  376. 'Connection' => 'keep-alive',
  377. 'X-Accel-Buffering' => 'no', // 禁用 Nginx 缓冲
  378. ]);
  379. }
  380. /**
  381. * 获取任务进度百分比
  382. */
  383. private function getTaskProgress($status) {
  384. switch ($status) {
  385. case \App\Models\MpGenerateVideoTask::STATUS_PENDING:
  386. return 10;
  387. case \App\Models\MpGenerateVideoTask::STATUS_PROCESSING:
  388. return 50;
  389. case \App\Models\MpGenerateVideoTask::STATUS_SUCCESS:
  390. return 100;
  391. case \App\Models\MpGenerateVideoTask::STATUS_FAILED:
  392. return 0;
  393. default:
  394. return 0;
  395. }
  396. }
  397. /**
  398. * 批量生成分镜视频
  399. */
  400. public function batchSetSegmentVideos(Request $request) {
  401. // 忽略所有超时限制
  402. set_time_limit(0);
  403. ini_set('max_execution_time', '0');
  404. $data = $request->all();
  405. // 验证参数
  406. $validator = Validator::make($data, [
  407. 'anime_id' => 'required|string',
  408. 'episode_number' => 'required|integer',
  409. ], [
  410. 'anime_id.required' => '动漫对话ID不能为空',
  411. 'episode_number.required' => '剧集序号不能为空',
  412. 'episode_number.integer' => '剧集序号必须是整数',
  413. ]);
  414. if ($validator->fails()) {
  415. Utils::throwError('1002:' . $validator->errors()->first());
  416. }
  417. $animeId = $data['anime_id'];
  418. $episodeId = $data['episode_id'];
  419. $episodeNumber = $data['episode_number'];
  420. // 获取所有分镜信息
  421. $segments = DB::table('mp_episode_segments')
  422. ->where('anime_id', $animeId)
  423. ->where('episode_id', $episodeId)
  424. ->where('video_url', '')
  425. ->orderBy('segment_number')
  426. ->limit(2)
  427. ->get();
  428. if ($segments->isEmpty()) {
  429. Utils::throwError('20003:未找到分镜数据');
  430. }
  431. // 批量创建视频任务
  432. $taskIds = [];
  433. $segmentTasks = [];
  434. foreach ($segments as $segment) {
  435. try {
  436. // 检查是否已有视频或正在生成中
  437. if (!empty($segment->video_url) || $segment->video_task_status === '生成中') {
  438. continue;
  439. }
  440. // 获取分镜内容
  441. $segmentContent = $segment->segment_content ?: '';
  442. $tailFrame = $segment->tail_frame ?: '';
  443. // 构建完整的提示词
  444. $fullPrompt = $segmentContent;
  445. if ($tailFrame) {
  446. $fullPrompt .= "\n尾帧描述:$tailFrame";
  447. }
  448. // 智能选择视频时长
  449. // $videoDuration = $this->AnimeService->calculateOptimalVideoDuration($segmentContent, $tailFrame);
  450. $videoDuration = -1;
  451. // 构建视频生成参数
  452. $videoParams = [
  453. 'model' => 'doubao-seedance-1-0-pro-250528',
  454. 'alias_segment_id' => $segment->segment_id,
  455. 'prompt' => $fullPrompt,
  456. 'video_duration' => $videoDuration,
  457. 'video_resolution' => '720P',
  458. 'seed' => -1,
  459. 'ratio' => '16:9',
  460. 'generate_audio' => false,
  461. 'draft' => false,
  462. 'watermark' => false,
  463. 'camera_fixed' => false,
  464. // 'callback_url' => 'http://mpaudio.yqsd.cn/api/video/seedanceCallback'
  465. ];
  466. // 如果分镜有图片,作为首帧
  467. if (!empty($segment->img_url)) {
  468. $videoParams['first_frame_url'] = $segment->img_url;
  469. }
  470. // 构建content数组
  471. $videoParams['content'] = [
  472. [
  473. 'type' => 'text',
  474. 'text' => $videoParams['prompt'],
  475. ]
  476. ];
  477. // 如果有首帧图片,添加到content中
  478. if (isset($videoParams['first_frame_url'])) {
  479. $videoParams['content'][] = [
  480. 'type' => 'image_url',
  481. 'image_url' => [
  482. 'url' => $videoParams['first_frame_url'],
  483. ],
  484. 'role' => 'first_frame',
  485. ];
  486. }
  487. // 创建视频任务
  488. $task = $this->AIVideoGenerationService->createSeedanceTask($videoParams);
  489. $taskIds[] = $task->id;
  490. // 更新分镜表的视频任务信息
  491. DB::table('mp_episode_segments')
  492. ->where('segment_id', $segment->segment_id)
  493. ->update([
  494. 'video_task_id' => $task->id,
  495. 'video_task_status' => '生成中',
  496. 'updated_at' => date('Y-m-d H:i:s')
  497. ]);
  498. $segmentTasks[] = [
  499. 'segment_id' => $segment->segment_id,
  500. 'task_id' => $task->id,
  501. 'segment_number' => $segment->segment_number,
  502. ];
  503. } catch (\Exception $e) {
  504. dLog('anime')->error('创建分镜视频任务失败: ' . $e->getMessage(), [
  505. 'segment_id' => $segment->segment_id,
  506. 'anime_id' => $animeId,
  507. 'episode_number' => $episodeNumber
  508. ]);
  509. continue;
  510. }
  511. }
  512. if (empty($taskIds)) {
  513. Utils::throwError('20003:没有需要生成视频的分镜');
  514. }
  515. // 记录开始时间(包含创建任务的时间)
  516. $startTime = time();
  517. $maxDuration = 1800; // 30分钟超时
  518. // 设置 SSE 响应头并开始长连接
  519. return response()->stream(function () use ($taskIds, $segmentTasks, $animeId, $episodeNumber, $episodeId, $startTime, $maxDuration) {
  520. // 发送初始任务创建信息
  521. echo "data: " . json_encode([
  522. 'type' => 'tasks_created',
  523. 'data' => [
  524. 'anime_id' => $animeId,
  525. 'episode_id' => $episodeId,
  526. 'episode_number' => $episodeNumber,
  527. 'total_tasks' => count($taskIds),
  528. 'task_ids' => $taskIds,
  529. 'segments' => $segmentTasks,
  530. 'start_time' => $startTime,
  531. 'max_duration' => $maxDuration
  532. ]
  533. ]) . "\n\n";
  534. ob_flush();
  535. flush();
  536. $checkInterval = 10; // 每10秒检查一次
  537. $completedTasks = [];
  538. $failedTasks = [];
  539. // 检查是否已经超时(包含创建任务的时间)
  540. while (time() - $startTime < $maxDuration) {
  541. try {
  542. $allCompleted = true;
  543. $currentStatus = [];
  544. foreach ($segmentTasks as $segmentTask) {
  545. $taskId = $segmentTask['task_id'];
  546. $segmentId = $segmentTask['segment_id'];
  547. // 跳过已完成或失败的任务
  548. if (in_array($taskId, $completedTasks) || in_array($taskId, $failedTasks)) {
  549. continue;
  550. }
  551. // 查询任务状态
  552. $task = \App\Models\MpGenerateVideoTask::find($taskId);
  553. if (!$task) {
  554. $failedTasks[] = $taskId;
  555. continue;
  556. }
  557. // 如果任务还在处理中,查询最新状态
  558. if ($task->status === 'processing') {
  559. $statusResult = $this->AIVideoGenerationService->querySeedanceTaskStatus($task);
  560. if (isset($statusResult['status'])) {
  561. // 更新任务状态
  562. $task->update([
  563. 'status' => $statusResult['status'],
  564. 'result_json' => $statusResult['result_json'] ?? [],
  565. 'result_url' => $statusResult['result_url'] ?? null,
  566. 'last_frame_url' => $statusResult['last_frame_url'] ?? '',
  567. 'error_message' => $statusResult['error_message'] ?? null,
  568. 'completed_at' => in_array($statusResult['status'], [
  569. 'success',
  570. 'failed'
  571. ]) ? now() : null
  572. ]);
  573. // 如果任务成功,更新分镜表
  574. if ($statusResult['status'] === 'success' && isset($statusResult['result_url'])) {
  575. DB::table('mp_episode_segments')
  576. ->where('segment_id', $segmentId)
  577. ->update([
  578. 'video_url' => $statusResult['result_url'],
  579. 'video_task_status' => '已完成',
  580. 'last_frame_url' => $statusResult['last_frame_url'] ?? '',
  581. 'updated_at' => date('Y-m-d H:i:s')
  582. ]);
  583. $completedTasks[] = $taskId;
  584. // 发送单个任务完成通知
  585. echo "data: " . json_encode([
  586. 'type' => 'task_completed',
  587. 'data' => [
  588. 'segment_id' => $segmentId,
  589. 'task_id' => $taskId,
  590. 'segment_number' => $segmentTask['segment_number'],
  591. 'status' => 'success',
  592. 'video_url' => $statusResult['result_url'],
  593. 'last_frame_url' => $statusResult['last_frame_url'] ?? '',
  594. 'completed_count' => count($completedTasks),
  595. 'total_count' => count($segmentTasks)
  596. ]
  597. ]) . "\n\n";
  598. ob_flush();
  599. flush();
  600. } elseif ($statusResult['status'] === 'failed') {
  601. DB::table('mp_episode_segments')
  602. ->where('segment_id', $segmentId)
  603. ->update([
  604. 'video_task_status' => '失败',
  605. 'updated_at' => date('Y-m-d H:i:s')
  606. ]);
  607. $failedTasks[] = $taskId;
  608. // 发送任务失败通知
  609. echo "data: " . json_encode([
  610. 'type' => 'task_failed',
  611. 'data' => [
  612. 'segment_id' => $segmentId,
  613. 'task_id' => $taskId,
  614. 'segment_number' => $segmentTask['segment_number'],
  615. 'status' => 'failed',
  616. 'error_message' => $statusResult['error_message'] ?? '视频生成失败',
  617. 'failed_count' => count($failedTasks),
  618. 'total_count' => count($segmentTasks)
  619. ]
  620. ]) . "\n\n";
  621. ob_flush();
  622. flush();
  623. }
  624. }
  625. }
  626. // 收集当前状态
  627. $currentStatus[] = [
  628. 'segment_id' => $segmentId,
  629. 'task_id' => $taskId,
  630. 'segment_number' => $segmentTask['segment_number'],
  631. 'status' => $task->status,
  632. 'progress' => $this->getTaskProgress($task->status),
  633. 'video_url' => $task->result_url,
  634. 'error_message' => $task->error_message
  635. ];
  636. // 检查是否还有未完成的任务
  637. if (!in_array($task->status, ['success', 'failed'])) {
  638. $allCompleted = false;
  639. }
  640. }
  641. // 发送整体进度更新
  642. echo "data: " . json_encode([
  643. 'type' => 'progress_update',
  644. 'data' => [
  645. 'anime_id' => $animeId,
  646. 'episode_id' => $episodeId,
  647. 'episode_number' => $episodeNumber,
  648. 'completed_count' => count($completedTasks),
  649. 'failed_count' => count($failedTasks),
  650. 'total_count' => count($segmentTasks),
  651. 'elapsed_time' => time() - $startTime,
  652. 'remaining_time' => max(0, $maxDuration - (time() - $startTime)),
  653. 'segments_status' => $currentStatus
  654. ]
  655. ]) . "\n\n";
  656. ob_flush();
  657. flush();
  658. // 如果所有任务都完成了,结束连接
  659. if ($allCompleted) {
  660. // 获取最终结果
  661. $finalSegments = DB::table('mp_episode_segments')
  662. ->where('anime_id', $animeId)
  663. ->where('episode_id', $episodeId)
  664. ->orderBy('segment_number')
  665. ->select('segment_id', 'segment_number', 'video_url', 'video_task_status', 'last_frame_url')
  666. ->get();
  667. echo "data: " . json_encode([
  668. 'type' => 'all_completed',
  669. 'data' => [
  670. 'anime_id' => $animeId,
  671. 'episode_id' => $episodeId,
  672. 'episode_number' => $episodeNumber,
  673. 'completed_count' => count($completedTasks),
  674. 'failed_count' => count($failedTasks),
  675. 'total_count' => count($segmentTasks),
  676. 'total_elapsed_time' => time() - $startTime,
  677. 'segments' => $finalSegments->map(function($segment) {
  678. return [
  679. 'segment_id' => $segment->segment_id,
  680. 'segment_number' => $segment->segment_number,
  681. 'video_url' => $segment->video_url,
  682. 'status' => $segment->video_task_status,
  683. 'last_frame_url' => $segment->last_frame_url
  684. ];
  685. })->toArray()
  686. ]
  687. ]) . "\n\n";
  688. ob_flush();
  689. flush();
  690. break;
  691. }
  692. // 在睡眠前再次检查是否超时,避免不必要的等待
  693. if (time() - $startTime >= $maxDuration) {
  694. break;
  695. }
  696. sleep($checkInterval);
  697. } catch (\Exception $e) {
  698. echo "data: " . json_encode([
  699. 'type' => 'error',
  700. 'message' => '查询任务状态失败: ' . $e->getMessage(),
  701. 'elapsed_time' => time() - $startTime,
  702. 'remaining_time' => max(0, $maxDuration - (time() - $startTime))
  703. ]) . "\n\n";
  704. ob_flush();
  705. flush();
  706. // 检查是否超时,如果超时则停止循环
  707. if (time() - $startTime >= $maxDuration) {
  708. break;
  709. }
  710. sleep($checkInterval);
  711. }
  712. }
  713. // 超时处理 - 只有在真正超时时才执行
  714. if (time() - $startTime >= $maxDuration) {
  715. // 获取当前所有任务状态
  716. $timeoutSegments = DB::table('mp_episode_segments')
  717. ->where('anime_id', $animeId)
  718. ->where('episode_id', $episodeId)
  719. ->orderBy('segment_number')
  720. ->select('segment_id', 'segment_number', 'video_task_id', 'video_task_status', 'video_url')
  721. ->get();
  722. echo "data: " . json_encode([
  723. 'type' => 'timeout',
  724. 'message' => '批量视频生成已超时30分钟,停止状态查询',
  725. 'data' => [
  726. 'anime_id' => $animeId,
  727. 'episode_id' => $episodeId,
  728. 'episode_number' => $episodeNumber,
  729. 'completed_count' => count($completedTasks),
  730. 'failed_count' => count($failedTasks),
  731. 'total_count' => count($segmentTasks),
  732. 'total_elapsed_time' => time() - $startTime,
  733. 'timeout_duration' => $maxDuration,
  734. 'segments' => $timeoutSegments->map(function($segment) {
  735. return [
  736. 'segment_id' => $segment->segment_id,
  737. 'segment_number' => $segment->segment_number,
  738. 'task_id' => $segment->video_task_id,
  739. 'status' => $segment->video_task_status,
  740. 'video_url' => $segment->video_url
  741. ];
  742. })->toArray()
  743. ]
  744. ]) . "\n\n";
  745. ob_flush();
  746. flush();
  747. }
  748. }, 200, [
  749. 'Content-Type' => 'text/event-stream',
  750. 'Cache-Control' => 'no-cache',
  751. 'Connection' => 'keep-alive',
  752. 'X-Accel-Buffering' => 'no', // 禁用 Nginx 缓冲
  753. ]);
  754. }
  755. // public function scriptList(Request $request) {
  756. // $data = $request->all();
  757. // $result = $this->AnimeService->scriptList($data);
  758. // return $this->success($result, [new AnimeTransformer(), 'newBuildScriptList']);
  759. // }
  760. }