MAGETWO-40057: Create Grid Table

magento · Aug 7, 2015 · dc19322 · dc19322
1 parent fe60ed7
commit dc19322
Showing 1 changed file with 26 additions and 6 deletions.
diff --git a/app/code/Magento/Customer/Setup/UpgradeData.php b/app/code/Magento/Customer/Setup/UpgradeData.php
@@ -6,7 +6,8 @@
 
 namespace Magento\Customer\Setup;
 
-use Magento\Framework\Module\Setup\Migration;
+use Magento\Customer\Model\Customer;
+use Magento\Framework\Indexer\IndexerRegistry;
 use Magento\Framework\Setup\UpgradeDataInterface;
 use Magento\Framework\Setup\ModuleContextInterface;
 use Magento\Framework\Setup\ModuleDataSetupInterface;
@@ -21,16 +22,31 @@ class UpgradeData implements UpgradeDataInterface
      *
      * @var CustomerSetupFactory
      */
-    private $customerSetupFactory;
+    protected $customerSetupFactory;
+
+    /**
+     * @var IndexerRegistry
+     */
+    protected $indexerRegistry;
+
+    /**
+     * @var \Magento\Eav\Model\Config
+     */
+    protected $eavConfig;
 
     /**
-     * Init
-     *
      * @param CustomerSetupFactory $customerSetupFactory
+     * @param IndexerRegistry $indexerRegistry
+     * @param \Magento\Eav\Model\Config $eavConfig
      */
-    public function __construct(CustomerSetupFactory $customerSetupFactory)
-    {
+    public function __construct(
+        CustomerSetupFactory $customerSetupFactory,
+        IndexerRegistry $indexerRegistry,
+        \Magento\Eav\Model\Config $eavConfig
+    ) {
         $this->customerSetupFactory = $customerSetupFactory;
+        $this->indexerRegistry = $indexerRegistry;
+        $this->eavConfig = $eavConfig;
     }
 
     /**
@@ -171,6 +187,10 @@ public function upgrade(ModuleDataSetupInterface $setup, ModuleContextInterface
             }
         }
 
+        $indexer = $this->indexerRegistry->get(Customer::CUSTOMER_GRID_INDEXER_ID);
+        $indexer->reindexAll();
+        $this->eavConfig->clear();
+
         $setup->endSetup();
     }
 }