ProcessBatchEpisodeGenerationCommand.php 11 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261
  1. <?php
  2. namespace App\Console\Commands;
  3. use Illuminate\Console\Command;
  4. use Illuminate\Support\Facades\DB;
  5. use App\Services\DeepSeek\DeepSeekService;
  6. use App\Facade\Site;
  7. /**
  8. * 批量生成分集定时任务
  9. * 每次执行处理一个待生成的剧集
  10. */
  11. class ProcessBatchEpisodeGenerationCommand extends Command
  12. {
  13. /**
  14. * The name and signature of the console command.
  15. *
  16. * @var string
  17. */
  18. protected $signature = 'batch:generate-episodes';
  19. /**
  20. * The console command description.
  21. *
  22. * @var string
  23. */
  24. protected $description = '批量生成分集定时任务,每5秒检查一次待处理任务,单次最多执行50秒';
  25. protected $deepSeekService;
  26. /**
  27. * Create a new command instance.
  28. *
  29. * @return void
  30. */
  31. public function __construct(DeepSeekService $deepSeekService)
  32. {
  33. parent::__construct();
  34. $this->deepSeekService = $deepSeekService;
  35. }
  36. /**
  37. * Execute the console command.
  38. *
  39. * @return int
  40. */
  41. public function handle()
  42. {
  43. dLog('command')->info('开始执行批量生成分集任务...');
  44. // 每5秒执行一次任务,直到50秒后停止(参考 CheckVideoGenerationTasksCommand 的循环逻辑)
  45. $time_start = time();
  46. while (true) {
  47. $time_diff = time() - $time_start;
  48. sleep(5);
  49. if ($time_diff > 50) {
  50. break;
  51. }
  52. // 处理一个待生成的剧集;返回 1 表示处理失败,结束本次执行
  53. $result = $this->processPendingEpisodeTask();
  54. if ($result === -1) break;
  55. }
  56. return 0;
  57. }
  58. /**
  59. * 处理一个待生成的剧集
  60. *
  61. * @return int 0-成功或无任务可处理(可继续下一轮),1-处理失败
  62. */
  63. private function processPendingEpisodeTask()
  64. {
  65. $anime_tasks = DB::table('mp_batch_episode_generation_details')->where('status', 'pending')->pluck('anime_id')->toArray();
  66. if (!$anime_tasks) return -1;
  67. foreach ($anime_tasks as $anime_id) {
  68. dLog('command')->info("~~~~~~开始执行($anime_id)任务~~~~~~");
  69. try {
  70. // 查找状态为 pending 的任务,按 anime_id 和 episode_number 排序
  71. $task = DB::table('mp_batch_episode_generation_details')
  72. ->where('status', 'pending')
  73. ->orderBy('anime_id')
  74. ->orderBy('episode_number')
  75. ->first();
  76. if (!$task) {
  77. dLog('command')->info('[' . date('Y-m-d H:i:s') . '] 没有待处理的任务');
  78. continue;
  79. }
  80. dLog('command')->info('[' . date('Y-m-d H:i:s') . '] 找到待处理任务 - Anime ID: ' . $task->anime_id . ', Episode: ' . $task->episode_number);
  81. // 检查前一集是否已完成(如果不是第1集)
  82. if ($task->episode_number > 1) {
  83. $prevEpisodeNumber = $task->episode_number - 1;
  84. // 检查前一集是否在批量任务中
  85. $prevTaskInBatch = DB::table('mp_batch_episode_generation_details')
  86. ->where('anime_id', $task->anime_id)
  87. ->where('episode_number', $prevEpisodeNumber)
  88. ->first();
  89. if ($prevTaskInBatch && $prevTaskInBatch->status !== 'completed') {
  90. dLog('command')->info('[' . date('Y-m-d H:i:s') . '] 前一集(第' . $prevEpisodeNumber . '集)尚未完成,跳过当前任务 (anime_id: ' . $task->anime_id . ')');
  91. continue;
  92. }
  93. // 检查前一集是否已在系统中生成(查询 mp_anime_episodes)
  94. $prevEpisodeExists = DB::table('mp_anime_episodes')
  95. ->where('anime_id', $task->anime_id)
  96. ->where('episode_number', $prevEpisodeNumber)
  97. ->where('is_default', 1)
  98. ->exists();
  99. if (!$prevEpisodeExists) {
  100. dLog('command')->error('[' . date('Y-m-d H:i:s') . '] 前一集(第' . $prevEpisodeNumber . '集)不存在,无法继续生成 (anime_id: ' . $task->anime_id . ')');
  101. // 标记为失败
  102. DB::table('mp_batch_episode_generation_details')
  103. ->where('id', $task->id)
  104. ->update([
  105. 'status' => 'failed',
  106. 'error_message' => "前一集(第{$prevEpisodeNumber}集)不存在,无法继续生成",
  107. 'updated_at' => now()
  108. ]);
  109. continue;
  110. }
  111. }
  112. // 检查当前集是否已经存在(避免重复生成)
  113. $currentEpisodeExists = DB::table('mp_anime_episodes')
  114. ->where('anime_id', $task->anime_id)
  115. ->where('episode_number', $task->episode_number)
  116. ->where('is_default', 1)
  117. ->exists();
  118. if ($currentEpisodeExists) {
  119. dLog('command')->info('[' . date('Y-m-d H:i:s') . '] 第' . $task->episode_number . '集已存在,跳过生成 (anime_id: ' . $task->anime_id . ')');
  120. // 标记为已完成
  121. DB::table('mp_batch_episode_generation_details')
  122. ->where('id', $task->id)
  123. ->update([
  124. 'status' => 'completed',
  125. 'completed_at' => now(),
  126. // 'updated_at' => now()
  127. ]);
  128. continue;
  129. }
  130. // 标记为处理中
  131. DB::table('mp_batch_episode_generation_details')
  132. ->where('id', $task->id)
  133. ->update([
  134. 'status' => 'processing',
  135. 'updated_at' => now()
  136. ]);
  137. // 设置用户上下文(绑定到容器)
  138. app()->instance('siteData', [
  139. 'uid' => $task->uid,
  140. 'cpid' => $task->cpid
  141. ]);
  142. // 解析请求数据
  143. $requestData = json_decode($task->request_data, true);
  144. $requestData['episode_number'] = $task->episode_number;
  145. // 使用"继续策划下一集"逻辑
  146. $requestData['prompt'] = '继续策划下一集';
  147. dLog('command')->info('[' . date('Y-m-d H:i:s') . '] 开始生成第' . $task->episode_number . '集... (anime_id: ' . $task->anime_id . ')');
  148. // 调用非流式生成方法,并记录实际执行耗时
  149. $chatStartTime = microtime(true);
  150. try {
  151. $result = $this->deepSeekService->chatForAceNonStream($requestData);
  152. } finally {
  153. $chatDuration = round(microtime(true) - $chatStartTime, 2);
  154. dLog('command')->info('[' . date('Y-m-d H:i:s') . '] chatForAceNonStream 执行完成,耗时: ' . $chatDuration . ' 秒 (anime_id: ' . $task->anime_id . ', 第' . $task->episode_number . '集)');
  155. }
  156. // 检查是否有错误
  157. if (isset($result['error']) && $result['error']) {
  158. dLog('command')->error('[' . date('Y-m-d H:i:s') . '] 生成失败: ' . $result['error'] . ' (anime_id: ' . $task->anime_id . ', 第' . $task->episode_number . '集)');
  159. // 更新重试次数
  160. $retryCount = $task->retry_count + 1;
  161. // 标记为失败
  162. DB::table('mp_batch_episode_generation_details')
  163. ->where('id', $task->id)
  164. ->update([
  165. 'status' => 'pending',
  166. 'error_message' => $result['error'],
  167. 'retry_count' => $retryCount,
  168. 'updated_at' => now()
  169. ]);
  170. // 记录错误日志
  171. logDB('batch_episode_generation', 'error', "anime_id: {$task->anime_id} 第{$task->episode_number}集生成失败", [
  172. 'anime_id' => $task->anime_id,
  173. 'episode_number' => $task->episode_number,
  174. 'error' => $result['error']
  175. ]);
  176. continue;
  177. }
  178. // 标记为完成
  179. DB::table('mp_batch_episode_generation_details')
  180. ->where('id', $task->id)
  181. ->update([
  182. 'status' => 'completed',
  183. 'result_data' => json_encode($result, JSON_UNESCAPED_UNICODE),
  184. 'completed_at' => now(),
  185. // 'updated_at' => now()
  186. ]);
  187. dLog('command')->info('[' . date('Y-m-d H:i:s') . '] 第' . $task->episode_number . '集生成成功 (anime_id: ' . $task->anime_id . ')');
  188. continue;
  189. } catch (\Exception $e) {
  190. $animeLogInfo = isset($task) ? ' (anime_id: ' . $task->anime_id . ', 第' . $task->episode_number . '集)' : '';
  191. dLog('command')->error('[' . date('Y-m-d H:i:s') . '] 生成失败: ' . $e->getMessage() . $animeLogInfo);
  192. if (isset($task)) {
  193. // 更新重试次数
  194. $retryCount = $task->retry_count + 1;
  195. // 标记为失败
  196. DB::table('mp_batch_episode_generation_details')
  197. ->where('id', $task->id)
  198. ->update([
  199. 'status' => 'pending',
  200. 'error_message' => $e->getMessage(),
  201. 'retry_count' => $retryCount,
  202. 'updated_at' => now()
  203. ]);
  204. // 记录错误日志
  205. logDB('batch_episode_generation', 'error', "anime_id: {$task->anime_id} 第{$task->episode_number}集生成失败", [
  206. 'anime_id' => $task->anime_id,
  207. 'episode_number' => $task->episode_number,
  208. 'error' => $e->getMessage(),
  209. 'trace' => $e->getTraceAsString()
  210. ]);
  211. }
  212. continue;
  213. }
  214. }
  215. return 0;
  216. }
  217. }