blob: a309a75c09589729111ee3e1702b7013c5c8e96f (
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
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
|
<?php
namespace SMW\MediaWiki\Connection;
use DatabaseBase;
use Psr\Log\LoggerAwareTrait;
use RuntimeException;
use SMW\Connection\ConnectionProvider as IConnectionProvider;
/**
* @license GNU GPL v2+
* @since 1.9
*
* @author Jeroen De Dauw < jeroendedauw@gmail.com >
*/
class LoadBalancerConnectionProvider implements IConnectionProvider {
use LoggerAwareTrait;
/**
* @var DatabaseBase|null
*/
protected $connection = null;
/**
* @var int|null
*/
protected $id = null;
/**
* @var string|array
*/
protected $groups;
/**
* @var string|boolean $wiki
*/
protected $wiki;
/**
* @since 1.9
*
* @param int $id
* @param string|array $groups
* @param string|boolean $wiki
*/
public function __construct( $id, $groups = [], $wiki = false ) {
$this->id = $id;
$this->groups = $groups;
$this->wiki = $wiki;
}
/**
* @see IConnectionProvider::getConnection
*
* @since 1.9
*
* @return DatabaseBase
* @throws RuntimeException
*/
public function getConnection() {
if ( $this->connection === null ) {
$this->connection = wfGetLB( $this->wiki )->getConnection( $this->id, $this->groups, $this->wiki );
}
if ( $this->connection instanceof DatabaseBase ) {
return $this->connection;
}
throw new RuntimeException( 'Expected a DatabaseBase instance' );
}
/**
* @see IConnectionProvider::releaseConnection
*
* @since 1.9
*/
public function releaseConnection() {
if ( $this->wiki !== false && $this->connection !== null ) {
wfGetLB( $this->wiki )->reuseConnection( $this->connection );
}
}
}
|