fixed some issues with IncrementalIndexer

jikan-me · Nov 8, 2024 · a87b461 · a87b461
1 parent cc5c12f
commit a87b461
Showing 1 changed file with 10 additions and 8 deletions.
diff --git a/app/Console/Commands/Indexer/IncrementalIndexer.php b/app/Console/Commands/Indexer/IncrementalIndexer.php
@@ -103,7 +103,9 @@ private function fetchIds(string $mediaType, array $idsToFetch, bool $resume): v
             $this->info("Resuming from index: $index");
         }
 
-        if ($index > 0 && !isset($this->ids[$index]))
+        $ids = array_merge($idsToFetch['sfw'], $idsToFetch['nsfw']);
+
+        if ($index > 0 && !isset($ids[$index]))
         {
             $index = 0;
             $this->warn('Invalid index; set back to 0');
@@ -112,7 +114,6 @@ private function fetchIds(string $mediaType, array $idsToFetch, bool $resume): v
         Storage::put("indexer/incremental/{$mediaType}_resume.save", 0);
 
         $this->info("$idCount $mediaType entries available");
-        $ids = array_merge($idsToFetch['sfw'], $idsToFetch['nsfw']);
 
         for ($i = $index; $i <= ($idCount - 1); $i++)
         {
@@ -184,7 +185,7 @@ public function handle(): int
         }
 
         // we want to handle signals from the OS
-        $this->trap(SIGTERM, fn () => $this->cancelled = true);
+        $this->trap([SIGTERM, SIGQUIT, SIGINT], fn () => $this->cancelled = true);
 
         $resume = $this->option('resume') ?? false;
         $onlyFailed = $this->option('failed') ?? false;
@@ -208,13 +209,14 @@ public function handle(): int
                 $idsToFetch = $this->getIdsToFetch($mediaType);
             }
 
+            if ($this->cancelled)
+            {
+                return 127;
+            }
+
             $idCount = count($idsToFetch);
-            if ($idCount == 0)
+            if ($idCount === 0)
             {
-                if ($this->cancelled)
-                {
-                    return 127;
-                }
                 continue;
             }