1 files changed, 247 insertions, 0 deletions
diff --git a/www/wiki/extensions/SemanticMediaWiki/src/SQLStore/QueryDependency/DependencyLinksTableUpdater.php b/www/wiki/extensions/SemanticMediaWiki/src/SQLStore/QueryDependency/DependencyLinksTableUpdater.php
new file mode 100644
index 00000000..7a623771
--- /dev/null
+++ b/www/wiki/extensions/SemanticMediaWiki/src/SQLStore/QueryDependency/DependencyLinksTableUpdater.php
@@ -0,0 +1,247 @@
+<?php
+
+namespace SMW\SQLStore\QueryDependency;
+
+use Psr\Log\LoggerAwareTrait;
+use SMW\DIWikiPage;
+use SMW\SQLStore\SQLStore;
+use SMW\Store;
+
+/**
+ * @license GNU GPL v2+
+ * @since 2.4
+ *
+ * @author mwjames
+ */
+class DependencyLinksTableUpdater {
+
+	use LoggerAwareTrait;
+
+	/**
+	 * @var array
+	 */
+	private static $updateList = [];
+
+	/**
+	 * @var Store
+	 */
+	private $store;
+
+	/**
+	 * @since 2.4
+	 *
+	 * @param Store $store
+	 */
+	public function __construct( Store $store ) {
+		$this->store = $store;
+	}
+
+	/**
+	 * @since 2.4
+	 *
+	 * @return Store
+	 */
+	public function getStore() {
+		return $this->store;
+	}
+
+	/**
+	 * @since 2.4
+	 */
+	public function clear() {
+		self::$updateList = [];
+	}
+
+	/**
+	 * @since 2.4
+	 *
+	 * @param integer $sid
+	 * @param array|null $dependencyList
+	 */
+	public function addToUpdateList( $sid, array $dependencyList = null ) {
+
+		if ( $sid == 0 || $dependencyList === null || $dependencyList === [] ) {
+			return null;
+		}
+
+		if ( !isset( self::$updateList[$sid] ) ) {
+			return self::$updateList[$sid] = $dependencyList;
+		}
+
+		self::$updateList[$sid] = array_merge( self::$updateList[$sid], $dependencyList );
+	}
+
+	/**
+	 * @since 2.4
+	 */
+	public function doUpdate() {
+		foreach ( self::$updateList as $sid => $dependencyList ) {
+
+			if ( $dependencyList === [] ) {
+				continue;
+			}
+
+			$this->updateDependencyList( $sid, $dependencyList );
+			self::$updateList[$sid] = [];
+		}
+	}
+
+	/**
+	 * @since 2.4
+	 *
+	 * @param array $dependencyList
+	 */
+	public function deleteDependenciesFromList( array $deleteIdList ) {
+
+		$context = [
+			'method' => __METHOD__,
+			'role' => 'developer',
+			'list' => implode( ' ,', $deleteIdList )
+		];
+
+		$this->logger->info( '[QueryDependency] Delete dependencies: {list}', $context );
+
+		$connection = $this->store->getConnection( 'mw.db' );
+		$connection->beginAtomicTransaction( __METHOD__ );
+
+		$connection->delete(
+			SQLStore::QUERY_LINKS_TABLE,
+			[
+				's_id' => $deleteIdList
+			],
+			__METHOD__
+		);
+
+		$connection->endAtomicTransaction( __METHOD__ );
+	}
+
+	/**
+	 * @since 2.4
+	 *
+	 * @param integer $sid
+	 * @param array $dependencyList
+	 */
+	private function updateDependencyList( $sid, array $dependencyList ) {
+
+		$connection = $this->store->getConnection( 'mw.db' );
+		$connection->beginAtomicTransaction( __METHOD__ );
+
+		// Before an insert, delete all entries that for the criteria which is
+		// cheaper then doing an individual upsert or selectRow, this also ensures
+		// that entries are self-corrected for dependencies matched
+		$connection->delete(
+			SQLStore::QUERY_LINKS_TABLE,
+			[
+				's_id' => $sid
+			],
+			__METHOD__
+		);
+
+		if ( $sid == 0 ) {
+			return $connection->endAtomicTransaction( __METHOD__ );
+		}
+
+		$inserts = [];
+
+		foreach ( $dependencyList as $dependency ) {
+
+			if ( !$dependency instanceof DIWikiPage ) {
+				continue;
+			}
+
+			$oid = $this->getId( $dependency );
+
+			// If the ID_TABLE didn't contained an valid ID then we create one ourselves
+			// to ensure that object entities are tracked from the start
+			// This can happen when a query is added with object reference that have not
+			// yet been referenced as annotation and therefore do not recognized as
+			// value annotation
+			if ( $oid < 1 && ( ( $oid = $this->createId( $dependency ) ) < 1 ) ) {
+				continue;
+			}
+
+			$inserts[$sid . $oid] = [
+				's_id' => $sid,
+				'o_id' => $oid
+			];
+		}
+
+		if ( $inserts === [] ) {
+			return $connection->endAtomicTransaction( __METHOD__ );
+		}
+
+		// MW's multi-array insert needs a numeric dimensional array but the key
+		// was used with a hash to avoid duplicate entries hence the re-copy
+		$inserts = array_values( $inserts );
+
+		$context = [
+			'method' => __METHOD__,
+			'role' => 'developer',
+			'id' => $sid
+		];
+
+		$this->logger->info( '[QueryDependency] Table insert: {id} ID', $context );
+
+		$connection->insert(
+			SQLStore::QUERY_LINKS_TABLE,
+			$inserts,
+			__METHOD__
+		);
+
+		$connection->endAtomicTransaction( __METHOD__ );
+	}
+
+	/**
+	 * @since 2.4
+	 *
+	 * @param DIWikiPage $subject, $subobjectName
+	 * @param string $subobjectName
+	 */
+	public function getId( DIWikiPage $subject, $subobjectName = '' ) {
+
+		if ( $subobjectName !== '' ) {
+			$subject = new DIWikiPage(
+				$subject->getDBkey(),
+				$subject->getNamespace(),
+				$subject->getInterwiki(),
+				$subobjectName
+			);
+		}
+
+		$id = $this->store->getObjectIds()->getId(
+			$subject
+		);
+
+		return $id;
+	}
+
+	/**
+	 * @since 2.4
+	 *
+	 * @param DIWikiPage $subject, $subobjectName
+	 * @param string $subobjectName
+	 */
+	public function createId( DIWikiPage $subject, $subobjectName = '' ) {
+
+		$id = $this->store->getObjectIds()->makeSMWPageID(
+			$subject->getDBkey(),
+			$subject->getNamespace(),
+			$subject->getInterwiki(),
+			$subobjectName,
+			false
+		);
+
+		$context = [
+			'method' => __METHOD__,
+			'role' => 'developer',
+			'id' => $id,
+			'origin' => $subject->getHash() . $subobjectName
+
+		];
+
+		$this->logger->info( '[QueryDependency] Table update: new {id} ID; {origin}', $context );
+
+		return $id;
+	}
+
+}