Merge pull request #33 from justbetter/feature/defer

Defer retrieve all

Author: VincentBean

src/Actions/Retrieval/RetrieveAllStock.php

@@ -3,17 +3,47 @@
 namespace JustBetter\MagentoStock\Actions\Retrieval;
 
 use Illuminate\Support\Carbon;
+use Illuminate\Support\Enumerable;
 use JustBetter\MagentoStock\Contracts\Retrieval\RetrievesAllStock;
 use JustBetter\MagentoStock\Jobs\Retrieval\RetrieveStockJob;
+use JustBetter\MagentoStock\Models\Stock;
 use JustBetter\MagentoStock\Repositories\BaseRepository;
 
 class RetrieveAllStock implements RetrievesAllStock
 {
-    public function retrieve(?Carbon $from): void
+    public function retrieve(?Carbon $from, bool $defer = true): void
     {
         $repository = BaseRepository::resolve();
 
-        $repository->skus($from)->each(fn (string $sku) => RetrieveStockJob::dispatch($sku));
+        if (! $defer) {
+            $repository->skus($from)->each(fn (string $sku) => RetrieveStockJob::dispatch($sku));
+
+            return;
+        }
+
+        $date = now();
+
+        $repository->skus($from)->chunk(250)->each(function (Enumerable $skus) use ($date): void {
+            $existing = Stock::query()
+                ->whereIn('sku', $skus)
+                ->pluck('sku');
+
+            Stock::query()
+                ->whereIn('sku', $existing)
+                ->update(['retrieve' => true]);
+
+            Stock::query()->insert(
+                $skus
+                    ->diff($existing)
+                    ->values()
+                    ->map(fn (string $sku): array => [
+                        'sku' => $sku,
+                        'retrieve' => true,
+                        'created_at' => $date,
+                        'updated_at' => $date,
+                    ])->toArray()
+            );
+        });
     }
 
     public static function bind(): void

src/Commands/Retrieval/RetrieveAllStockCommand.php

@@ -8,7 +8,7 @@
 
 class RetrieveAllStockCommand extends Command
 {
-    protected $signature = 'magento-stock:retrieve-all {from?}';
+    protected $signature = 'magento-stock:retrieve-all {from?} {--queue}';
 
     protected $description = 'Retrieve all (modified) stock';
 
@@ -17,11 +17,14 @@ public function handle(): int
         /** @var ?string $from */
         $from = $this->argument('from');
 
+        /** @var bool $defer */
+        $defer = ! $this->option('queue');
+
         $date = $from !== null
             ? Carbon::parse($from)
             : null;
 
-        RetrieveAllStockJob::dispatch($date);
+        RetrieveAllStockJob::dispatch($date, $defer);
 
         return static::SUCCESS;
     }

src/Contracts/Retrieval/RetrievesAllStock.php

@@ -6,5 +6,5 @@
 
 interface RetrievesAllStock
 {
-    public function retrieve(?Carbon $from): void;
+    public function retrieve(?Carbon $from, bool $defer = true): void;
 }

src/Jobs/Retrieval/RetrieveAllStockJob.php

@@ -16,13 +16,13 @@ class RetrieveAllStockJob implements ShouldBeUnique, ShouldQueue
     use InteractsWithQueue;
     use Queueable;
 
-    public function __construct(public ?Carbon $from = null)
+    public function __construct(public ?Carbon $from = null, public bool $defer = true)
     {
-        $this->onQueue(config('magento-stock.queue'));
+        $this->onQueue(config()->string('magento-stock.queue'));
     }
 
     public function handle(RetrievesAllStock $stock): void
     {
-        $stock->retrieve($this->from);
+        $stock->retrieve($this->from, $this->defer);
     }
 }

tests/Actions/Retrieval/RetrieveAllStockTest.php

@@ -5,6 +5,7 @@
 use Illuminate\Support\Facades\Bus;
 use JustBetter\MagentoStock\Actions\Retrieval\RetrieveAllStock;
 use JustBetter\MagentoStock\Jobs\Retrieval\RetrieveStockJob;
+use JustBetter\MagentoStock\Models\Stock;
 use JustBetter\MagentoStock\Tests\Fakes\FakeRepository;
 use JustBetter\MagentoStock\Tests\TestCase;
 use PHPUnit\Framework\Attributes\Test;
@@ -19,8 +20,33 @@ public function it_dispatches_jobs(): void
 
         /** @var RetrieveAllStock $action */
         $action = app(RetrieveAllStock::class);
-        $action->retrieve(null);
+        $action->retrieve(null, false);
 
         Bus::assertDispatched(RetrieveStockJob::class);
     }
+
+    #[Test]
+    public function it_defers_retrievals(): void
+    {
+        config()->set('magento-stock.repository', FakeRepository::class);
+
+        Bus::fake();
+
+        Stock::query()->create(['sku' => 'sku_1', 'retrieve' => false]);
+
+        /** @var RetrieveAllStock $action */
+        $action = app(RetrieveAllStock::class);
+        $action->retrieve(null, true);
+
+        Bus::assertNotDispatched(RetrieveStockJob::class);
+
+        $stocks = Stock::query()
+            ->where('retrieve', '=', true)
+            ->pluck('sku');
+
+        $this->assertEquals([
+            'sku_1',
+            'sku_2',
+        ], $stocks->toArray());
+    }
 }