B2B-2469: Improve category children loading

Author: rganin

app/code/Magento/CatalogGraphQl/Model/Resolver/Categories/DataProvider/Category/CollectionProcessor/CatalogProcessor.php

@@ -8,7 +8,6 @@
 namespace Magento\CatalogGraphQl\Model\Resolver\Categories\DataProvider\Category\CollectionProcessor;
 
 use Magento\Catalog\Api\CategoryRepositoryInterface;
-use Magento\Catalog\Model\Category;
 use Magento\Catalog\Model\ResourceModel\Category\Collection;
 use Magento\CatalogGraphQl\Model\Resolver\Categories\DataProvider\Category\CollectionProcessorInterface;
 use Magento\Framework\Api\SearchCriteriaInterface;

app/code/Magento/CatalogGraphQl/Model/Resolver/Products/DataProvider/CategoryTree.php

@@ -7,26 +7,20 @@
 
 namespace Magento\CatalogGraphQl\Model\Resolver\Products\DataProvider;
 
-use Exception;
 use GraphQL\Language\AST\FieldNode;
 use GraphQL\Language\AST\NodeKind;
-use Iterator;
 use Magento\Catalog\Api\Data\CategoryInterface;
-use Magento\Catalog\Model\Category;
 use Magento\Catalog\Model\ResourceModel\Category\Collection;
 use Magento\Catalog\Model\ResourceModel\Category\CollectionFactory;
 use Magento\CatalogGraphQl\Model\AttributesJoiner;
 use Magento\CatalogGraphQl\Model\Category\DepthCalculator;
-use Magento\CatalogGraphQl\Model\Category\LevelCalculator;
 use Magento\CatalogGraphQl\Model\Resolver\Categories\DataProvider\Category\CollectionProcessorInterface;
-use Magento\Framework\DB\Select;
 use Magento\Framework\DB\Sql\Expression;
 use Magento\Framework\Api\Search\SearchCriteria;
 use Magento\Framework\EntityManager\MetadataPool;
 use Magento\Framework\Exception\LocalizedException;
 use Magento\Framework\GraphQl\Schema\Type\ResolveInfo;
 use Magento\GraphQl\Model\Query\ContextInterface;
-use Magento\Store\Api\Data\StoreInterface;
 
 /**
  * Category tree data provider

app/code/Magento/CatalogGraphQl/Model/Resolver/Products/DataProvider/CategoryTree/Wrapper/NodeWrapper.php

@@ -9,7 +9,6 @@
 
 use Magento\Catalog\Model\Category;
 use Magento\CatalogGraphQl\Model\Category\Hydrator;
-use Magento\Framework\App\ObjectManager;
 
 /**
  * Tree node forgery for category tree wrapper.

app/code/Magento/CatalogGraphQl/Model/Resolver/Products/DataProvider/ExtractDataFromCategoryTree.php

@@ -33,18 +33,18 @@ public function __construct(NodeWrapperFactory $nodeWrapperFactory)
      * Build result tree from collection
      *
      * @param Collection $collection
-     * @param array $topLevelCategories
+     * @param array $topLevelCategoryIds
      * @return array
      */
-    public function buildTree(Collection $collection, array $topLevelCategories) : array
+    public function buildTree(Collection $collection, array $topLevelCategoryIds) : array
     {
         $wrapper = $this->nodeWrapperFactory->create();
         /** @var Category $item */
         foreach ($collection->getItems() as $item) {
             $wrapper->wrap($item);
         }
         $tree = [];
-        foreach ($topLevelCategories as $topLevelCategory) {
+        foreach ($topLevelCategoryIds as $topLevelCategory) {
             $tree[] = $wrapper->getNodeById($topLevelCategory)->renderArray();
         }
         return $this->sortTree($tree);