SqlSearchResultSet.php

Go to the documentation of this file.

<?php
 
use MediaWiki\MediaWikiServices;
use MediaWiki\Title\Title;
use Wikimedia\Rdbms\IResultWrapper;
 
class SqlSearchResultSet extends SearchResultSet {
    protected $resultSet;
    protected $terms;
    protected $totalHits;
 
    public function __construct( IResultWrapper $resultSet, array $terms, $total = null ) {
        parent::__construct();
        $this->resultSet = $resultSet;
        $this->terms = $terms;
        $this->totalHits = $total;
    }
 
    public function termMatches() {
        return $this->terms;
    }
 
    public function numRows() {
        if ( $this->resultSet === false ) {
            return 0;
        }
 
        return $this->resultSet->numRows();
    }
 
    public function extractResults() {
        if ( $this->resultSet === false ) {
            return [];
        }
 
        if ( $this->results === null ) {
            $this->results = [];
            $this->resultSet->rewind();
            $terms = MediaWikiServices::getInstance()->getContentLanguage()
                ->convertForSearchResult( $this->terms );
            foreach ( $this->resultSet as $row ) {
                $result = new SqlSearchResult(
                    Title::makeTitle( $row->page_namespace, $row->page_title ),
                    $terms
                );
                $this->augmentResult( $result );
                $this->results[] = $result;
            }
        }
        return $this->results;
    }
 
    public function getTotalHits() {
        if ( $this->totalHits !== null ) {
            return $this->totalHits;
        } else {
            // Special:Search expects a number here.
            return $this->numRows();
        }
    }
}