ADO-379: Uses correct primary key for catalog_category_entity table in enterprise vs opensource magento

Author: Priyambad Pandey

app/code/Meta/Catalog/Model/Category/CategoryUtility/CategoryUtilities.php

@@ -34,6 +34,8 @@
 use Meta\BusinessExtension\Helper\FBEHelper;
 use Meta\BusinessExtension\Model\System\Config as SystemConfig;
 use Meta\Catalog\Helper\Product\Identifier as ProductIdentifier;
+use Magento\Framework\EntityManager\MetadataPool;
+use Magento\Catalog\Api\Data\CategoryInterface;
 
 /**
  * @SuppressWarnings(PHPMD.CouplingBetweenObjects)
@@ -93,6 +95,11 @@ class CategoryUtilities
      */
     private ResourceConnection $resourceConnection;
 
+    /**
+     * @var MetadataPool
+     */
+    private MetadataPool $metadataPool;
+
     /**
      * Constructor
      * @param ProductCollectionFactory $productCollectionFactory
@@ -104,6 +111,7 @@ class CategoryUtilities
      * @param CategoryImageService $imageService
      * @param EavConfig $eavConfig
      * @param ResourceConnection $resourceConnection
+     * @param MetadataPool $metadataPool
      */
     public function __construct(
         ProductCollectionFactory    $productCollectionFactory,
@@ -114,7 +122,8 @@ public function __construct(
         ProductIdentifier           $productIdentifier,
         CategoryImageService        $imageService,
         EavConfig                   $eavConfig,
-        ResourceConnection          $resourceConnection
+        ResourceConnection          $resourceConnection,
+        MetadataPool                $metadataPool
     ) {
         $this->categoryCollection = $categoryCollection;
         $this->categoryRepository = $categoryRepository;
@@ -125,6 +134,7 @@ public function __construct(
         $this->imageService = $imageService;
         $this->eavConfig = $eavConfig;
         $this->resourceConnection = $resourceConnection;
+        $this->metadataPool = $metadataPool;
     }
     /**
      * Fetch products for product category
@@ -382,13 +392,16 @@ public function saveFBProductSetID(Category $category, string $setId, $storeId):
                 $categoryEntityVarcharTable = $this->resourceConnection->getTableName(
                     'catalog_category_entity_varchar'
                 );
+                
+                $categoryLinkField = $this->metadataPool->getMetadata(CategoryInterface::class)->getLinkField();
+
                 if ($category->getData(SystemConfig::META_PRODUCT_SET_ID) == null) {
                     $this->resourceConnection->getConnection()->insert(
                         $categoryEntityVarcharTable,
                         [
                             'attribute_id' => $productSetAttributeId,
                             'store_id' => $storeId,
-                            'entity_id' => $category->getId(),
+                            $categoryLinkField => $category->getId(),
                             'value' => $setId
                         ]
                     );
@@ -398,13 +411,13 @@ public function saveFBProductSetID(Category $category, string $setId, $storeId):
                         [
                             'attribute_id' => $productSetAttributeId,
                             'store_id' => $storeId,
-                            'entity_id' => $category->getId(),
+                            $categoryLinkField => $category->getId(),
                             'value' => $setId
                         ],
                         [
                             'attribute_id = ?' => $productSetAttributeId,
                             'store_id = ?' => $storeId,
-                            'entity_id = ?' => $category->getId(),
+                            sprintf('%s = ?', $categoryLinkField) => $category->getId()
                         ]
                     );
                 }