X-Git-Url: https://scripts.mit.edu/gitweb/autoinstallsdev/mediawiki.git/blobdiff_plain/19e297c21b10b1b8a3acad5e73fc71dcb35db44a..6932310fd58ebef145fa01eb76edf7150284d8ea:/maintenance/fixUserRegistration.php

diff --git a/maintenance/fixUserRegistration.php b/maintenance/fixUserRegistration.php
index d4ff7c23..37fd44fb 100644
--- a/maintenance/fixUserRegistration.php
+++ b/maintenance/fixUserRegistration.php
@@ -18,38 +18,74 @@
  * 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 fixes the user_registration field.
+ *
+ * @ingroup Maintenance
+ */
 class FixUserRegistration extends Maintenance {
 	public function __construct() {
 		parent::__construct();
-		$this->mDescription = "Fix the user_registration field";
+		$this->addDescription( 'Fix the user_registration field' );
+		$this->setBatchSize( 1000 );
 	}
 
 	public function execute() {
-		$dbr = wfGetDB( DB_SLAVE );
-		$dbw = wfGetDB( DB_MASTER );
+		$dbw = $this->getDB( DB_MASTER );
 
-		// Get user IDs which need fixing
-		$res = $dbr->select( 'user', 'user_id', 'user_registration IS NULL', __METHOD__ );
-		foreach ( $res as $row ) {
-			$id = $row->user_id;
-			// Get first edit time
-			$timestamp = $dbr->selectField( 'revision', 'MIN(rev_timestamp)', array( 'rev_user' => $id ), __METHOD__ );
-			// Update
-			if ( !empty( $timestamp ) ) {
-				$dbw->update( 'user', array( 'user_registration' => $timestamp ), array( 'user_id' => $id ), __METHOD__ );
-				$this->output( "$id $timestamp\n" );
-			} else {
-				$this->output( "$id NULL\n" );
+		$lastId = 0;
+		do {
+			// Get user IDs which need fixing
+			$res = $dbw->select(
+				'user',
+				'user_id',
+				[
+					'user_id > ' . $dbw->addQuotes( $lastId ),
+					'user_registration IS NULL'
+				],
+				__METHOD__,
+				[
+					'LIMIT' => $this->mBatchSize,
+					'ORDER BY' => 'user_id',
+				]
+			);
+			foreach ( $res as $row ) {
+				$id = $row->user_id;
+				$lastId = $id;
+				// Get first edit time
+				$timestamp = $dbw->selectField(
+					'revision',
+					'MIN(rev_timestamp)',
+					[ 'rev_user' => $id ],
+					__METHOD__
+				);
+				// Update
+				if ( $timestamp !== null ) {
+					$dbw->update(
+						'user',
+						[ 'user_registration' => $timestamp ],
+						[ 'user_id' => $id ],
+						__METHOD__
+					);
+					$user = User::newFromId( $id );
+					$user->invalidateCache();
+					$this->output( "Set registration for #$id to $timestamp\n" );
+				} else {
+					$this->output( "Could not find registration for #$id NULL\n" );
+				}
 			}
-		}
-		$this->output( "\n" );
+			$this->output( "Waiting for replica DBs..." );
+			wfWaitForSlaves();
+			$this->output( " done.\n" );
+		} while ( $res->numRows() >= $this->mBatchSize );
 	}
 }
 
 $maintClass = "FixUserRegistration";
-require_once( RUN_MAINTENANCE_IF_MAIN );
+require_once RUN_MAINTENANCE_IF_MAIN;