MpGenerateVideoTask.php 2.6 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192
  1. <?php
  2. namespace App\Models;
  3. use Illuminate\Database\Eloquent\Model;
  4. class MpGenerateVideoTask extends Model
  5. {
  6. protected $table = 'mp_generate_video_tasks';
  7. const STATUS_PENDING = 'pending';
  8. const STATUS_PROCESSING = 'processing';
  9. const STATUS_SUCCESS = 'success';
  10. const STATUS_FAILED = 'failed';
  11. protected $fillable = [
  12. 'id',
  13. 'task_id',
  14. 'prompt',
  15. 'first_frame_url',
  16. 'tail_frame_url',
  17. 'ref_image_url',
  18. 'seed',
  19. 'frames',
  20. 'status',
  21. 'result_url',
  22. 'compressed_url',
  23. 'error_message',
  24. 'extra_params',
  25. 'started_at',
  26. 'completed_at',
  27. 'video_resolution',
  28. 'result_json',
  29. 'api_type', // 新增:API类型标识(seedance_1_5_pro, jimeng_ai等)
  30. 'last_frame_url', // 新增:尾帧图片URL(用于return_last_frame功能)
  31. 'alias_segment_id', // 新增:关联分镜ID
  32. 'alias_act_id', // 新增: 关联片段ID
  33. 'charge_info', // 新增: 计费关联数据
  34. ];
  35. protected $casts = [
  36. 'extra_params' => 'array',
  37. 'result_json' => 'array',
  38. 'charge_info' => 'array',
  39. ];
  40. /**
  41. * 更新任务状态
  42. * @param string $status
  43. * @param array $data
  44. * @return void
  45. */
  46. public function updateStatus(string $status, array $data = []): void
  47. {
  48. $updateData = ['status' => $status];
  49. if (isset($data['result_json'])) {
  50. $updateData['result_json'] = $data['result_json'];
  51. }
  52. if (isset($data['result_url'])) {
  53. $updateData['result_url'] = $data['result_url'];
  54. }
  55. if (isset($data['compressed_url'])) {
  56. $updateData['compressed_url'] = $data['compressed_url'];
  57. }
  58. if (isset($data['last_frame_url'])) {
  59. $updateData['last_frame_url'] = $data['last_frame_url'];
  60. }
  61. if (isset($data['error_message'])) {
  62. $updateData['error_message'] = $data['error_message'];
  63. }
  64. if (isset($data['extra_params'])) {
  65. $updateData['extra_params'] = $data['extra_params'];
  66. }
  67. if ($status === self::STATUS_PROCESSING) {
  68. if (isset($data['task_id'])) $updateData['task_id'] = $data['task_id'];
  69. if (!$this->started_at) $updateData['started_at'] = now();
  70. }
  71. if (in_array($status, [self::STATUS_SUCCESS, self::STATUS_FAILED]) && !$this->completed_at) {
  72. $updateData['completed_at'] = now();
  73. }
  74. $this->update($updateData);
  75. }
  76. }