* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
* http://www.gnu.org/copyleft/gpl.html
*
+ * @file
* @ingroup Maintenance
*/
-require_once( dirname(__FILE__) . '/Maintenance.php' );
+require_once( __DIR__ . '/Maintenance.php' );
+/**
+ * Maintenance script that re-assigns users from an old group to a new one.
+ *
+ * @ingroup Maintenance
+ */
class MigrateUserGroup extends Maintenance {
public function __construct() {
parent::__construct();
$newGroup = $this->getArg( 1 );
$dbw = wfGetDB( DB_MASTER );
$start = $dbw->selectField( 'user_groups', 'MIN(ug_user)',
- array('ug_group' => $oldGroup), __FUNCTION__ );
+ array( 'ug_group' => $oldGroup ), __FUNCTION__ );
$end = $dbw->selectField( 'user_groups', 'MAX(ug_user)',
- array('ug_group' => $oldGroup), __FUNCTION__ );
- if( $start === null ) {
+ array( 'ug_group' => $oldGroup ), __FUNCTION__ );
+ if ( $start === null ) {
$this->error( "Nothing to do - no users in the '$oldGroup' group", true );
}
# Do remaining chunk
$blockStart = $start;
$blockEnd = $start + $this->mBatchSize - 1;
// Migrate users over in batches...
- while( $blockEnd <= $end ) {
+ while ( $blockEnd <= $end ) {
$this->output( "Doing users $blockStart to $blockEnd\n" );
- $dbw->begin();
+ $dbw->begin( __METHOD__ );
$dbw->update( 'user_groups',
- array('ug_group' => $newGroup),
- array('ug_group' => $oldGroup,
- "ug_user BETWEEN $blockStart AND $blockEnd" )
+ array( 'ug_group' => $newGroup ),
+ array( 'ug_group' => $oldGroup,
+ "ug_user BETWEEN $blockStart AND $blockEnd" ),
+ __METHOD__,
+ array( 'IGNORE' )
+ );
+ $count += $dbw->affectedRows();
+ $dbw->delete( 'user_groups',
+ array( 'ug_group' => $oldGroup,
+ "ug_user BETWEEN $blockStart AND $blockEnd" ),
+ __METHOD__
);
$count += $dbw->affectedRows();
- $dbw->commit();
+ $dbw->commit( __METHOD__ );
$blockStart += $this->mBatchSize;
$blockEnd += $this->mBatchSize;
- wfWaitForSlaves( 5 );
+ wfWaitForSlaves();
}
$this->output( "Done! $count user(s) in group '$oldGroup' are now in '$newGroup' instead.\n" );
}
}
$maintClass = "MigrateUserGroup";
-require_once( DO_MAINTENANCE );
+require_once( RUN_MAINTENANCE_IF_MAIN );