1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
<?php
/**
* Copyright © Magento, Inc. All rights reserved.
* See COPYING.txt for license details.
*/
namespace Magento\Framework\EntityManager\Sequence;
use Psr\Log\LoggerInterface;
use Magento\Framework\EntityManager\MetadataPool;
/**
* Class SequenceManager
*/
class SequenceManager
{
/**
* @var SequenceRegistry
*/
private $sequenceRegistry;
/**
* @var MetadataPool
*/
private $metadataPool;
/**
* @var LoggerInterface
*/
private $logger;
/**
* @var \Magento\Framework\App\ResourceConnection
*/
private $appResource;
/**
* @param MetadataPool $metadataPool
* @param SequenceRegistry $sequenceRegistry
* @param LoggerInterface $logger
* @param \Magento\Framework\App\ResourceConnection $appResource
*/
public function __construct(
MetadataPool $metadataPool,
SequenceRegistry $sequenceRegistry,
LoggerInterface $logger,
\Magento\Framework\App\ResourceConnection $appResource
) {
$this->metadataPool = $metadataPool;
$this->sequenceRegistry = $sequenceRegistry;
$this->logger = $logger;
$this->appResource = $appResource;
}
/**
* Forces creation of a sequence value.
*
* @param string $entityType
* @param string|int $identifier
*
* @return int
*
* @throws \Exception
*/
public function force($entityType, $identifier)
{
$sequenceInfo = $this->sequenceRegistry->retrieve($entityType);
if (!isset($sequenceInfo['sequenceTable'])) {
throw new \Exception(
'TODO: use correct Exception class' . PHP_EOL . ' Sequence table doesn\'t exists'
);
}
try {
$metadata = $this->metadataPool->getMetadata($entityType);
$connection = $this->appResource->getConnectionByName(
$metadata->getEntityConnectionName()
);
return $connection->insert(
$this->appResource->getTableName($sequenceInfo['sequenceTable']),
['sequence_value' => $identifier]
);
} catch (\Exception $e) {
$this->logger->critical($e->getMessage(), $e->getTrace());
throw new \Exception('TODO: use correct Exception class' . PHP_EOL . $e->getMessage());
}
}
/**
* @param string $entityType
* @param int $identifier
* @return int
* @throws \Exception
*/
public function delete($entityType, $identifier)
{
$metadata = $this->metadataPool->getMetadata($entityType);
$sequenceInfo = $this->sequenceRegistry->retrieve($entityType);
if (!isset($sequenceInfo['sequenceTable'])) {
throw new \Exception('TODO: use correct Exception class' . PHP_EOL . ' Sequence table doesn\'t exists');
}
try {
$connection = $this->appResource->getConnectionByName($metadata->getEntityConnectionName());
return $connection->delete(
$this->appResource->getTableName($sequenceInfo['sequenceTable']),
['sequence_value = ?' => $identifier]
);
} catch (\Exception $e) {
$this->logger->critical($e->getMessage(), $e->getTrace());
throw new \Exception('TODO: use correct Exception class' . PHP_EOL . $e->getMessage());
}
}
}