SequenceFactory.php 2.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485
  1. <?php
  2. /**
  3. * Copyright © Magento, Inc. All rights reserved.
  4. * See COPYING.txt for license details.
  5. */
  6. namespace Magento\Framework\EntityManager\Sequence;
  7. use Magento\Framework\DB\Sequence\SequenceInterface;
  8. use Magento\Framework\ObjectManagerInterface;
  9. /**
  10. * Class SequenceFactory
  11. */
  12. class SequenceFactory
  13. {
  14. /**
  15. * @var ObjectManagerInterface
  16. */
  17. protected $objectManager;
  18. /**
  19. * @var SequenceRegistry
  20. */
  21. protected $sequenceRegistry;
  22. /**
  23. * @var string
  24. */
  25. protected $instanceName;
  26. /**
  27. * @param SequenceRegistry $sequenceRegistry
  28. * @param ObjectManagerInterface $objectManager
  29. * @param string $instanceName
  30. */
  31. public function __construct(
  32. SequenceRegistry $sequenceRegistry,
  33. ObjectManagerInterface $objectManager,
  34. $instanceName = \Magento\Framework\EntityManager\Sequence\Sequence::class
  35. ) {
  36. $this->sequenceRegistry = $sequenceRegistry;
  37. $this->objectManager = $objectManager;
  38. $this->instanceName = $instanceName;
  39. }
  40. /**
  41. * Creates sequence instance
  42. *
  43. * @param string $entityType
  44. * @param array $config
  45. * @return SequenceInterface
  46. */
  47. public function create($entityType, $config)
  48. {
  49. if ($this->sequenceRegistry->retrieve($entityType) === false) {
  50. if (isset($config[$entityType]['sequence'])) {
  51. $this->sequenceRegistry->register(
  52. $entityType,
  53. $config[$entityType]['sequence']
  54. );
  55. } elseif (isset($config[$entityType]['sequenceTable'])) {
  56. if (isset($config[$entityType]['connectionName'])) {
  57. $connectionName = $config[$entityType]['connectionName'];
  58. } else {
  59. $connectionName = 'default';
  60. }
  61. $this->sequenceRegistry->register(
  62. $entityType,
  63. $this->objectManager->create(
  64. $this->instanceName,
  65. [
  66. 'connectionName' => $connectionName,
  67. 'sequenceTable' => $config[$entityType]['sequenceTable'],
  68. ]
  69. ),
  70. $config[$entityType]['sequenceTable']
  71. );
  72. } else {
  73. $this->sequenceRegistry->register($entityType);
  74. }
  75. }
  76. return $this->sequenceRegistry->retrieve($entityType)['sequence'];
  77. }
  78. }