1 files changed, 89 insertions, 0 deletions
diff --git a/www/wiki/extensions/Translate/webservices/QueryAggregator.php b/www/wiki/extensions/Translate/webservices/QueryAggregator.php
new file mode 100644
index 00000000..6cc6465b
--- /dev/null
+++ b/www/wiki/extensions/Translate/webservices/QueryAggregator.php
@@ -0,0 +1,89 @@
+<?php
+/**
+ * Web service utility class.
+ *
+ * @file
+ * @author Niklas Laxström
+ * @license GPL-2.0-or-later
+ */
+
+/**
+ * Runs multiple web service queries asynchronously to save time.
+ *
+ * @ingroup TranslationWebService
+ * @since 2015.02
+ */
+class QueryAggregator {
+	protected $queries = [];
+	protected $responses = [];
+	protected $timeout = 0;
+	protected $hasRun = false;
+
+	/**
+	 * Register a query to be run.
+	 * @param TranslationQuery $query
+	 * @return mixed Query id that can be used to fetch results.
+	 */
+	public function addQuery( TranslationQuery $query ) {
+		$this->queries[] = $query;
+
+		$this->timeout = max( $query->getTimeout(), $this->timeout );
+		return count( $this->queries ) - 1;
+	}
+
+	/**
+	 * Returns a response for a query.
+	 * @param mixed $id Query id.
+	 * @return TranslationQueryResponse
+	 * @throws RuntimeException if called before run() has been called.
+	 */
+	public function getResponse( $id ) {
+		if ( !$this->hasRun ) {
+			throw new RuntimeException( 'Tried to get response before queries ran' );
+		}
+
+		return TranslationQueryResponse::newFromMultiHttp(
+			$this->responses[$id],
+			$this->queries[$id]
+		);
+	}
+
+	/**
+	 * Runs all the queries.
+	 */
+	public function run() {
+		global $wgSitename;
+
+		$version = TRANSLATE_VERSION;
+
+		$http = new MultiHttpClient( [
+			'reqTimeout' => $this->timeout,
+			'connTimeout' => 3,
+			'userAgent' => "MediaWiki Translate extension $version for $wgSitename"
+		] );
+		$responses = $http->runMulti( $this->getMultiHttpQueries( $this->queries ) );
+		foreach ( $responses as $index => $response ) {
+			$this->responses[$index] = $response;
+		}
+		$this->hasRun = true;
+	}
+
+	/**
+	 * Formats queries for format used by MultiHttpClient class.
+	 * @param TranslationQuery[] $queries
+	 * @return array[]
+	 */
+	protected function getMultiHttpQueries( $queries ) {
+		$converter = function ( TranslationQuery $q ) {
+			return [
+				'url' => $q->getUrl(),
+				'method' => $q->getMethod(),
+				'query' => $q->getQueryParameters(),
+				'body' => $q->getBody(),
+				'headers' => $q->getHeaders(),
+			];
+		};
+
+		return array_map( $converter, $queries );
+	}
+}