B2B-2658: Implement GraphQL Resolver Cache for Customer query

Author: rganin

app/code/Magento/GraphQlResolverCache/Model/Resolver/Result/ValueProcessor.php

@@ -124,14 +124,20 @@ public function preProcessParentValue(array &$value): void
      */
     private function hydrateData(&$value)
     {
-        $key = $this->defaultFlagGetter->getFlagFromValue($value) ?? null;
-        if ($value && $key) {
-            if (isset($this->processedValues[$key])) {
-                $value = $this->processedValues[$key];
-            } elseif (isset($this->hydrators[$key]) && $this->hydrators[$key] instanceof HydratorInterface) {
-                $this->hydrators[$key]->hydrate($value);
-                $this->defaultFlagSetter->unsetFlagFromValue($value);
-                $this->processedValues[$key] = $value;
+        $reference = $this->defaultFlagGetter->getFlagFromValue($value) ?? null;
+        if (isset($reference['cacheKey']) && isset($reference['index'])) {
+            $cacheKey = $reference['cacheKey'];
+            $index = $reference['index'];
+            if ($value && $cacheKey) {
+                if (isset($this->processedValues[$cacheKey][$index])) {
+                    $value = $this->processedValues[$cacheKey][$index];
+                } elseif (isset($this->hydrators[$cacheKey])
+                    && $this->hydrators[$cacheKey] instanceof HydratorInterface
+                ) {
+                    $this->hydrators[$cacheKey]->hydrate($value);
+                    $this->defaultFlagSetter->unsetFlagFromValue($value);
+                    $this->processedValues[$cacheKey][$index] = $value;
+                }
             }
         }
     }

app/code/Magento/GraphQlResolverCache/Model/Resolver/Result/ValueProcessor/FlagGetter/FlagGetterInterface.php

@@ -18,5 +18,5 @@ interface FlagGetterInterface
      * @param array $value
      * @return string|null
      */
-    public function getFlagFromValue(&$value): ?string;
+    public function getFlagFromValue($value): ?array;
 }

app/code/Magento/GraphQlResolverCache/Model/Resolver/Result/ValueProcessor/FlagGetter/Generic.php

@@ -17,7 +17,7 @@ class Generic implements FlagGetterInterface
     /**
      * @inheritdoc
      */
-    public function getFlagFromValue(&$value): ?string
+    public function getFlagFromValue($value): ?array
     {
         return $value[ValueProcessorInterface::VALUE_HYDRATION_REFERENCE_KEY] ?? null;
     }

app/code/Magento/GraphQlResolverCache/Model/Resolver/Result/ValueProcessor/FlagSetter/Enumerable.php

@@ -20,7 +20,10 @@ class Enumerable implements FlagSetterInterface
     public function setFlagOnValue(&$value, string $flagValue): void
     {
         foreach ($value as $key => $data) {
-            $value[$key][ValueProcessorInterface::VALUE_HYDRATION_REFERENCE_KEY] = $flagValue;
+            $value[$key][ValueProcessorInterface::VALUE_HYDRATION_REFERENCE_KEY] = [
+                'cacheKey' => $flagValue,
+                'index' => $key
+            ];
         }
     }
 

app/code/Magento/GraphQlResolverCache/Model/Resolver/Result/ValueProcessor/FlagSetter/Generic.php

@@ -19,7 +19,10 @@ class Generic implements FlagSetterInterface
      */
     public function setFlagOnValue(&$value, string $flagValue): void
     {
-        $value[ValueProcessorInterface::VALUE_HYDRATION_REFERENCE_KEY] = $flagValue;
+        $value[ValueProcessorInterface::VALUE_HYDRATION_REFERENCE_KEY] = [
+            'cacheKey' => $flagValue,
+            'index' => 0
+        ];
     }
 
     /**