X-Git-Url: https://git.heureux-cyclage.org/?a=blobdiff_plain;ds=sidebyside;f=maintenance%2FupdateArticleCount.php;h=213195df0f42fb8161b363607f4d8e46e8a60ea3;hb=c16f2ef0853f8cdc344ed2a89a5d0a27ef0e3e63;hp=470647a4c4f9e0c705bd4ddcab1621042d1faba9;hpb=5775d5e5c15098da4c1ebc121e678cee95ecaafa;p=lhc%2Fweb%2Fwiklou.git

diff --git a/maintenance/updateArticleCount.php b/maintenance/updateArticleCount.php
index 470647a4c4..213195df0f 100644
--- a/maintenance/updateArticleCount.php
+++ b/maintenance/updateArticleCount.php
@@ -35,24 +35,30 @@ class UpdateArticleCount extends Maintenance {
 
 	public function __construct() {
 		parent::__construct();
-		$this->mDescription = "Count of the number of articles and update the site statistics table";
+		$this->addDescription( 'Count of the number of articles and update the site statistics table' );
 		$this->addOption( 'update', 'Update the site_stats table with the new count' );
+		$this->addOption( 'use-master', 'Count using the master database' );
 	}
 
 	public function execute() {
 		$this->output( "Counting articles..." );
 
-		$counter = new SiteStatsInit( false );
+		if ( $this->hasOption( 'use-master' ) ) {
+			$dbr = $this->getDB( DB_MASTER );
+		} else {
+			$dbr = $this->getDB( DB_REPLICA, 'vslow' );
+		}
+		$counter = new SiteStatsInit( $dbr );
 		$result = $counter->articles();
 
 		$this->output( "found {$result}.\n" );
 		if ( $this->hasOption( 'update' ) ) {
 			$this->output( "Updating site statistics table... " );
-			$dbw = wfGetDB( DB_MASTER );
+			$dbw = $this->getDB( DB_MASTER );
 			$dbw->update(
 				'site_stats',
-				array( 'ss_good_articles' => $result ),
-				array( 'ss_row_id' => 1 ),
+				[ 'ss_good_articles' => $result ],
+				[ 'ss_row_id' => 1 ],
 				__METHOD__
 			);
 			$this->output( "done.\n" );