Add cache to AuthCodeRepository

Author: cicnavi

src/Repositories/AuthCodeRepository.php

@@ -32,6 +32,8 @@
 
 class AuthCodeRepository extends AbstractDatabaseRepository implements AuthCodeRepositoryInterface
 {
+    final public const TABLE_NAME = 'oidc_auth_code';
+
     public function __construct(
         ModuleConfig $moduleConfig,
         Database $database,
@@ -43,8 +45,6 @@ public function __construct(
         parent::__construct($moduleConfig, $database, $protocolCache);
     }
 
-    final public const TABLE_NAME = 'oidc_auth_code';
-
     public function getTableName(): string
     {
         return $this->database->applyPrefix(self::TABLE_NAME);
@@ -79,6 +79,14 @@ public function persistNewAuthCode(OAuth2AuthCodeEntityInterface $authCodeEntity
             $stmt,
             $this->preparePdoState($authCodeEntity->getState()),
         );
+
+        $this->protocolCache?->set(
+            $authCodeEntity->getState(),
+            $this->helpers->dateTime()->getSecondsToExpirationTime(
+                $authCodeEntity->getExpiryDateTime()->getTimestamp(),
+            ),
+            $this->getCacheKey((string)$authCodeEntity->getIdentifier()),
+        );
     }
 
     /**
@@ -87,22 +95,38 @@ public function persistNewAuthCode(OAuth2AuthCodeEntityInterface $authCodeEntity
      */
     public function findById(string $codeId): ?AuthCodeEntityInterface
     {
-        $stmt = $this->database->read(
-            "SELECT * FROM {$this->getTableName()} WHERE id = :id",
-            [
-                'id' => $codeId,
-            ],
-        );
-
-        if (empty($rows = $stmt->fetchAll())) {
-            return null;
+        /** @var ?array $data */
+        $data = $this->protocolCache?->get(null, $this->getCacheKey($codeId));
+
+        if (!is_array($data)) {
+            $stmt = $this->database->read(
+                "SELECT * FROM {$this->getTableName()} WHERE id = :id",
+                [
+                    'id' => $codeId,
+                ],
+            );
+
+            if (empty($rows = $stmt->fetchAll())) {
+                return null;
+            }
+
+            /** @var array $data */
+            $data = current($rows);
         }
 
-        /** @var array $data */
-        $data = current($rows);
         $data['client'] = $this->clientRepository->findById((string)$data['client_id']);
 
-        return $this->authCodeEntityFactory->fromState($data);
+        $authCodeEntity = $this->authCodeEntityFactory->fromState($data);
+
+        $this->protocolCache?->set(
+            $authCodeEntity->getState(),
+            $this->helpers->dateTime()->getSecondsToExpirationTime(
+                $authCodeEntity->getExpiryDateTime()->getTimestamp(),
+            ),
+            $this->getCacheKey((string)$authCodeEntity->getIdentifier()),
+        );
+
+        return $authCodeEntity;
     }
 
     /**
@@ -177,6 +201,14 @@ private function update(AuthCodeEntity $authCodeEntity): void
             $stmt,
             $this->preparePdoState($authCodeEntity->getState()),
         );
+
+        $this->protocolCache?->set(
+            $authCodeEntity->getState(),
+            $this->helpers->dateTime()->getSecondsToExpirationTime(
+                $authCodeEntity->getExpiryDateTime()->getTimestamp(),
+            ),
+            $this->getCacheKey((string)$authCodeEntity->getIdentifier()),
+        );
     }
 
     protected function preparePdoState(array $state): array