Merge pull request #5 from alchemy-fr/feature/preserve-payloads

Add option to prevent payload files from being deleted
alchemy-fr · Nov 14, 2016 · ae83862 · ae83862
2 parents 32ef82b + ea4e88d
commit ae83862
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 4 deletions.
diff --git a/src/WorkerBundle/Commands/DispatchingConsumerCommand.php b/src/WorkerBundle/Commands/DispatchingConsumerCommand.php
@@ -12,8 +12,10 @@
 namespace Alchemy\WorkerBundle\Commands;
 
 use Alchemy\Worker\MessageDispatcher;
+use Alchemy\Worker\WorkerInvoker;
 use Symfony\Component\Console\Command\Command;
 use Symfony\Component\Console\Input\InputInterface;
+use Symfony\Component\Console\Input\InputOption;
 use Symfony\Component\Console\Output\OutputInterface;
 
 class DispatchingConsumerCommand extends Command
@@ -23,21 +25,29 @@ class DispatchingConsumerCommand extends Command
      */
     private $messageDispatcher;
 
+    /**
+     * @var WorkerInvoker
+     */
+    private $workerInvoker;
+
     /**
      * @param MessageDispatcher $messageDispatcher
+     * @param WorkerInvoker $workerInvoker
      */
-    public function __construct(MessageDispatcher $messageDispatcher)
+    public function __construct(MessageDispatcher $messageDispatcher, WorkerInvoker $workerInvoker)
     {
         parent::__construct();
 
         $this->messageDispatcher = $messageDispatcher;
+        $this->workerInvoker = $workerInvoker;
     }
 
     protected function configure()
     {
         parent::configure();
 
-        $this->setName('workers:run-dispatcher');
+        $this->setName('workers:run-dispatcher')
+            ->addOption('preserve-payload', 'p', InputOption::VALUE_NONE);
     }
 
     /**
@@ -47,6 +57,10 @@ protected function configure()
      */
     protected function execute(InputInterface $input, OutputInterface $output)
     {
+        if ($input->getOption('preserve-payload')) {
+            $this->workerInvoker->preservePayloads();
+        }
+
         while (true) {
             $this->messageDispatcher->run();
         }

diff --git a/src/WorkerBundle/Commands/InvokeWorkerCommand.php b/src/WorkerBundle/Commands/InvokeWorkerCommand.php
@@ -14,6 +14,7 @@
 use Alchemy\Worker\WorkerResolver;
 use Symfony\Component\Console\Command\Command;
 use Symfony\Component\Console\Input\InputInterface;
+use Symfony\Component\Console\Input\InputOption;
 use Symfony\Component\Console\Output\OutputInterface;
 
 class InvokeWorkerCommand extends Command
@@ -40,7 +41,8 @@ protected function configure()
 
         $this->setName('workers:run-worker')
             ->addArgument('type')
-            ->addArgument('body');
+            ->addArgument('body')
+            ->addOption('preserve-payload', 'p', InputOption::VALUE_NONE);
     }
 
     protected function execute(InputInterface $input, OutputInterface $output)
@@ -66,6 +68,8 @@ protected function execute(InputInterface $input, OutputInterface $output)
 
         $worker->process($body);
 
-        unlink($input->getArgument('body'));
+        if (! $input->getOption('preserve-payload')) {
+            unlink($input->getArgument('body'));
+        }
     }
 }
diff --git a/src/WorkerBundle/Resources/config/services.yml b/src/WorkerBundle/Resources/config/services.yml
@@ -42,6 +42,7 @@ services:
     class: Alchemy\WorkerBundle\Commands\DispatchingConsumerCommand
     arguments:
       - "@alchemy_worker.message_dispatcher"
+      - "@alchemy_worker.worker_invoker"
     tags:
       - { name: console.command }