Merge pull request #35 from 21TORR/next

Add `DispatchAfterRunTask` and improve output of queue command

Author: carlgu

CHANGELOG.md

@@ -1,3 +1,10 @@
+2.3.0
+=====
+
+* (improvement) Always show key of task in queue tasks command.
+* (feature) Add `DispatchAfterRunTask` to be able to redispatch tasks after the given run. You can use it in the Scheduler to reliably redispatch tasks.
+
+
 2.2.0
 =====
 

src/Command/QueueTasksCommand.php

@@ -167,6 +167,10 @@ private function formatTaskLabel (Task $task) : string
 			);
 		}
 
-		return $metaData->label;
+		return \sprintf(
+			"%s (<fg=yellow>%s</>)",
+			$metaData->label,
+			$metaData->getKey(),
+		);
 	}
 }

src/Task/DispatchAfterRunTask/DispatchAfterRunTask.php

@@ -0,0 +1,34 @@
+<?php declare(strict_types=1);
+
+namespace Torr\TaskManager\Task\DispatchAfterRunTask;
+
+use Torr\TaskManager\Task\Task;
+use Torr\TaskManager\Task\TaskMetaData;
+
+/**
+ * This task takes another task and puts it into the queue.
+ *
+ * This task is supposed to be worked on synchronously, as it is pretty lightweight and only
+ * redispatches the given task.
+ */
+readonly class DispatchAfterRunTask extends Task
+{
+	public function __construct (
+		public Task $task,
+		public array|string $transportNames = [],
+	)
+	{
+		parent::__construct();
+	}
+
+	/**
+	 *
+	 */
+	#[\Override]
+	public function getMetaData () : TaskMetaData
+	{
+		return new TaskMetaData(
+			\sprintf("Redispatch task '%s' after the current run", $this->task->getMetaData()->label),
+		);
+	}
+}

src/Task/DispatchAfterRunTask/DispatchAfterRunTaskHandler.php

@@ -0,0 +1,32 @@
+<?php declare(strict_types=1);
+
+namespace Torr\TaskManager\Task\DispatchAfterRunTask;
+
+use Symfony\Component\Messenger\Attribute\AsMessageHandler;
+use Symfony\Component\Messenger\Stamp\TransportNamesStamp;
+use Torr\TaskManager\Manager\TaskManager;
+
+/**
+ * @final
+ */
+readonly class DispatchAfterRunTaskHandler
+{
+	/**
+	 */
+	public function __construct (
+		private TaskManager $taskManager,
+	) {}
+
+	/**
+	 *
+	 */
+	#[AsMessageHandler]
+	public function onDispatchAfterRunTask (DispatchAfterRunTask $task) : void
+	{
+		$stamps = !empty($task->transportNames)
+			? [new TransportNamesStamp($task->transportNames)]
+			: [];
+
+		$this->taskManager->enqueue($task->task, $stamps);
+	}
+}