MC-39463: GraphQL caches urlResolver response and can return the old value after the url rewrite update

Author: thiaramus

app/code/Magento/UrlRewrite/Model/UrlRewrite.php

@@ -7,8 +7,17 @@
 
 namespace Magento\UrlRewrite\Model;
 
+use Magento\Catalog\Api\CategoryRepositoryInterface;
+use Magento\Catalog\Api\ProductRepositoryInterface;
+use Magento\Catalog\Model\Category;
+use Magento\Catalog\Model\Product;
+use Magento\Cms\Api\PageRepositoryInterface;
+use Magento\Cms\Model\Page;
 use Magento\Framework\App\ObjectManager;
+use Magento\Framework\EntityManager\EventManager;
+use Magento\Framework\Indexer\CacheContext;
 use Magento\Framework\Serialize\Serializer\Json;
+use Magento\UrlRewrite\Controller\Adminhtml\Url\Rewrite;
 
 /**
  * UrlRewrite model class
@@ -92,4 +101,58 @@ public function setMetadata($metadata)
         }
         return $this->setData(\Magento\UrlRewrite\Service\V1\Data\UrlRewrite::METADATA, $metadata);
     }
+
+    private function opt1() {
+        $map = [
+            Rewrite::ENTITY_TYPE_PRODUCT => Product::CACHE_TAG,
+            Rewrite::ENTITY_TYPE_CATEGORY => Category::CACHE_TAG,
+            Rewrite::ENTITY_TYPE_CMS_PAGE => Page::CACHE_TAG
+        ];
+
+        if ($this->getEntityType() !== Rewrite::ENTITY_TYPE_CUSTOM) {
+            $cacheKey = $map[$this->getEntityType()];
+
+            $cacheContext = ObjectManager::getInstance()->get(CacheContext::class);
+            $eventManager = ObjectManager::getInstance()->get(EventManager::class);
+
+            $cacheContext->registerEntities($cacheKey, [$this->getEntityId()]);
+            $eventManager->dispatch('clean_cache_by_tags', ['object' => $cacheContext]);
+        }
+    }
+
+    private function opt2() {
+        $map = [
+            Rewrite::ENTITY_TYPE_PRODUCT => function ($prodId) {
+                /** @var ProductRepositoryInterface $productRepository */
+                $productRepository = ObjectManager::getInstance()->get(ProductRepositoryInterface::class);
+                return $productRepository->getById($prodId);
+            },
+            Rewrite::ENTITY_TYPE_CATEGORY => function ($catId) {
+                /** @var CategoryRepositoryInterface $productRepository */
+                $categoryRepository = ObjectManager::getInstance()->get(CategoryRepositoryInterface::class);
+                return $categoryRepository->get($catId);
+            },
+            Rewrite::ENTITY_TYPE_CMS_PAGE => function ($cmsId) {
+                /** @var PageRepositoryInterface $productRepository */
+                $pageRepository = ObjectManager::getInstance()->get(PageRepositoryInterface::class);
+                return $pageRepository->getById($cmsId);
+            },
+            Rewrite::ENTITY_TYPE_CUSTOM => false
+        ];
+
+        $getter = $map[$this->getEntityType()];
+
+        if ($getter) {
+            $entity = $getter($this->getEntityId());
+
+            $entityManager = ObjectManager::getInstance()->get(EventManager::class);
+            $entityManager->dispatch('clean_cache_by_tags', ['object' => $entity]);
+        }
+    }
+
+    public function afterSave()
+    {
+        $this->opt1();
+        return parent::afterSave(); // TODO: Change the autogenerated stub
+    }
 }