2024-06-17 12:44:12 +00:00
|
|
|
<?php
|
|
|
|
|
|
|
|
|
|
namespace Appwrite\Platform\Tasks;
|
|
|
|
|
|
|
|
|
|
use Appwrite\Event\Func;
|
2024-07-19 09:22:50 +00:00
|
|
|
use Swoole\Coroutine as Co;
|
2024-10-01 14:30:47 +00:00
|
|
|
use Utopia\Queue\Connection\Redis;
|
2024-06-17 12:44:12 +00:00
|
|
|
|
|
|
|
|
class ScheduleExecutions extends ScheduleBase
|
|
|
|
|
{
|
|
|
|
|
public const UPDATE_TIMER = 3; // seconds
|
|
|
|
|
public const ENQUEUE_TIMER = 4; // seconds
|
|
|
|
|
|
|
|
|
|
public static function getName(): string
|
|
|
|
|
{
|
|
|
|
|
return 'schedule-executions';
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public static function getSupportedResource(): string
|
|
|
|
|
{
|
|
|
|
|
return 'execution';
|
|
|
|
|
}
|
|
|
|
|
|
2024-09-03 03:07:53 +00:00
|
|
|
public static function getCollectionId(): string
|
|
|
|
|
{
|
|
|
|
|
return 'executions';
|
|
|
|
|
}
|
|
|
|
|
|
2024-10-01 14:30:47 +00:00
|
|
|
protected function enqueueResources(array $pools, callable $getConsoleDB): void
|
2024-06-17 12:44:12 +00:00
|
|
|
{
|
2024-10-01 14:30:47 +00:00
|
|
|
[$connection,$pool, $dbForConsole] = $getConsoleDB();
|
|
|
|
|
$this->connections->add($connection, $pool);
|
|
|
|
|
|
|
|
|
|
$queuePool = $pools['pools-queue-queue']['pool'];
|
|
|
|
|
$queueConnection = $queuePool->get();
|
|
|
|
|
$this->connections->add($queueConnection, $queuePool);
|
|
|
|
|
|
|
|
|
|
$queueForFunctions = new Func(new Redis($queueConnection));
|
2024-07-18 12:03:24 +00:00
|
|
|
$intervalEnd = (new \DateTime())->modify('+' . self::ENQUEUE_TIMER . ' seconds');
|
2024-06-17 13:12:02 +00:00
|
|
|
|
2024-06-17 12:44:12 +00:00
|
|
|
foreach ($this->schedules as $schedule) {
|
2024-06-17 13:12:02 +00:00
|
|
|
if (!$schedule['active']) {
|
2024-07-01 13:35:37 +00:00
|
|
|
$dbForConsole->deleteDocument(
|
|
|
|
|
'schedules',
|
|
|
|
|
$schedule['$id'],
|
|
|
|
|
);
|
|
|
|
|
|
2024-09-07 17:31:17 +00:00
|
|
|
unset($this->schedules[$schedule['$internalId']]);
|
2024-06-17 12:44:12 +00:00
|
|
|
continue;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
$scheduledAt = new \DateTime($schedule['schedule']);
|
2024-07-18 12:03:24 +00:00
|
|
|
if ($scheduledAt <= $intervalEnd) {
|
2024-06-17 12:44:12 +00:00
|
|
|
continue;
|
|
|
|
|
}
|
|
|
|
|
|
2024-09-07 10:20:23 +00:00
|
|
|
$data = $dbForConsole->getDocument(
|
|
|
|
|
'schedules',
|
|
|
|
|
$schedule['$id'],
|
|
|
|
|
)->getAttribute('data', []);
|
2024-07-18 12:03:24 +00:00
|
|
|
|
2024-09-07 10:20:23 +00:00
|
|
|
$delay = $scheduledAt->getTimestamp() - (new \DateTime())->getTimestamp();
|
2024-07-19 09:22:50 +00:00
|
|
|
|
2024-10-01 14:30:47 +00:00
|
|
|
\go(function () use ($queueForFunctions, $schedule, $delay, $data, $dbForConsole) {
|
2024-07-19 09:22:50 +00:00
|
|
|
Co::sleep($delay);
|
2024-07-18 12:03:24 +00:00
|
|
|
|
2024-09-07 10:20:23 +00:00
|
|
|
$queueForFunctions->setType('schedule')
|
2024-07-18 12:03:24 +00:00
|
|
|
// Set functionId instead of function as we don't have $dbForProject
|
|
|
|
|
// TODO: Refactor to use function instead of functionId
|
|
|
|
|
->setFunctionId($schedule['resource']['functionId'])
|
|
|
|
|
->setExecution($schedule['resource'])
|
2024-09-07 10:20:23 +00:00
|
|
|
->setMethod($data['method'] ?? 'POST')
|
|
|
|
|
->setPath($data['path'] ?? '/')
|
|
|
|
|
->setHeaders($data['headers'] ?? [])
|
|
|
|
|
->setBody($data['body'] ?? '')
|
2024-07-18 12:03:24 +00:00
|
|
|
->setProject($schedule['project'])
|
2024-09-07 10:20:23 +00:00
|
|
|
->setUserId($data['userId'] ?? '')
|
2024-07-18 12:03:24 +00:00
|
|
|
->trigger();
|
2024-06-17 12:44:12 +00:00
|
|
|
|
2024-10-01 14:30:47 +00:00
|
|
|
$dbForConsole->deleteDocument(
|
|
|
|
|
'schedules',
|
|
|
|
|
$schedule['$id'],
|
|
|
|
|
);
|
|
|
|
|
});
|
2024-09-20 16:30:05 +00:00
|
|
|
|
2024-09-07 17:31:17 +00:00
|
|
|
unset($this->schedules[$schedule['$internalId']]);
|
2024-06-17 12:44:12 +00:00
|
|
|
}
|
2024-06-17 13:12:02 +00:00
|
|
|
|
2024-10-01 14:30:47 +00:00
|
|
|
$this->connections->reclaim();
|
2024-06-17 12:44:12 +00:00
|
|
|
}
|
|
|
|
|
}
|