TaskBackgroundJob.php 1.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263
  1. <?php
  2. declare(strict_types=1);
  3. /**
  4. * @copyright Copyright (c) 2023 Marcel Klehr <mklehr@gmx.net>
  5. *
  6. * @author Marcel Klehr <mklehr@gmx.net>
  7. *
  8. * @license GNU AGPL version 3 or any later version
  9. *
  10. * This program is free software: you can redistribute it and/or modify
  11. * it under the terms of the GNU Affero General Public License as
  12. * published by the Free Software Foundation, either version 3 of the
  13. * License, or (at your option) any later version.
  14. *
  15. * This program is distributed in the hope that it will be useful,
  16. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  18. * GNU Affero General Public License for more details.
  19. *
  20. * You should have received a copy of the GNU Affero General Public License
  21. * along with this program. If not, see <http://www.gnu.org/licenses/>.
  22. */
  23. namespace OC\TextToImage;
  24. use OCP\AppFramework\Utility\ITimeFactory;
  25. use OCP\BackgroundJob\QueuedJob;
  26. use OCP\EventDispatcher\IEventDispatcher;
  27. use OCP\TextToImage\Events\TaskFailedEvent;
  28. use OCP\TextToImage\Events\TaskSuccessfulEvent;
  29. use OCP\TextToImage\IManager;
  30. class TaskBackgroundJob extends QueuedJob {
  31. public function __construct(
  32. ITimeFactory $timeFactory,
  33. private IManager $text2imageManager,
  34. private IEventDispatcher $eventDispatcher,
  35. ) {
  36. parent::__construct($timeFactory);
  37. // We want to avoid overloading the machine with these jobs
  38. // so we only allow running one job at a time
  39. $this->setAllowParallelRuns(false);
  40. }
  41. /**
  42. * @param array{taskId: int} $argument
  43. * @inheritDoc
  44. */
  45. protected function run($argument) {
  46. $taskId = $argument['taskId'];
  47. $task = $this->text2imageManager->getTask($taskId);
  48. try {
  49. $this->text2imageManager->runTask($task);
  50. $event = new TaskSuccessfulEvent($task);
  51. } catch (\Throwable $e) {
  52. $event = new TaskFailedEvent($task, $e->getMessage());
  53. }
  54. $this->eventDispatcher->dispatchTyped($event);
  55. }
  56. }