1 files changed, 110 insertions, 9 deletions

diff --git a/maintenance/populateParentId.php b/maintenance/populateParentId.php
index 01730033..bf81cb68 100644
--- a/maintenance/populateParentId.php
+++ b/maintenance/populateParentId.php
@@ -1,18 +1,119 @@
 <?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
  */
 
-require_once 'commandLine.inc';
-require_once 'populateParentId.inc';
-	
-$db =& wfGetDB( DB_MASTER );
-if ( !$db->tableExists( 'revision' ) ) {
-	echo "revision table does not exist\n";
-	exit( 1 );
+require_once( dirname(__FILE__) . '/Maintenance.php' );
+
+class PopulateParentId extends Maintenance {
+	public function __construct() {
+		parent::__construct();
+		$this->mDescription = "Populates rev_parent_id";
+		$this->setBatchSize( 200 );
+	}
+
+	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
+		$end += $this->mBatchSize - 1;
+		$blockStart = intval( $start );
+		$blockEnd = intval( $start ) + $this->mBatchSize - 1;
+		$count = 0;
+		$changed = 0;
+		while( $blockEnd <= $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 - 1;
+			$blockEnd += $this->mBatchSize - 1;
+			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;
+		}
+	}
 }
 
-populate_rev_parent_id( $db );
+$maintClass = "PopulateParentId";
+require_once( DO_MAINTENANCE );