ConnectionManager.php

Go to the documentation of this file.

<?php
namespace Wikimedia\Rdbms;
 
use InvalidArgumentException;
 
class ConnectionManager {
 
    private $loadBalancer;
 
    private $domain;
 
    private $groups = [];
 
    public function __construct( ILoadBalancer $loadBalancer, $domain = false, array $groups = [] ) {
        if ( !is_string( $domain ) && $domain !== false ) {
            throw new InvalidArgumentException( '$dbName must be a string, or false.' );
        }
 
        $this->loadBalancer = $loadBalancer;
        $this->domain = $domain;
        $this->groups = $groups;
    }
 
    private function getConnection( $i, ?array $groups = null, int $flags = 0 ) {
        $groups = $groups ?? $this->groups;
        return $this->loadBalancer->getConnection( $i, $groups, $this->domain, $flags );
    }
 
    private function getConnectionRef( $i, array $groups = null ) {
        $groups = $groups ?? $this->groups;
        return $this->loadBalancer->getConnectionRef( $i, $groups, $this->domain );
    }
 
    private function getLazyConnectionRef( $i, array $groups = null ) {
        $groups = $groups ?? $this->groups;
        return $this->loadBalancer->getLazyConnectionRef( $i, $groups, $this->domain );
    }
 
    public function getWriteConnection( int $flags = 0 ) {
        return $this->getConnection( DB_PRIMARY, null, $flags );
    }
 
    public function getReadConnection( ?array $groups = null, int $flags = 0 ) {
        $groups = $groups ?? $this->groups;
        return $this->getConnection( DB_REPLICA, $groups, $flags );
    }
 
    public function releaseConnection( IDatabase $db ) {
        $this->loadBalancer->reuseConnection( $db );
    }
 
    public function getWriteConnectionRef() {
        return $this->getConnectionRef( DB_PRIMARY );
    }
 
    public function getReadConnectionRef( array $groups = null ) {
        $groups = $groups ?? $this->groups;
        return $this->getConnectionRef( DB_REPLICA, $groups );
    }
 
    public function getLazyReadConnectionRef( array $groups = null ) {
        $groups = $groups ?? $this->groups;
        return $this->getLazyConnectionRef( DB_REPLICA, $groups );
    }
 
}