7b791474a5
Now largely automated: VARS=$(grep -o "'[A-Za-z0-9_]*'" includes/MainConfigNames.php | \ tr "\n" '|' | sed "s/|$/\n/;s/'//g") sed -i -E "s/'($VARS)'/MainConfigNames::\1/g" \ $(grep -ERIl "'($VARS)'" includes/) Then git add -p with lots of error-prone manual checking. Then semi-manually add all the necessary "use" lines: vim $(grep -L 'use MediaWiki\\MainConfigNames;' \ $(git diff --cached --name-only --diff-filter=M HEAD^)) I didn't bother fixing lines that were over 100 characters unless they were over 120 and triggered phpcs. Bug: T305805 Change-Id: I74e0ab511abecb276717ad4276a124760a268147
150 lines
4.1 KiB
PHP
150 lines
4.1 KiB
PHP
<?php
|
|
/**
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License along
|
|
* with this program; if not, write to the Free Software Foundation, Inc.,
|
|
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
* http://www.gnu.org/copyleft/gpl.html
|
|
*
|
|
* @file
|
|
*/
|
|
|
|
namespace MediaWiki\User;
|
|
|
|
use MediaWiki\Config\ServiceOptions;
|
|
use MediaWiki\DAO\WikiAwareEntity;
|
|
use MediaWiki\MainConfigNames;
|
|
use Psr\Log\LoggerInterface;
|
|
use Wikimedia\Rdbms\ILBFactory;
|
|
use Wikimedia\Rdbms\ILoadBalancer;
|
|
|
|
/**
|
|
* ActorStore factory for various domains.
|
|
*
|
|
* @package MediaWiki\User
|
|
* @since 1.36
|
|
*/
|
|
class ActorStoreFactory {
|
|
|
|
/** @internal */
|
|
public const CONSTRUCTOR_OPTIONS = [
|
|
MainConfigNames::SharedDB,
|
|
MainConfigNames::SharedTables,
|
|
];
|
|
|
|
/** @var ILBFactory */
|
|
private $loadBalancerFactory;
|
|
|
|
/** @var UserNameUtils */
|
|
private $userNameUtils;
|
|
|
|
/** @var LoggerInterface */
|
|
private $logger;
|
|
|
|
/** @var string|false */
|
|
private $sharedDB;
|
|
|
|
/** @var string[] */
|
|
private $sharedTables;
|
|
|
|
/** @var ActorStore[] */
|
|
private $storeCache = [];
|
|
|
|
/**
|
|
* @param ServiceOptions $options
|
|
* @param ILBFactory $loadBalancerFactory
|
|
* @param UserNameUtils $userNameUtils
|
|
* @param LoggerInterface $logger
|
|
*/
|
|
public function __construct(
|
|
ServiceOptions $options,
|
|
ILBFactory $loadBalancerFactory,
|
|
UserNameUtils $userNameUtils,
|
|
LoggerInterface $logger
|
|
) {
|
|
$options->assertRequiredOptions( self::CONSTRUCTOR_OPTIONS );
|
|
$this->loadBalancerFactory = $loadBalancerFactory;
|
|
$this->sharedDB = $options->get( MainConfigNames::SharedDB );
|
|
$this->sharedTables = $options->get( MainConfigNames::SharedTables );
|
|
$this->userNameUtils = $userNameUtils;
|
|
$this->logger = $logger;
|
|
}
|
|
|
|
/**
|
|
* @param string|false $wikiId
|
|
* @return ActorNormalization
|
|
*/
|
|
public function getActorNormalization( $wikiId = WikiAwareEntity::LOCAL ): ActorNormalization {
|
|
return $this->getActorStore( $wikiId );
|
|
}
|
|
|
|
/**
|
|
* @param string|false $wikiId
|
|
* @return ActorStore
|
|
*/
|
|
public function getActorStore( $wikiId = WikiAwareEntity::LOCAL ): ActorStore {
|
|
// During the transition from User, we still have old User objects
|
|
// representing users from a different wiki, so we still have IDatabase::getDomainId
|
|
// passed as $wikiId, so we need to remap it back to LOCAL.
|
|
if ( is_string( $wikiId ) && $this->loadBalancerFactory->getLocalDomainID() === $wikiId ) {
|
|
$wikiId = WikiAwareEntity::LOCAL;
|
|
}
|
|
|
|
$storeCacheKey = $this->getStoreCacheKey( $wikiId );
|
|
if ( !isset( $this->storeCache[$storeCacheKey] ) ) {
|
|
$this->storeCache[$storeCacheKey] = new ActorStore(
|
|
$this->getLoadBalancerForTable( 'actor', $wikiId ),
|
|
$this->userNameUtils,
|
|
$this->logger,
|
|
$wikiId
|
|
);
|
|
}
|
|
return $this->storeCache[$storeCacheKey];
|
|
}
|
|
|
|
/**
|
|
* @param bool $wikiId
|
|
* @return UserIdentityLookup
|
|
*/
|
|
public function getUserIdentityLookup(
|
|
$wikiId = WikiAwareEntity::LOCAL
|
|
): UserIdentityLookup {
|
|
return $this->getActorStore( $wikiId );
|
|
}
|
|
|
|
/**
|
|
* @param string|false $wikiId
|
|
* @return string
|
|
*/
|
|
private function getStoreCacheKey( $wikiId ): string {
|
|
return $wikiId === WikiAwareEntity::LOCAL ? 'LOCAL' : $wikiId;
|
|
}
|
|
|
|
/**
|
|
* Returns a load balancer for the database that has the $table
|
|
* for the given $wikiId.
|
|
*
|
|
* @param string $table
|
|
* @param bool $wikiId
|
|
* @return ILoadBalancer
|
|
*/
|
|
private function getLoadBalancerForTable(
|
|
string $table,
|
|
$wikiId = WikiAwareEntity::LOCAL
|
|
): ILoadBalancer {
|
|
if ( $this->sharedDB && in_array( $table, $this->sharedTables ) ) {
|
|
// The main LB is already properly set up for shared databases early in Setup.php
|
|
return $this->loadBalancerFactory->getMainLB();
|
|
}
|
|
return $this->loadBalancerFactory->getMainLB( $wikiId );
|
|
}
|
|
}
|