Use ConnectionProvider

miraheze · Nov 20, 2024 · 3906658 · 3906658
1 parent 334d0dd
commit 3906658
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/maintenance/insertMissingLocalUserRows.php b/maintenance/insertMissingLocalUserRows.php
@@ -49,14 +49,11 @@ public function execute() {
 		$dryRun = $this->getOption( 'dry-run', false );
 		$all = $this->getOption( 'all-wikis', false );
 
-		$centralDB = $this->getServiceContainer()->getDBLoadBalancerFactory()
-			->getMainLB( $this->getConfig()->get( 'CentralAuthDatabase' ) )
-			->getMaintenanceConnectionRef( DB_REPLICA, [], $this->getConfig()->get( 'CentralAuthDatabase' ) );
+		$connectionProvider = $this->getServiceContainer()->getConnectionProvider();
+		$centralDB = $connectionProvider->getReplicaDatabase( $this->getConfig()->get( 'CentralAuthDatabase' ) );
 
 		foreach ( ( $all ? $this->getConfig()->get( MainConfigNames::LocalDatabases ) : [ WikiMap::getCurrentWikiId() ] ) as $wiki ) {
-			$lb = $this->getServiceContainer()->getDBLoadBalancerFactory()->getMainLB( $wiki );
-
-			$res = $lb->getMaintenanceConnectionRef( DB_REPLICA, [], $wiki )->select(
+			$res = $connectionProvider->getReplicaDatabase( $wiki )->select(
 				'user',
 				[ 'user_name' ],
 				[