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         /**
  10          * @var ResultAugmentor
  11          */
  12         private $rowAugmentor;
  13
  14         /**
  15          * @param ResultAugmentor $augmentor Per-result augmentor to use.
  16          */
  17         public function __construct( ResultAugmentor $augmentor ) {
  18                 $this->rowAugmentor = $augmentor;
  19         }
  20
  21         /**
  22          * Produce data to augment search result set.
  23          * @param ISearchResultSet $resultSet
  24          * @return array Data for all results
  25          */
  26         public function augmentAll( ISearchResultSet $resultSet ) {
  27                 $data = [];
  28                 foreach ( $resultSet->extractResults() as $result ) {
  29                         $id = $result->getTitle()->getArticleID();
  30                         if ( !$id ) {
  31                                 continue;
  32                         }
  33                         $data[$id] = $this->rowAugmentor->augment( $result );
  34                 }
  35                 return $data;
  36         }
  37 }