SearchResultSetTrait.php

Go to the documentation of this file.

<?php
 
trait SearchResultSetTrait {
    private $extraData = [];
 
    public function setAugmentedData( $name, $data ) {
        foreach ( $data as $id => $resultData ) {
            $this->extraData[$id][$name] = $resultData;
        }
    }
 
    public function augmentResult( SearchResult $result ) {
        $id = $result->getTitle()->getArticleID();
        if ( $id === -1 ) {
            return;
        }
        $result->setExtensionData( function () use ( $id ) {
            return $this->extraData[$id] ?? [];
        } );
    }
 
    public function getOffset() {
        return null;
    }
 
    final public function getIterator() {
        return new ArrayIterator( $this->extractResults() );
    }
}