1 files changed, 73 insertions, 0 deletions
diff --git a/www/wiki/extensions/SemanticMediaWiki/src/Connection/ConnectionManager.php b/www/wiki/extensions/SemanticMediaWiki/src/Connection/ConnectionManager.php
new file mode 100644
index 00000000..4e16ffee
--- /dev/null
+++ b/www/wiki/extensions/SemanticMediaWiki/src/Connection/ConnectionManager.php
@@ -0,0 +1,73 @@
+<?php
+
+namespace SMW\Connection;
+
+use RuntimeException;
+
+/**
+ * @license GNU GPL v2+
+ * @since 2.1
+ *
+ * @author mwjames
+ */
+class ConnectionManager {
+
+	/**
+	 * By design this variable is static to ensure that ConnectionProvider
+	 * instances are only initialized once per request.
+	 *
+	 * @var array
+	 */
+	private static $connectionProviders = [];
+
+	/**
+	 * @since 2.1
+	 *
+	 * @param string|null $id
+	 *
+	 * @return mixed
+	 * @throws RuntimeException
+	 */
+	public function getConnection( $id = null ) {
+		return $this->findConnectionProvider( strtolower( $id ) )->getConnection();
+	}
+
+	/**
+	 * @since 2.1
+	 */
+	public function releaseConnections() {
+		foreach ( self::$connectionProviders as $connectionProvider ) {
+			$connectionProvider->releaseConnection();
+		}
+	}
+
+	/**
+	 * @since 2.1
+	 *
+	 * @param string $id
+	 * @param ConnectionProvider $connectionProvider
+	 */
+	public function registerConnectionProvider( $id, ConnectionProvider $connectionProvider ) {
+		self::$connectionProviders[strtolower( $id )] = $connectionProvider;
+	}
+
+	/**
+	 * @since 3.0
+	 *
+	 * @param string $id
+	 * @param callable $callback
+	 */
+	public function registerCallbackConnection( $id, callable $callback ) {
+		self::$connectionProviders[strtolower( $id )] = new CallbackConnectionProvider( $callback );
+	}
+
+	private function findConnectionProvider( $id ) {
+
+		if ( isset( self::$connectionProviders[$id] ) ) {
+			return self::$connectionProviders[$id];
+		}
+
+		throw new RuntimeException( "{$id} is missing a registered connection provider" );
+	}
+
+}