X-Git-Url: https://git.heureux-cyclage.org/?a=blobdiff_plain;f=includes%2Fsearch%2FSearchEngine.php;h=ea86a4b580687a478dde9d4102acd0b7d1b37281;hb=46c17ddb1ff10a1e073742c9d8b831cd915802f3;hp=bd48e2136479be0aa6e98bc254356a8d93b07bc0;hpb=51755499a7b2b4fd677bbc6a05d1b2d6c5f09a64;p=lhc%2Fweb%2Fwiklou.git

diff --git a/includes/search/SearchEngine.php b/includes/search/SearchEngine.php
index bd48e21364..ea86a4b580 100644
--- a/includes/search/SearchEngine.php
+++ b/includes/search/SearchEngine.php
@@ -77,7 +77,9 @@ abstract class SearchEngine {
 	 * @return SearchResultSet|Status|null
 	 */
 	public function searchText( $term ) {
-		return $this->doSearchText( $term );
+		return $this->maybePaginate( function () use ( $term ) {
+			return $this->doSearchText( $term );
+		} );
 	}
 
 	/**
@@ -132,7 +134,9 @@ abstract class SearchEngine {
 	 * @return SearchResultSet|null
 	 */
 	public function searchTitle( $term ) {
-		return $this->doSearchTitle( $term );
+		return $this->maybePaginate( function () use ( $term ) {
+			return $this->doSearchTitle( $term );
+		} );
 	}
 
 	/**
@@ -146,6 +150,40 @@ abstract class SearchEngine {
 		return null;
 	}
 
+	/**
+	 * Performs an overfetch and shrink operation to determine if
+	 * the next page is available for search engines that do not
+	 * explicitly implement their own pagination.
+	 *
+	 * @param Closure $fn Takes no arguments
+	 * @return SearchResultSet|Status<SearchResultSet>|null Result of calling $fn
+	 */
+	private function maybePaginate( Closure $fn ) {
+		if ( $this instanceof PaginatingSearchEngine ) {
+			return $fn();
+		}
+		$this->limit++;
+		try {
+			$resultSetOrStatus = $fn();
+		} finally {
+			$this->limit--;
+		}
+
+		$resultSet = null;
+		if ( $resultSetOrStatus instanceof SearchResultSet ) {
+			$resultSet = $resultSetOrStatus;
+		} elseif ( $resultSetOrStatus instanceof Status &&
+			$resultSetOrStatus->getValue() instanceof SearchResultSet
+		) {
+			$resultSet = $resultSetOrStatus->getValue();
+		}
+		if ( $resultSet ) {
+			$resultSet->shrink( $this->limit );
+		}
+
+		return $resultSetOrStatus;
+	}
+
 	/**
 	 * @since 1.18
 	 * @param string $feature
@@ -206,6 +244,8 @@ abstract class SearchEngine {
 	 * search=test&prefix=Main_Page/Archive -> test prefix:Main Page/Archive
 	 * @param string $term
 	 * @return string
+	 * @deprecated since 1.32 this should now be handled internally by the
+	 * search engine
 	 */
 	public function transformSearchTerm( $term ) {
 		return $term;
@@ -467,7 +507,7 @@ abstract class SearchEngine {
 	 *
 	 * @todo This isn't ideal, we'd really like to have content-specific handling here
 	 * @param Title $t Title we're indexing
-	 * @param Content $c Content of the page to index
+	 * @param Content|null $c Content of the page to index
 	 * @return string
 	 */
 	public function getTextFromContent( Title $t, Content $c = null ) {
@@ -523,6 +563,22 @@ abstract class SearchEngine {
 		return $search;
 	}
 
+	/**
+	 * Perform an overfetch of completion search results. This allows
+	 * determining if another page of results is available.
+	 *
+	 * @param string $search
+	 * @return SearchSuggestionSet
+	 */
+	protected function completionSearchBackendOverfetch( $search ) {
+		$this->limit++;
+		try {
+			return $this->completionSearchBackend( $search );
+		} finally {
+			$this->limit--;
+		}
+	}
+
 	/**
 	 * Perform a completion search.
 	 * Does not resolve namespaces and does not check variants.
@@ -560,7 +616,8 @@ abstract class SearchEngine {
 			return SearchSuggestionSet::emptySuggestionSet(); // Return empty result
 		}
 		$search = $this->normalizeNamespaces( $search );
-		return $this->processCompletionResults( $search, $this->completionSearchBackend( $search ) );
+		$suggestions = $this->completionSearchBackendOverfetch( $search );
+		return $this->processCompletionResults( $search, $suggestions );
 	}
 
 	/**
@@ -574,8 +631,8 @@ abstract class SearchEngine {
 		}
 		$search = $this->normalizeNamespaces( $search );
 
-		$results = $this->completionSearchBackend( $search );
-		$fallbackLimit = $this->limit - $results->getSize();
+		$results = $this->completionSearchBackendOverfetch( $search );
+		$fallbackLimit = 1 + $this->limit - $results->getSize();
 		if ( $fallbackLimit > 0 ) {
 			global $wgContLang;
 
@@ -614,15 +671,26 @@ abstract class SearchEngine {
 	 * @return SearchSuggestionSet
 	 */
 	protected function processCompletionResults( $search, SearchSuggestionSet $suggestions ) {
+		// We over-fetched to determine pagination. Shrink back down if we have extra results
+		// and mark if pagination is possible
+		$suggestions->shrink( $this->limit );
+
 		$search = trim( $search );
 		// preload the titles with LinkBatch
-		$titles = $suggestions->map( function ( SearchSuggestion $sugg ) {
+		$lb = new LinkBatch( $suggestions->map( function ( SearchSuggestion $sugg ) {
 			return $sugg->getSuggestedTitle();
-		} );
-		$lb = new LinkBatch( $titles );
+		} ) );
 		$lb->setCaller( __METHOD__ );
 		$lb->execute();
 
+		$diff = $suggestions->filter( function ( SearchSuggestion $sugg ) {
+			return $sugg->getSuggestedTitle()->isKnown();
+		} );
+		if ( $diff > 0 ) {
+			MediaWikiServices::getInstance()->getStatsdDataFactory()
+				->updateCount( 'search.completion.missing', $diff );
+		}
+
 		$results = $suggestions->map( function ( SearchSuggestion $sugg ) {
 			return $sugg->getSuggestedTitle()->getPrefixedText();
 		} );
@@ -830,7 +898,6 @@ abstract class SearchEngine {
 		$setAugmentors = [];
 		$rowAugmentors = [];
 		Hooks::run( "SearchResultsAugment", [ &$setAugmentors, &$rowAugmentors ] );
-
 		if ( !$setAugmentors && !$rowAugmentors ) {
 			// We're done here
 			return;