SaveMultiple.php 3.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124
  1. <?php
  2. /**
  3. * Copyright © Magento, Inc. All rights reserved.
  4. * See COPYING.txt for license details.
  5. */
  6. declare(strict_types=1);
  7. namespace Magento\Inventory\Model\ResourceModel\StockSourceLink;
  8. use Magento\Framework\App\ResourceConnection;
  9. use Magento\Inventory\Model\ResourceModel\StockSourceLink as StockSourceLinkResourceModel;
  10. use Magento\Inventory\Model\StockSourceLink;
  11. use Magento\InventoryApi\Api\Data\StockSourceLinkInterface;
  12. /**
  13. * Implementation of StockSourceLink save multiple operation for specific db layer
  14. * Save Multiple used here for performance efficient purposes over single save operation
  15. */
  16. class SaveMultiple
  17. {
  18. /**
  19. * @var ResourceConnection
  20. */
  21. private $resourceConnection;
  22. /**
  23. * @param ResourceConnection $resourceConnection
  24. */
  25. public function __construct(
  26. ResourceConnection $resourceConnection
  27. ) {
  28. $this->resourceConnection = $resourceConnection;
  29. }
  30. /**
  31. * Multiple save StockSourceLinks
  32. *
  33. * @param StockSourceLinkInterface[] $links
  34. * @return void
  35. */
  36. public function execute(array $links)
  37. {
  38. if (!count($links)) {
  39. return;
  40. }
  41. $connection = $this->resourceConnection->getConnection();
  42. $tableName = $this->resourceConnection->getTableName(
  43. StockSourceLinkResourceModel::TABLE_NAME_STOCK_SOURCE_LINK
  44. );
  45. $columnsSql = $this->buildColumnsSqlPart([
  46. StockSourceLink::SOURCE_CODE,
  47. StockSourceLink::STOCK_ID,
  48. StockSourceLink::PRIORITY,
  49. ]);
  50. $valuesSql = $this->buildValuesSqlPart($links);
  51. $onDuplicateSql = $this->buildOnDuplicateSqlPart([
  52. StockSourceLink::PRIORITY,
  53. ]);
  54. $bind = $this->getSqlBindData($links);
  55. $insertSql = sprintf(
  56. 'INSERT INTO %s (%s) VALUES %s %s',
  57. $tableName,
  58. $columnsSql,
  59. $valuesSql,
  60. $onDuplicateSql
  61. );
  62. $connection->query($insertSql, $bind);
  63. }
  64. /**
  65. * @param array $columns
  66. * @return string
  67. */
  68. private function buildColumnsSqlPart(array $columns): string
  69. {
  70. $connection = $this->resourceConnection->getConnection();
  71. $processedColumns = array_map([$connection, 'quoteIdentifier'], $columns);
  72. return implode(', ', $processedColumns);
  73. }
  74. /**
  75. * @param StockSourceLinkInterface[] $links
  76. * @return string
  77. */
  78. private function buildValuesSqlPart(array $links): string
  79. {
  80. $sql = rtrim(str_repeat('(?, ?, ?), ', count($links)), ', ');
  81. return $sql;
  82. }
  83. /**
  84. * @param StockSourceLinkInterface[] $links
  85. * @return array
  86. */
  87. private function getSqlBindData(array $links): array
  88. {
  89. $bind = [];
  90. foreach ($links as $link) {
  91. $bind = array_merge($bind, [
  92. $link->getSourceCode(),
  93. $link->getStockId(),
  94. $link->getPriority(),
  95. ]);
  96. }
  97. return $bind;
  98. }
  99. /**
  100. * @param array $fields
  101. * @return string
  102. */
  103. private function buildOnDuplicateSqlPart(array $fields): string
  104. {
  105. $connection = $this->resourceConnection->getConnection();
  106. $processedFields = [];
  107. foreach ($fields as $field) {
  108. $processedFields[] = sprintf('%1$s = VALUES(%1$s)', $connection->quoteIdentifier($field));
  109. }
  110. return 'ON DUPLICATE KEY UPDATE ' . implode(', ', $processedFields);
  111. }
  112. }