B2B-2677: [MediaGallery] Implement data caching for GraphQL results on resolver level

Author: joanhe

app/code/Magento/CatalogGraphQl/Model/Resolver/Cache/Product/MediaGallery/ResolverCacheIdentity.php

@@ -31,7 +31,7 @@ public function getIdentities($resolvedData, ?array $parentResolvedData = null):
         /** @var Product $mediaGalleryEntryProduct */
         $mediaGalleryEntryProduct = array_pop($resolvedData)['model'];
         return [
-            sprintf('%s_%s', self::CACHE_TAG, $mediaGalleryEntryProduct->getSku())
+            sprintf('%s_%s', self::CACHE_TAG, $mediaGalleryEntryProduct->getId())
         ];
     }
 }

app/code/Magento/CatalogGraphQl/Model/Resolver/Cache/Product/MediaGallery/TagsStrategy.php

@@ -36,7 +36,7 @@ public function getTags($object)
             $this->mediaGalleryChangeDetector->isChanged($object)
         ) {
             return [
-                sprintf('%s_%s', ResolverCacheIdentity::CACHE_TAG, $object->getSku())
+                sprintf('%s_%s', ResolverCacheIdentity::CACHE_TAG, $object->getId())
             ];
         }
 

app/code/Magento/CatalogGraphQl/Observer/AfterImportDataObserver.php

@@ -7,6 +7,8 @@
 
 namespace Magento\CatalogGraphQl\Observer;
 
+use Magento\Catalog\Model\ProductRepository;
+use Magento\Framework\Api\SearchCriteriaBuilder;
 use Magento\Framework\Event\Observer;
 use Magento\CatalogGraphQl\Model\Resolver\Cache\Product\MediaGallery\ResolverCacheIdentity;
 use Magento\Framework\Event\ObserverInterface;
@@ -22,12 +24,29 @@ class AfterImportDataObserver implements ObserverInterface
      */
     private $graphQlResolverCache;
 
+    /**
+     * @var ProductRepository
+     */
+    private $productRepository;
+
+    /**
+     * @var SearchCriteriaBuilder
+     */
+    private $criteriaBuilder;
+
     /**
      * @param GraphQlResolverCache $graphQlResolverCache
+     * @param ProductRepository $productRepository
+     * @param SearchCriteriaBuilder $criteriaBuilder
      */
-    public function __construct(GraphQlResolverCache $graphQlResolverCache)
-    {
+    public function __construct(
+        GraphQlResolverCache $graphQlResolverCache,
+        ProductRepository $productRepository,
+        SearchCriteriaBuilder $criteriaBuilder
+    ) {
         $this->graphQlResolverCache = $graphQlResolverCache;
+        $this->productRepository = $productRepository;
+        $this->criteriaBuilder = $criteriaBuilder;
     }
 
     /**
@@ -48,10 +67,13 @@ public function execute(Observer $observer)
         }
 
         $productSkusToInvalidate = array_merge(...$productSkusToInvalidate);
+        $products = $this->productRepository->getList(
+            $this->criteriaBuilder->addFilter('sku', $productSkusToInvalidate, 'in')->create()
+        )->getItems();
 
-        $tags = array_map(function ($productSku) {
-            return sprintf('%s_%s', ResolverCacheIdentity::CACHE_TAG, $productSku);
-        }, $productSkusToInvalidate);
+        $tags = array_map(function ($product) {
+            return sprintf('%s_%s', ResolverCacheIdentity::CACHE_TAG, $product->getId());
+        }, $products);
 
         $this->graphQlResolverCache->clean(
             \Zend_Cache::CLEANING_MODE_MATCHING_ANY_TAG,

dev/tests/api-functional/testsuite/Magento/GraphQl/Catalog/ResolverCache/MediaGalleryTest.php

@@ -466,7 +466,7 @@ private function getExpectedCacheTags(ProductInterface $product): array
         $cacheIdPrefix = $this->graphQlResolverCache->getLowLevelFrontend()->getOption('cache_id_prefix');
 
         return [
-            $cacheIdPrefix . strtoupper(ResolverCacheIdentity::CACHE_TAG . '_' . $product->getSku()),
+            $cacheIdPrefix . strtoupper(ResolverCacheIdentity::CACHE_TAG . '_' . $product->getId()),
             $cacheIdPrefix . strtoupper(GraphQlResolverCache::CACHE_TAG),
             $cacheIdPrefix . 'MAGE',
         ];