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 ??= $this->groups;
        return $this->loadBalancer->getConnection( $i, $groups, $this->domain, $flags );
    }
 
    public function getWriteConnection( int $flags = 0 ) {
        return $this->getConnection( DB_PRIMARY, null, $flags );
    }
 
    public function getReadConnection( ?array $groups = null, int $flags = 0 ) {
        $groups ??= $this->groups;
        return $this->getConnection( DB_REPLICA, $groups, $flags );
    }
}