LYNX-290: Bulk quote_shipping_rate delete (#174)

Author: sivaschenko

app/code/Magento/Quote/Model/ResourceModel/Quote/Address/Rate/Collection.php

@@ -5,10 +5,12 @@
  */
 namespace Magento\Quote\Model\ResourceModel\Quote\Address\Rate;
 
+use Magento\Quote\Model\ResourceModel\Quote\Address\Rate;
+
 /**
  * Quote addresses shipping rates collection
  *
- * @author      Magento Core Team <core@magentocommerce.com>
+ * @SuppressWarnings(PHPMD.CouplingBetweenObjects)
  */
 class Collection extends \Magento\Framework\Model\ResourceModel\Db\VersionControl\Collection
 {
@@ -24,15 +26,21 @@ class Collection extends \Magento\Framework\Model\ResourceModel\Db\VersionContro
      */
     private $_carrierFactory;
 
+    /**
+     * @var Delete
+     */
+    private Delete $deleteRates;
+
     /**
      * @param \Magento\Framework\Data\Collection\EntityFactory $entityFactory
      * @param \Psr\Log\LoggerInterface $logger
      * @param \Magento\Framework\Data\Collection\Db\FetchStrategyInterface $fetchStrategy
      * @param \Magento\Framework\Event\ManagerInterface $eventManager
      * @param \Magento\Framework\Model\ResourceModel\Db\VersionControl\Snapshot $entitySnapshot
      * @param \Magento\Shipping\Model\CarrierFactoryInterface $carrierFactory
-     * @param \Magento\Framework\DB\Adapter\AdapterInterface $connection
-     * @param \Magento\Framework\Model\ResourceModel\Db\AbstractDb $resource
+     * @param Delete $deleteRates
+     * @param \Magento\Framework\DB\Adapter\AdapterInterface|null $connection
+     * @param \Magento\Framework\Model\ResourceModel\Db\AbstractDb|null $resource
      */
     public function __construct(
         \Magento\Framework\Data\Collection\EntityFactory $entityFactory,
@@ -41,8 +49,9 @@ public function __construct(
         \Magento\Framework\Event\ManagerInterface $eventManager,
         \Magento\Framework\Model\ResourceModel\Db\VersionControl\Snapshot $entitySnapshot,
         \Magento\Shipping\Model\CarrierFactoryInterface $carrierFactory,
+        Delete $deleteRates,
         \Magento\Framework\DB\Adapter\AdapterInterface $connection = null,
-        \Magento\Framework\Model\ResourceModel\Db\AbstractDb $resource = null
+        \Magento\Framework\Model\ResourceModel\Db\AbstractDb $resource = null,
     ) {
         parent::__construct(
             $entityFactory,
@@ -53,6 +62,7 @@ public function __construct(
             $connection,
             $resource
         );
+        $this->deleteRates = $deleteRates;
         $this->_carrierFactory = $carrierFactory;
     }
 
@@ -112,4 +122,27 @@ public function addItem(\Magento\Framework\DataObject $rate)
         }
         return parent::addItem($rate);
     }
+
+    /**
+     * @inheritdoc
+     */
+    public function save()
+    {
+        $itemsToDelete = [];
+        $itemsToSave = [];
+        /** @var Rate $item */
+        foreach ($this->getItems() as $item) {
+            if ($item->isDeleted()) {
+                $itemsToDelete[] = $item;
+            } else {
+                $itemsToSave[] = $item;
+            }
+        }
+        $this->deleteRates->execute($itemsToDelete);
+        /** @var Rate $item */
+        foreach ($itemsToSave as $item) {
+            $item->save();
+        }
+        return $this;
+    }
 }

app/code/Magento/Quote/Model/ResourceModel/Quote/Address/Rate/Delete.php

@@ -0,0 +1,65 @@
+<?php
+/************************************************************************
+ *
+ * Copyright 2023 Adobe
+ * All Rights Reserved.
+ *
+ * NOTICE: All information contained herein is, and remains
+ * the property of Adobe and its suppliers, if any. The intellectual
+ * and technical concepts contained herein are proprietary to Adobe
+ * and its suppliers and are protected by all applicable intellectual
+ * property laws, including trade secret and copyright laws.
+ * Dissemination of this information or reproduction of this material
+ * is strictly forbidden unless prior written permission is obtained
+ * from Adobe.
+ * ***********************************************************************
+ */
+declare(strict_types=1);
+
+namespace Magento\Quote\Model\ResourceModel\Quote\Address\Rate;
+
+use Magento\Framework\App\ResourceConnection;
+use Magento\Quote\Model\Quote\Address\Rate;
+
+class Delete
+{
+    private const TABLE = 'quote_shipping_rate';
+    private const FIELD_RATE_ID = 'rate_id';
+
+    /**
+     * @var ResourceConnection
+     */
+    private ResourceConnection $resourceConnection;
+
+    /**
+     * @param ResourceConnection $resourceConnection
+     */
+    public function __construct(ResourceConnection $resourceConnection)
+    {
+        $this->resourceConnection = $resourceConnection;
+    }
+
+    /**
+     * Remove shipping rates
+     *
+     * @param Rate[] $rates
+     * @return void
+     */
+    public function execute(array $rates): void
+    {
+        if (empty($rates)) {
+            return;
+        }
+        $this->resourceConnection->getConnection()->delete(
+            $this->resourceConnection->getTableName(self::TABLE),
+            [
+                self::FIELD_RATE_ID . ' IN (?)' => array_map(
+                    function ($rate) {
+                        return $rate->getId();
+                    },
+                    $rates
+                )
+            ]
+        );
+    }
+}