Fix related users saving (#29)

abrouter · Aug 10, 2022 · 2777120 · 2777120
1 parent a0059c9
commit 2777120
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 6 deletions.
diff --git a/src/DB/Managers/ParallelRunningStateManager.php b/src/DB/Managers/ParallelRunningStateManager.php
@@ -10,6 +10,8 @@
 
 class ParallelRunningStateManager
 {
+    private const EXPIRES = 3600 * 24 * 180;
+
     private ParallelRunningStateCachedRepository $parallelRunningStateCachedRepository;
     private KvStorageConfigAccessor $kvStorageConfigAccessor;
 
@@ -25,7 +27,8 @@ public function setReadyToServe()
     {
         $this->kvStorageConfigAccessor->getKvStorage()->put(
             ParallelRunningDictionary::IS_RUNNING_KEY,
-            ParallelRunningDictionary::IS_RUNNING_VALUE_TRUE
+            ParallelRunningDictionary::IS_RUNNING_VALUE_TRUE,
+            self::EXPIRES
         );
         $this->parallelRunningStateCachedRepository->clearCacheServing();
     }
@@ -34,7 +37,8 @@ public function setStopServing()
     {
         $this->kvStorageConfigAccessor->getKvStorage()->put(
             ParallelRunningDictionary::IS_RUNNING_KEY,
-            ParallelRunningDictionary::IS_RUNNING_VALUE_STOPPED
+            ParallelRunningDictionary::IS_RUNNING_VALUE_STOPPED,
+            self::EXPIRES
         );
         $this->parallelRunningStateCachedRepository->clearCacheServing();
     }
@@ -43,7 +47,8 @@ public function setInitialized()
     {
         $this->kvStorageConfigAccessor->getKvStorage()->put(
             ParallelRunningDictionary::IS_INITIAZLIED_KEY,
-            ParallelRunningDictionary::IS_INITIAZLIED_TRUE_VALUE
+            ParallelRunningDictionary::IS_INITIAZLIED_TRUE_VALUE,
+            self::EXPIRES
         );
         $this->parallelRunningStateCachedRepository->clearCacheInitialized();
     }

diff --git a/src/DB/Managers/RelatedUsersCacheManager.php b/src/DB/Managers/RelatedUsersCacheManager.php
@@ -10,6 +10,8 @@
 
 class RelatedUsersCacheManager
 {
+    private const EXPIRES = 3600 * 24 * 180;
+
     private KvStorageConfigAccessor $kvStorage;
 
     public function __construct(
@@ -23,6 +25,10 @@ public function store(array $relatedUsers): void
         $this
             ->kvStorage
             ->getKvStorage()
-            ->put(ParallelRunningDictionary::RELATED_USERS_KEY, json_encode($relatedUsers));
+            ->put(
+                ParallelRunningDictionary::RELATED_USERS_KEY,
+                json_encode($relatedUsers),
+                self::EXPIRES
+            );
     }
 }
diff --git a/src/Events/Handlers/RelatedUsersStatisticsInterceptor.php b/src/Events/Handlers/RelatedUsersStatisticsInterceptor.php
@@ -7,6 +7,7 @@
 use Abrouter\Client\Contracts\TaskContract;
 use Abrouter\Client\DB\Managers\RelatedUsersCacheManager;
 use Abrouter\Client\DB\RelatedUsersStore;
+use Abrouter\Client\DB\Repositories\RelatedUsersCacheRepository;
 use Abrouter\Client\Events\HandlerInterface;
 use Abrouter\Client\Services\ExperimentsParallelRun\ParallelRunSwitch;
 use Abrouter\Client\Services\Statistics\SendEventTask;
@@ -19,14 +20,18 @@ class RelatedUsersStatisticsInterceptor implements HandlerInterface
 
     private RelatedUsersCacheManager $relatedUsersCacheManager;
 
+    private RelatedUsersCacheRepository $relatedUsersCacheRepository;
+
     public function __construct(
         ParallelRunSwitch $parallelRunSwitch,
         RelatedUsersStore $relatedUsersStore,
-        RelatedUsersCacheManager $relatedUsersCacheManager
+        RelatedUsersCacheManager $relatedUsersCacheManager,
+        RelatedUsersCacheRepository $relatedUsersCacheRepository
     ) {
         $this->parallelRunSwitch = $parallelRunSwitch;
         $this->relatedUsersStore = $relatedUsersStore;
         $this->relatedUsersCacheManager = $relatedUsersCacheManager;
+        $this->relatedUsersCacheRepository = $relatedUsersCacheRepository;
     }
 
     public function handle(TaskContract $taskContract): bool
@@ -42,12 +47,14 @@ public function handle(TaskContract $taskContract): bool
         $userId = $taskContract->getEventDTO()->getBaseEventDTO()->getUserId();
         $temporaryUserId = $taskContract->getEventDTO()->getBaseEventDTO()->getTemporaryUserId();
 
+        $this->relatedUsersStore::load($this->relatedUsersCacheRepository->getAll());
+
         $this->relatedUsersStore->get()->append($userId, $temporaryUserId);
+
         $this->relatedUsersCacheManager->store(
             $this->relatedUsersStore->get()->getAll()
         );
 
-
         return true;
     }
 }