ObjectCacheFactory.php

Go to the documentation of this file.

<?php
use MediaWiki\Config\ServiceOptions;
use MediaWiki\Http\Telemetry;
use MediaWiki\Logger\Spi;
use MediaWiki\MainConfigNames;
use Wikimedia\Stats\StatsFactory;
 
class ObjectCacheFactory {
    public const CONSTRUCTOR_OPTIONS = [
        MainConfigNames::SQLiteDataDir,
        MainConfigNames::UpdateRowsPerQuery,
        MainConfigNames::MemCachedServers,
        MainConfigNames::MemCachedPersistent,
        MainConfigNames::MemCachedTimeout,
        MainConfigNames::CachePrefix,
        MainConfigNames::ObjectCaches,
        MainConfigNames::MainCacheType,
    ];
 
    private ServiceOptions $options;
    private StatsFactory $stats;
    private Spi $logger;
    private $instances = [];
    private string $domainId;
    private $dbLoadBalancerFactory;
 
    public function __construct(
        ServiceOptions $options,
        StatsFactory $stats,
        Spi $loggerSpi,
        callable $dbLoadBalancerFactory,
        string $domainId
    ) {
        $options->assertRequiredOptions( self::CONSTRUCTOR_OPTIONS );
        $this->options = $options;
        $this->stats = $stats;
        $this->logger = $loggerSpi;
        $this->dbLoadBalancerFactory = $dbLoadBalancerFactory;
        $this->domainId = $domainId;
    }
 
    private function getDefaultKeyspace(): string {
        $cachePrefix = $this->options->get( MainConfigNames::CachePrefix );
        if ( is_string( $cachePrefix ) && $cachePrefix !== '' ) {
            return $cachePrefix;
        }
 
        return $this->domainId;
    }
 
    private function newFromId( $id ): BagOStuff {
        if ( $id === CACHE_ANYTHING ) {
            $id = ObjectCache::getAnythingId();
        }
 
        if ( !isset( $this->options->get( MainConfigNames::ObjectCaches )[$id] ) ) {
            // Always recognize these
            if ( $id === CACHE_NONE ) {
                return new EmptyBagOStuff();
            } elseif ( $id === CACHE_HASH ) {
                return new HashBagOStuff();
            } elseif ( $id === CACHE_ACCEL ) {
                return ObjectCache::makeLocalServerCache( $this->getDefaultKeyspace() );
            }
 
            throw new InvalidArgumentException( "Invalid object cache type \"$id\" requested. " .
                "It is not present in \$wgObjectCaches." );
        }
 
        return $this->newFromParams( $this->options->get( MainConfigNames::ObjectCaches )[$id] );
    }
 
    public function getInstance( $id ): BagOStuff {
        if ( !isset( $this->instances[$id] ) ) {
            $this->instances[$id] = $this->newFromId( $id );
        }
 
        return $this->instances[$id];
    }
 
    public function newFromParams( array $params ): BagOStuff {
        $logger = $this->logger->getLogger( $params['loggroup'] ?? 'objectcache' );
        // Apply default parameters and resolve the logger instance
        $params += [
            'logger' => $logger,
            'keyspace' => $this->getDefaultKeyspace(),
            'asyncHandler' => [ DeferredUpdates::class, 'addCallableUpdate' ],
            'reportDupes' => true,
            'stats' => $this->stats,
        ];
 
        if ( isset( $params['factory'] ) ) {
            $args = $params['args'] ?? [ $params ];
 
            return call_user_func( $params['factory'], ...$args );
        }
 
        if ( !isset( $params['class'] ) ) {
            throw new InvalidArgumentException(
                'No "factory" nor "class" provided; got "' . print_r( $params, true ) . '"'
            );
        }
 
        $class = $params['class'];
 
        // Normalization and DI for SqlBagOStuff
        if ( is_a( $class, SqlBagOStuff::class, true ) ) {
            $this->prepareSqlBagOStuffFromParams( $params );
        }
 
        // Normalization and DI for MemcachedBagOStuff
        if ( is_subclass_of( $class, MemcachedBagOStuff::class ) ) {
            $this->prepareMemcachedBagOStuffFromParams( $params );
        }
 
        // Normalization and DI for MultiWriteBagOStuff
        if ( is_a( $class, MultiWriteBagOStuff::class, true ) ) {
            $this->prepareMultiWriteBagOStuffFromParams( $params );
        }
        if ( is_a( $class, RESTBagOStuff::class, true ) ) {
            $this->prepareRESTBagOStuffFromParams( $params );
        }
 
        return new $class( $params );
    }
 
    private function prepareSqlBagOStuffFromParams( array &$params ): void {
        if ( isset( $params['globalKeyLB'] ) ) {
            throw new InvalidArgumentException(
                'globalKeyLB in $wgObjectCaches is no longer supported' );
        }
        if ( isset( $params['server'] ) && !isset( $params['servers'] ) ) {
            $params['servers'] = [ $params['server'] ];
            unset( $params['server'] );
        }
        if ( isset( $params['servers'] ) ) {
            // In the past it was not required to set 'dbDirectory' in $wgObjectCaches
            foreach ( $params['servers'] as &$server ) {
                if ( $server['type'] === 'sqlite' && !isset( $server['dbDirectory'] ) ) {
                    $server['dbDirectory'] = $this->options->get( MainConfigNames::SQLiteDataDir );
                }
            }
        } elseif ( isset( $params['cluster'] ) ) {
            $cluster = $params['cluster'];
            $dbLbFactory = $this->dbLoadBalancerFactory;
            $params['loadBalancerCallback'] = static function () use ( $cluster, $dbLbFactory ) {
                return $dbLbFactory()->getExternalLB( $cluster );
            };
            $params += [ 'dbDomain' => false ];
        } else {
            $dbLbFactory = $this->dbLoadBalancerFactory;
            $params['loadBalancerCallback'] = static function () use ( $dbLbFactory ) {
                return $dbLbFactory()->getMainLb();
            };
            $params += [ 'dbDomain' => false ];
        }
        $params += [ 'writeBatchSize' => $this->options->get( MainConfigNames::UpdateRowsPerQuery ) ];
    }
 
    private function prepareMemcachedBagOStuffFromParams( array &$params ): void {
        $params += [
            'servers' => $this->options->get( MainConfigNames::MemCachedServers ),
            'persistent' => $this->options->get( MainConfigNames::MemCachedPersistent ),
            'timeout' => $this->options->get( MainConfigNames::MemCachedTimeout ),
        ];
    }
 
    private function prepareMultiWriteBagOStuffFromParams( array &$params ): void {
        // Phan warns about foreach with non-array because it
        // thinks any key can be Closure|IBufferingStatsdDataFactory
        '@phan-var array{caches:array[]} $params';
        foreach ( $params['caches'] ?? [] as $i => $cacheInfo ) {
            // Ensure logger, keyspace, asyncHandler, etc are injected just as if
            // one of these was configured without MultiWriteBagOStuff.
            $params['caches'][$i] = $this->newFromParams( $cacheInfo );
        }
    }
 
    private function prepareRESTBagOStuffFromParams( array &$params ): void {
        $params['telemetry'] = Telemetry::getInstance();
    }
 
    public function getLocalServerInstance( $fallback = CACHE_NONE ): BagOStuff {
        $cache = $this->getInstance( CACHE_ACCEL );
        if ( $cache instanceof EmptyBagOStuff ) {
            if ( is_array( $fallback ) ) {
                $fallback = $fallback['fallback'] ?? CACHE_NONE;
            }
            $cache = $this->getInstance( $fallback );
        }
 
        return $cache;
    }
 
    public function clear(): void {
        $this->instances = [];
    }
 
    public function setInstanceForTesting( $cacheId, BagOStuff $cache ): void {
        if ( !defined( 'MW_PHPUNIT_TEST' ) ) {
            throw new LogicException( __METHOD__ . ' can not be called outside of tests' );
        }
        $this->instances[$cacheId] = $cache;
    }
}