From 393bf9f5d1f2915b3fc82a8a4092976c6cc618fe Mon Sep 17 00:00:00 2001
From: peterjaap <peterjaap@elgentos.nl>
Date: Tue, 16 Apr 2019 14:48:17 +0200
Subject: [PATCH] Added LCP indexer to avoid race conditions, upped version to
 0.3.0 because it might be breaking somewhat

---
 Model/Indexer/Prewarm.php                     | 93 +++++++++++++++++++
 Observer/ProductSaveAfter.php                 | 18 ++--
 .../Action/AfterUpdateAttributesPlugin.php    | 15 ++-
 Plugin/StockItemSaveAround.php                | 29 ++++++
 etc/di.xml                                    |  5 +
 etc/indexer.xml                               |  9 ++
 etc/module.xml                                |  2 +-
 etc/mview.xml                                 | 21 +++++
 8 files changed, 180 insertions(+), 12 deletions(-)
 create mode 100644 Model/Indexer/Prewarm.php
 create mode 100644 Plugin/StockItemSaveAround.php
 create mode 100644 etc/indexer.xml
 create mode 100644 etc/mview.xml

diff --git a/Model/Indexer/Prewarm.php b/Model/Indexer/Prewarm.php
new file mode 100644
index 0000000..7f8bc66
--- /dev/null
+++ b/Model/Indexer/Prewarm.php
@@ -0,0 +1,93 @@
+<?php
+
+namespace Elgentos\LargeConfigProducts\Model\Indexer;
+
+use Elgentos\LargeConfigProducts\Model\PublisherNotifier;
+use Magento\Catalog\Model\ResourceModel\Product\Collection as ProductCollection;
+use Magento\Framework\App\Area;
+use Magento\Framework\App\State;
+use Magento\Framework\Indexer\ActionInterface as IndexerActionInterface;
+use Magento\Framework\Message\ManagerInterface;
+use Magento\Framework\Mview\ActionInterface as MviewActionInterface;
+use Magento\Store\Model\StoreManagerInterface;
+use Symfony\Component\Console\Output\ConsoleOutput;
+use Magento\Catalog\Model\ResourceModel\Product\CollectionFactory as ProductCollectionFactory;
+
+class Prewarm implements IndexerActionInterface, MviewActionInterface
+{
+    /**
+     * @var ProductCollectionFactory
+     */
+    public $productCollectionFactory;
+    private $storeManager;
+    private $messageManager;
+    private $output;
+    /**
+     * @var State
+     */
+    private $state;
+
+    /**
+     * @var PublisherNotifier
+     */
+    protected $publisherNotifier;
+
+    /**
+     * Product constructor.
+     * @param StoreManagerInterface $storeManager
+     * @param ManagerInterface $messageManager
+     * @param ConsoleOutput $output
+     * @param State $state
+     * @param PublisherNotifier $publisherNotifier
+     * @param ProductCollectionFactory $productCollectionFactory
+     */
+    public function __construct(
+        StoreManagerInterface $storeManager,
+        ManagerInterface $messageManager,
+        ConsoleOutput $output,
+        State $state,
+        PublisherNotifier $publisherNotifier,
+        ProductCollectionFactory $productCollectionFactory
+    ) {
+        $this->publisherNotifier = $publisherNotifier;
+        $this->storeManager = $storeManager;
+        $this->messageManager = $messageManager;
+        $this->output = $output;
+        $this->state = $state;
+        $this->productCollectionFactory = $productCollectionFactory;
+    }
+
+    public function execute($productIds)
+    {
+        try {
+            $this->state->setAreaCode(Area::AREA_GLOBAL);
+        } catch (\Exception $e) {
+
+        }
+
+        if (!is_array($productIds)) {
+            /** @var ProductCollection $collection */
+            $collection = $this->productCollectionFactory->create();
+            $productIds = $collection->addAttributeToFilter('type_id', 'configurable')->getAllIds();
+        }
+
+        foreach ($productIds as $productId) {
+            $this->publisherNotifier->notify([$productId]);
+        }
+    }
+
+    public function executeFull()
+    {
+        $this->execute(null);
+    }
+
+    public function executeList(array $ids)
+    {
+        $this->execute($ids);
+    }
+
+    public function executeRow($id)
+    {
+        $this->execute([$id]);
+    }
+}
diff --git a/Observer/ProductSaveAfter.php b/Observer/ProductSaveAfter.php
index f1d4e6f..7c32695 100644
--- a/Observer/ProductSaveAfter.php
+++ b/Observer/ProductSaveAfter.php
@@ -2,17 +2,20 @@
 
 namespace Elgentos\LargeConfigProducts\Observer;
 
-use Elgentos\LargeConfigProducts\Model\PublisherNotifier;
 use Magento\Framework\Event\ObserverInterface;
+use Magento\Framework\Indexer\IndexerRegistry;
 
 class ProductSaveAfter implements ObserverInterface
 {
+    private $indexer;
+
     /**
      * ProductSaveAfter constructor.
-     * @param PublisherNotifier $publisherNotifier
+     * @param IndexerRegistry $indexerRegistry
      */
-    public function __construct(PublisherNotifier $publisherNotifier) {
-        $this->publisherNotifier = $publisherNotifier;
+    public function __construct(IndexerRegistry $indexerRegistry)
+    {
+        $this->indexer = $indexerRegistry->get('elgentos_lcp_prewarm');
     }
 
     /**
@@ -20,7 +23,10 @@ public function __construct(PublisherNotifier $publisherNotifier) {
      */
     public function execute(
         \Magento\Framework\Event\Observer $observer
-    ) {
-        $this->publisherNotifier->notify([$observer->getProduct()->getId()]);
+    )
+    {
+        if (!$this->indexer->isScheduled()) {
+            $this->indexer->reindexRow($observer->getProduct()->getId());
+        }
     }
 }
\ No newline at end of file
diff --git a/Plugin/Product/Action/AfterUpdateAttributesPlugin.php b/Plugin/Product/Action/AfterUpdateAttributesPlugin.php
index d95c591..608eb78 100644
--- a/Plugin/Product/Action/AfterUpdateAttributesPlugin.php
+++ b/Plugin/Product/Action/AfterUpdateAttributesPlugin.php
@@ -2,17 +2,20 @@
 
 namespace Elgentos\LargeConfigProducts\Plugin\Product\Action;
 
-use Elgentos\LargeConfigProducts\Model\PublisherNotifier;
 use Magento\Catalog\Model\Product\Action as ProductAction;
+use Magento\Framework\Indexer\IndexerRegistry;
 
 class AfterUpdateAttributesPlugin {
 
+    private $indexer;
+
     /**
      * AfterUpdateAttributesPlugin constructor.
-     * @param PublisherNotifier $publisherNotifier
+     * @param IndexerRegistry $indexerRegistry
      */
-    public function __construct(PublisherNotifier $publisherNotifier) {
-        $this->publisherNotifier = $publisherNotifier;
+    public function __construct(IndexerRegistry $indexerRegistry)
+    {
+        $this->indexer = $indexerRegistry->get('elgentos_lcp_prewarm');
     }
 
     /**
@@ -30,7 +33,9 @@ public function afterUpdateAttributes(
         $attrData,
         $storeId
     ) {
-        $this->publisherNotifier->notify($productIds);
+        if (!$this->indexer->isScheduled()) {
+            $this->indexer->reindexList($productIds);
+        }
 
         return $action;
     }
diff --git a/Plugin/StockItemSaveAround.php b/Plugin/StockItemSaveAround.php
new file mode 100644
index 0000000..028b551
--- /dev/null
+++ b/Plugin/StockItemSaveAround.php
@@ -0,0 +1,29 @@
+<?php
+
+namespace Elgentos\LargeConfigProducts\Plugin;
+
+use Magento\Framework\Indexer\IndexerRegistry;
+
+class StockItemSaveAround
+{
+    private $indexer;
+
+    public function __construct(IndexerRegistry $indexerRegistry)
+    {
+        $this->indexer = $indexerRegistry->get('elgentos_lcp_prewarm');
+    }
+
+    public function aroundSave(
+        \Magento\CatalogInventory\Model\ResourceModel\Stock\Item $stockItemModel,
+        \Closure $proceed,
+        \Magento\CatalogInventory\Api\Data\StockItemInterface $stockItem
+    ) {
+        $stockItemModel->addCommitCallback(function () use ($stockItem) {
+            if (!$this->indexer->isScheduled()) {
+                $this->indexer->reindexRow($stockItem->getProductId());
+            }
+        });
+
+        return $proceed($stockItem);
+    }
+}
diff --git a/etc/di.xml b/etc/di.xml
index dc2ff3e..0a9f591 100644
--- a/etc/di.xml
+++ b/etc/di.xml
@@ -15,6 +15,11 @@
         </arguments>
     </type>
 
+<!--     Not sure if this one is even needed, comment it out for now -->
+<!--    <type name="Magento\CatalogInventory\Model\ResourceModel\Stock\Item">-->
+<!--        <plugin name="elgentosLCPStockItemPlugin" type="Elgentos\LargeConfigProducts\Plugin\StockItemSaveAround"/>-->
+<!--    </type>-->
+
     <type name="Magento\Catalog\Model\Product\Action">
         <plugin sortOrder="1" name="elgentosLargeConfigProductsAfterUpdateAttributes"
                 type="Elgentos\LargeConfigProducts\Plugin\Product\Action\AfterUpdateAttributesPlugin" />
diff --git a/etc/indexer.xml b/etc/indexer.xml
new file mode 100644
index 0000000..de0ef07
--- /dev/null
+++ b/etc/indexer.xml
@@ -0,0 +1,9 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<config xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:noNamespaceSchemaLocation="urn:magento:framework:Indexer/etc/indexer.xsd">
+    <indexer id="elgentos_lcp_prewarm" view_id="elgentos_lcp_prewarm" class="Elgentos\LargeConfigProducts\Model\Indexer\Prewarm">
+        <title translate="true">Large Config Products prewarmer</title>
+        <description translate="true">
+            Adds products to the LCP message queue to be prewarmed.
+        </description>
+    </indexer>
+</config>
diff --git a/etc/module.xml b/etc/module.xml
index 90b5144..c80a869 100644
--- a/etc/module.xml
+++ b/etc/module.xml
@@ -1,7 +1,7 @@
 <?xml version="1.0"?>
 <config xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
         xsi:noNamespaceSchemaLocation="urn:magento:framework:Module/etc/module.xsd">
-    <module name="Elgentos_LargeConfigProducts" setup_version="0.1.11">
+    <module name="Elgentos_LargeConfigProducts" setup_version="0.3.0">
         <sequence>
             <module name="Magento_ConfigurableProduct" />
             <module name="Magento_Framework" />
diff --git a/etc/mview.xml b/etc/mview.xml
new file mode 100644
index 0000000..08d6e68
--- /dev/null
+++ b/etc/mview.xml
@@ -0,0 +1,21 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<config xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:noNamespaceSchemaLocation="urn:magento:framework:Mview/etc/mview.xsd">
+    <view id="elgentos_lcp_prewarm" class="Elgentos\LargeConfigProducts\Model\Indexer\Prewarm" group="indexer">
+        <subscriptions>
+            <table name="catalog_product_entity" entity_column="entity_id" />
+            <table name="catalog_product_entity_datetime" entity_column="entity_id" />
+            <table name="catalog_product_entity_decimal" entity_column="entity_id" />
+            <table name="catalog_product_entity_gallery" entity_column="entity_id" />
+            <table name="catalog_product_entity_int" entity_column="entity_id" />
+            <table name="catalog_product_entity_media_gallery" entity_column="value_id" />
+            <table name="catalog_product_entity_media_gallery_value" entity_column="entity_id" />
+            <table name="catalog_product_entity_text" entity_column="entity_id" />
+            <table name="catalog_product_entity_tier_price" entity_column="entity_id" />
+            <table name="catalog_product_entity_varchar" entity_column="entity_id" />
+            <table name="catalog_product_website" entity_column="product_id" />
+            <table name="catalog_category_product" entity_column="product_id" />
+            <table name="cataloginventory_stock_item" entity_column="product_id" />
+            <table name="catalog_product_index_price" entity_column="entity_id" />
+        </subscriptions>
+    </view>
+</config>
\ No newline at end of file