Removed flush of repositories

src/Command/RunCommand.php

@@ -11,6 +11,7 @@
 
 namespace Task\TaskBundle\Command;
 
+use Doctrine\ORM\EntityManagerInterface;
 use Symfony\Component\Console\Command\Command;
 use Symfony\Component\Console\Input\InputInterface;
 use Symfony\Component\Console\Output\OutputInterface;
@@ -32,17 +33,28 @@ class RunCommand extends Command
      */
     private $scheduler;
 
+    /**
+     * @var EntityManagerInterface
+     */
+    private $entityManager;
+
     /**
      * @param string $name
      * @param TaskRunnerInterface $runner
      * @param TaskSchedulerInterface $scheduler
+     * @param EntityManagerInterface $entityManager
      */
-    public function __construct($name, TaskRunnerInterface $runner, TaskSchedulerInterface $scheduler)
-    {
+    public function __construct(
+        $name,
+        TaskRunnerInterface $runner,
+        TaskSchedulerInterface $scheduler,
+        EntityManagerInterface $entityManager = null
+    ) {
         parent::__construct($name);
 
         $this->runner = $runner;
         $this->scheduler = $scheduler;
+        $this->entityManager = $entityManager;
     }
 
     /**
@@ -60,5 +72,9 @@ protected function execute(InputInterface $input, OutputInterface $output)
     {
         $this->runner->runTasks();
         $this->scheduler->scheduleTasks();
+
+        if ($this->entityManager) {
+            $this->entityManager->flush();
+        }
     }
 }

src/Command/ScheduleTaskCommand.php

@@ -11,6 +11,7 @@
 
 namespace Task\TaskBundle\Command;
 
+use Doctrine\ORM\EntityManagerInterface;
 use Symfony\Component\Console\Command\Command;
 use Symfony\Component\Console\Input\InputArgument;
 use Symfony\Component\Console\Input\InputInterface;
@@ -28,15 +29,22 @@ class ScheduleTaskCommand extends Command
      */
     private $scheduler;
 
+    /**
+     * @var EntityManagerInterface
+     */
+    private $entityManager;
+
     /**
      * @param string $name
      * @param TaskSchedulerInterface $runner
+     * @param EntityManagerInterface $entityManager
      */
-    public function __construct($name, TaskSchedulerInterface $runner)
+    public function __construct($name, TaskSchedulerInterface $runner, EntityManagerInterface $entityManager = null)
     {
         parent::__construct($name);
 
         $this->scheduler = $runner;
+        $this->entityManager = $entityManager;
     }
 
     /**
@@ -84,5 +92,9 @@ protected function execute(InputInterface $input, OutputInterface $output)
         }
 
         $taskBuilder->schedule();
+
+        if ($this->entityManager) {
+            $this->entityManager->flush();
+        }
     }
 }

src/DependencyInjection/TaskExtension.php

@@ -14,6 +14,7 @@
 use Symfony\Component\Config\FileLocator;
 use Symfony\Component\DependencyInjection\ContainerBuilder;
 use Symfony\Component\DependencyInjection\Loader;
+use Symfony\Component\DependencyInjection\Reference;
 use Symfony\Component\HttpKernel\DependencyInjection\Extension;
 
 /**
@@ -38,5 +39,13 @@ public function load(array $configs, ContainerBuilder $container)
         if ($config['run']['mode'] === 'listener') {
             $loader->load('listener.xml');
         }
+
+        if ('doctrine' === $config['storage']) {
+            // FIXME move to compiler pass
+            $container->getDefinition('task.command.schedule_task')
+                ->addArgument(new Reference('doctrine.orm.entity_manager'));
+            $container->getDefinition('task.command.run')
+                ->addArgument(new Reference('doctrine.orm.entity_manager'));
+        }
     }
 }

src/Entity/TaskExecutionRepository.php

@@ -34,7 +34,7 @@ public function create(TaskInterface $task, \DateTime $scheduleTime)
     /**
      * {@inheritdoc}
      */
-    public function persist(TaskExecutionInterface $execution)
+    public function save(TaskExecutionInterface $execution)
     {
         $this->_em->persist($execution);
 
@@ -51,16 +51,6 @@ public function remove(TaskExecutionInterface $execution)
         return $this;
     }
 
-    /**
-     * {@inheritdoc}
-     */
-    public function flush()
-    {
-        $this->_em->flush();
-
-        return $this;
-    }
-
     /**
      * {@inheritdoc}
      */

src/Entity/TaskRepository.php

@@ -39,7 +39,7 @@ public function findByUuid($uuid)
     /**
      * {@inheritdoc}
      */
-    public function persist(TaskInterface $task)
+    public function save(TaskInterface $task)
     {
         $this->_em->persist($task);
 
@@ -56,16 +56,6 @@ public function remove(TaskInterface $task)
         return $this;
     }
 
-    /**
-     * {@inheritdoc}
-     */
-    public function flush()
-    {
-        $this->_em->flush();
-
-        return $this;
-    }
-
     /**
      * {@inheritdoc}
      */

src/Resources/config/command.xml

@@ -7,6 +7,7 @@
             <argument type="string">task:run</argument>
             <argument type="service" id="task.runner"/>
             <argument type="service" id="task.scheduler"/>
+            <!-- add entity_manager if doctrine storage is enabled -->
 
             <tag name="console.command"/>
         </service>
@@ -21,6 +22,7 @@
         <service id="task.command.schedule_task" class="Task\TaskBundle\Command\ScheduleTaskCommand">
             <argument type="string">task:schedule</argument>
             <argument type="service" id="task.scheduler"/>
+            <!-- add entity_manager if doctrine storage is enabled -->
 
             <tag name="console.command"/>
         </service>

tests/Functional/BaseCommandTestCase.php

@@ -104,8 +104,7 @@ protected function createTask($workload, CronExpression $cronExpression = null,
         if ($cronExpression) {
             $task->setInterval($cronExpression, new \DateTime(), new \DateTime('+1 year'));
         }
-        $this->taskRepository->persist($task);
-        $this->taskRepository->flush();
+        $this->taskRepository->save($task);
 
         return $task;
     }
@@ -123,8 +122,9 @@ protected function createTaskExecution(TaskInterface $task, \DateTime $scheduleT
     {
         $execution = $this->taskExecutionRepository->create($task, $scheduleTime);
         $execution->setStatus($status);
-        $this->taskExecutionRepository->persist($execution);
-        $this->taskExecutionRepository->flush();
+        $this->taskExecutionRepository->save($execution);
+
+        $this->getEntityManager()->flush();
 
         return $execution;
     }

tests/Functional/Command/DebugTasksCommandTest.php

@@ -35,7 +35,7 @@ public function testExecute()
         $executions[1]->setResult(strrev($executions[1]->getWorkload()));
         $executions[1]->setDuration(0.0001);
 
-        $this->taskExecutionRepository->flush();
+        $this->getEntityManager()->flush();
 
         $this->commandTester->execute(
             [
@@ -62,7 +62,7 @@ public function testExecutePaginated()
             $this->createTaskExecution($task, new \DateTime('+1 hour')),
         ];
 
-        $this->taskExecutionRepository->flush();
+        $this->getEntityManager()->flush();
 
         $this->commandTester->execute(
             [