AbstractExportTask.php 2.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889
  1. <?php
  2. namespace General\Services\Export;
  3. use General\Helpers\AliOSSHelper;
  4. use Illuminate\Database\Eloquent\Model;
  5. /**
  6. * 导出任务
  7. * $model 对象必须有name,query_condition,status,file_path,is_file_deleted 逻辑不能变
  8. */
  9. abstract class AbstractExportTask
  10. {
  11. private $model;
  12. private $query_data;
  13. const ready = 1;
  14. const running = 2;
  15. const complete = 3;
  16. const failure = 4;
  17. public function __construct(Model $model, ExportQueryData $query_data = null)
  18. {
  19. $this->model = $model;
  20. $this->query_data = $query_data;
  21. }
  22. abstract public function getTaskData(): array;
  23. /**
  24. * 创建导出任务
  25. */
  26. public function addTask(array $query_params)
  27. {
  28. if ($this->query_data) {
  29. $params = array_merge($this->getTaskData(), [
  30. 'name' => $this->query_data->memu_name,
  31. 'query_params' => json_encode($query_params, JSON_UNESCAPED_UNICODE),
  32. 'query_condition' => $this->query_data->getQueryParamsContent($query_params),
  33. 'status' => self::ready,
  34. ]);
  35. $this->setModel($this->model::create($params));
  36. }
  37. }
  38. public function setModel(Model $model)
  39. {
  40. $this->model = $model;
  41. }
  42. public function getModel(): Model
  43. {
  44. return $this->model;
  45. }
  46. /**
  47. * 设置状态
  48. */
  49. public function setStatus(int $status)
  50. {
  51. $this->model->status = $status;
  52. $this->model->save();
  53. }
  54. /**
  55. * 保存文件
  56. */
  57. public function saveExportTaskFile(string $local_path, string $file_name)
  58. {
  59. $service = new AliOSSHelper;
  60. $oss_path = $service->uploadFile("reports/" . $file_name, $local_path . '/' . $file_name);
  61. if ($oss_path) {
  62. $this->model->file_path = "reports/" . $file_name;
  63. $this->setStatus(self::complete);
  64. } else {
  65. $this->task->setStatus(self::failure);
  66. }
  67. }
  68. /**
  69. * 删除文件
  70. */
  71. public function delExportTaskFile()
  72. {
  73. $service = new AliOSSHelper;
  74. $service->delFile($this->model->file_path);
  75. $this->model->is_file_deleted = 1;
  76. $this->model->save();
  77. }
  78. }