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 ISearchResultSet $resultSet
* @return array Data for all results
*/
public function augmentAll( ISearchResultSet $resultSet ) {
$data = [];
foreach ( $resultSet->extractResults() as $result ) {
$id = $result->getTitle()->getArticleID();
if ( !$id ) {
continue;
}
$data[$id] = $this->rowAugmentor->augment( $result );
}
return $data;
}
}
|
