X-Git-Url: https://git.cyclocoop.org/?a=blobdiff_plain;f=maintenance%2FupdateArticleCount.php;h=55f535d20e3401eea9a0165ea010959cd8a699f1;hb=4e004124cd0860b4eec0306de53b2ea738f6a881;hp=4d49dd2dbbc90350f9372243d821cf13c29a3b8f;hpb=ba25ed9ee004985826eeb4635ebf740da95ef297;p=lhc%2Fweb%2Fwiklou.git

diff --git a/maintenance/updateArticleCount.php b/maintenance/updateArticleCount.php
index 4d49dd2dbb..55f535d20e 100644
--- a/maintenance/updateArticleCount.php
+++ b/maintenance/updateArticleCount.php
@@ -23,7 +23,7 @@
  * @author Rob Church <robchur@gmail.com>
  */
 
-require_once( __DIR__ . '/Maintenance.php' );
+require_once __DIR__ . '/Maintenance.php';
 
 /**
  * Maintenance script to provide a better count of the number of articles
@@ -37,25 +37,37 @@ 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" );
 		}
 	}
 }
 
 $maintClass = "UpdateArticleCount";
-require_once( RUN_MAINTENANCE_IF_MAIN );
+require_once RUN_MAINTENANCE_IF_MAIN;