LBFactorySingle.php

Go to the documentation of this file.

<?php
namespace Wikimedia\Rdbms;
 
use InvalidArgumentException;
use BadMethodCallException;
 
class LBFactorySingle extends LBFactory {
    private $lb;
 
    public function __construct( array $conf ) {
        parent::__construct( $conf );
 
        if ( !isset( $conf['connection'] ) ) {
            throw new InvalidArgumentException( "Missing 'connection' argument." );
        }
 
        $lb = new LoadBalancerSingle( array_merge( $this->baseLoadBalancerParams(), $conf ) );
        $this->initLoadBalancer( $lb );
        $this->lb = $lb;
    }
 
    public static function newFromConnection( IDatabase $db, array $params = [] ) {
        return new static( array_merge(
            [ 'localDomain' => $db->getDomainID() ],
            $params,
            [ 'connection' => $db ]
        ) );
    }
 
    public function newMainLB( $domain = false ) {
        return $this->lb;
    }
 
    public function getMainLB( $domain = false ) {
        return $this->lb;
    }
 
    public function newExternalLB( $cluster ) {
        throw new BadMethodCallException( "Method is not supported." );
    }
 
    public function getExternalLB( $cluster ) {
        throw new BadMethodCallException( "Method is not supported." );
    }
 
    public function getAllMainLBs() {
        return [ 'DEFAULT' => $this->lb ];
    }
 
    public function getAllExternalLBs() {
        return [];
    }
 
    public function forEachLB( $callback, array $params = [] ) {
        if ( isset( $this->lb ) ) { // may not be set during _destruct()
            $callback( $this->lb, ...$params );
        }
    }
}