QueryAggregator.php

<?php
declare( strict_types = 1 );
 
namespace MediaWiki\Extension\Translate\WebService;
 
use MediaWiki\MediaWikiServices;
use RuntimeException;
use TranslateUtils;
 
class QueryAggregator {
    protected $queries = [];
    protected $responses = [];
    protected $timeout = 0;
    protected $hasRun = false;
 
    public function addQuery( TranslationQuery $query ) {
        $this->queries[] = $query;
 
        $this->timeout = max( $query->getTimeout(), $this->timeout );
        return count( $this->queries ) - 1;
    }
 
    public function getResponse( $id ): TranslationQueryResponse {
        if ( !$this->hasRun ) {
            throw new RuntimeException( 'Tried to get response before queries ran' );
        }
 
        return new TranslationQueryResponse( $this->responses[$id], $this->queries[$id] );
    }
 
    public function run(): void {
        global $wgSitename;
 
        $version = TranslateUtils::getVersion();
 
        $clientOptions = [
            'reqTimeout' => $this->timeout,
            'connTimeout' => 3,
            'userAgent' => "MediaWiki Translate extension $version for $wgSitename"
        ];
 
        $httpRequestFactory = MediaWikiServices::getInstance()->getHttpRequestFactory();
 
        $http = $httpRequestFactory->createMultiClient( $clientOptions );
        $responses = $http->runMulti( $this->getMultiHttpQueries( $this->queries ) );
        foreach ( $responses as $index => $response ) {
            $this->responses[$index] = $response;
        }
        $this->hasRun = true;
    }
 
    protected function getMultiHttpQueries( array $queries ): array {
        $converter = static function ( TranslationQuery $q ) {
            return [
                'url' => $q->getUrl(),
                'method' => $q->getMethod(),
                'query' => $q->getQueryParameters(),
                'body' => $q->getBody(),
                'headers' => $q->getHeaders(),
            ];
        };
 
        return array_map( $converter, $queries );
    }
}