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
|
<?php
namespace MediaWiki\Extension\AbuseFilter;
use Wikimedia\Rdbms\DBError;
use Wikimedia\Rdbms\IDatabase;
use Wikimedia\Rdbms\LBFactory;
class CentralDBManager {
public const SERVICE_NAME = 'AbuseFilterCentralDBManager';
/** @var LBFactory */
private $loadBalancerFactory;
/** @var string|false */
private $dbName;
/** @var bool */
private $filterIsCentral;
/**
* @param LBFactory $loadBalancerFactory
* @param string|false|null $dbName
* @param bool $filterIsCentral
*/
public function __construct( LBFactory $loadBalancerFactory, $dbName, bool $filterIsCentral ) {
$this->loadBalancerFactory = $loadBalancerFactory;
// Use false to agree with LoadBalancer
$this->dbName = $dbName ?: false;
$this->filterIsCentral = $filterIsCentral;
}
/**
* @param int $index DB_PRIMARY/DB_REPLICA
* @return IDatabase
* @throws DBError
* @throws CentralDBNotAvailableException
*/
public function getConnection( int $index ): IDatabase {
if ( !is_string( $this->dbName ) ) {
throw new CentralDBNotAvailableException( '$wgAbuseFilterCentralDB is not configured' );
}
return $this->loadBalancerFactory
->getMainLB( $this->dbName )
->getConnection( $index, [], $this->dbName );
}
/**
* @return string
* @throws CentralDBNotAvailableException
*/
public function getCentralDBName(): string {
if ( !is_string( $this->dbName ) ) {
throw new CentralDBNotAvailableException( '$wgAbuseFilterCentralDB is not configured' );
}
return $this->dbName;
}
/**
* Whether this database is the central one.
* @todo Deprecate the config in favour of just checking whether the current DB is the same
* as $wgAbuseFilterCentralDB.
* @return bool
*/
public function filterIsCentral(): bool {
return $this->filterIsCentral;
}
}
|
