private function handleResultFromHook($srchres, $namespaces, $search, $limit)
{
$rescorer = new SearchExactMatchRescorer();
return $rescorer->rescore($search, $namespaces, $srchres, $limit);
}
/**
* Process completion search results.
* Resolves the titles and rescores.
* @param SearchSuggestionSet $suggestions
* @return SearchSuggestionSet
*/
protected function processCompletionResults($search, SearchSuggestionSet $suggestions)
{
$search = trim($search);
// preload the titles with LinkBatch
$titles = $suggestions->map(function (SearchSuggestion $sugg) {
return $sugg->getSuggestedTitle();
});
$lb = new LinkBatch($titles);
$lb->setCaller(__METHOD__);
$lb->execute();
$results = $suggestions->map(function (SearchSuggestion $sugg) {
return $sugg->getSuggestedTitle()->getPrefixedText();
});
if ($this->offset === 0) {
// Rescore results with an exact title match
// NOTE: in some cases like cross-namespace redirects
// (frequently used as shortcuts e.g. WP:WP on huwiki) some
// backends like Cirrus will return no results. We should still
// try an exact title match to workaround this limitation
$rescorer = new SearchExactMatchRescorer();
$rescoredResults = $rescorer->rescore($search, $this->namespaces, $results, $this->limit);
} else {
// No need to rescore if offset is not 0
// The exact match must have been returned at position 0
// if it existed.
$rescoredResults = $results;
}
if (count($rescoredResults) > 0) {
$found = array_search($rescoredResults[0], $results);
if ($found === false) {
// If the first result is not in the previous array it
// means that we found a new exact match
$exactMatch = SearchSuggestion::fromTitle(0, Title::newFromText($rescoredResults[0]));
$suggestions->prepend($exactMatch);
$suggestions->shrink($this->limit);
} else {
// if the first result is not the same we need to rescore
if ($found > 0) {
$suggestions->rescore($found);
}
}
}
return $suggestions;
}
private function handleResultFromHook($srchres, $namespaces, $search, $limit, $offset)
{
if ($offset === 0) {
// Only perform exact db match if offset === 0
// This is still far from perfect but at least we avoid returning the
// same title afain and again when the user is scrolling with a query
// that matches a title in the db.
$rescorer = new SearchExactMatchRescorer();
$srchres = $rescorer->rescore($search, $namespaces, $srchres, $limit);
}
return $srchres;
}
/**
* Process completion search results.
* Resolves the titles and rescores.
* @param SearchSuggestionSet $suggestions
* @return SearchSuggestionSet
*/
protected function processCompletionResults($search, SearchSuggestionSet $suggestions)
{
if ($suggestions->getSize() == 0) {
// If we don't have anything, don't bother
return $suggestions;
}
$search = trim($search);
// preload the titles with LinkBatch
$titles = $suggestions->map(function (SearchSuggestion $sugg) {
return $sugg->getSuggestedTitle();
});
$lb = new LinkBatch($titles);
$lb->setCaller(__METHOD__);
$lb->execute();
$results = $suggestions->map(function (SearchSuggestion $sugg) {
return $sugg->getSuggestedTitle()->getPrefixedText();
});
// Rescore results with an exact title match
$rescorer = new SearchExactMatchRescorer();
$rescoredResults = $rescorer->rescore($search, $this->namespaces, $results, $this->limit);
if (count($rescoredResults) > 0) {
$found = array_search($rescoredResults[0], $results);
if ($found === false) {
// If the first result is not in the previous array it
// means that we found a new exact match
$exactMatch = SearchSuggestion::fromTitle(0, Title::newFromText($rescoredResults[0]));
$suggestions->prepend($exactMatch);
$suggestions->shrink($this->limit);
} else {
// if the first result is not the same we need to rescore
if ($found > 0) {
$suggestions->rescore($found);
}
}
}
return $suggestions;
}
/**
* This implementation will run the completion suggester if it's enabled and if the
* query is for NS_MAIN. Fallback to SearchEngine default implemention otherwise.
*
* @param string $search the user query
* @return SearchSuggestionSet the suggestions
*/
public function searchSuggestions($search)
{
$config = ConfigFactory::getDefaultInstance()->makeConfig('CirrusSearch');
$useCompletionSuggester = $config->getElement('CirrusSearchUseCompletionSuggester');
$context = RequestContext::getMain();
$request = $context->getRequest();
// Allow experimentation with query parameters
if ($request && $request->getVal('cirrusUseCompletionSuggester') === 'yes') {
$useCompletionSuggester = true;
}
if (!$useCompletionSuggester) {
// Completion suggester is not enabled, fallback to
// default implementation
return $this->searchSuggestionsPrefixSearchFallback($search);
}
// We use Title to extract namespace from a Title string
// We append a random letter behind just in case the search
// string ends with ':'.
$title = Title::newFromText($search . "A");
if ($title->getNamespace() != NS_MAIN || count($this->namespaces) != 1 || reset($this->namespaces) != NS_MAIN) {
// Fallback to prefix search if we are not on content namespace
return $this->searchSuggestionsPrefixSearchFallback($search);
}
$user = $context->getUser();
// offset is omitted, searchSuggestion does not support
// scrolling results
$suggester = new CompletionSuggester($this->connection, $this->limit, $config, $this->namespaces, $user, $this->indexBaseName);
// Not really useful, mostly for testing purpose
$variants = $request->getArray('cirrusCompletionSuggesterVariant');
if (empty($variants)) {
global $wgContLang;
$variants = $wgContLang->autoConvertToAllVariants($search);
} else {
if (count($variants) > 3) {
// We should not allow too many variants
$variants = array_slice($variants, 0, 3);
}
}
$response = $suggester->suggest($search, $variants);
if ($response->isOK()) {
// Errors will be logged, let's try the exact db match
$suggestions = $response->getValue();
} else {
$suggestions = SearchSuggestionSet::emptySuggestionSet();
}
// preload the titles with LinkBatch
$titles = $suggestions->map(function ($sugg) {
return $sugg->getSuggestedTitle();
});
$lb = new LinkBatch($titles);
$lb->setCaller(__METHOD__);
$lb->execute();
$results = $suggestions->map(function ($sugg) {
return $sugg->getSuggestedTitle()->getPrefixedText();
});
// now we can trim
$search = trim($search);
// Rescore results with an exact title match
$rescorer = new SearchExactMatchRescorer();
$rescoredResults = $rescorer->rescore($search, $this->namespaces, $results, $this->limit);
if (count($rescoredResults) > 0) {
$found = array_search($rescoredResults[0], $results);
if ($found === false) {
// If the first result is not in the previous array it
// means that we found a new exact match
$exactMatch = SearchSuggestion::fromTitle(0, Title::newFromText($rescoredResults[0]));
$suggestions->prepend($exactMatch);
$suggestions->shrink($this->limit);
} else {
// if the first result is not the same we need to rescore
if ($found > 0) {
$suggestions->rescore($found);
}
}
}
return $suggestions;
}