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
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
|
<?php
/**
* Database factory class, this will determine whether to use the main database
* or an external database defined in configuration file
*/
class MWEchoDbFactory {
/**
* The wiki to access the database for
* @var string|bool
*/
protected $wiki;
/**
* The cluster for the database
* @var string|bool
*/
protected $cluster;
/**
* @param string|bool
* @param string|bool
*/
public function __construct( $cluster = false, $wiki = false ) {
$this->cluster = $cluster;
$this->wiki = $wiki;
}
/**
* Create a db factory instance from default Echo configuration
* DO NOT use singleton in here because job queue may run
* against multiple wikis, having a singleton would result in
* wrong db configuration. In addition, singleton is not necessary
* because it's actually handled inside core database object
*
* @return MWEchoDbFactory
*/
public static function newFromDefault() {
global $wgEchoCluster;
return new self( $wgEchoCluster );
}
/**
* Get the database load balancer
* @param $wiki string|bool The wiki ID, or false for the current wiki
* @return LoadBalancer
*/
protected function getLB() {
// Use the external db defined for Echo
if ( $this->cluster ) {
$lb = wfGetLBFactory()->getExternalLB( $this->cluster, $this->wiki );
} else {
$lb = wfGetLB( $this->wiki );
}
return $lb;
}
/**
* Get the database connection for Echo
* @param $db int Index of the connection to get
* @param $groups mixed Query groups.
* @return DatabaseBase
*/
public function getEchoDb( $db, $groups = array() ) {
return $this->getLB()->getConnection( $db, $groups, $this->wiki );
}
/**
* Wrapper function for wfGetDB, some extensions like MobileFrontend is
* using this to issue sql queries against Echo database directly. This
* is totally not accepted and should be updated to use Echo database access
* objects
*
* @deprecated Use newFromDefault() instead to create a db factory
* @param $db int Index of the connection to get
* @param $groups mixed Query groups.
* @param $wiki string|bool The wiki ID, or false for the current wiki
* @return DatabaseBase
*/
public static function getDB( $db, $groups = array(), $wiki = false ) {
global $wgEchoCluster;
// Use the external db defined for Echo
if ( $wgEchoCluster ) {
$lb = wfGetLBFactory()->getExternalLB( $wgEchoCluster, $wiki );
} else {
$lb = wfGetLB( $wiki );
}
return $lb->getConnection( $db, $groups, $wiki );
}
/**
* Wait for the slaves of the database
*/
public function waitForSlaves() {
$this->waitFor( $this->getMasterPosition() );
}
/**
* Get the current master position for the wiki and echo
* db when they have at least one slave in their cluster.
*
* @return array
*/
public function getMasterPosition() {
$position = array(
'wikiDb' => false,
'echoDb' => false,
);
$lb = wfGetLB();
if ( $lb->getServerCount() > 1 ) {
$position['wikiDb'] = $lb->getMasterPos();
};
if ( $this->cluster ) {
$lb = $this->getLB();
if ( $lb->getServerCount() > 1 ) {
$position['echoDb'] = $lb->getMasterPos();
}
}
return $position;
}
/**
* Recieves the output of self::getMasterPosition. Waits
* for slaves to catch up to the master position at that
* point.
*
* @param array $position
*/
public function waitFor( array $position ) {
if ( $position['wikiDb'] ) {
wfGetLB()->waitFor( $position['wikiDb'] );
}
if ( $position['echoDb'] ) {
$this->getLB()->waitFor( $position['echoDb'] );
}
}
}
|
GitWeb