1 files changed, 144 insertions, 0 deletions
diff --git a/www/wiki/extensions/SemanticMediaWiki/src/Elastic/Indexer/IndexerRecoveryJob.php b/www/wiki/extensions/SemanticMediaWiki/src/Elastic/Indexer/IndexerRecoveryJob.php
new file mode 100644
index 00000000..7304fc1d
--- /dev/null
+++ b/www/wiki/extensions/SemanticMediaWiki/src/Elastic/Indexer/IndexerRecoveryJob.php
@@ -0,0 +1,144 @@
+<?php
+
+namespace SMW\Elastic\Indexer;
+
+use SMW\ApplicationFactory;
+use SMW\DIWikiPage;
+use SMW\MediaWiki\Job;
+use SMW\Elastic\Connection\Client as ElasticClient;
+use SMW\Elastic\ElasticFactory;
+use SMW\SQLStore\ChangeOp\ChangeDiff;
+use Title;
+
+/**
+ * @license GNU GPL v2+
+ * @since 3.0
+ *
+ * @author mwjames
+ */
+class IndexerRecoveryJob extends Job {
+
+	/**
+	 * @since 3.0
+	 *
+	 * @param Title $title
+	 * @param array $params job parameters
+	 */
+	public function __construct( Title $title, $params = [] ) {
+		parent::__construct( 'smw.elasticIndexerRecovery', $title, $params );
+		$this->removeDuplicates = true;
+	}
+
+	/**
+	 * @see Job::run
+	 *
+	 * @since  3.0
+	 */
+	public function allowRetries() {
+		return false;
+	}
+
+	/**
+	 * @see Job::run
+	 *
+	 * @since  3.0
+	 */
+	public function run() {
+
+		$applicationFactory = ApplicationFactory::getInstance();
+		$store = $applicationFactory->getStore( '\SMW\SQLStore\SQLStore' );
+
+		$connection = $store->getConnection( 'elastic' );
+
+		// Make sure a node is available
+		if ( $connection->hasLock( ElasticClient::TYPE_DATA ) || !$connection->ping() ) {
+
+			if ( $connection->hasLock( ElasticClient::TYPE_DATA ) ) {
+				$this->params['retryCount'] = 0;
+			}
+
+			return $this->requeueRetry( $connection->getConfig() );
+		}
+
+		$elasticFactory = $applicationFactory->singleton( 'ElasticFactory' );
+
+		$this->indexer = $elasticFactory->newIndexer(
+			$store
+		);
+
+		$this->indexer->setOrigin( __METHOD__ );
+
+		$this->indexer->setLogger(
+			$applicationFactory->getMediaWikiLogger( 'smw-elastic' )
+		);
+
+		if ( $this->hasParameter( 'delete' ) ) {
+			$this->delete( $this->getParameter( 'delete' ) );
+		}
+
+		if ( $this->hasParameter( 'create' ) ) {
+			$this->create( $this->getParameter( 'create' ) );
+		}
+
+		if ( $this->hasParameter( 'index' ) ) {
+			$this->index(
+				$connection,
+				$applicationFactory->getCache(),
+				$this->getParameter( 'index' )
+			);
+		}
+
+		return true;
+	}
+
+	private function requeueRetry( $config ) {
+
+		// Give up!
+		if ( $this->getParameter( 'retryCount' ) >= $config->dotGet( 'indexer.job.recovery.retries' ) ) {
+			return true;
+		}
+
+		if ( !isset( $this->params['retryCount'] ) ) {
+			$this->params['retryCount'] = 1;
+		} else {
+			$this->params['retryCount']++;
+		}
+
+		if ( !isset( $this->params['createdAt'] ) ) {
+			$this->params['createdAt'] = time();
+		}
+
+		$job = new self( $this->title, $this->params );
+		$job->setDelay( 60 * 10 );
+
+		$job->insert();
+	}
+
+	private function delete( array $idList ) {
+		$this->indexer->delete( $idList );
+	}
+
+	private function create( $hash ) {
+		$this->indexer->create( DIWikiPage::doUnserialize( $hash ) );
+	}
+
+	private function index( $connection, $cache, $hash ) {
+
+		$subject = DIWikiPage::doUnserialize( $hash );
+		$text = '';
+
+		$changeDiff = ChangeDiff::fetch(
+			$cache,
+			$subject
+		);
+
+		if ( $connection->getConfig()->dotGet( 'indexer.raw.text', false ) ) {
+			$text = $this->indexer->fetchNativeData( $subject );
+		}
+
+		if ( $changeDiff !== false ) {
+			$this->indexer->index( $changeDiff, $text );
+		}
+	}
+
+}