Save Asynchronous Operations with one Batch improvement

app/code/Magento/AsynchronousOperations/Api/SaveMultipleOperationsInterface.php

@@ -0,0 +1,27 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+
+declare(strict_types=1);
+
+namespace Magento\AsynchronousOperations\Api;
+
+use Magento\AsynchronousOperations\Api\Data\OperationInterface;
+
+/**
+ * Interface for saving multiple operations
+ *
+ * @api
+ */
+interface SaveMultipleOperationsInterface
+{
+    /**
+     * Save Operations for Bulk
+     *
+     * @param OperationInterface[] $operations
+     * @return void
+     */
+    public function execute(array $operations): void;
+}

app/code/Magento/AsynchronousOperations/Model/MassSchedule.php

@@ -16,6 +16,7 @@
 use Magento\Framework\Bulk\BulkManagementInterface;
 use Magento\Framework\DataObject\IdentityGeneratorInterface;
 use Magento\Framework\Encryption\Encryptor;
+use Magento\AsynchronousOperations\Api\SaveMultipleOperationsInterface;
 use Magento\Framework\Exception\BulkException;
 use Magento\Framework\Exception\LocalizedException;
 use Psr\Log\LoggerInterface;
@@ -67,6 +68,11 @@ class MassSchedule
      */
     private $encryptor;
 
+    /**
+     * @var SaveMultipleOperationsInterface
+     */
+    private $saveMultipleOperations;
+
     /**
      * Initialize dependencies.
      *
@@ -78,6 +84,7 @@ class MassSchedule
      * @param OperationRepository $operationRepository
      * @param UserContextInterface $userContext
      * @param Encryptor $encryptor
+     * @param SaveMultipleOperationsInterface $saveMultipleOperations
      */
     public function __construct(
         IdentityGeneratorInterface $identityService,
@@ -87,7 +94,8 @@ public function __construct(
         LoggerInterface $logger,
         OperationRepository $operationRepository,
         UserContextInterface $userContext,
-        Encryptor $encryptor
+        Encryptor $encryptor,
+        SaveMultipleOperationsInterface $saveMultipleOperations
     ) {
         $this->identityService = $identityService;
         $this->itemStatusInterfaceFactory = $itemStatusInterfaceFactory;
@@ -97,6 +105,7 @@ public function __construct(
         $this->operationRepository = $operationRepository;
         $this->userContext = $userContext;
         $this->encryptor = $encryptor;
+        $this->saveMultipleOperations = $saveMultipleOperations;
     }
 
     /**
@@ -131,6 +140,7 @@ public function publishMass($topicName, array $entitiesArray, $groupId = null, $
 
         $operations = [];
         $requestItems = [];
+        $singleOperationsData = [];
         $bulkException = new BulkException();
         foreach ($entitiesArray as $key => $entityParams) {
             /** @var \Magento\AsynchronousOperations\Api\Data\ItemStatusInterface $requestItem */
@@ -139,6 +149,7 @@ public function publishMass($topicName, array $entitiesArray, $groupId = null, $
             try {
                 $operation = $this->operationRepository->createByTopic($topicName, $entityParams, $groupId);
                 $operations[] = $operation;
+                $singleOperationsData[] = $operation->getData();
                 $requestItem->setId($key);
                 $requestItem->setStatus(ItemStatusInterface::STATUS_ACCEPTED);
                 $requestItem->setDataHash(
@@ -159,6 +170,7 @@ public function publishMass($topicName, array $entitiesArray, $groupId = null, $
             }
         }
 
+        $this->saveMultipleOperations->execute($singleOperationsData);
         if (!$this->bulkManagement->scheduleBulk($groupId, $operations, $bulkDescription, $userId)) {
             throw new LocalizedException(
                 __('Something went wrong while processing the request.')

app/code/Magento/AsynchronousOperations/Model/OperationManagement.php

@@ -10,7 +10,7 @@
 use Magento\Framework\EntityManager\EntityManager;
 
 /**
- * Class OperationManagement
+ * Class for managing Bulk Operations
  */
 class OperationManagement implements \Magento\Framework\Bulk\OperationManagementInterface
 {
@@ -45,7 +45,7 @@ public function __construct(
         $this->operationFactory = $operationFactory;
         $this->logger = $logger;
     }
-    
+
     /**
      * @inheritDoc
      */

app/code/Magento/AsynchronousOperations/Model/ResourceModel/Operation.php

@@ -7,17 +7,21 @@
 namespace Magento\AsynchronousOperations\Model\ResourceModel;
 
 /**
- * Class Operation
+ * Resource class for Bulk Operations
  */
 class Operation extends \Magento\Framework\Model\ResourceModel\Db\AbstractDb
 {
+
+    public const TABLE_NAME = "magento_operation";
+    public const TABLE_PRIMARY_KEY = "id";
+
     /**
      * Initialize banner sales rule resource model
      *
      * @return void
      */
     protected function _construct()
     {
-        $this->_init('magento_operation', 'id');
+        $this->_init(self::TABLE_NAME, self::TABLE_PRIMARY_KEY);
     }
 }

app/code/Magento/AsynchronousOperations/Model/ResourceModel/Operation/OperationRepository.php

@@ -67,9 +67,11 @@ public function __construct(
     }
 
     /**
-     * @param $topicName
-     * @param $entityParams
-     * @param $groupId
+     * Creating topic for publishing
+     *
+     * @param string $topicName
+     * @param mixed $entityParams
+     * @param string $groupId
      * @return mixed
      */
     public function createByTopic($topicName, $entityParams, $groupId)
@@ -93,6 +95,6 @@ public function createByTopic($topicName, $entityParams, $groupId)
 
         /** @var \Magento\AsynchronousOperations\Api\Data\OperationInterface $operation */
         $operation = $this->operationFactory->create($data);
-        return $this->entityManager->save($operation);
+        return $operation;
     }
 }

app/code/Magento/AsynchronousOperations/Model/SaveMultipleOperations.php

@@ -0,0 +1,53 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+
+declare(strict_types=1);
+
+namespace Magento\AsynchronousOperations\Model;
+
+use Magento\AsynchronousOperations\Api\Data\OperationInterface;
+use Magento\AsynchronousOperations\Api\SaveMultipleOperationsInterface;
+use Magento\AsynchronousOperations\Model\ResourceModel\Operation as OperationResource;
+use Magento\Framework\Exception\CouldNotSaveException;
+
+/**
+ * Implementation for saving multiple operations
+ */
+class SaveMultipleOperations implements SaveMultipleOperationsInterface
+{
+
+    /**
+     * @var OperationResource
+     */
+    private $operationResource;
+
+    /**
+     * BulkSummary constructor.
+     *
+     * @param OperationResource $operationResource
+     */
+    public function __construct(
+        OperationResource $operationResource
+    ) {
+        $this->operationResource = $operationResource;
+    }
+
+    /**
+     * @inheritDoc
+     */
+    public function execute(array $operations): void
+    {
+        try {
+            $connection = $this->operationResource->getConnection();
+            $connection->insertMultiple(
+                $this->operationResource->getTable(OperationResource::TABLE_NAME),
+                $operations
+            );
+        } catch (\Exception $exception) {
+            throw new CouldNotSaveException(__($exception->getMessage()));
+        }
+    }
+}

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

@@ -7,6 +7,7 @@
 -->
 <config xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:noNamespaceSchemaLocation="urn:magento:framework:ObjectManager/etc/config.xsd">
     <preference for="Magento\AsynchronousOperations\Api\Data\BulkSummaryInterface" type="Magento\AsynchronousOperations\Model\BulkSummary" />
+    <preference for="Magento\AsynchronousOperations\Api\SaveMultipleOperationsInterface" type="Magento\AsynchronousOperations\Model\SaveMultipleOperations" />
     <preference for="Magento\AsynchronousOperations\Api\Data\OperationInterface" type="Magento\AsynchronousOperations\Model\Operation" />
     <preference for="Magento\AsynchronousOperations\Api\Data\OperationListInterface" type="Magento\AsynchronousOperations\Model\OperationList" />
     <preference for="Magento\Framework\Bulk\BulkManagementInterface" type="Magento\AsynchronousOperations\Model\BulkManagement" />