RenameUserJob.php

Go to the documentation of this file.

<?php
 
class RenameUserJob extends Job {
    public function __construct( Title $title, $params = [], $id = 0 ) {
        parent::__construct( 'renameUser', $title, $params, $id );
    }
 
    public function run() {
        global $wgUpdateRowsPerQuery;
 
        $table = $this->params['table'];
        $column = $this->params['column'];
        $oldname = $this->params['oldname'];
        $newname = $this->params['newname'];
        $count = $this->params['count'];
        if ( isset( $this->params['userID'] ) ) {
            $userID = $this->params['userID'];
            $uidColumn = $this->params['uidColumn'];
        } else {
            $userID = null;
            $uidColumn = null;
        }
        if ( isset( $this->params['timestampColumn'] ) ) {
            $timestampColumn = $this->params['timestampColumn'];
            $minTimestamp = $this->params['minTimestamp'];
            $maxTimestamp = $this->params['maxTimestamp'];
        } else {
            $timestampColumn = null;
            $minTimestamp = null;
            $maxTimestamp = null;
        }
        $uniqueKey = isset( $this->params['uniqueKey'] ) ? $this->params['uniqueKey'] : null;
        $keyId = isset( $this->params['keyId'] ) ? $this->params['keyId'] : null;
        $logId = isset( $this->params['logId'] ) ? $this->params['logId'] : null;
 
        $dbw = wfGetDB( DB_MASTER );
        if ( $logId ) {
            # Block until the transaction that inserted this job commits.
            # The atomic section is for sanity as FOR UPDATE does not lock in auto-commit mode
            # per http://dev.mysql.com/doc/refman/5.7/en/innodb-locking-reads.html.
            $dbw->startAtomic( __METHOD__ );
            $committed = $dbw->selectField( 'logging',
                '1',
                [ 'log_id' => $logId ],
                __METHOD__,
                [ 'FOR UPDATE' ]
            );
            $dbw->endAtomic( __METHOD__ );
            # If the transaction inserting this job was rolled back, detect that
            if ( $committed === false ) { // rollback happened?
                throw new LogicException( 'Cannot run job if the account rename failed.' );
            }
        }
 
        # Flush any state snapshot data (and release the lock above)
        $dbw->commit( __METHOD__, 'flush' );
 
        # Conditions like "*_user_text = 'x'
        $conds = [ $column => $oldname ];
        # If user ID given, add that to condition to avoid rename collisions
        if ( $userID !== null ) {
            $conds[$uidColumn] = $userID;
        }
        # Bound by timestamp if given
        if ( $timestampColumn !== null ) {
            $conds[] = "$timestampColumn >= " . $dbw->addQuotes( $minTimestamp );
            $conds[] = "$timestampColumn <= " . $dbw->addQuotes( $maxTimestamp );
        # Bound by unique key if given (B/C)
        } elseif ( $uniqueKey !== null && $keyId !== null ) {
            $conds[$uniqueKey] = $keyId;
        } else {
            throw new InvalidArgumentException( 'Expected ID batch or time range' );
        }
 
        $affectedCount = 0;
        # Actually update the rows for this job...
        if ( $uniqueKey !== null ) {
            # Select the rows to update by PRIMARY KEY
            $ids = $dbw->selectFieldValues( $table, $uniqueKey, $conds, __METHOD__ );
            # Update these rows by PRIMARY KEY to avoid slave lag
            foreach ( array_chunk( $ids, $wgUpdateRowsPerQuery ) as $batch ) {
                $dbw->commit( __METHOD__, 'flush' );
                wfWaitForSlaves();
 
                $dbw->update( $table,
                    [ $column => $newname ],
                    [ $column => $oldname, $uniqueKey => $batch ],
                    __METHOD__
                );
                $affectedCount += $dbw->affectedRows();
            }
        } else {
            # Update the chunk of rows directly
            $dbw->update( $table,
                [ $column => $newname ],
                $conds,
                __METHOD__
            );
            $affectedCount += $dbw->affectedRows();
        }
 
        # Special case: revisions may be deleted while renaming...
        if ( $affectedCount < $count && $table === 'revision' && $timestampColumn !== null ) {
            # If some revisions were not renamed, they may have been deleted.
            # Do a pass on the archive table to get these straglers...
            $ids = $dbw->selectFieldValues(
                'archive',
                'ar_id',
                [
                    'ar_user_text' => $oldname,
                    'ar_user' => $userID,
                    // No user,rev_id index, so use timestamp to bound
                    // the rows. This can use the user,timestamp index.
                    "ar_timestamp >= '$minTimestamp'",
                    "ar_timestamp <= '$maxTimestamp'"
                ],
                __METHOD__
            );
            foreach ( array_chunk( $ids, $wgUpdateRowsPerQuery ) as $batch ) {
                $dbw->commit( __METHOD__, 'flush' );
                wfWaitForSlaves();
 
                $dbw->update(
                    'archive',
                    [ 'ar_user_text' => $newname ],
                    [ 'ar_user_text' => $oldname, 'ar_id' => $batch ],
                    __METHOD__
                );
            }
        }
        # Special case: revisions may be restored while renaming...
        if ( $affectedCount < $count && $table === 'archive' && $timestampColumn !== null ) {
            # If some revisions were not renamed, they may have been restored.
            # Do a pass on the revision table to get these straglers...
            $ids = $dbw->selectFieldValues(
                'revision',
                'rev_id',
                [
                    'rev_user_text' => $oldname,
                    'rev_user' => $userID,
                    // No user,rev_id index, so use timestamp to bound
                    // the rows. This can use the user,timestamp index.
                    "rev_timestamp >= '$minTimestamp'",
                    "rev_timestamp <= '$maxTimestamp'"
                ],
                __METHOD__
            );
            foreach ( array_chunk( $ids, $wgUpdateRowsPerQuery ) as $batch ) {
                $dbw->commit( __METHOD__, 'flush' );
                wfWaitForSlaves();
 
                $dbw->update(
                    'revision',
                    [ 'rev_user_text' => $newname ],
                    [ 'rev_user_text' => $oldname, 'rev_id' => $batch ],
                    __METHOD__
                );
            }
        }
 
        return true;
    }
}