DBAccessBase.php

Go to the documentation of this file.

<?php
 
use MediaWiki\MediaWikiServices;
use Wikimedia\Rdbms\IDatabase;
use Wikimedia\Rdbms\LoadBalancer;
 
abstract class DBAccessBase implements IDBAccessObject {
    protected $wiki = false;
 
    public function __construct( $wiki = false ) {
        $this->wiki = $wiki;
    }
 
    protected function getConnection( $id, array $groups = [] ) {
        $loadBalancer = $this->getLoadBalancer();
 
        return $loadBalancer->getConnection( $id, $groups, $this->wiki );
    }
 
    protected function releaseConnection( IDatabase $db ) {
        if ( $this->wiki !== false ) {
            $loadBalancer = $this->getLoadBalancer();
            $loadBalancer->reuseConnection( $db );
        }
    }
 
    public function getLoadBalancer() {
        $lbFactory = MediaWikiServices::getInstance()->getDBLoadBalancerFactory();
        return $lbFactory->getMainLB( $this->wiki );
    }
}