Page MenuHomeWickedGov Phorge
Files F5972854

populateLocalAndGlobalIds.php
No OneTemporary
Actions

Size
3 KB
Referenced Files
None
Subscribers
None

populateLocalAndGlobalIds.php
View Options

<?php
namespace MediaWiki\Extension\CentralAuth\Maintenance;
$IP = getenv( 'MW_INSTALL_PATH' );
if ( $IP === false ) {
$IP = __DIR__ . '/../../..';
}
require_once "$IP/maintenance/Maintenance.php";
use MediaWiki\Extension\CentralAuth\CentralAuthServices;
use MediaWiki\Maintenance\Maintenance;
use MediaWiki\WikiMap\WikiMap;
use Wikimedia\Rdbms\SelectQueryBuilder;
class PopulateLocalAndGlobalIds extends Maintenance {
public function __construct() {
parent::__construct();
$this->requireExtension( 'CentralAuth' );
$this->addDescription(
"Populate the localuser.lu_local_id and localuser.lu_global_id fields"
);
$this->setBatchSize( 1000 );
}
public function execute() {
$databaseManager = CentralAuthServices::getDatabaseManager();
$dbr = $databaseManager->getCentralReplicaDB();
$dbw = $databaseManager->getCentralPrimaryDB();
$lastGlobalId = -1;
// Skip people in global rename queue
$wiki = WikiMap::getCurrentWikiId();
$globalRenames = $dbr->newSelectQueryBuilder()
->select( 'ru_oldname' )
->from( 'renameuser_status' )
->caller( __METHOD__ )
->fetchFieldValues();
$ldbr = $databaseManager->getLocalDB( DB_REPLICA, $wiki );
$this->output( "Populating fields for wiki $wiki... \n" );
do {
$rows = $dbr->newSelectQueryBuilder()
->select( [ 'lu_name', 'gu_id' ] )
->from( 'localuser' )
->join( 'globaluser', null, 'gu_name = lu_name' )
->where( [
// Start from where we left off in last batch
$dbr->expr( 'gu_id', '>=', $lastGlobalId ),
'lu_wiki' => $wiki,
// Only pick records not already populated
'lu_local_id' => null
] )
->orderBy( 'gu_id', SelectQueryBuilder::SORT_ASC )
->limit( $this->mBatchSize )
->caller( __METHOD__ )
->fetchResultSet();
$numRows = $rows->numRows();
$globalUidToLocalName = [];
foreach ( $rows as $row ) {
if ( in_array( $row->lu_name, $globalRenames ) ) {
$this->output(
"User " . $row->lu_name . " not migrated (pending rename)\n"
);
continue;
}
$globalUidToLocalName[$row->gu_id] = $row->lu_name;
}
if ( !$globalUidToLocalName ) {
continue;
}
$localNameToUid = [];
$localIds = $ldbr->newSelectQueryBuilder()
->select( [ 'user_id', 'user_name' ] )
->from( 'user' )
->where( [ 'user_name' => array_values( $globalUidToLocalName ) ] )
->caller( __METHOD__ )
->fetchResultSet();
foreach ( $localIds as $lid ) {
$localNameToUid[$lid->user_name] = $lid->user_id;
}
$updated = 0;
foreach ( $globalUidToLocalName as $gid => $uname ) {
// Save progress so we know where to start our next batch
$lastGlobalId = $gid;
$dbw->newUpdateQueryBuilder()
->update( 'localuser' )
->set( [
'lu_local_id' => $localNameToUid[$uname],
'lu_global_id' => $gid
] )
->where( [ 'lu_name' => $uname, 'lu_wiki' => $wiki ] )
->caller( __METHOD__ )
->execute();
if ( !$dbw->affectedRows() ) {
$this->output(
"Update failed for global user $lastGlobalId for wiki $wiki \n"
);
} else {
// Count number of records actually updated
$updated++;
}
}
$this->output(
"Updated $updated records. Last user: $lastGlobalId; Wiki: $wiki \n"
);
$this->waitForReplication();
} while ( $numRows >= $this->mBatchSize );
$this->output( "Completed $wiki \n" );
}
}
$maintClass = PopulateLocalAndGlobalIds::class;
require_once RUN_MAINTENANCE_IF_MAIN;

File Metadata

Mime Type
text/x-php
Expires
Sat, Oct 3, 16:12 (1 d, 6 h ago)
Storage Engine
local-disk
Storage Format
Raw Data
Storage Handle
ca/ba/79a75f72cd81b8ba5647111cc441
Default Alt Text
populateLocalAndGlobalIds.php (3 KB)

Event Timeline