blob: ddfd98716d8da8814f046813b541ed88a4d07e5e (
plain) (
blame)
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
|
<?php
use Wikimedia\Rdbms\IDatabase;
/**
* Deferrable Update for closure/callback updates that should use auto-commit mode
* @since 1.28
*/
class AutoCommitUpdate implements DeferrableUpdate, DeferrableCallback {
/** @var IDatabase */
private $dbw;
/** @var string */
private $fname;
/** @var callable|null */
private $callback;
/**
* @param IDatabase $dbw DB handle; update aborts if a transaction now this rolls back
* @param string $fname Caller name (usually __METHOD__)
* @param callable $callback Callback that takes (IDatabase, method name string)
* @param IDatabase[] $conns Abort if a transaction now on one of these rolls back [optional]
*/
public function __construct( IDatabase $dbw, $fname, callable $callback, array $conns = [] ) {
$this->dbw = $dbw;
$this->fname = $fname;
$this->callback = $callback;
// Register DB connections for which uncommitted changes are related to this update
$conns[] = $dbw;
foreach ( $conns as $conn ) {
if ( $conn->trxLevel() ) {
$conn->onTransactionResolution( [ $this, 'cancelOnRollback' ], $fname );
}
}
}
public function doUpdate() {
if ( !$this->callback ) {
return;
}
$autoTrx = $this->dbw->getFlag( DBO_TRX );
$this->dbw->clearFlag( DBO_TRX );
try {
/** @var Exception $e */
$e = null;
( $this->callback )( $this->dbw, $this->fname );
} catch ( Exception $e ) {
}
if ( $autoTrx ) {
$this->dbw->setFlag( DBO_TRX );
}
if ( $e ) {
throw $e;
}
}
/**
* @private This method is public so that it works with onTransactionResolution()
* @param int $trigger
*/
public function cancelOnRollback( $trigger ) {
if ( $trigger === IDatabase::TRIGGER_ROLLBACK ) {
$this->callback = null;
}
}
public function getOrigin() {
return $this->fname;
}
}
|
