diff options
Diffstat (limited to 'maintenance/populateParentId.php')
| -rw-r--r-- | maintenance/populateParentId.php | 119 |
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 ); |