blob: 8b62989b53ab284c06e94e9e1b58fb6d2363bf62 (
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
|
<?php
use Wikimedia\Rdbms\IDatabase;
/**
* Deferrable Update for closure/callback updates via IDatabase::doAtomicSection()
* @since 1.27
*/
class AtomicSectionUpdate implements DeferrableUpdate, DeferrableCallback {
/** @var IDatabase */
private $dbw;
/** @var string */
private $fname;
/** @var callable|null */
private $callback;
/**
* @param IDatabase $dbw
* @param string $fname Caller name (usually __METHOD__)
* @param callable $callback
* @see IDatabase::doAtomicSection()
*/
public function __construct( IDatabase $dbw, $fname, callable $callback ) {
$this->dbw = $dbw;
$this->fname = $fname;
$this->callback = $callback;
if ( $this->dbw->trxLevel() ) {
$this->dbw->onTransactionResolution( [ $this, 'cancelOnRollback' ], $fname );
}
}
public function doUpdate() {
if ( $this->callback ) {
$this->dbw->doAtomicSection( $this->fname, $this->callback );
}
}
public function cancelOnRollback( $trigger ) {
if ( $trigger === IDatabase::TRIGGER_ROLLBACK ) {
$this->callback = null;
}
}
public function getOrigin() {
return $this->fname;
}
}
|