1 files changed, 213 insertions, 0 deletions
diff --git a/www/wiki/extensions/SemanticMediaWiki/src/MediaWiki/JobQueue.php b/www/wiki/extensions/SemanticMediaWiki/src/MediaWiki/JobQueue.php
new file mode 100644
index 00000000..2e580459
--- /dev/null
+++ b/www/wiki/extensions/SemanticMediaWiki/src/MediaWiki/JobQueue.php
@@ -0,0 +1,213 @@
+<?php
+
+namespace SMW\MediaWiki;
+
+use JobQueueGroup;
+
+/**
+ * MediaWiki's JobQueue contains mostly final methods making it difficult to use
+ * an instance during tests hence this class provides a reduced interface with
+ * mockable methods.
+ *
+ * @license GNU GPL v2+
+ * @since 3.0
+ *
+ * @author mwjames
+ */
+class JobQueue {
+
+	/**
+	 * @var JobQueueGroup
+	 */
+	private $jobQueueGroup;
+
+	/**
+	 * @var boolean
+	 */
+	private $disableCache = false;
+
+	/**
+	 * @since 3.0
+	 *
+	 * @param JobQueueGroup $jobQueueGroup
+	 */
+	public function __construct( JobQueueGroup $jobQueueGroup ) {
+		$this->jobQueueGroup = $jobQueueGroup;
+	}
+
+	/**
+	 * @since 3.0
+	 *
+	 * @param boolean $disableCache
+	 */
+	public function disableCache( $disableCache = true ) {
+		$this->disableCache = (bool)$disableCache;
+	}
+
+	/**
+	 * @since 3.0
+	 *
+	 * @param string $type
+	 *
+	 * @return boolean
+	 */
+	public function isDelayedJobsEnabled( $type ) {
+		return $this->jobQueueGroup->get( $this->mapLegacyType( $type ) )->delayedJobsEnabled();
+	}
+
+	/**
+	 * @since 3.0
+	 *
+	 * @param array $list
+	 *
+	 * @return []
+	 */
+	public function runFromQueue( array $list ) {
+
+		$log = [];
+
+		foreach ( $list as $type => $amount ) {
+
+			if ( $amount == 0 || $amount === false ) {
+				continue;
+			}
+
+			$jobs = array_fill( 0, $amount, $type );
+			$log[$type] = [];
+
+			foreach ( $jobs as $job ) {
+				$j = $this->pop( $job );
+
+				if ( $j === false ) {
+					break;
+				}
+
+				$log[$type][] = $j->getTitle()->getPrefixedDBKey();
+
+				$j->run();
+				$this->ack( $j );
+			}
+		}
+
+		return $log;
+	}
+
+	/**
+	 * @since 3.0
+	 *
+	 * @param string $type
+	 *
+	 * @return Job|boolean
+	 */
+	public function pop( $type ) {
+		return $this->jobQueueGroup->get( $this->mapLegacyType( $type ) )->pop();
+	}
+
+	/**
+	 * Acknowledge that a job was completed
+	 *
+	 * @since 3.0
+	 *
+	 * @param Job $job
+	 */
+	public function ack( \Job $job ) {
+		$this->jobQueueGroup->get( $job->getType() )->ack( $job );
+	}
+
+	/**
+	 * @since 3.0
+	 *
+	 * @param string $type
+	 */
+	public function delete( $type ) {
+
+		$jobQueue = $this->jobQueueGroup->get( $this->mapLegacyType( $type ) );
+		$jobQueue->delete();
+
+		if ( $this->disableCache ) {
+			$jobQueue->flushCaches();
+			$this->disableCache = false;
+		}
+	}
+
+	/**
+	 * @since 3.0
+	 *
+	 * @param Job|Job[] $jobs
+	 */
+	public function push( $jobs ) {
+		$this->jobQueueGroup->push( $jobs );
+	}
+
+	/**
+	 * @since 3.0
+	 *
+	 * @param Job|Job[] $jobs
+	 */
+	public function lazyPush( $jobs ) {
+
+		if ( !method_exists( $this->jobQueueGroup, 'lazyPush' ) ) {
+			return $this->push( $jobs );
+		}
+
+		$this->jobQueueGroup->lazyPush( $jobs );
+	}
+
+	/**
+	 * @since 3.0
+	 *
+	 * @return array
+	 */
+	public function getQueueSizes() {
+		return $this->jobQueueGroup->getQueueSizes();
+	}
+
+	/**
+	 * @since 3.0
+	 *
+	 * @param string $type
+	 *
+	 * @return integer
+	 */
+	public function getQueueSize( $type ) {
+
+		$jobQueue = $this->jobQueueGroup->get( $this->mapLegacyType( $type ) );
+
+		if ( $this->disableCache ) {
+			$jobQueue->flushCaches();
+			$this->disableCache = false;
+		}
+
+		return $jobQueue->getSize();
+	}
+
+	/**
+	 * @since 3.0
+	 *
+	 * @param string $type
+	 *
+	 * @return boolean
+	 */
+	public function hasPendingJob( $type ) {
+		return $this->getQueueSize( $type ) > 0;
+	}
+
+	/**
+	 * @note FIXME Remove with 3.1
+	 * @since 3.0
+	 *
+	 * @param string $type
+	 *
+	 * @return string
+	 */
+	public static function mapLegacyType( $type ) {
+
+		// Legacy names
+		if ( strpos( $type, 'SMW\\' ) !== false ) {
+			$type = 'smw.' . lcfirst( str_replace( [ 'SMW\\', 'Job' ], '', $type ) );
+		}
+
+		return $type;
+	}
+
+}