X-Git-Url: https://git.heureux-cyclage.org/?a=blobdiff_plain;f=maintenance%2FrebuildLocalisationCache.php;h=5915a6b02dd6c9dd504c3427fe2484baefaf374b;hb=500a1401d224455b562fbafd08d63b13023824e3;hp=bc61eb165f883ed75f0404d0c0d3364e9fc90dc4;hpb=cfeca6aaa5aa82dc08538073ca55070b2f7dd960;p=lhc%2Fweb%2Fwiklou.git

diff --git a/maintenance/rebuildLocalisationCache.php b/maintenance/rebuildLocalisationCache.php
index bc61eb165f..5915a6b02d 100644
--- a/maintenance/rebuildLocalisationCache.php
+++ b/maintenance/rebuildLocalisationCache.php
@@ -5,65 +5,143 @@
  * using $wgLocalisationCacheConf['manualRecache'] = true;
  *
  * Usage:
- *    php rebuildLocalisationCache.php [--force]
+ *    php rebuildLocalisationCache.php [--force] [--threads=N]
  *
  * Use --force to rebuild all files, even the ones that are not out of date.
  * Use --threads=N to fork more threads.
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with this program; if not, write to the Free Software Foundation, Inc.,
+ * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
+ * http://www.gnu.org/copyleft/gpl.html
+ *
+ * @ingroup Maintenance
  */
 
-require( dirname(__FILE__).'/commandLine.inc' );
-ini_set( 'memory_limit', '200M' );
+require_once( dirname( __FILE__ ) . '/Maintenance.php' );
 
-$force = isset( $options['force'] );
-$threads = intval( isset( $options['threads'] ) ? $options['threads'] : 1 );
+class RebuildLocalisationCache extends Maintenance {
+	public function __construct() {
+		parent::__construct();
+		$this->mDescription = "Rebuild the localisation cache";
+		$this->addOption( 'force', 'Rebuild all files, even ones not out of date' );
+		$this->addOption( 'threads', 'Fork more than one thread', false, true );
+		$this->addOption( 'outdir', 'Override the output directory (normally $wgCacheDirectory)', 
+			false, true );
+	}
 
-$conf = $wgLocalisationCacheConf;
-$conf['manualRecache'] = false; // Allow fallbacks to create CDB files
-if ( $force ) {
-	$conf['forceRecache'] = true;
-}
-$lc = new LocalisationCache_BulkLoad( $conf );
+	public function memoryLimit() {
+		return '1000M';
+	}
 
-$codes = array_keys( Language::getLanguageNames( true ) );
-sort( $codes );
+	public function execute() {
+		global $wgLocalisationCacheConf;
 
-// Initialise and split into chunks
-$numRebuilt = 0;
-$total = count($codes);
-$chunks = array_chunk( $codes, ceil(count($codes)/$threads) );
-$pids = array();
+		$force = $this->hasOption( 'force' );
+		$threads = $this->getOption( 'threads', 1 );
+		if ( $threads < 1 || $threads != intval( $threads ) ) {
+			$this->output( "Invalid thread count specified; running single-threaded.\n" );
+			$threads = 1;
+		}
+		if ( $threads > 1 && wfIsWindows() ) {
+			$this->output( "Threaded rebuild is not supported on Windows; running single-threaded.\n" );
+			$threads = 1;
+		}
+		if ( $threads > 1 && !function_exists( 'pcntl_fork' ) ) {
+			$this->output( "PHP pcntl extension is not present; running single-threaded.\n" );
+			$threads = 1;
+		}
 
-foreach ( $chunks as $codes ) {
-	// Do not fork for only one thread
-	$pid = ( $threads > 1 ) ? pcntl_fork() : -1;
+		$conf = $wgLocalisationCacheConf;
+		$conf['manualRecache'] = false; // Allow fallbacks to create CDB files
+		if ( $force ) {
+			$conf['forceRecache'] = true;
+		}
+		if ( $this->hasOption( 'outdir' ) ) {
+			$conf['storeDirectory'] = $this->getOption( 'outdir' );
+		}
+		$lc = new LocalisationCache_BulkLoad( $conf );
 
-	if ( $pid === 0 ) {
-		// Child
-		doRebuild( $codes, $numRebuilt, $lc, $force );
-		exit();
-	} elseif ($pid === -1) {
-		// Fork failed or one thread, do it serialized
-		doRebuild( $codes, $numRebuilt, $lc, $force );
-	} else {
-		// Main thread
-		$pids[] = $pid;
-	}
-}
+		$codes = array_keys( Language::fetchLanguageNames( null, 'mwfile' ) );
+		sort( $codes );
 
-// Wait for all children
-foreach ( $pids as $pid ) pcntl_waitpid($pid, $status);
+		// Initialise and split into chunks
+		$numRebuilt = 0;
+		$total = count( $codes );
+		$chunks = array_chunk( $codes, ceil( count( $codes ) / $threads ) );
+		$pids = array();
+		foreach ( $chunks as $codes ) {
+			// Do not fork for only one thread
+			$pid = ( $threads > 1 ) ? pcntl_fork() : -1;
 
-echo "$numRebuilt languages rebuilt out of $total.\n";
-if ( $numRebuilt == 0 ) {
-	echo "Use --force to rebuild the caches which are still fresh.\n";
-}
+			if ( $pid === 0 ) {
+				// Child, reseed because there is no bug in PHP:
+				// http://bugs.php.net/bug.php?id=42465
+				mt_srand( getmypid() );
+				$numRebuilt = $this->doRebuild( $codes, $lc, $force );
+				// Abuse the exit value for the count of rebuild languages
+				exit( $numRebuilt );
+			} elseif ( $pid === -1 ) {
+				// Fork failed or one thread, do it serialized
+				$numRebuilt += $this->doRebuild( $codes, $lc, $force );
+			} else {
+				// Main thread
+				$pids[] = $pid;
+			}
+		}
+		// Wait for all children
+		foreach ( $pids as $pid ) {
+			$status = 0;
+			pcntl_waitpid( $pid, $status );
+			// Fetch the count from the return value
+			$numRebuilt += pcntl_wexitstatus( $status );
+		}
 
-function doRebuild( $codes, &$numRebuilt, $lc, $force ) {
-	foreach ( $codes as $code ) {
-		if ( $force || $lc->isExpired( $code ) ) {
-			echo "Rebuilding $code...\n";
-			$lc->recache( $code );
-			$numRebuilt++;
+		$this->output( "$numRebuilt languages rebuilt out of $total\n" );
+		if ( $numRebuilt === 0 ) {
+			$this->output( "Use --force to rebuild the caches which are still fresh.\n" );
 		}
 	}
+
+	/**
+	 * Helper function to rebuild list of languages codes. Prints the code
+	 * for each language which is rebuilt.
+	 * @param $codes array List of language codes to rebuild.
+	 * @param $lc LocalisationCache Instance of LocalisationCache_BulkLoad (?)
+	 * @param $force bool Rebuild up-to-date languages
+	 * @return int Number of rebuilt languages
+	 */
+	private function doRebuild( $codes, $lc, $force ) {
+		$numRebuilt = 0;
+		foreach ( $codes as $code ) {
+			if ( $force || $lc->isExpired( $code ) ) {
+				$this->output( "Rebuilding $code...\n" );
+				$lc->recache( $code );
+				$numRebuilt++;
+			}
+		}
+		return $numRebuilt;
+	}
+
+	/**
+	 * Sets whether a run of this maintenance script has the force parameter set
+	 *
+	 * @param bool $forced
+	 */
+	public function setForce( $forced = true ) {
+		$this->mOptions['force'] = $forced;
+	}
 }
+
+$maintClass = "RebuildLocalisationCache";
+require_once( RUN_MAINTENANCE_IF_MAIN );