Store related users in the Redis (#28)

Author: yatsenkolesh

src/DB/Dictionary/ParallelRunningDictionary.php

@@ -8,6 +8,7 @@ class ParallelRunningDictionary
 {
     public const IS_INITIAZLIED_KEY = 'parallelRunningIsInitialized';
     public const IS_RUNNING_KEY = 'parallelRunningReadyToServe';
+    public const RELATED_USERS_KEY = 'parallelRunningRelatedUsers';
 
     public const IS_INITIAZLIED_TRUE_VALUE = 'initialized';
     public const IS_RUNNING_VALUE_TRUE = 'ready';

src/DB/Managers/RelatedUsersCacheManager.php

@@ -0,0 +1,28 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Abrouter\Client\DB\Managers;
+
+use Abrouter\Client\Config\Accessors\KvStorageConfigAccessor;
+use Abrouter\Client\DB\Dictionary\ParallelRunningDictionary;
+use Abrouter\Client\Services\KvStorage\KvStorage;
+
+class RelatedUsersCacheManager
+{
+    private KvStorageConfigAccessor $kvStorage;
+
+    public function __construct(
+        KvStorageConfigAccessor $kvStorage
+    ) {
+        $this->kvStorage = $kvStorage;
+    }
+
+    public function store(array $relatedUsers): void
+    {
+        $this
+            ->kvStorage
+            ->getKvStorage()
+            ->put(ParallelRunningDictionary::RELATED_USERS_KEY, json_encode($relatedUsers));
+    }
+}

src/DB/RelatedUsersStore.php

@@ -27,4 +27,9 @@ public static function load(array $relatedUsersList): void
     {
         self::$store = new RelatedUsersCollection($relatedUsersList);
     }
+
+    public static function isLoaded(): bool
+    {
+        return self::$store !== null;
+    }
 }

src/DB/Repositories/RelatedUsersCacheRepository.php

@@ -0,0 +1,26 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Abrouter\Client\DB\Repositories;
+
+use Abrouter\Client\Config\Accessors\KvStorageConfigAccessor;
+use Abrouter\Client\DB\Dictionary\ParallelRunningDictionary;
+use Abrouter\Client\Services\KvStorage\KvStorage;
+
+class RelatedUsersCacheRepository
+{
+    private KvStorageConfigAccessor $kvStorage;
+
+    public function __construct(
+        KvStorageConfigAccessor $kvStorage
+    ) {
+        $this->kvStorage = $kvStorage;
+    }
+
+    public function getAll(): array
+    {
+        $json = $this->kvStorage->getKvStorage()->get(ParallelRunningDictionary::RELATED_USERS_KEY);
+        return $json === null ? [] : json_decode($json, true);
+    }
+}

src/Events/Handlers/RelatedUsersStatisticsInterceptor.php

@@ -5,6 +5,7 @@
 namespace Abrouter\Client\Events\Handlers;
 
 use Abrouter\Client\Contracts\TaskContract;
+use Abrouter\Client\DB\Managers\RelatedUsersCacheManager;
 use Abrouter\Client\DB\RelatedUsersStore;
 use Abrouter\Client\Events\HandlerInterface;
 use Abrouter\Client\Services\ExperimentsParallelRun\ParallelRunSwitch;
@@ -16,12 +17,16 @@ class RelatedUsersStatisticsInterceptor implements HandlerInterface
 
     private RelatedUsersStore $relatedUsersStore;
 
+    private RelatedUsersCacheManager $relatedUsersCacheManager;
+
     public function __construct(
         ParallelRunSwitch $parallelRunSwitch,
-        RelatedUsersStore $relatedUsersStore
+        RelatedUsersStore $relatedUsersStore,
+        RelatedUsersCacheManager $relatedUsersCacheManager
     ) {
         $this->parallelRunSwitch = $parallelRunSwitch;
         $this->relatedUsersStore = $relatedUsersStore;
+        $this->relatedUsersCacheManager = $relatedUsersCacheManager;
     }
 
     public function handle(TaskContract $taskContract): bool
@@ -38,6 +43,10 @@ public function handle(TaskContract $taskContract): bool
         $temporaryUserId = $taskContract->getEventDTO()->getBaseEventDTO()->getTemporaryUserId();
 
         $this->relatedUsersStore->get()->append($userId, $temporaryUserId);
+        $this->relatedUsersCacheManager->store(
+            $this->relatedUsersStore->get()->getAll()
+        );
+
 
         return true;
     }

src/Services/ExperimentsParallelRun/ParallelRunInitializer.php

@@ -7,23 +7,33 @@
 use Abrouter\Client\DB\Managers\ParallelRunningStateManager;
 use Abrouter\Client\DB\RelatedUsersStore;
 use Abrouter\Client\DB\Repositories\ParallelRunningStateCachedRepository;
+use Abrouter\Client\DB\Repositories\RelatedUsersCacheRepository;
 
 class ParallelRunInitializer
 {
     private ParallelRunningStateCachedRepository $parallelRunningStateCachedRepository;
 
     private ParallelRunningStateManager $parallelRunningStateManager;
 
+    private RelatedUsersCacheRepository $relatedUsersCacheRepository;
+
     public function __construct(
         ParallelRunningStateCachedRepository $parallelRunningStateCachedRepository,
-        ParallelRunningStateManager $parallelRunningStateManager
+        ParallelRunningStateManager $parallelRunningStateManager,
+        RelatedUsersCacheRepository $relatedUsersCacheRepository
     ) {
         $this->parallelRunningStateCachedRepository = $parallelRunningStateCachedRepository;
         $this->parallelRunningStateManager = $parallelRunningStateManager;
+        $this->relatedUsersCacheRepository = $relatedUsersCacheRepository;
     }
 
     public function initializeIfNot(): bool
     {
+        if (!RelatedUsersStore::isLoaded()) {
+            //initialization related users
+            RelatedUsersStore::load($this->relatedUsersCacheRepository->getAll());
+        }
+
         //if parallel running ready to serve
         if ($this->parallelRunningStateCachedRepository->isReady()) {
             return true;
@@ -34,9 +44,6 @@ public function initializeIfNot(): bool
             return false;
         }
 
-        //initialization
-        RelatedUsersStore::load([]);
-
         $this->parallelRunningStateManager->setInitialized();
         $this->parallelRunningStateManager->setReadyToServe();
 

tests/Integration/DB/Managers/RelatedUsersCacheManagerTest.php

@@ -0,0 +1,29 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Abrouter\Client\Tests\Integration\DB\Managers;
+
+use Abrouter\Client\DB\Dictionary\ParallelRunningDictionary;
+use Abrouter\Client\DB\RedisConnection;
+use Abrouter\Client\DB\Repositories\RelatedUsersCacheRepository;
+use Abrouter\Client\Tests\Integration\IntegrationTestCase;
+
+class RelatedUsersCacheManagerTest extends IntegrationTestCase
+{
+    public function testSavingSomeUsers()
+    {
+        $this->configureParallelRun('default');
+        $this->clearRedis();
+
+        $relatedUsersList = ['test' => ['test2']];
+        $relatedUsersCacheRepository = $this->getContainer()->make(RelatedUsersCacheRepository::class);
+        $redis = $this->getContainer()->make(RedisConnection::class);
+        $redis->getConnection()->set(
+            ParallelRunningDictionary::RELATED_USERS_KEY,
+            json_encode($relatedUsersList)
+        );
+
+        $this->assertEquals($relatedUsersCacheRepository->getAll(), $relatedUsersList);
+    }
+}

tests/Integration/DB/Repositories/RelatedUsersCacheRepositoryTest.php

@@ -0,0 +1,31 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Abrouter\Client\Tests\Integration\DB\Repositories;
+
+use Abrouter\Client\DB\Dictionary\ParallelRunningDictionary;
+use Abrouter\Client\DB\Managers\RelatedUsersCacheManager;
+use Abrouter\Client\DB\RedisConnection;
+use Abrouter\Client\Tests\Integration\IntegrationTestCase;
+
+class RelatedUsersCacheRepositoryTest extends IntegrationTestCase
+{
+    public function testGet()
+    {
+        $this->configureParallelRun('default');
+        $this->clearRedis();
+
+        $usersList = [
+            'test' => ['test'],
+        ];
+        $relatedUsersCacheManager = $this->getContainer()->make(RelatedUsersCacheManager::class);
+        $relatedUsersCacheManager->store($usersList);
+
+        $redis = $this->getContainer()->make(RedisConnection::class);
+        $this->assertEquals(
+            $usersList,
+            json_decode($redis->getConnection()->get(ParallelRunningDictionary::RELATED_USERS_KEY), true)
+        );
+    }
+}