added lazy loading of product wallet.

Author: rez1dent3

config/config.php

@@ -38,6 +38,7 @@
 use Bavix\Wallet\Services\CastService;
 use Bavix\Wallet\Services\ConsistencyService;
 use Bavix\Wallet\Services\DiscountService;
+use Bavix\Wallet\Services\EagerLoaderService;
 use Bavix\Wallet\Services\ExchangeService;
 use Bavix\Wallet\Services\PrepareService;
 use Bavix\Wallet\Services\PurchaseService;
@@ -100,6 +101,7 @@
         'cast' => CastService::class,
         'consistency' => ConsistencyService::class,
         'discount' => DiscountService::class,
+        'eager_loader' => EagerLoaderService::class,
         'exchange' => ExchangeService::class,
         'prepare' => PrepareService::class,
         'purchase' => PurchaseService::class,

src/Internal/Repository/WalletRepository.php

@@ -83,6 +83,22 @@ public function getBySlug(string $holderType, int|string $holderId, string $slug
         ]);
     }
 
+    /**
+     * @param array<int|string> $holderIds
+     *
+     * @return Wallet[]
+     */
+    public function findDefaultAll(string $holderType, array $holderIds): array
+    {
+        return $this->wallet->newQuery()
+            ->where('slug', config('wallet.wallet.default.slug', 'default'))
+            ->where('holder_type', $holderType)
+            ->whereIn('holder_id', $holderIds)
+            ->get()
+            ->all()
+        ;
+    }
+
     /**
      * @param array<string, int|string> $attributes
      */

src/Internal/Repository/WalletRepositoryInterface.php

@@ -17,6 +17,13 @@ public function findByUuid(string $uuid): ?Wallet;
 
     public function findBySlug(string $holderType, int|string $holderId, string $slug): ?Wallet;
 
+    /**
+     * @param array<int|string> $holderIds
+     *
+     * @return Wallet[]
+     */
+    public function findDefaultAll(string $holderType, array $holderIds): array;
+
     /**
      * @throws ModelNotFoundException
      */

src/Services/EagerLoaderService.php

@@ -0,0 +1,47 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Bavix\Wallet\Services;
+
+use Bavix\Wallet\Internal\Dto\BasketDtoInterface;
+use Bavix\Wallet\Internal\Repository\WalletRepositoryInterface;
+
+final class EagerLoaderService implements EagerLoaderServiceInterface
+{
+    public function __construct(
+        private CastServiceInterface $castService,
+        private WalletRepositoryInterface $walletRepository
+    ) {
+    }
+
+    public function loadWalletsByBasket(BasketDtoInterface $basketDto): void
+    {
+        $products = [];
+        /** @var array<array-key, array<array-key, int|string>> $productGroupIds */
+        $productGroupIds = [];
+        foreach ($basketDto->items() as $index => $item) {
+            $model = $this->castService->getModel($item->getProduct());
+            if (!$model->relationLoaded('wallet')) {
+                $products[$index] = $item->getProduct();
+                $productGroupIds[$model->getMorphClass()][$index] = $model->getKey();
+            }
+        }
+
+        foreach ($productGroupIds as $holderType => $holderIds) {
+            $allWallets = $this->walletRepository->findDefaultAll($holderType, array_unique($holderIds));
+            $wallets = [];
+            foreach ($allWallets as $wallet) {
+                $wallets[$wallet->holder_id] = $wallet;
+            }
+
+            foreach ($holderIds as $index => $holderId) {
+                $wallet = $wallets[$holderId] ?? null;
+                if ($wallet !== null) {
+                    $model = $this->castService->getModel($products[$index]);
+                    $model->setRelation('wallet', $wallet);
+                }
+            }
+        }
+    }
+}

src/Services/EagerLoaderServiceInterface.php

@@ -0,0 +1,12 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Bavix\Wallet\Services;
+
+use Bavix\Wallet\Internal\Dto\BasketDtoInterface;
+
+interface EagerLoaderServiceInterface
+{
+    public function loadWalletsByBasket(BasketDtoInterface $basketDto): void;
+}

src/Traits/CartPay.php

@@ -23,6 +23,7 @@
 use Bavix\Wallet\Services\BasketServiceInterface;
 use Bavix\Wallet\Services\CastServiceInterface;
 use Bavix\Wallet\Services\ConsistencyServiceInterface;
+use Bavix\Wallet\Services\EagerLoaderServiceInterface;
 use Bavix\Wallet\Services\PrepareServiceInterface;
 use Bavix\Wallet\Services\PurchaseServiceInterface;
 use Bavix\Wallet\Services\TransferServiceInterface;
@@ -55,6 +56,7 @@ public function payFreeCart(CartInterface $cart): array
             $basketDto = $cart->getBasketDto();
             $basketService = app(BasketServiceInterface::class);
             $availabilityAssembler = app(AvailabilityDtoAssemblerInterface::class);
+            app(EagerLoaderServiceInterface::class)->loadWalletsByBasket($basketDto);
             if (!$basketService->availability($availabilityAssembler->create($this, $basketDto, false))) {
                 throw new ProductEnded(
                     app(TranslatorServiceInterface::class)->get('wallet::errors.product_stock'),
@@ -111,6 +113,7 @@ public function payCart(CartInterface $cart, bool $force = false): array
             $basketDto = $cart->getBasketDto();
             $basketService = app(BasketServiceInterface::class);
             $availabilityAssembler = app(AvailabilityDtoAssemblerInterface::class);
+            app(EagerLoaderServiceInterface::class)->loadWalletsByBasket($basketDto);
             if (!$basketService->availability($availabilityAssembler->create($this, $basketDto, $force))) {
                 throw new ProductEnded(
                     app(TranslatorServiceInterface::class)->get('wallet::errors.product_stock'),
@@ -188,6 +191,7 @@ public function refundCart(CartInterface $cart, bool $force = false, bool $gifts
     {
         return app(AtomicServiceInterface::class)->block($this, function () use ($cart, $force, $gifts) {
             $basketDto = $cart->getBasketDto();
+            app(EagerLoaderServiceInterface::class)->loadWalletsByBasket($basketDto);
             $transfers = app(PurchaseServiceInterface::class)->already($this, $basketDto, $gifts);
             if (count($transfers) !== $basketDto->total()) {
                 throw new ModelNotFoundException(

src/WalletServiceProvider.php

@@ -76,6 +76,8 @@
 use Bavix\Wallet\Services\ConsistencyServiceInterface;
 use Bavix\Wallet\Services\DiscountService;
 use Bavix\Wallet\Services\DiscountServiceInterface;
+use Bavix\Wallet\Services\EagerLoaderService;
+use Bavix\Wallet\Services\EagerLoaderServiceInterface;
 use Bavix\Wallet\Services\ExchangeService;
 use Bavix\Wallet\Services\ExchangeServiceInterface;
 use Bavix\Wallet\Services\PrepareService;
@@ -198,6 +200,10 @@ private function services(array $configure, array $cache): void
             $configure['consistency'] ?? ConsistencyService::class
         );
         $this->app->singleton(DiscountServiceInterface::class, $configure['discount'] ?? DiscountService::class);
+        $this->app->singleton(
+            EagerLoaderServiceInterface::class,
+            $configure['eager_loader'] ?? EagerLoaderService::class
+        );
         $this->app->singleton(ExchangeServiceInterface::class, $configure['exchange'] ?? ExchangeService::class);
         $this->app->singleton(PrepareServiceInterface::class, $configure['prepare'] ?? PrepareService::class);
         $this->app->singleton(PurchaseServiceInterface::class, $configure['purchase'] ?? PurchaseService::class);

tests/Units/Domain/EagerLoadingTest.php

@@ -4,9 +4,12 @@
 
 namespace Bavix\Wallet\Test\Units\Domain;
 
+use Bavix\Wallet\Objects\Cart;
 use Bavix\Wallet\Test\Infra\Factories\BuyerFactory;
+use Bavix\Wallet\Test\Infra\Factories\ItemFactory;
 use Bavix\Wallet\Test\Infra\Factories\UserMultiFactory;
 use Bavix\Wallet\Test\Infra\Models\Buyer;
+use Bavix\Wallet\Test\Infra\Models\Item;
 use Bavix\Wallet\Test\Infra\Models\UserMulti;
 use Bavix\Wallet\Test\Infra\TestCase;
 use Illuminate\Database\Eloquent\Collection;
@@ -85,4 +88,31 @@ public function testMultiWallets(): void
         self::assertTrue($user->getWallet('hello')->relationLoaded('holder'));
         self::assertTrue($user->is($user->getWallet('hello')->holder));
     }
+
+    public function testEagerLoaderPay(): void
+    {
+        /** @var Buyer $buyer */
+        $buyer = BuyerFactory::new()->create();
+        /** @var Item[] $products */
+        $products = ItemFactory::times(50)->create([
+            'quantity' => 10,
+            'price' => 1,
+        ]);
+        $productIds = [];
+        foreach ($products as $product) {
+            $productIds[] = $product->getKey();
+            self::assertSame(0, $product->balanceInt);
+        }
+
+        $products = Item::query()->whereKey($productIds)->get()->all();
+
+        $cart = app(Cart::class);
+        foreach ($products as $product) {
+            $cart = $cart->withItem($product, 10);
+        }
+
+        $transfers = $buyer->forcePayCart($cart);
+        self::assertSame((int) -$cart->getTotal($buyer), $buyer->balanceInt);
+        self::assertCount(500, $transfers);
+    }
 }