04a1760b54
* Allow for DI of config and PoolCounterConnectionManager. * Manage the PoolCounterConnectionManager singleton without using a global/static field. * Allow for test overrides (in ParserOutputAccessTest) without needing to bring all of ObjectFactory and 'factory' and thus exposing class constructors to stable interface (except not really since the args are hardcoded in practice). Bug: T201223 Change-Id: I514fee20b388f04f9c85c5a1373845d621c65395
60 lines
1.7 KiB
PHP
60 lines
1.7 KiB
PHP
<?php
|
|
namespace MediaWiki\PoolCounter;
|
|
|
|
use PoolCounter;
|
|
use PoolCounterNull;
|
|
|
|
/**
|
|
* @since 1.40
|
|
*/
|
|
class PoolCounterFactory {
|
|
private ?PoolCounterConnectionManager $manager = null;
|
|
private ?array $typeConfigs;
|
|
private array $clientConf;
|
|
|
|
/**
|
|
* @internal For use by ServiceWiring
|
|
* @param array|null $typeConfigs See $wgPoolCounterConf
|
|
* @param array $clientConf See $wgPoolCountClientConf
|
|
*/
|
|
public function __construct( ?array $typeConfigs, array $clientConf ) {
|
|
$this->typeConfigs = $typeConfigs;
|
|
$this->clientConf = $clientConf;
|
|
}
|
|
|
|
private function getClientManager(): PoolCounterConnectionManager {
|
|
$this->manager ??= new PoolCounterConnectionManager( $this->clientConf );
|
|
return $this->manager;
|
|
}
|
|
|
|
/**
|
|
* Get a PoolCounter.
|
|
*
|
|
* @internal This should only be called from PoolCounterWork
|
|
* @param string $type The class of actions to limit concurrency for (task type)
|
|
* @param string $key
|
|
* @return PoolCounter
|
|
*/
|
|
public function create( string $type, string $key ): PoolCounter {
|
|
$conf = $this->typeConfigs[$type] ?? null;
|
|
if ( $conf === null ) {
|
|
return new PoolCounterNull();
|
|
}
|
|
|
|
$class = $conf['class'] ?? null;
|
|
if ( $class === 'PoolCounter_Client' ) {
|
|
// Since 1.16: Introduce PoolCounter_Client in PoolCounter extension.
|
|
// Since 1.40: Move to core as symbolic name, discourage use of class name.
|
|
$class = PoolCounterClient::class;
|
|
}
|
|
/** @var PoolCounter $poolCounter */
|
|
$poolCounter = new $class( $conf, $type, $key );
|
|
|
|
// Support subclass for back-compat with the extension
|
|
if ( $poolCounter instanceof PoolCounterClient ) {
|
|
$poolCounter->setManager( $this->getClientManager() );
|
|
}
|
|
|
|
return $poolCounter;
|
|
}
|
|
}
|