Rewrite failed-jobs command

Author: codebarista

src/Console/Commands/FailedJobs.php

@@ -16,67 +16,100 @@ class FailedJobs extends Command implements Isolatable, PromptsForMissingInput
     /**
      * @var string
      */
-    protected $signature = 'codebarista:failed-jobs';
+    protected $signature = 'codebarista:failed-jobs {action}';
 
     /**
      * @var string
      */
-    protected $description = 'Retry or delete failed jobs';
+    protected $description = 'Retry, delete or reset failed jobs';
+
+    protected function promptForMissingArgumentsUsing(): array
+    {
+        return [
+            'action' => ['Which action should be executed?', 'retry, delete or reset'],
+        ];
+    }
 
     public function handle(RedisJobRepository $repository): int
     {
-        $action = $this->choice('What action should be taken?', [
-            'delete',
-            'retry',
-        ]);
-
-        return match ($action) {
-            'delete' => $this->deleteFailedJobs($repository),
-            'retry' => $this->retryFailedJobs($repository),
+        $amount = $repository->totalFailed();
+
+        return match ($this->argument('action')) {
+            'delete' => $this->deleteFailedJobs($repository, $amount),
+            'retry' => $this->retryFailedJobs($repository, $amount),
+            'reset' => $this->resetFailedJobs($repository),
             default => self::INVALID,
         };
+
     }
 
-    private function deleteFailedJobs(RedisJobRepository $repository): int
+    private function deleteFailedJobs(RedisJobRepository $repository, int $amount): int
     {
-        if (! $this->confirm('Really delete all failed jobs?')) {
-            $this->components->info('Process terminated by user');
+        if ($this->confirm(sprintf('Really delete %d failed %s?', $amount, Str::plural('job', $amount)))) {
+            $failedJobs = $this->getFailedJobs($repository, $amount);
+            $counter = 0;
 
-            return self::SUCCESS;
+            foreach ($failedJobs as $jobId) {
+                $counter += $repository->deleteFailed($jobId);
+            }
+
+            $this->resetFailedJobs($repository);
+
+            $this->components->info(sprintf('%d of %d %s deleted.',
+                $counter, $amount, Str::plural('failed job', $amount)
+            ));
         }
 
-        $i = $repository->getFailed()->count();
+        return self::SUCCESS;
+    }
+
+    private function retryFailedJobs(RedisJobRepository $repository, int $amount): int
+    {
+        $chunk = (int) $this->ask('How many failed jobs should be retried at once?', $amount);
 
-        // What a hack...but it works
-        $repository->recentFailedJobExpires = 0;
-        $repository->trimRecentJobs();
+        $failedJobs = $this->getFailedJobs($repository, $chunk);
+        $counter = 0;
 
-        $repository->failedJobExpires = 0;
-        $repository->trimFailedJobs();
+        foreach ($failedJobs as $jobId) {
+            dispatch(new RetryFailedJob($jobId));
+            $counter += $repository->deleteFailed($jobId);
+        }
 
-        $this->components->info(
-            sprintf('%d %s deleted.', $i, Str::plural(' job', $i))
-        );
+        $this->components->info(sprintf('%d of %d %s retried.',
+            $counter, $amount, Str::plural('failed job', $amount)
+        ));
 
         return self::SUCCESS;
     }
 
-    private function retryFailedJobs(RedisJobRepository $repository): int
+    private function resetFailedJobs(RedisJobRepository $repository): int
     {
-        if (! $this->confirm('Really retry all failed jobs?')) {
-            $this->components->info('Process terminated by user');
+        if ($this->confirm('Reset all failed jobs and their counters?')) {
+            $repository->recentFailedJobExpires = 0;
+            $repository->trimRecentJobs();
 
-            return self::SUCCESS;
+            $repository->failedJobExpires = 0;
+            $repository->trimFailedJobs();
         }
 
-        $i = $repository->getFailed()->each(function ($job) {
-            dispatch(new RetryFailedJob($job->id));
-        })->count();
+        return self::SUCCESS;
+    }
 
-        $this->components->info(
-            sprintf('%d %s retried.', $i, Str::plural(' job', $i))
-        );
+    private function getFailedJobs(RedisJobRepository $repository, int $amount): array
+    {
+        $failedJobs = $repository->getFailed($afterIndex = -1);
+        $jobs = [];
+
+        while ($failedJobs->count() > 0) {
+            foreach ($failedJobs as $failedJob) {
+                $jobs[] = $failedJob->id;
+            }
+            $failedJobs = $repository->getFailed($afterIndex += 50);
+        }
 
-        return self::SUCCESS;
+        // sort all jobs in descending order
+        $jobs = array_reverse($jobs);
+
+        return array_slice($jobs, 0, $amount);
     }
 }