lazy fork state

Author: rez1dent3

src/Internal/Service/StateService.php

@@ -11,18 +11,30 @@ final class StateService implements StateServiceInterface
      */
     private array $forks = [];
 
-    public function fork(string $uuid, string $value): void
+    /**
+     * @var array<string, callable>
+     */
+    private array $forkCallables = [];
+
+    public function fork(string $uuid, callable $value): void
     {
-        $this->forks[$uuid] ??= $value;
+        $this->forkCallables[$uuid] ??= $value;
     }
 
     public function get(string $uuid): ?string
     {
+        if ($this->forkCallables[$uuid] ?? null) {
+            $callable = $this->forkCallables[$uuid];
+            unset($this->forkCallables[$uuid]);
+
+            $this->forks[$uuid] = $callable();
+        }
+
         return $this->forks[$uuid] ?? null;
     }
 
     public function drop(string $uuid): void
     {
-        unset($this->forks[$uuid]);
+        unset($this->forks[$uuid], $this->forkCallables[$uuid]);
     }
 }

src/Internal/Service/StateServiceInterface.php

@@ -6,7 +6,10 @@
 
 interface StateServiceInterface
 {
-    public function fork(string $uuid, string $value): void;
+    /**
+     * @param callable(): string $value
+     */
+    public function fork(string $uuid, callable $value): void;
 
     public function get(string $uuid): ?string;
 

src/Services/AtomicService.php

@@ -52,7 +52,7 @@ public function blocks(array $objects, callable $callback): mixed
 
         $callable = function () use ($blockObjects, $callback) {
             foreach ($blockObjects as $uuid => $wallet) {
-                $this->stateService->fork($uuid, $this->bookkeeperService->amount($wallet));
+                $this->stateService->fork($uuid, fn () => $this->bookkeeperService->amount($wallet));
             }
             return $this->databaseService->transaction($callback);
         };