1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
|
<?php
use MediaWiki\HookContainer\HookContainer;
use MediaWiki\MediaWikiServices;
use Wikimedia\ObjectFactory;
use Wikimedia\Rdbms\IDatabase;
use Wikimedia\Rdbms\ILoadBalancer;
/**
* Factory class for SearchEngine.
* Allows to create engine of the specific type.
*/
class SearchEngineFactory {
/**
* Configuration for SearchEngine classes.
* @var SearchEngineConfig
*/
private $config;
/** @var HookContainer */
private $hookContainer;
public function __construct( SearchEngineConfig $config, HookContainer $hookContainer ) {
$this->config = $config;
$this->hookContainer = $hookContainer;
}
/**
* Create SearchEngine of the given type.
*
* @param string|null $type
* @return SearchEngine
*/
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 );
}
$mappings = $this->config->getSearchMappings();
if ( isset( $mappings[$class] ) ) {
$spec = $mappings[$class];
} else {
// Convert non mapped classes to ObjectFactory spec
$spec = [ 'class' => $class ];
}
$args = [];
if ( isset( $spec['class'] ) && is_subclass_of( $spec['class'], SearchDatabase::class ) ) {
$args['extraArgs'][] = $lb;
}
/** @var SearchEngine $engine */
$engine = ObjectFactory::getObjectFromSpec( $spec, $args );
$engine->setHookContainer( $this->hookContainer );
return $engine;
}
/**
* @param IDatabase|ILoadBalancer $dbOrLb
* @return string SearchEngine subclass name
* @since 1.28
*/
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;
}
}
}
|
