blob: cd7956e02b08296b8549ad4f9c86c54d993b591d (
plain)
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
|
<?php
namespace Aws\DynamoDb;
use Aws\DynamoDb\Exception\DynamoDbException;
/**
* The locking connection adds locking logic to the read operation.
*/
class LockingSessionConnection extends StandardSessionConnection
{
public function __construct(DynamoDbClient $client, array $config = [])
{
parent::__construct($client, $config);
}
/**
* {@inheritdoc}
* Retries the request until the lock can be acquired
*/
public function read($id)
{
// Create the params for the UpdateItem operation so that a lock can be
// set and item returned (via ReturnValues) in a one, atomic operation.
$params = [
'TableName' => $this->getTableName(),
'Key' => $this->formatKey($id),
'Expected' => ['lock' => ['Exists' => false]],
'AttributeUpdates' => ['lock' => ['Value' => ['N' => '1']]],
'ReturnValues' => 'ALL_NEW',
];
// Acquire the lock and fetch the item data.
$timeout = time() + $this->getMaxLockWaitTime();
while (true) {
try {
$item = [];
$result = $this->client->updateItem($params);
if (isset($result['Attributes'])) {
foreach ($result['Attributes'] as $key => $value) {
$item[$key] = current($value);
}
}
return $item;
} catch (DynamoDbException $e) {
if ($e->getAwsErrorCode() === 'ConditionalCheckFailedException'
&& time() < $timeout
) {
usleep(rand(
$this->getMinLockRetryMicrotime(),
$this->getMaxLockRetryMicrotime()
));
} else {
break;
}
}
}
}
}
|
