MpGeneratePicTask.php 2.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586
  1. <?php
  2. namespace App\Models;
  3. use Illuminate\Database\Eloquent\Factories\HasFactory;
  4. use Illuminate\Database\Eloquent\Model;
  5. class MpGeneratePicTask extends Model
  6. {
  7. use HasFactory;
  8. protected $fillable = [
  9. 'task_id',
  10. 'alias_segment_id',
  11. 'prompt',
  12. 'width',
  13. 'height',
  14. 'image_num',
  15. 'scale',
  16. 'ref_img_url',
  17. 'mask_img_url',
  18. 'extra_params',
  19. 'status',
  20. 'result_url',
  21. 'error_message',
  22. 'started_at',
  23. 'completed_at',
  24. 'created_at',
  25. 'updated_at',
  26. 'result_json',
  27. 'model'
  28. ];
  29. protected $casts = [
  30. 'result_url' => 'array',
  31. 'ref_img_url' => 'array',
  32. 'mask_img_url' => 'array',
  33. 'extra_params' => 'array',
  34. 'result_json' => 'array',
  35. 'started_at' => 'datetime',
  36. 'completed_at' => 'datetime',
  37. ];
  38. const STATUS_PENDING = 'pending';
  39. const STATUS_PROCESSING = 'processing';
  40. const STATUS_SUCCESS = 'success';
  41. const STATUS_FAILED = 'failed';
  42. /**
  43. * 更新任务状态
  44. *
  45. * @param string $status
  46. * @param array $data
  47. * @return bool
  48. */
  49. public function updateStatus(string $status, array $data = [])
  50. {
  51. $updateData = ['status' => $status];
  52. if (!$this->created_at) {
  53. $updateData['created_at'] = now();
  54. }
  55. $updateData['updated_at'] = now();
  56. if (isset($data['extra_params'])) $updateData['extra_params'] = $data['extra_params'];
  57. if (isset($data['result_json'])) $updateData['result_json'] = $data['result_json'];
  58. if ($status === self::STATUS_PROCESSING) {
  59. if (isset($data['task_id'])) $updateData['task_id'] = $data['task_id'];
  60. if (!$this->started_at) $updateData['started_at'] = now();
  61. }
  62. if (in_array($status, [self::STATUS_SUCCESS, self::STATUS_FAILED])) {
  63. $updateData['completed_at'] = now();
  64. if (isset($data['result_url'])) {
  65. $updateData['result_url'] = $data['result_url'];
  66. }
  67. if (isset($data['error_message'])) {
  68. $updateData['error_message'] = $data['error_message'];
  69. }
  70. }
  71. return $this->update($updateData);
  72. }
  73. }