tao-test/app/taoTaskQueue/model/Service/RestartStuckTaskService.php

84 lines
2.7 KiB
PHP

<?php
/**
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU General Public License
* as published by the Free Software Foundation; under version 2
* of the License (non-upgradable).
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
*
* Copyright (c) 2021 (original work) Open Assessment Technologies SA;
*/
declare(strict_types=1);
namespace oat\taoTaskQueue\model\Service;
use InvalidArgumentException;
use oat\oatbox\service\ConfigurableService;
use oat\tao\model\taskQueue\QueueDispatcherInterface;
use oat\tao\model\taskQueue\TaskLogInterface;
use oat\taoTaskQueue\model\QueueBroker\RdsQueueBroker;
use oat\taoTaskQueue\model\StuckTask;
class RestartStuckTaskService extends ConfigurableService
{
public function restart(StuckTask $stuckTask): void
{
$taskLogEntity = $stuckTask->getTaskLog();
$broker = $this->getQueueDispatcher()
->getQueue($stuckTask->getQueueName())
->getBroker();
if (!$broker instanceof RdsQueueBroker) {
throw new InvalidArgumentException(
sprintf(
'Broker %s for queue %s is not supported. Supported only %s',
$broker->getBrokerId(),
$stuckTask->getQueueName(),
RdsQueueBroker::class
)
);
}
if ($stuckTask->isOrphan()) {
$callback = $taskLogEntity->getTaskName();
$this->getTaskLog()->getBroker()->updateStatus(
$taskLogEntity->getId(),
TaskLogInterface::STATUS_CANCELLED
);
$this->getQueueDispatcher()->createTask(
new $callback(),
$taskLogEntity->getParameters(),
$taskLogEntity->getLabel()
);
return;
}
$broker->changeTaskVisibility($stuckTask->getTaskId(), true);
$this->getTaskLog()->setStatus($stuckTask->getTaskId(), TaskLogInterface::STATUS_ENQUEUED);
}
private function getTaskLog(): TaskLogInterface
{
return $this->getServiceLocator()->get(TaskLogInterface::SERVICE_ID);
}
private function getQueueDispatcher(): QueueDispatcherInterface
{
return $this->getServiceLocator()->get(QueueDispatcherInterface::SERVICE_ID);
}
}