LinksUpdate: Replace the deprecated function wfGetDB with IConnectionProvider

Change-Id: I432151384f69dcb33671fd6b4c257960e294ef52

1 files changed, 5 insertions, 1 deletions

diff --git a/includes/deferred/LinksUpdate/LinksUpdate.php b/includes/deferred/LinksUpdate/LinksUpdate.php
index 75d996443bcc..0970966d175f 100644
--- a/includes/deferred/LinksUpdate/LinksUpdate.php
+++ b/includes/deferred/LinksUpdate/LinksUpdate.php
@@ -41,6 +41,7 @@ use MediaWiki\Title\Title;
 use MediaWiki\User\UserIdentity;
 use RefreshLinksJob;
 use RuntimeException;
+use Wikimedia\Rdbms\IConnectionProvider;
 use Wikimedia\Rdbms\IDatabase;
 use Wikimedia\ScopedCallback;
 
@@ -83,6 +84,8 @@ class LinksUpdate extends DataUpdate {
 	/** @var LinksTableGroup */
 	private $tableFactory;
 
+	private IConnectionProvider $connectionProvider;
+
 	/**
 	 * @param PageIdentity $page The page we're updating
 	 * @param ParserOutput $parserOutput Output from a full parse of this page
@@ -116,6 +119,7 @@ class LinksUpdate extends DataUpdate {
 		);
 		// TODO: this does not have to be called in LinksDeletionUpdate
 		$this->tableFactory->setParserOutput( $parserOutput );
+		$this->connectionProvider = $services->getDBLoadBalancerFactory();
 	}
 
 	public function setTransactionTicket( $ticket ) {
@@ -537,7 +541,7 @@ class LinksUpdate extends DataUpdate {
 	 */
 	protected function getDB() {
 		if ( !$this->db ) {
-			$this->db = wfGetDB( DB_PRIMARY );
+			$this->db = $this->connectionProvider->getPrimaryDatabase();
 		}
 
 		return $this->db;