Add a retry-attempts option.

swarrot · Sep 18, 2014 · 98df5fb · 98df5fb
1 parent 045194b
commit 98df5fb
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/Command/SwarrotCommand.php b/Command/SwarrotCommand.php
@@ -63,6 +63,7 @@ protected function configure()
         }
         if (array_key_exists('retry', $this->processorStack)) {
             $this->addOption('no-retry', 'R', InputOption::VALUE_NONE, 'Deactivate retry.');
+            $this->addOption('retry-attempts', null, InputOption::VALUE_REQUIRED, 'Number of maximum retry attempts (if it exists, override the extra data parameter)');
         }
     }
 
@@ -149,6 +150,10 @@ protected function getOptions(InputInterface $input)
             if (isset($this->extras['retry_attempts'])) {
                 $attempts = $this->extras['retry_attempts'];
             }
+
+            if ($input->hasOption('retry-attempts')) {
+                $attempts = (int) $input->getOption('retry-attempts');
+            }
             $options['retry_attempts'] = $attempts;
         }