MAGETWO-33564: [GitHub] Database Schema: Incorrect Unique Indexes #1002

Author: RuslanKostiv1

app/code/Magento/CatalogInventory/Setup/InstallSchema.php

@@ -254,10 +254,10 @@ public function install(SchemaSetupInterface $setup, ModuleContextInterface $con
             ->addIndex(
                 $installer->getIdxName(
                     'cataloginventory_stock_item',
-                    ['product_id', 'website_id'],
+                    ['product_id', 'stock_id'],
                     \Magento\Framework\DB\Adapter\AdapterInterface::INDEX_TYPE_UNIQUE
                 ),
-                ['product_id', 'website_id'],
+                ['product_id', 'stock_id'],
                 ['type' => \Magento\Framework\DB\Adapter\AdapterInterface::INDEX_TYPE_UNIQUE]
             )
             ->addIndex(

app/code/Magento/CatalogInventory/Setup/UpgradeSchema.php

@@ -0,0 +1,69 @@
+<?php
+/**
+ * Copyright © 2016 Magento. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+
+namespace Magento\CatalogInventory\Setup;
+
+use Magento\Framework\Setup\UpgradeSchemaInterface;
+use Magento\Framework\Setup\ModuleContextInterface;
+use Magento\Framework\Setup\SchemaSetupInterface;
+
+class UpgradeSchema implements UpgradeSchemaInterface
+{
+    /**
+     * @var string
+     */
+    private $productCompositeKeyVersion = '2.2.0';
+
+    /**
+     * {@inheritdoc}
+     */
+    public function upgrade(SchemaSetupInterface $setup, ModuleContextInterface $context)
+    {
+        $setup->startSetup();
+
+        if (version_compare($context->getVersion(), $this->productCompositeKeyVersion, '<')) {
+            $this->upgradeProductCompositeKey($setup);
+        }
+
+        $setup->endSetup();
+    }
+
+    /**
+     * @param SchemaSetupInterface $setup
+     * @return void
+     */
+    private function upgradeProductCompositeKey(SchemaSetupInterface $setup)
+    {
+        $oldCompositeKeyColumns = ['product_id', 'website_id'];
+        $newCompositeKeyColumns = ['product_id', 'stock_id'];
+
+        $oldIndexName = $setup->getIdxName(
+            \Magento\CatalogInventory\Model\Stock\Item::ENTITY,
+            $oldCompositeKeyColumns,
+            \Magento\Framework\DB\Adapter\AdapterInterface::INDEX_TYPE_UNIQUE
+        );
+
+        $newIndexName = $setup->getIdxName(
+            \Magento\CatalogInventory\Model\Stock\Item::ENTITY,
+            $newCompositeKeyColumns,
+            \Magento\Framework\DB\Adapter\AdapterInterface::INDEX_TYPE_UNIQUE
+        );
+
+        // Drop a key based on the following columns: "product_id","website_id"
+        $setup->getConnection()->dropIndex(
+            $setup->getTable(\Magento\CatalogInventory\Model\Stock\Item::ENTITY),
+            $oldIndexName
+        );
+
+        // Create a key based on the following columns: "product_id","stock_id"
+        $setup->getConnection()->addIndex(
+            \Magento\CatalogInventory\Model\Stock\Item::ENTITY,
+            $newIndexName,
+            $newCompositeKeyColumns,
+            \Magento\Framework\DB\Adapter\AdapterInterface::INDEX_TYPE_UNIQUE
+        );
+    }
+}

app/code/Magento/CatalogInventory/etc/module.xml

@@ -6,7 +6,7 @@
  */
 -->
 <config xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:noNamespaceSchemaLocation="urn:magento:framework:Module/etc/module.xsd">
-    <module name="Magento_CatalogInventory" setup_version="2.0.2">
+    <module name="Magento_CatalogInventory" setup_version="2.2.0">
         <sequence>
             <module name="Magento_Catalog"/>
         </sequence>