SearchEngineFactory.php

Go to the documentation of this file.

<?php

use Wikimedia\Rdbms\IDatabase;
use Wikimedia\Rdbms\ILoadBalancer;
use MediaWiki\MediaWikiServices;

class SearchEngineFactory {
    private $config;

    public function __construct( SearchEngineConfig $config ) {
        $this->config = $config;
    }

    public function create( $type = null ) {
        $configuredClass = $this->config->getSearchType();
        $alternativesClasses = $this->config->getSearchTypes();

        $lb = MediaWikiServices::getInstance()->getDBLoadBalancer();
        if ( $type !== null && in_array( $type, $alternativesClasses ) ) {
            $class = $type;
        } elseif ( $configuredClass !== null ) {
            $class = $configuredClass;
        } else {
            $class = self::getSearchEngineClass( $lb );
        }

        if ( is_subclass_of( $class, SearchDatabase::class ) ) {
            return new $class( $lb );
        } else {
            return new $class();
        }
    }

    public static function getSearchEngineClass( $dbOrLb ) {
        $type = ( $dbOrLb instanceof IDatabase )
            ? $dbOrLb->getType()
            : $dbOrLb->getServerType( $dbOrLb->getWriterIndex() );

        switch ( $type ) {
            case 'sqlite':
                return SearchSqlite::class;
            case 'mysql':
                return SearchMySQL::class;
            case 'postgres':
                return SearchPostgres::class;
            default:
                return SearchEngineDummy::class;
        }
    }
}