Code Coverage for /workspace/src/includes/search/PerRowAugmentor.php

1	<?php
2
3	/**
4	* Perform augmentation of each row and return composite result,
5	* indexed by ID.
6	*/
7	class PerRowAugmentor implements ResultSetAugmentor {
8
9	private ResultAugmentor $rowAugmentor;
10
11	/**
12	* @param ResultAugmentor $augmentor Per-result augmentor to use.
13	*/
14	public function __construct( ResultAugmentor $augmentor ) {
15	$this->rowAugmentor = $augmentor;
16	}
17
18	/**
19	* Produce data to augment search result set.
20	* @param ISearchResultSet $resultSet
21	* @return array Data for all results
22	*/
23	public function augmentAll( ISearchResultSet $resultSet ) {
24	$data = [];
25	foreach ( $resultSet->extractResults() as $result ) {
26	$id = $result->getTitle()->getArticleID();
27	if ( !$id ) {
28	continue;
29	}
30	$data[$id] = $this->rowAugmentor->augment( $result );
31	}
32	return $data;
33	}
34	}