Update TableBuilder.php

Fix to be compatible with both CE and EE.

Author: jarnooravainen

app/code/Magento/Catalog/Model/Indexer/Product/Flat/TableBuilder.php

@@ -74,6 +74,7 @@ public function build($storeId, $changedIds, $valueFieldSuffix)
         $attributes = $this->_productIndexerHelper->getAttributes();
         $eavAttributes = $this->_productIndexerHelper->getTablesStructure($attributes);
         $entityTableColumns = $eavAttributes[$entityTableName];
+        $linkField = $this->getMetadataPool()->getMetadata(\Magento\Catalog\Api\Data\ProductInterface::class)->getLinkField();
 
         $temporaryEavAttributes = $eavAttributes;
 
@@ -101,11 +102,11 @@ public function build($storeId, $changedIds, $valueFieldSuffix)
             $temporaryTableName = $this->_getTemporaryTableName($tableName);
 
             //Add primary key to temporary table for increase speed of joins in future
-            $this->_addPrimaryKeyToTable($temporaryTableName, 'row_id');
+            $this->_addPrimaryKeyToTable($temporaryTableName, $linkField);
 
             //Create temporary table for composite attributes
             if (isset($valueTables[$temporaryTableName . $valueFieldSuffix])) {
-                $this->_addPrimaryKeyToTable($temporaryTableName . $valueFieldSuffix, 'row_id');
+                $this->_addPrimaryKeyToTable($temporaryTableName . $valueFieldSuffix, $linkField);
             }
 
             //Fill temporary tables with attributes grouped by it type