X-Git-Url: https://git.heureux-cyclage.org/?a=blobdiff_plain;f=maintenance%2FpopulateParentId.php;h=45255b020c578120e551a17d859a51733a58729d;hb=f980fa58a14b66ccd875d481c632ee3213b375b9;hp=1dfc2298f9009082dbf0e0db75d44f5521f98164;hpb=3eeccabcccbba67ed4e5c2acf6e1428a2c7ed464;p=lhc%2Fweb%2Fwiklou.git

diff --git a/maintenance/populateParentId.php b/maintenance/populateParentId.php
index 1dfc2298f9..45255b020c 100644
--- a/maintenance/populateParentId.php
+++ b/maintenance/populateParentId.php
@@ -1,64 +1,118 @@
 <?php
-
 /*
  * Makes the required database updates for rev_parent_id
  * to be of any use. It can be used for some simple tracking
  * and to find new page edits by users.
+ *
+ * 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
  */
 
-define( 'BATCH_SIZE', 500 );
-
-require_once 'commandLine.inc';
-	
-$db =& wfGetDB( DB_MASTER );
-if ( !$db->tableExists( 'revision' ) ) {
-	echo "revision table does not exist\n";
-	exit( 1 );
-}
+require_once( dirname( __FILE__ ) . '/Maintenance.php' );
 
-populate_rev_parent_id( $db );
+class PopulateParentId extends Maintenance {
+	public function __construct() {
+		parent::__construct();
+		$this->mDescription = "Populates rev_parent_id";
+		$this->setBatchSize( 200 );
+	}
 
-function populate_rev_parent_id( $db ) {
-	echo "Populating rev_parent_id column\n";
-	$start = $db->selectField( 'revision', 'MIN(rev_id)', false, __FUNCTION__ );
-	$end = $db->selectField( 'revision', 'MAX(rev_id)', false, __FUNCTION__ );
-	$blockStart = $start;
-	$blockEnd = $start + BATCH_SIZE - 1;
-	$count = 0;
-	while( $blockEnd <= $end ) {
-		$cond = "rev_id BETWEEN $blockStart AND $blockEnd";
-		$res = $db->select( 'revision', array('rev_id','rev_page'), $cond, __FUNCTION__ );
-		# Go through and update rev_parent_id from these rows.
-		# Assume that the previous revision of the title was
-		# the original previous revision of the title when the
-		# edit was made...
-		foreach( $res as $row ) {
-			$previousID = $db->selectField( 'revision', 'rev_id', 
-				array( 'rev_page' => $row->rev_page, "rev_id < '{$row->rev_id}'" ), 
-				__FUNCTION__,
-				array( 'ORDER BY' => 'rev_id DESC' ) );
-			# Update the row...
-			$db->update( 'revision',
-				array( 'rev_parent_id' => intval($previousID) ),
-				array( 'rev_id' => $row->rev_id ),
-				__FUNCTION__ );
-			$count++;
+	public function execute() {
+		$db = wfGetDB( DB_MASTER );
+		if ( !$db->tableExists( 'revision' ) ) {
+			$this->error( "revision table does not exist", true );
+		}
+		$this->output( "Populating rev_parent_id column\n" );
+		$start = $db->selectField( 'revision', 'MIN(rev_id)', false, __FUNCTION__ );
+		$end = $db->selectField( 'revision', 'MAX(rev_id)', false, __FUNCTION__ );
+		if ( is_null( $start ) || is_null( $end ) ) {
+			$this->output( "...revision table seems to be empty.\n" );
+			$db->insert( 'updatelog',
+				array( 'ul_key' => 'populate rev_parent_id' ),
+				__METHOD__,
+				'IGNORE' );
+			return;
+		}
+		# Do remaining chunk
+		$blockStart = intval( $start );
+		$blockEnd = intval( $start ) + $this->mBatchSize - 1;
+		$count = 0;
+		$changed = 0;
+		while ( $blockStart <= $end ) {
+			$this->output( "...doing rev_id from $blockStart to $blockEnd\n" );
+			$cond = "rev_id BETWEEN $blockStart AND $blockEnd";
+			$res = $db->select( 'revision',
+				array( 'rev_id', 'rev_page', 'rev_timestamp', 'rev_parent_id' ),
+				$cond, __METHOD__ );
+			# Go through and update rev_parent_id from these rows.
+			# Assume that the previous revision of the title was
+			# the original previous revision of the title when the
+			# edit was made...
+			foreach ( $res as $row ) {
+				# First, check rows with the same timestamp other than this one
+				# with a smaller rev ID. The highest ID "wins". This avoids loops
+				# as timestamp can only decrease and never loops with IDs (from parent to parent)
+				$previousID = $db->selectField( 'revision', 'rev_id',
+					array( 'rev_page' => $row->rev_page, 'rev_timestamp' => $row->rev_timestamp,
+						"rev_id < " . intval( $row->rev_id ) ),
+					__METHOD__,
+					array( 'ORDER BY' => 'rev_id DESC' ) );
+				# If there are none, check the the highest ID with a lower timestamp
+				if ( !$previousID ) {
+					# Get the highest older timestamp
+					$lastTimestamp = $db->selectField( 'revision', 'rev_timestamp',
+						array( 'rev_page' => $row->rev_page, "rev_timestamp < " . $db->addQuotes( $row->rev_timestamp ) ),
+						__METHOD__,
+						array( 'ORDER BY' => 'rev_timestamp DESC' ) );
+					# If there is one, let the highest rev ID win
+					if ( $lastTimestamp ) {
+						$previousID = $db->selectField( 'revision', 'rev_id',
+							array( 'rev_page' => $row->rev_page, 'rev_timestamp' => $lastTimestamp ),
+							__METHOD__,
+							array( 'ORDER BY' => 'rev_id DESC' ) );
+					}
+				}
+				$previousID = intval( $previousID );
+				if ( $previousID != $row->rev_parent_id )
+					$changed++;
+				# Update the row...
+				$db->update( 'revision',
+					array( 'rev_parent_id' => $previousID ),
+					array( 'rev_id' => $row->rev_id ),
+					__METHOD__ );
+				$count++;
+			}
+			$blockStart += $this->mBatchSize;
+			$blockEnd += $this->mBatchSize;
+			wfWaitForSlaves( 5 );
+		}
+		$logged = $db->insert( 'updatelog',
+			array( 'ul_key' => 'populate rev_parent_id' ),
+			__METHOD__,
+			'IGNORE' );
+		if ( $logged ) {
+			$this->output( "rev_parent_id population complete ... {$count} rows [{$changed} changed]\n" );
+			return true;
+		} else {
+			$this->output( "Could not insert rev_parent_id population row.\n" );
+			return false;
 		}
-		$blockStart += BATCH_SIZE;
-		$blockEnd += BATCH_SIZE;
-		wfWaitForSlaves( 5 );
-	}
-	$logged = $db->insert( 'updatelog',
-		array( 'ul_key' => 'populate rev_parent_id' ),
-		__FUNCTION__,
-		'IGNORE' );
-	if( $logged ) {
-		echo "rev_parent_id population complete ... {$count} rows\n";
-		return true;
-	} else {
-		echo "Could not insert rev_parent_id population row.\n";
-		return false;
 	}
 }
 
-
+$maintClass = "PopulateParentId";
+require_once( DO_MAINTENANCE );