From ee76cc512ccb9dcc968345e821f50f65454fbaae Mon Sep 17 00:00:00 2001 From: Marcel Klehr Date: Thu, 2 Nov 2023 12:07:10 +0100 Subject: fix(TextToImage): Always set completionExpectedAt in run/schedule methods Signed-off-by: Marcel Klehr --- lib/private/TextToImage/Manager.php | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'lib') diff --git a/lib/private/TextToImage/Manager.php b/lib/private/TextToImage/Manager.php index a1f0f97b0c5..f68f657277f 100644 --- a/lib/private/TextToImage/Manager.php +++ b/lib/private/TextToImage/Manager.php @@ -119,6 +119,9 @@ class Manager implements IManager { $this->logger->debug('Trying to run Text2Image provider '.$provider::class); try { $task->setStatus(Task::STATUS_RUNNING); + $completionExpectedAt = new \DateTime('now'); + $completionExpectedAt->add(new \DateInterval('PT'.$provider->getExpectedRuntime().'S')); + $task->setCompletionExpectedAt($completionExpectedAt); if ($task->getId() === null) { $this->logger->debug('Inserting Text2Image task into DB'); $taskEntity = $this->taskMapper->insert(DbTask::fromPublicTask($task)); @@ -204,6 +207,9 @@ class Manager implements IManager { } $this->logger->debug('Scheduling Text2Image Task'); $task->setStatus(Task::STATUS_SCHEDULED); + $completionExpectedAt = new \DateTime('now'); + $completionExpectedAt->add(new \DateInterval('PT'.$this->getPreferredProviders()[0]->getExpectedRuntime().'S')); + $task->setCompletionExpectedAt($completionExpectedAt); $taskEntity = DbTask::fromPublicTask($task); $this->taskMapper->insert($taskEntity); $task->setId($taskEntity->getId()); -- cgit v1.2.3