Code Coverage |
||||||||||
Lines |
Functions and Methods |
Classes and Traits |
||||||||
| Total | |
0.00% |
0 / 45 |
|
0.00% |
0 / 3 |
CRAP | |
0.00% |
0 / 1 |
| InterwikiSearcher | |
0.00% |
0 / 45 |
|
0.00% |
0 / 3 |
110 | |
0.00% |
0 / 1 |
| __construct | |
0.00% |
0 / 3 |
|
0.00% |
0 / 1 |
2 | |||
| getInterwikiResults | |
0.00% |
0 / 41 |
|
0.00% |
0 / 1 |
72 | |||
| recordQueryCacheMetrics | |
0.00% |
0 / 1 |
|
0.00% |
0 / 1 |
2 | |||
| 1 | <?php |
| 2 | |
| 3 | namespace CirrusSearch; |
| 4 | |
| 5 | use CirrusSearch\Fallbacks\FallbackRunner; |
| 6 | use CirrusSearch\Parser\BasicQueryClassifier; |
| 7 | use CirrusSearch\Parser\NamespacePrefixParser; |
| 8 | use CirrusSearch\Profile\SearchProfileService; |
| 9 | use CirrusSearch\Search\CrossProjectBlockScorerFactory; |
| 10 | use CirrusSearch\Search\FullTextResultsType; |
| 11 | use CirrusSearch\Search\MSearchRequests; |
| 12 | use CirrusSearch\Search\SearchContext; |
| 13 | use CirrusSearch\Search\SearchQuery; |
| 14 | use CirrusSearch\Search\SearchQueryBuilder; |
| 15 | use CirrusSearch\Search\TitleHelper; |
| 16 | use MediaWiki\MediaWikiServices; |
| 17 | use MediaWiki\Status\Status; |
| 18 | use MediaWiki\User\User; |
| 19 | use Wikimedia\Stats\StatsFactory; |
| 20 | |
| 21 | /** |
| 22 | * Performs searches using Elasticsearch -- on interwikis! |
| 23 | * |
| 24 | * @license GPL-2.0-or-later |
| 25 | */ |
| 26 | class InterwikiSearcher extends Searcher { |
| 27 | |
| 28 | /** |
| 29 | * @param Connection $connection |
| 30 | * @param SearchConfig $config |
| 31 | * @param int[]|null $namespaces Namespace numbers to search, or null for all of them |
| 32 | * @param User|null $user |
| 33 | * @param CirrusDebugOptions|null $debugOptions |
| 34 | * @param NamespacePrefixParser|null $namespacePrefixParser |
| 35 | * @param InterwikiResolver|null $interwikiResolver |
| 36 | * @param TitleHelper|null $titleHelper |
| 37 | * @param CirrusSearchHookRunner|null $cirrusSearchHookRunner |
| 38 | */ |
| 39 | public function __construct( |
| 40 | Connection $connection, |
| 41 | SearchConfig $config, |
| 42 | ?array $namespaces = null, |
| 43 | ?User $user = null, |
| 44 | ?CirrusDebugOptions $debugOptions = null, |
| 45 | ?NamespacePrefixParser $namespacePrefixParser = null, |
| 46 | ?InterwikiResolver $interwikiResolver = null, |
| 47 | ?TitleHelper $titleHelper = null, |
| 48 | ?CirrusSearchHookRunner $cirrusSearchHookRunner = null |
| 49 | ) { |
| 50 | $maxResults = $config->get( CirrusConfigNames::NumCrossProjectSearchResults ); |
| 51 | parent::__construct( $connection, 0, $maxResults, $config, $namespaces, $user, false, |
| 52 | $debugOptions, $namespacePrefixParser, $interwikiResolver, $titleHelper, $cirrusSearchHookRunner ); |
| 53 | } |
| 54 | |
| 55 | /** |
| 56 | * Fetch search results, from caches, if there's any |
| 57 | * @param SearchQuery $query original search query |
| 58 | * @return Status |
| 59 | */ |
| 60 | public function getInterwikiResults( SearchQuery $query ): Status { |
| 61 | $sources = MediaWikiServices::getInstance() |
| 62 | ->getService( InterwikiResolver::SERVICE ) |
| 63 | ->getSisterProjectConfigs(); |
| 64 | if ( !$sources ) { |
| 65 | // Nothing to search for |
| 66 | return Status::newGood( [] ); |
| 67 | } |
| 68 | |
| 69 | $iwQueries = []; |
| 70 | foreach ( $sources as $interwiki => $config ) { |
| 71 | $iwQueries[$interwiki] = SearchQueryBuilder::forCrossProjectSearch( $config, $query ) |
| 72 | ->build(); |
| 73 | } |
| 74 | |
| 75 | $blockScorer = CrossProjectBlockScorerFactory::load( $this->config ); |
| 76 | $msearches = new MSearchRequests(); |
| 77 | foreach ( $iwQueries as $interwiki => $iwQuery ) { |
| 78 | $context = SearchContext::fromSearchQuery( $iwQuery, |
| 79 | FallbackRunner::create( $iwQuery, $this->interwikiResolver ), $this->cirrusSearchHookRunner ); |
| 80 | // The local wiki might have some special routing rules that do |
| 81 | // different things, but we can't assume those exist on other |
| 82 | // wikis. Always use defaults for interwiki search. |
| 83 | $context->setProfileContext( SearchProfileService::CONTEXT_DEFAULT ); |
| 84 | $this->searchContext = $context; |
| 85 | $this->setResultsType( new FullTextResultsType( |
| 86 | $this->searchContext->getFetchPhaseBuilder(), |
| 87 | $query->getParsedQuery()->isQueryOfClass( BasicQueryClassifier::COMPLEX_QUERY ), |
| 88 | $this->titleHelper, [], |
| 89 | $this->searchContext->getConfig()->getElement( CirrusConfigNames::DeduplicateInMemory ) === true ) ); |
| 90 | $this->config = $context->getConfig(); |
| 91 | $this->limit = $iwQuery->getLimit(); |
| 92 | $this->offset = $iwQuery->getOffset(); |
| 93 | $this->buildFullTextSearch( $query->getParsedQuery()->getQueryWithoutNsHeader() ); |
| 94 | $this->indexBaseName = $context->getConfig()->get( CirrusConfigNames::IndexBaseName ); |
| 95 | $search = $this->buildSearch(); |
| 96 | if ( $this->searchContext->areResultsPossible() ) { |
| 97 | $msearches->addRequest( $interwiki, $search ); |
| 98 | } |
| 99 | } |
| 100 | |
| 101 | $searchDescription = "{$this->searchContext->getSearchType()} search for '{$this->searchContext->getOriginalSearchTerm()}'"; |
| 102 | if ( $this->searchContext->getDebugOptions()->isCirrusDumpQuery() ) { |
| 103 | return $msearches->dumpQuery( $searchDescription ); |
| 104 | } |
| 105 | $mresponses = $this->searchMulti( $msearches ); |
| 106 | if ( $mresponses->hasFailure() ) { |
| 107 | return $mresponses->getFailure(); |
| 108 | } |
| 109 | |
| 110 | if ( $this->searchContext->getDebugOptions()->isReturnRaw() ) { |
| 111 | return $mresponses->dumpResults( $searchDescription ); |
| 112 | } |
| 113 | |
| 114 | return $mresponses->transformAndGetMulti( $this->searchContext->getResultsType(), array_keys( $iwQueries ), |
| 115 | static function ( array $v ) use ( $blockScorer ) { |
| 116 | return $blockScorer->reorder( $v ); |
| 117 | } ); |
| 118 | } |
| 119 | |
| 120 | protected function recordQueryCacheMetrics( StatsFactory $requestStats, string $cacheStatus, ?string $type = null ): void { |
| 121 | parent::recordQueryCacheMetrics( $requestStats, $cacheStatus, "interwiki" ); |
| 122 | } |
| 123 | } |