ACPT-1181: reset mutable state after request

Author: JacobBrownAustin

app/code/Magento/Authorization/Model/CompositeUserContext.php

@@ -7,6 +7,7 @@
 namespace Magento\Authorization\Model;
 
 use Magento\Framework\ObjectManager\Helper\Composite as CompositeHelper;
+use Magento\Framework\ObjectManager\ResetAfterRequestInterface;
 
 /**
  * User context.
@@ -17,7 +18,7 @@
  * @api
  * @since 100.0.2
  */
-class CompositeUserContext implements \Magento\Authorization\Model\UserContextInterface
+class CompositeUserContext implements \Magento\Authorization\Model\UserContextInterface, ResetAfterRequestInterface
 {
     /**
      * @var UserContextInterface[]
@@ -92,4 +93,12 @@ protected function getUserContext()
         }
         return $this->chosenUserContext;
     }
+
+    /**
+     * @inheritDoc
+     */
+    public function _resetState(): void
+    {
+        $this->chosenUserContext = null;
+    }
 }

app/code/Magento/Bundle/Model/Plugin/Frontend/ProductIdentitiesExtender.php

@@ -9,11 +9,12 @@
 
 use Magento\Bundle\Model\Product\Type as BundleType;
 use Magento\Catalog\Model\Product as CatalogProduct;
+use Magento\Framework\ObjectManager\ResetAfterRequestInterface;
 
 /**
  * Add child identities to product identities on storefront.
  */
-class ProductIdentitiesExtender
+class ProductIdentitiesExtender implements ResetAfterRequestInterface
 {
     /**
      * @var BundleType
@@ -68,4 +69,12 @@ private function getChildrenIds($entityId): array
 
         return $this->cacheChildrenIds[$entityId];
     }
+
+    /**
+     * @inheritDoc
+     */
+    public function _resetState(): void
+    {
+        $this->cacheChildrenIds = [];
+    }
 }

app/code/Magento/Bundle/Model/Plugin/ProductIdentitiesExtender.php

@@ -9,11 +9,12 @@
 
 use Magento\Bundle\Model\Product\Type as BundleType;
 use Magento\Catalog\Model\Product as CatalogProduct;
+use Magento\Framework\ObjectManager\ResetAfterRequestInterface;
 
 /**
  * Add parent identities to product identities.
  */
-class ProductIdentitiesExtender
+class ProductIdentitiesExtender implements ResetAfterRequestInterface
 {
     /**
      * @var BundleType
@@ -68,4 +69,12 @@ private function getParentIdsByChild($entityId): array
 
         return $this->cacheParentIdsByChild[$entityId];
     }
+
+    /**
+     * @inheritDoc
+     */
+    public function _resetState(): void
+    {
+        $this->cacheParentIdsByChild = [];
+    }
 }

app/code/Magento/Bundle/Model/Product/SelectionProductsDisabledRequired.php

@@ -10,6 +10,7 @@
 use Magento\Framework\EntityManager\MetadataPool;
 use Magento\Catalog\Model\Product\Attribute\Source\Status;
 use Magento\Bundle\Model\ResourceModel\Selection as BundleSelection;
+use Magento\Framework\ObjectManager\ResetAfterRequestInterface;
 use Magento\Store\Model\StoreManagerInterface;
 use Magento\Catalog\Model\ResourceModel\Product\CollectionFactory as ProductCollectionFactory;
 use Magento\Catalog\Model\Product;
@@ -18,7 +19,7 @@
 /**
  * Class to return ids of options and child products when all products in required option are disabled in bundle product
  */
-class SelectionProductsDisabledRequired
+class SelectionProductsDisabledRequired implements ResetAfterRequestInterface
 {
     /**
      * @var BundleSelection
@@ -161,4 +162,12 @@ private function getCacheKey(int $bundleId, int $websiteId): string
     {
         return $bundleId . '-' . $websiteId;
     }
+
+    /**
+     * @inheritDoc
+     */
+    public function _resetState(): void
+    {
+        $this->productsDisabledRequired = [];
+    }
 }

app/code/Magento/Bundle/Pricing/Adjustment/Calculator.php

@@ -12,6 +12,7 @@
 use Magento\Bundle\Pricing\Price\BundleSelectionFactory;
 use Magento\Bundle\Pricing\Price\BundleSelectionPrice;
 use Magento\Catalog\Model\Product;
+use Magento\Framework\ObjectManager\ResetAfterRequestInterface;
 use Magento\Framework\Pricing\Adjustment\Calculator as CalculatorBase;
 use Magento\Framework\Pricing\Amount\AmountFactory;
 use Magento\Framework\Pricing\Amount\AmountInterface;
@@ -25,7 +26,7 @@
  * Bundle price calculator
  * @SuppressWarnings(PHPMD.CouplingBetweenObjects)
  */
-class Calculator implements BundleCalculatorInterface
+class Calculator implements BundleCalculatorInterface, ResetAfterRequestInterface
 {
     /**
      * @var CalculatorBase
@@ -425,4 +426,12 @@ public function processOptions($option, $selectionPriceList, $searchMin = true)
         }
         return $result;
     }
+
+    /**
+     * @inheritDoc
+     */
+    public function _resetState(): void
+    {
+        $this->optionAmount = [];
+    }
 }

app/code/Magento/Bundle/Pricing/Adjustment/DefaultSelectionPriceListProvider.php

@@ -11,13 +11,14 @@
 use Magento\Catalog\Model\Product;
 use Magento\Bundle\Model\Product\Price;
 use Magento\Catalog\Helper\Data as CatalogData;
+use Magento\Framework\ObjectManager\ResetAfterRequestInterface;
 use Magento\Store\Model\StoreManagerInterface;
 use Magento\Store\Api\WebsiteRepositoryInterface;
 
 /**
  * Provide lightweight implementation which uses price index
  */
-class DefaultSelectionPriceListProvider implements SelectionPriceListProviderInterface
+class DefaultSelectionPriceListProvider implements SelectionPriceListProviderInterface, ResetAfterRequestInterface
 {
     /**
      * @var BundleSelectionFactory
@@ -245,4 +246,12 @@ private function getBundleOptions(Product $saleableItem)
     {
         return $saleableItem->getTypeInstance()->getOptionsCollection($saleableItem);
     }
+
+    /**
+     * @inheritDoc
+     */
+    public function _resetState(): void
+    {
+        $this->priceList = [];
+    }
 }

app/code/Magento/Bundle/Pricing/Price/BundleOptions.php

@@ -8,14 +8,15 @@
 namespace Magento\Bundle\Pricing\Price;
 
 use Magento\Bundle\Pricing\Adjustment\BundleCalculatorInterface;
+use Magento\Framework\ObjectManager\ResetAfterRequestInterface;
 use Magento\Framework\Pricing\SaleableInterface;
 use Magento\Framework\Pricing\Amount\AmountInterface;
 use Magento\Catalog\Model\Product;
 
 /**
  * Bundle option price calculation model.
  */
-class BundleOptions
+class BundleOptions implements ResetAfterRequestInterface
 {
     /**
      * @var BundleCalculatorInterface
@@ -135,4 +136,12 @@ public function getOptionSelectionAmount(
 
         return $this->optionSelectionAmountCache[$cacheKey];
     }
+
+    /**
+     * @inheritDoc
+     */
+    public function _resetState(): void
+    {
+        $this->optionSelectionAmountCache = [];
+    }
 }

app/code/Magento/Bundle/Pricing/Price/BundleRegularPrice.php

@@ -7,14 +7,15 @@
 namespace Magento\Bundle\Pricing\Price;
 
 use Magento\Bundle\Pricing\Adjustment\BundleCalculatorInterface;
+use Magento\Framework\ObjectManager\ResetAfterRequestInterface;
 use Magento\Framework\Pricing\Amount\AmountInterface;
 use Magento\Catalog\Pricing\Price\CustomOptionPrice;
 use Magento\Bundle\Model\Product\Price;
 
 /**
  * Bundle product regular price model
  */
-class BundleRegularPrice extends \Magento\Catalog\Pricing\Price\RegularPrice implements RegularPriceInterface
+class BundleRegularPrice extends \Magento\Catalog\Pricing\Price\RegularPrice implements RegularPriceInterface, ResetAfterRequestInterface
 {
     /**
      * @var BundleCalculatorInterface
@@ -72,4 +73,13 @@ public function getMinimalPrice()
     {
         return $this->getAmount();
     }
+
+    /**
+     * @inheritDoc
+     */
+    public function _resetState(): void
+    {
+        $this->maximalPrice = null;
+        $this->amount = [];
+    }
 }

app/code/Magento/BundleGraphQl/Model/Resolver/BundleItemLinks.php

@@ -7,37 +7,43 @@
 
 namespace Magento\BundleGraphQl\Model\Resolver;
 
-use Magento\Framework\Exception\LocalizedException;
-use Magento\Framework\GraphQl\Schema\Type\ResolveInfo;
 use Magento\BundleGraphQl\Model\Resolver\Links\Collection;
+use Magento\BundleGraphQl\Model\Resolver\Links\CollectionFactory;
+use Magento\Framework\App\ObjectManager;
+use Magento\Framework\Exception\LocalizedException;
 use Magento\Framework\GraphQl\Config\Element\Field;
 use Magento\Framework\GraphQl\Query\Resolver\ValueFactory;
 use Magento\Framework\GraphQl\Query\ResolverInterface;
+use Magento\Framework\GraphQl\Schema\Type\ResolveInfo;
 
 /**
  * @inheritdoc
  */
 class BundleItemLinks implements ResolverInterface
 {
     /**
-     * @var Collection
+     * @var CollectionFactory
      */
-    private $linkCollection;
+    private CollectionFactory $linkCollectionFactory;
 
     /**
      * @var ValueFactory
      */
-    private $valueFactory;
+    private ValueFactory $valueFactory;
 
     /**
-     * @param Collection $linkCollection
+     * @param Collection $linkCollection Deprecated. Use $linkCollectionFactory instead
      * @param ValueFactory $valueFactory
+     * @param CollectionFactory|null $linkCollectionFactory
+     * @SuppressWarnings(PHPMD.UnusedFormalParameter)
      */
     public function __construct(
         Collection $linkCollection,
-        ValueFactory $valueFactory
+        ValueFactory $valueFactory,
+        CollectionFactory $linkCollectionFactory = null
     ) {
-        $this->linkCollection = $linkCollection;
+        $this->linkCollectionFactory = $linkCollectionFactory
+            ?: ObjectManager::getInstance()->get(CollectionFactory::class);
         $this->valueFactory = $valueFactory;
     }
 
@@ -49,12 +55,11 @@ public function resolve(Field $field, $context, ResolveInfo $info, array $value
         if (!isset($value['option_id']) || !isset($value['parent_id'])) {
             throw new LocalizedException(__('"option_id" and "parent_id" values should be specified'));
         }
-
-        $this->linkCollection->addIdFilters((int)$value['option_id'], (int)$value['parent_id']);
-        $result = function () use ($value) {
-            return $this->linkCollection->getLinksForOptionId((int)$value['option_id']);
+        $linkCollection = $this->linkCollectionFactory->create();
+        $linkCollection->addIdFilters((int)$value['option_id'], (int)$value['parent_id']);
+        $result = function () use ($value, $linkCollection) {
+            return $linkCollection->getLinksForOptionId((int)$value['option_id']);
         };
-
         return $this->valueFactory->create($result);
     }
 }

app/code/Magento/BundleGraphQl/Model/Resolver/BundleItems.php

@@ -7,54 +7,60 @@
 
 namespace Magento\BundleGraphQl\Model\Resolver;
 
-use Magento\Framework\GraphQl\Schema\Type\ResolveInfo;
 use Magento\Bundle\Model\Product\Type;
 use Magento\BundleGraphQl\Model\Resolver\Options\Collection;
+use Magento\BundleGraphQl\Model\Resolver\Options\CollectionFactory;
 use Magento\Catalog\Api\Data\ProductInterface;
+use Magento\Framework\App\ObjectManager;
 use Magento\Framework\EntityManager\MetadataPool;
 use Magento\Framework\GraphQl\Config\Element\Field;
 use Magento\Framework\GraphQl\Query\Resolver\ValueFactory;
 use Magento\Framework\GraphQl\Query\ResolverInterface;
+use Magento\Framework\GraphQl\Schema\Type\ResolveInfo;
 
 /**
  * @inheritdoc
  */
 class BundleItems implements ResolverInterface
 {
     /**
-     * @var Collection
+     * @var CollectionFactory
      */
-    private $bundleOptionCollection;
+    private CollectionFactory $bundleOptionCollectionFactory;
 
     /**
      * @var ValueFactory
      */
-    private $valueFactory;
+    private ValueFactory $valueFactory;
 
     /**
      * @var MetadataPool
      */
-    private $metadataPool;
+    private MetadataPool $metadataPool;
 
     /**
-     * @param Collection $bundleOptionCollection
+     * @param Collection $bundleOptionCollection Deprecated. Use $bundleOptionCollectionFactory
      * @param ValueFactory $valueFactory
      * @param MetadataPool $metadataPool
+     * @param CollectionFactory|null $bundleOptionCollectionFactory
+     * @SuppressWarnings(PHPMD.UnusedFormalParameter)
      */
     public function __construct(
         Collection $bundleOptionCollection,
         ValueFactory $valueFactory,
-        MetadataPool $metadataPool
+        MetadataPool $metadataPool,
+        CollectionFactory $bundleOptionCollectionFactory = null
     ) {
-        $this->bundleOptionCollection = $bundleOptionCollection;
+        $this->bundleOptionCollectionFactory = $bundleOptionCollectionFactory
+            ?: ObjectManager::getInstance()->get(CollectionFactory::class);
         $this->valueFactory = $valueFactory;
         $this->metadataPool = $metadataPool;
     }
 
     /**
      * Fetch and format bundle option items.
      *
-     * {@inheritDoc}
+     * @inheritDoc
      */
     public function resolve(Field $field, $context, ResolveInfo $info, array $value = null, array $args = null)
     {
@@ -68,17 +74,15 @@ public function resolve(Field $field, $context, ResolveInfo $info, array $value
             };
             return $this->valueFactory->create($result);
         }
-
-        $this->bundleOptionCollection->addParentFilterData(
+        $bundleOptionCollection = $this->bundleOptionCollectionFactory->create();
+        $bundleOptionCollection->addParentFilterData(
             (int)$value[$linkField],
             (int)$value['entity_id'],
             $value[ProductInterface::SKU]
         );
-
-        $result = function () use ($value, $linkField) {
-            return $this->bundleOptionCollection->getOptionsByParentId((int)$value[$linkField]);
+        $result = function () use ($value, $linkField, $bundleOptionCollection) {
+            return $bundleOptionCollection->getOptionsByParentId((int)$value[$linkField]);
         };
-
         return $this->valueFactory->create($result);
     }
 }