MC-38995: Customer group price is not working in product query graphql

Author: Oleksandr Iegorov

app/code/Magento/CatalogGraphQl/Model/Resolver/Products/DataProvider/Product/CollectionProcessor/PriceDataProcessor.php

@@ -0,0 +1,45 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+declare(strict_types=1);
+
+namespace Magento\CatalogGraphQl\Model\Resolver\Products\DataProvider\Product\CollectionProcessor;
+
+use Magento\Catalog\Model\ResourceModel\Product\Collection;
+use Magento\CatalogGraphQl\Model\Resolver\Products\DataProvider\Product\CollectionProcessorInterface;
+use Magento\Framework\Api\SearchCriteriaInterface;
+use Magento\GraphQl\Model\Query\ContextInterface;
+
+/**
+ * Add price data to product collection results
+ */
+class PriceDataProcessor implements CollectionProcessorInterface
+{
+    /**
+     * Process to add price data to product collection.
+     *
+     * @param Collection $collection
+     * @param SearchCriteriaInterface $searchCriteria
+     * @param array $attributeNames
+     * @param ContextInterface|null $context
+     * @return Collection
+     * @SuppressWarnings(PHPMD.UnusedFormalParameter)
+     */
+    public function process(
+        Collection $collection,
+        SearchCriteriaInterface $searchCriteria,
+        array $attributeNames,
+        ?ContextInterface $context = null
+    ): Collection {
+        if ($context) {
+            $customerGroupId = $context->getExtensionAttributes()->getCustomerGroupId();
+            if ($customerGroupId !== null) {
+                $collection->addPriceData($customerGroupId);
+            }
+        }
+
+        return $collection;
+    }
+}

app/code/Magento/CatalogGraphQl/etc/di.xml

@@ -51,6 +51,7 @@
                 <item name="stock" xsi:type="object">Magento\CatalogGraphQl\Model\Resolver\Products\DataProvider\Product\CollectionProcessor\StockProcessor</item>
                 <item name="visibility" xsi:type="object">Magento\CatalogGraphQl\Model\Resolver\Products\DataProvider\Product\CollectionProcessor\VisibilityStatusProcessor</item>
                 <item name="mediaGallery" xsi:type="object">Magento\CatalogGraphQl\Model\Resolver\Products\DataProvider\Product\CollectionProcessor\MediaGalleryProcessor</item>
+                <item name="priceData" xsi:type="object">Magento\CatalogGraphQl\Model\Resolver\Products\DataProvider\Product\CollectionProcessor\PriceDataProcessor</item>
             </argument>
         </arguments>
     </type>

app/code/Magento/CustomerGraphQl/Model/Context/AddCustomerGroupToContext.php

@@ -10,7 +10,6 @@
 use Magento\Authorization\Model\UserContextInterface;
 use Magento\GraphQl\Model\Query\ContextParametersInterface;
 use Magento\GraphQl\Model\Query\ContextParametersProcessorInterface;
-use Magento\Customer\Model\Session as CustomerSession;
 use Magento\Customer\Model\Group;
 use Magento\Customer\Api\CustomerRepositoryInterface;
 use Magento\Framework\Exception\LocalizedException;
@@ -20,25 +19,17 @@
  */
 class AddCustomerGroupToContext implements ContextParametersProcessorInterface
 {
-    /**
-     * @var CustomerSession
-     */
-    private $customerSession;
-
     /**
      * @var CustomerRepositoryInterface
      */
     private $customerRepository;
 
     /**
-     * @param CustomerSession $customerSession
      * @param CustomerRepositoryInterface $customerRepository
      */
     public function __construct(
-        CustomerSession $customerSession,
         CustomerRepositoryInterface $customerRepository
     ) {
-        $this->customerSession = $customerSession;
         $this->customerRepository = $customerRepository;
     }
 
@@ -47,7 +38,6 @@ public function __construct(
      */
     public function execute(ContextParametersInterface $contextParameters): ContextParametersInterface
     {
-        $customerSession = $this->customerSession;
         $customerGroupId = null;
         $extensionAttributes = $contextParameters->getExtensionAttributesData();
         if ($contextParameters->getUserType() === UserContextInterface::USER_TYPE_GUEST) {
@@ -61,7 +51,6 @@ public function execute(ContextParametersInterface $contextParameters): ContextP
             }
         }
         if ($customerGroupId !== null) {
-            $customerSession->setCustomerGroupId($customerGroupId);
             $contextParameters->addExtensionAttribute('customer_group_id', (int) $customerGroupId);
         }
         return $contextParameters;