summaryrefslogtreecommitdiff
path: root/www/wiki/includes/search/PerRowAugmentor.php
blob: a3979f7b71e903e710b33476f2db3c2b027741b6 (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
<?php

/**
 * Perform augmentation of each row and return composite result,
 * indexed by ID.
 */
class PerRowAugmentor implements ResultSetAugmentor {

	/**
	 * @var ResultAugmentor
	 */
	private $rowAugmentor;

	/**
	 * @param ResultAugmentor $augmentor Per-result augmentor to use.
	 */
	public function __construct( ResultAugmentor $augmentor ) {
		$this->rowAugmentor = $augmentor;
	}

	/**
	 * Produce data to augment search result set.
	 * @param SearchResultSet $resultSet
	 * @return array Data for all results
	 */
	public function augmentAll( SearchResultSet $resultSet ) {
		$data = [];
		foreach ( $resultSet->extractResults() as $result ) {
			$id = $result->getTitle()->getArticleID();
			if ( !$id ) {
				continue;
			}
			$data[$id] = $this->rowAugmentor->augment( $result );
		}
		return $data;
	}
}