add tagged service task compilerpass

Author: alexander-schranz

src/DependencyInjection/HandlerCompilerPass.php

@@ -16,6 +16,7 @@ class HandlerCompilerPass implements CompilerPassInterface
     const REGISTRY_ID = 'task.handler_registry';
     const HANDLER_TAG = 'task.handler';
     const ADD_FUNCTION_NAME = 'add';
+    const HANDLER_NAME_ATTRIBUTE = 'handler-name';
 
     /**
      * {@inheritdoc}
@@ -33,7 +34,7 @@ public function process(ContainerBuilder $container)
             foreach ($tags as $attributes) {
                 $definition->addMethodCall(
                     self::ADD_FUNCTION_NAME,
-                    [$attributes['handler-name'], new Reference($id)]
+                    [$attributes[self::HANDLER_NAME_ATTRIBUTE], new Reference($id)]
                 );
             }
         }

src/DependencyInjection/TaskCompilerPass.php

@@ -0,0 +1,64 @@
+<?php
+
+namespace Task\TaskBundle\DependencyInjection;
+
+use Symfony\Component\DependencyInjection\Compiler\CompilerPassInterface;
+use Symfony\Component\DependencyInjection\ContainerBuilder;
+use Symfony\Component\DependencyInjection\Reference;
+
+/**
+ * Compiler pass which collects worker services.
+ *
+ * @author @wachterjohannes <[email protected]>
+ */
+class TaskCompilerPass implements CompilerPassInterface
+{
+    const SCHEDULER_ID = 'task.scheduler';
+    const INTERVAL_TAG = 'task.interval';
+    const KEY_ATTRIBUTE = 'key';
+    const INTERVAL_ATTRIBUTE = 'interval';
+    const WORKLOAD_ATTRIBUTE = 'workload';
+    const CREATE_FUNCTION_NAME = 'createTaskAndSchedule';
+
+    /**
+     * {@inheritdoc}
+     */
+    public function process(ContainerBuilder $container)
+    {
+        if (!$container->has(self::SCHEDULER_ID)) {
+            return;
+        }
+
+        $schedulerDefinition = $container->getDefinition(self::SCHEDULER_ID);
+
+        $taggedServices = $container->findTaggedServiceIds(self::INTERVAL_TAG);
+        foreach ($taggedServices as $id => $tags) {
+            $handlerDefinition = $container->getDefinition($id);
+            $tag = $handlerDefinition->getTag(HandlerCompilerPass::HANDLER_TAG);
+
+            $handler = $tag[0][HandlerCompilerPass::HANDLER_NAME_ATTRIBUTE];
+
+            // remove all tasks with $id and not completed
+            foreach ($tags as $attributes) {
+                $interval = $attributes[self::INTERVAL_ATTRIBUTE];
+                $workload = isset($attributes[self::WORKLOAD_ATTRIBUTE]) ? $attributes[self::WORKLOAD_ATTRIBUTE] : null;
+                $key = isset($attributes[self::KEY_ATTRIBUTE]) ? $attributes[self::KEY_ATTRIBUTE] : null;
+
+                if (!$key) {
+                    $key = $handler . '_' . $interval . '_' . serialize($workload);
+                }
+
+
+                $schedulerDefinition->addMethodCall(
+                    self::CREATE_FUNCTION_NAME,
+                    [
+                        $handler,
+                        $interval,
+                        $workload,
+                        $key,
+                    ]
+                );
+            }
+        }
+    }
+}

src/Entity/Task.php

@@ -16,6 +16,11 @@ class Task
      */
     private $uuid;
 
+    /**
+     * @var string
+     */
+    private $key;
+
     /**
      * @var TaskInterface
      */
@@ -102,4 +107,21 @@ public function setCompleted($completed)
     {
         $this->completed = $completed;
     }
+
+    /**
+     * @return string
+     */
+    public function getKey()
+    {
+        return $this->key;
+    }
+
+    /**
+     * @param string $key
+     * @return $this
+     */
+    public function setKey($key)
+    {
+        $this->key = $key;
+    }
 }

src/Resources/config/doctrine/Task.orm.xml

@@ -12,6 +12,7 @@
 
         <field name="task" type="object"/>
         <field name="uuid" type="string" length="100"/>
+        <field name="key" type="string" length="255" nullable="true" column="task_key"/>
         <field name="executionDate" type="datetime"/>
         <field name="completed" type="boolean"/>
 

src/Storage/DoctrineStorage.php

@@ -32,6 +32,22 @@ public function __construct(EntityManagerInterface $entityManager, TaskRepositor
     public function store(TaskInterface $task)
     {
         $entity = new TaskEntity();
+
+        if ($task->getKey()) {
+            $oldEntity = $this->taskRepository->findOneBy(
+                [
+                    'key' => $task->getKey(),
+                    'completed' => false,
+                ]
+            );
+
+            if ($oldEntity) {
+                // TODO update task (warning execution date should not be changed)
+
+                return;
+            }
+        }
+
         $this->setTask($entity, $task);
 
         $this->entityManager->persist($entity);
@@ -89,6 +105,7 @@ private function setTask(TaskEntity $entity, TaskInterface $task)
     {
         $entity->setTask($task);
         $entity->setUuid($task->getUuid());
+        $entity->setKey($task->getKey());
         $entity->setCompleted($task->isCompleted());
         $entity->setExecutionDate($task->getExecutionDate());
     }

src/TaskBundle.php

@@ -5,6 +5,7 @@
 use Symfony\Component\DependencyInjection\ContainerBuilder;
 use Symfony\Component\HttpKernel\Bundle\Bundle;
 use Task\TaskBundle\DependencyInjection\HandlerCompilerPass;
+use Task\TaskBundle\DependencyInjection\TaskCompilerPass;
 
 /**
  * Integrates php-task into symfony.
@@ -18,5 +19,6 @@ public function build(ContainerBuilder $container)
         parent::build($container);
 
         $container->addCompilerPass(new HandlerCompilerPass());
+        $container->addCompilerPass(new TaskCompilerPass());
     }
 }