*
* @license GNU GPL v2+
* @author Jeroen De Dauw < jeroendedauw@gmail.com >
+ * @author Daniel Kinzler
*/
class DBSiteStore implements SiteStore {
protected $sites = null;
/**
- * @since 1.25
- * @param null $sitesTable Unused since 1.27
+ * @var LoadBalancer
*/
- public function __construct( $sitesTable = null ) {
- if ( $sitesTable !== null ) {
- throw new InvalidArgumentException(
- __METHOD__ . ': $sitesTable parameter must be null'
- );
- }
+ private $dbLoadBalancer;
+
+ /**
+ * @since 1.27
+ *
+ * @todo: inject some kind of connection manager that is aware of the target wiki,
+ * instead of injecting a LoadBalancer.
+ *
+ * @param LoadBalancer $dbLoadBalancer
+ */
+ public function __construct( LoadBalancer $dbLoadBalancer ) {
+ $this->dbLoadBalancer = $dbLoadBalancer;
}
/**
protected function loadSites() {
$this->sites = new SiteList();
- $dbr = wfGetDB( DB_SLAVE );
+ $dbr = $this->dbLoadBalancer->getConnection( DB_REPLICA );
$res = $dbr->select(
'sites',
- array(
+ [
'site_id',
'site_global_key',
'site_type',
'site_data',
'site_forward',
'site_config',
- ),
+ ],
'',
__METHOD__,
- array( 'ORDER BY' => 'site_global_key' )
+ [ 'ORDER BY' => 'site_global_key' ]
);
foreach ( $res as $row ) {
// Batch load the local site identifiers.
$ids = $dbr->select(
'site_identifiers',
- array(
+ [
'si_site',
'si_type',
'si_key',
- ),
- array(),
+ ],
+ [],
__METHOD__
);
* @return bool Success indicator
*/
public function saveSite( Site $site ) {
- return $this->saveSites( array( $site ) );
+ return $this->saveSites( [ $site ] );
}
/**
return true;
}
- $dbw = wfGetDB( DB_MASTER );
+ $dbw = $this->dbLoadBalancer->getConnection( DB_MASTER );
$dbw->startAtomic( __METHOD__ );
$success = true;
- $internalIds = array();
- $localIds = array();
+ $internalIds = [];
+ $localIds = [];
foreach ( $sites as $site ) {
if ( $site->getInternalId() !== null ) {
$internalIds[] = $site->getInternalId();
}
- $fields = array(
+ $fields = [
// Site data
'site_global_key' => $site->getGlobalId(), // TODO: check not null
'site_type' => $site->getType(),
// Site config
'site_forward' => $site->shouldForward() ? 1 : 0,
'site_config' => serialize( $site->getExtraConfig() ),
- );
+ ];
$rowId = $site->getInternalId();
if ( $rowId !== null ) {
$success = $dbw->update(
- 'sites', $fields, array( 'site_id' => $rowId ), __METHOD__
+ 'sites', $fields, [ 'site_id' => $rowId ], __METHOD__
) && $success;
} else {
$rowId = $dbw->nextSequenceValue( 'sites_site_id_seq' );
foreach ( $site->getLocalIds() as $idType => $ids ) {
foreach ( $ids as $id ) {
- $localIds[] = array( $rowId, $idType, $id );
+ $localIds[] = [ $rowId, $idType, $id ];
}
}
}
- if ( $internalIds !== array() ) {
+ if ( $internalIds !== [] ) {
$dbw->delete(
'site_identifiers',
- array( 'si_site' => $internalIds ),
+ [ 'si_site' => $internalIds ],
__METHOD__
);
}
foreach ( $localIds as $localId ) {
$dbw->insert(
'site_identifiers',
- array(
+ [
'si_site' => $localId[0],
'si_type' => $localId[1],
'si_key' => $localId[2],
- ),
+ ],
__METHOD__
);
}
* @return bool Success
*/
public function clear() {
- $dbw = wfGetDB( DB_MASTER );
+ $dbw = $this->dbLoadBalancer->getConnection( DB_MASTER );
$dbw->startAtomic( __METHOD__ );
$ok = $dbw->delete( 'sites', '*', __METHOD__ );