X-Git-Url: https://git.heureux-cyclage.org/?a=blobdiff_plain;f=maintenance%2FupdateArticleCount.php;h=55f535d20e3401eea9a0165ea010959cd8a699f1;hb=6dfc09c48b684578519c07e2a245f2f5c12883ec;hp=7964a21f82234d8a83087187e57dee97c18cd392;hpb=e7651f865d31dd6203828cf509383da2ea7515c3;p=lhc%2Fweb%2Fwiklou.git

diff --git a/maintenance/updateArticleCount.php b/maintenance/updateArticleCount.php
index 7964a21f82..55f535d20e 100644
--- a/maintenance/updateArticleCount.php
+++ b/maintenance/updateArticleCount.php
@@ -37,22 +37,34 @@ class UpdateArticleCount extends Maintenance {
 		parent::__construct();
 		$this->mDescription = "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 = wfGetDB( DB_MASTER );
+		} else {
+			$dbr = wfGetDB( DB_SLAVE, '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->update( 'site_stats', array( 'ss_good_articles' => $result ), array( 'ss_row_id' => 1 ), __METHOD__ );
+			$dbw->update(
+				'site_stats',
+				array( 'ss_good_articles' => $result ),
+				array( 'ss_row_id' => 1 ),
+				__METHOD__
+			);
 			$this->output( "done.\n" );
 		} else {
-			$this->output( "To update the site statistics table, run the script with the --update option.\n" );
+			$this->output( "To update the site statistics table, run the script "
+				. "with the --update option.\n" );
 		}
 	}
 }