StateServiceInterface

Author: rez1dent3

src/Internal/Decorator/StorageServiceLockDecorator.php

@@ -7,12 +7,14 @@
 use Bavix\Wallet\Internal\Exceptions\LockProviderNotFoundException;
 use Bavix\Wallet\Internal\Exceptions\RecordNotFoundException;
 use Bavix\Wallet\Internal\Service\LockServiceInterface;
+use Bavix\Wallet\Internal\Service\StateServiceInterface;
 use Bavix\Wallet\Internal\Service\StorageServiceInterface;
 
 final class StorageServiceLockDecorator implements StorageServiceInterface
 {
     public function __construct(
         private StorageServiceInterface $storageService,
+        private StateServiceInterface $stateService,
         private LockServiceInterface $lockService
     ) {
     }
@@ -29,7 +31,7 @@ public function missing(string $uuid): bool
 
     public function get(string $uuid): string
     {
-        return $this->storageService->get($uuid);
+        return $this->stateService->get($uuid) ?? $this->storageService->get($uuid);
     }
 
     public function sync(string $uuid, float|int|string $value): bool

src/Internal/Service/StateService.php

@@ -0,0 +1,28 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Bavix\Wallet\Internal\Service;
+
+final class StateService implements StateServiceInterface
+{
+    /**
+     * @var array<string, string>
+     */
+    private array $forks = [];
+
+    public function fork(string $uuid, string $value): void
+    {
+        $this->forks[$uuid] = $value;
+    }
+
+    public function get(string $uuid): ?string
+    {
+        return $this->forks[$uuid] ?? null;
+    }
+
+    public function drop(string $uuid): void
+    {
+        unset($this->forks[$uuid]);
+    }
+}

src/Internal/Service/StateServiceInterface.php

@@ -0,0 +1,14 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Bavix\Wallet\Internal\Service;
+
+interface StateServiceInterface
+{
+    public function fork(string $uuid, string $value): void;
+
+    public function get(string $uuid): ?string;
+
+    public function drop(string $uuid): void;
+}

src/Models/Wallet.php

@@ -14,6 +14,7 @@
 use Bavix\Wallet\Internal\Exceptions\LockProviderNotFoundException;
 use Bavix\Wallet\Internal\Exceptions\TransactionFailedException;
 use Bavix\Wallet\Internal\Service\MathServiceInterface;
+use Bavix\Wallet\Internal\Service\StateServiceInterface;
 use Bavix\Wallet\Internal\Service\UuidFactoryServiceInterface;
 use Bavix\Wallet\Services\AtomicServiceInterface;
 use Bavix\Wallet\Services\RegulatorServiceInterface;
@@ -114,6 +115,8 @@ public function setNameAttribute(string $name): void
     public function refreshBalance(): bool
     {
         return app(AtomicServiceInterface::class)->block($this, function () {
+            app(StateServiceInterface::class)->drop($this->uuid);
+
             $whatIs = $this->getBalanceAttribute();
             $balance = $this->getAvailableBalanceAttribute();
             if (app(MathServiceInterface::class)->compare($whatIs, $balance) === 0) {

src/Services/AtomicService.php

@@ -10,6 +10,7 @@
 use Bavix\Wallet\Internal\Exceptions\TransactionFailedException;
 use Bavix\Wallet\Internal\Service\DatabaseServiceInterface;
 use Bavix\Wallet\Internal\Service\LockServiceInterface;
+use Bavix\Wallet\Internal\Service\StateServiceInterface;
 use Illuminate\Database\RecordsNotFoundException;
 
 /**
@@ -19,6 +20,7 @@ final class AtomicService implements AtomicServiceInterface
 {
     public function __construct(
         private DatabaseServiceInterface $databaseService,
+        private StateServiceInterface $stateService,
         private LockServiceInterface $lockService,
         private CastServiceInterface $castService
     ) {
@@ -37,7 +39,17 @@ public function blocks(array $objects, callable $callback): mixed
         $callable = fn () => $this->databaseService->transaction($callback);
         foreach ($objects as $object) {
             $wallet = $this->castService->getWallet($object);
-            $callable = fn () => $this->lockService->block($wallet->uuid, $callable);
+            $callable = fn () => $this->lockService->block(
+                $wallet->uuid,
+                function () use ($wallet, $callable) {
+                    try {
+                        $this->stateService->fork($wallet->uuid, $wallet->balance);
+                        return $callable();
+                    } finally {
+                        $this->stateService->drop($wallet->uuid);
+                    }
+                }
+            );
         }
 
         return $callable();

src/WalletServiceProvider.php

@@ -52,6 +52,8 @@
 use Bavix\Wallet\Internal\Service\LockServiceInterface;
 use Bavix\Wallet\Internal\Service\MathService;
 use Bavix\Wallet\Internal\Service\MathServiceInterface;
+use Bavix\Wallet\Internal\Service\StateService;
+use Bavix\Wallet\Internal\Service\StateServiceInterface;
 use Bavix\Wallet\Internal\Service\StorageService;
 use Bavix\Wallet\Internal\Service\StorageServiceInterface;
 use Bavix\Wallet\Internal\Service\TranslatorService;
@@ -209,6 +211,7 @@ private function internal(array $configure): void
         $this->app->singleton(JsonServiceInterface::class, $configure['json'] ?? JsonService::class);
         $this->app->singleton(LockServiceInterface::class, $configure['lock'] ?? LockService::class);
         $this->app->singleton(MathServiceInterface::class, $configure['math'] ?? MathService::class);
+        $this->app->singleton(StateServiceInterface::class, $configure['state'] ?? StateService::class);
         $this->app->singleton(TranslatorServiceInterface::class, $configure['translator'] ?? TranslatorService::class);
         $this->app->singleton(UuidFactoryServiceInterface::class, $configure['uuid'] ?? UuidFactoryService::class);
     }