summaryrefslogtreecommitdiff
path: root/www/wiki/includes/search/SqlSearchResultSet.php
blob: 53d09e82b1017858c42df0b2d1d4679fdeed77bf (plain)
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
<?php

use Wikimedia\Rdbms\ResultWrapper;

/**
 * This class is used for different SQL-based search engines shipped with MediaWiki
 * @ingroup Search
 */
class SqlSearchResultSet extends SearchResultSet {
	protected $resultSet;
	protected $terms;
	protected $totalHits;

	function __construct( ResultWrapper $resultSet, $terms, $total = null ) {
		$this->resultSet = $resultSet;
		$this->terms = $terms;
		$this->totalHits = $total;
	}

	function termMatches() {
		return $this->terms;
	}

	function numRows() {
		if ( $this->resultSet === false ) {
			return false;
		}

		return $this->resultSet->numRows();
	}

	function next() {
		if ( $this->resultSet === false ) {
			return false;
		}

		$row = $this->resultSet->fetchObject();
		if ( $row === false ) {
			return false;
		}

		return SearchResult::newFromTitle(
			Title::makeTitle( $row->page_namespace, $row->page_title ), $this
		);
	}

	function rewind() {
		if ( $this->resultSet ) {
			$this->resultSet->rewind();
		}
	}

	function free() {
		if ( $this->resultSet === false ) {
			return false;
		}

		$this->resultSet->free();
	}

	function getTotalHits() {
		if ( !is_null( $this->totalHits ) ) {
			return $this->totalHits;
		} else {
			// Special:Search expects a number here.
			return $this->numRows();
		}
	}
}