save vary data to db

Author: David Verholen

Plugin/CacheIdentifier/SaveVaryData.php

@@ -0,0 +1,46 @@
+<?php
+declare(strict_types=1);
+/**
+ * SaveVaryData
+ *
+ * @copyright Copyright © 2019 brandung GmbH & Co. KG. All rights reserved.
+ * @author    david.verholen@brandung.de
+ */
+
+namespace Firegento\CacheWarmup\Plugin\CacheIdentifier;
+
+use Firegento\CacheWarmup\Api\CacheVaryDataRepositoryInterface;
+use Magento\Framework\App\Http\Context;
+
+class SaveVaryData
+{
+    /**
+     * @var Context
+     */
+    private $httpContext;
+    /**
+     * @var CacheVaryDataRepositoryInterface
+     */
+    private $varyDataRepository;
+
+    public function __construct(Context $httpContext, CacheVaryDataRepositoryInterface $varyDataRepository)
+    {
+        $this->httpContext = $httpContext;
+        $this->varyDataRepository = $varyDataRepository;
+    }
+
+    /**
+     * Adds a theme key to identifier for a built-in cache if user-agent theme rule is actual
+     *
+     * @param \Magento\Framework\App\PageCache\Identifier $identifier
+     * @param string $result
+     * @return string
+     * @SuppressWarnings(PHPMD.UnusedFormalParameter)
+     */
+    public function afterGetValue(\Magento\Framework\App\PageCache\Identifier $identifier, $result): string
+    {
+        $varyData = $this->httpContext->toArray()['data'];
+        $this->varyDataRepository->save($varyData);
+        return $result;
+    }
+}

Setup/InstallSchema.php

@@ -61,25 +61,25 @@ private function getCacheTagTableDefinition(SchemaSetupInterface $setup): \Magen
     private function getVaryDataTableDefinition(SchemaSetupInterface $setup): \Magento\Framework\DB\Ddl\Table
     {
         return $setup->getConnection()->newTable($setup->getTable('cache_vary_data'))
-            ->addColumn(
-                'id',
-                \Magento\Framework\DB\Ddl\Table::TYPE_INTEGER,
-                null,
-                [
-                    'identity' => true,
-                    'nullable' => false,
-                    'primary' => true,
-                    'unsigned' => true,
-                ],
-                'Cache Vary Data ID'
-            )
             ->addColumn(
                 'vary_data',
                 \Magento\Framework\DB\Ddl\Table::TYPE_TEXT,
                 255,
-                ['nullable' => false],
+                [
+                    'nullable' => false,
+                    'primary' => true
+                ],
                 'Cache Vary Data'
             )
+            ->addIndex(
+                $setup->getIdxName(
+                    'cache_vary_data',
+                    ['vary_data'],
+                    \Magento\Framework\DB\Adapter\AdapterInterface::INDEX_TYPE_UNIQUE
+                ),
+                ['vary_data'],
+                ['type' => \Magento\Framework\DB\Adapter\AdapterInterface::INDEX_TYPE_UNIQUE]
+            )
             ->setComment('Cache Vary Data Table');
     }
 

etc/di.xml

@@ -33,4 +33,9 @@
             type="Firegento\CacheWarmup\Model\Tag\CacheTag"
     />
 
+    <preference
+            for="Firegento\CacheWarmup\Api\CacheVaryDataRepositoryInterface"
+            type="Firegento\CacheWarmup\Model\CacheVaryDataRepository"
+    />
+
 </config>

etc/frontend/di.xml

@@ -0,0 +1,6 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<config xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:noNamespaceSchemaLocation="urn:magento:framework:ObjectManager/etc/config.xsd">
+    <type name="Magento\Framework\App\PageCache\Identifier">
+        <plugin name="saveVaryData" type="\Firegento\CacheWarmup\Plugin\CacheIdentifier\SaveVaryData"/>
+    </type>
+</config>