ProcessBatchEpisodeGenerationCommand.php 9.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252
  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. }
  55. return 0;
  56. }
  57. /**
  58. * 处理一个待生成的剧集
  59. *
  60. * @return int 0-成功或无任务可处理(可继续下一轮),1-处理失败
  61. */
  62. private function processPendingEpisodeTask()
  63. {
  64. $anime_tasks = DB::table('mp_batch_episode_generation_details')->where('status', 'pending')->pluck('anime_id')->toArray();
  65. foreach ($anime_tasks as $anime_id) {
  66. dLog('command')->info("~~~~~~开始执行($anime_id)任务~~~~~~");
  67. try {
  68. // 查找状态为 pending 的任务,按 anime_id 和 episode_number 排序
  69. $task = DB::table('mp_batch_episode_generation_details')
  70. ->where('status', 'pending')
  71. ->orderBy('anime_id')
  72. ->orderBy('episode_number')
  73. ->first();
  74. if (!$task) {
  75. dLog('command')->info('[' . date('Y-m-d H:i:s') . '] 没有待处理的任务');
  76. return 0;
  77. }
  78. dLog('command')->info('[' . date('Y-m-d H:i:s') . '] 找到待处理任务 - Anime ID: ' . $task->anime_id . ', Episode: ' . $task->episode_number);
  79. // 检查前一集是否已完成(如果不是第1集)
  80. if ($task->episode_number > 1) {
  81. $prevEpisodeNumber = $task->episode_number - 1;
  82. // 检查前一集是否在批量任务中
  83. $prevTaskInBatch = DB::table('mp_batch_episode_generation_details')
  84. ->where('anime_id', $task->anime_id)
  85. ->where('episode_number', $prevEpisodeNumber)
  86. ->first();
  87. if ($prevTaskInBatch && $prevTaskInBatch->status !== 'completed') {
  88. dLog('command')->info('[' . date('Y-m-d H:i:s') . '] 前一集(第' . $prevEpisodeNumber . '集)尚未完成,跳过当前任务');
  89. return 0;
  90. }
  91. // 检查前一集是否已在系统中生成(查询 mp_anime_episodes)
  92. $prevEpisodeExists = DB::table('mp_anime_episodes')
  93. ->where('anime_id', $task->anime_id)
  94. ->where('episode_number', $prevEpisodeNumber)
  95. ->where('is_default', 1)
  96. ->exists();
  97. if (!$prevEpisodeExists) {
  98. dLog('command')->error('[' . date('Y-m-d H:i:s') . '] 前一集(第' . $prevEpisodeNumber . '集)不存在,无法继续生成');
  99. // 标记为失败
  100. DB::table('mp_batch_episode_generation_details')
  101. ->where('id', $task->id)
  102. ->update([
  103. 'status' => 'failed',
  104. 'error_message' => "前一集(第{$prevEpisodeNumber}集)不存在,无法继续生成",
  105. 'updated_at' => now()
  106. ]);
  107. return 1;
  108. }
  109. }
  110. // 检查当前集是否已经存在(避免重复生成)
  111. $currentEpisodeExists = DB::table('mp_anime_episodes')
  112. ->where('anime_id', $task->anime_id)
  113. ->where('episode_number', $task->episode_number)
  114. ->where('is_default', 1)
  115. ->exists();
  116. if ($currentEpisodeExists) {
  117. dLog('command')->info('[' . date('Y-m-d H:i:s') . '] 第' . $task->episode_number . '集已存在,跳过生成');
  118. // 标记为已完成
  119. DB::table('mp_batch_episode_generation_details')
  120. ->where('id', $task->id)
  121. ->update([
  122. 'status' => 'completed',
  123. 'completed_at' => now(),
  124. // 'updated_at' => now()
  125. ]);
  126. return 0;
  127. }
  128. // 标记为处理中
  129. DB::table('mp_batch_episode_generation_details')
  130. ->where('id', $task->id)
  131. ->update([
  132. 'status' => 'processing',
  133. 'updated_at' => now()
  134. ]);
  135. // 设置用户上下文(绑定到容器)
  136. app()->instance('siteData', [
  137. 'uid' => $task->uid,
  138. 'cpid' => $task->cpid
  139. ]);
  140. // 解析请求数据
  141. $requestData = json_decode($task->request_data, true);
  142. $requestData['episode_number'] = $task->episode_number;
  143. // 使用"继续策划下一集"逻辑
  144. $requestData['prompt'] = '继续策划下一集';
  145. dLog('command')->info('[' . date('Y-m-d H:i:s') . '] 开始生成第' . $task->episode_number . '集...');
  146. // 调用非流式生成方法
  147. $result = $this->deepSeekService->chatForAceNonStream($requestData);
  148. // 检查是否有错误
  149. if (isset($result['error']) && $result['error']) {
  150. dLog('command')->error('[' . date('Y-m-d H:i:s') . '] 生成失败: ' . $result['error']);
  151. // 更新重试次数
  152. $retryCount = $task->retry_count + 1;
  153. // 标记为失败
  154. DB::table('mp_batch_episode_generation_details')
  155. ->where('id', $task->id)
  156. ->update([
  157. 'status' => 'pending',
  158. 'error_message' => $result['error'],
  159. 'retry_count' => $retryCount,
  160. 'updated_at' => now()
  161. ]);
  162. // 记录错误日志
  163. logDB('batch_episode_generation', 'error', "第{$task->episode_number}集生成失败", [
  164. 'anime_id' => $task->anime_id,
  165. 'episode_number' => $task->episode_number,
  166. 'error' => $result['error']
  167. ]);
  168. return 1;
  169. }
  170. // 标记为完成
  171. DB::table('mp_batch_episode_generation_details')
  172. ->where('id', $task->id)
  173. ->update([
  174. 'status' => 'completed',
  175. 'result_data' => json_encode($result, JSON_UNESCAPED_UNICODE),
  176. 'completed_at' => now(),
  177. // 'updated_at' => now()
  178. ]);
  179. dLog('command')->info('[' . date('Y-m-d H:i:s') . '] 第' . $task->episode_number . '集生成成功');
  180. return 0;
  181. } catch (\Exception $e) {
  182. dLog('command')->error('[' . date('Y-m-d H:i:s') . '] 生成失败: ' . $e->getMessage());
  183. if (isset($task)) {
  184. // 更新重试次数
  185. $retryCount = $task->retry_count + 1;
  186. // 标记为失败
  187. DB::table('mp_batch_episode_generation_details')
  188. ->where('id', $task->id)
  189. ->update([
  190. 'status' => 'pending',
  191. 'error_message' => $e->getMessage(),
  192. 'retry_count' => $retryCount,
  193. 'updated_at' => now()
  194. ]);
  195. // 记录错误日志
  196. logDB('batch_episode_generation', 'error', "第{$task->episode_number}集生成失败", [
  197. 'anime_id' => $task->anime_id,
  198. 'episode_number' => $task->episode_number,
  199. 'error' => $e->getMessage(),
  200. 'trace' => $e->getTraceAsString()
  201. ]);
  202. }
  203. return 1;
  204. }
  205. }
  206. return 0;
  207. }
  208. }