updateCollation.php

Go to the documentation of this file.

<?php
// @codeCoverageIgnoreStart
require_once __DIR__ . '/Maintenance.php';
// @codeCoverageIgnoreEnd
 
use MediaWiki\Collation\Collation;
use MediaWiki\Logger\LoggerFactory;
use MediaWiki\MainConfigNames;
use MediaWiki\Maintenance\Maintenance;
use MediaWiki\Storage\NameTableStore;
use MediaWiki\Title\NamespaceInfo;
use MediaWiki\Title\Title;
use Wikimedia\Rdbms\IMaintainableDatabase;
use Wikimedia\Rdbms\IReadableDatabase;
use Wikimedia\Rdbms\IResultWrapper;
 
class UpdateCollation extends Maintenance {
    public $sizeHistogram = [];
 
    private $numRowsProcessed = 0;
 
    private $force;
 
    private $dryRun;
 
    private $verboseStats;
 
    private $collation;
 
    private $collationName;
 
    private $targetTable;
 
    private $table;
 
    private bool $normalization = false;
 
    private $dbr;
 
    private $dbw;
 
    private NamespaceInfo $namespaceInfo;
    private NameTableStore $collationNameStore;
 
    public function __construct() {
        parent::__construct();
 
        $this->addDescription( <<<TEXT
This script will find all rows in the categorylinks table whose collation is
out-of-date (collation_name is not the same as \$wgCategoryCollation) and
repopulate cl_sortkey using the page title and cl_sortkey_prefix. If all
collations are up-to-date, it will do nothing.
TEXT
        );
 
        $this->setBatchSize( 100 );
        $this->addOption( 'force', 'Run on all rows, even if the collation is ' .
            'supposed to be up-to-date.', false, false, 'f' );
        $this->addOption( 'previous-collation', 'Set the previous value of ' .
            '$wgCategoryCollation here to speed up this script, especially if your ' .
            'categorylinks table is large. This will only update rows with that ' .
            'collation, though, so it may miss out-of-date rows with a different, ' .
            'even older collation.', false, true );
        $this->addOption( 'target-collation', 'Set this to the new collation type to ' .
            'use instead of $wgCategoryCollation. Usually you should not use this, ' .
            'you should just update $wgCategoryCollation in LocalSettings.php.',
            false, true );
        $this->addOption( 'table', 'Table relative to which updates are generated. This ' .
            'table will be updated in place, unless --target-table is set. Defaults to ' .
            'categorylinks.', false, true );
        $this->addOption( 'target-table', 'Copy rows from table into the ' .
            'specified table instead of updating them in place.', false, true );
        $this->addOption( 'only-migrate-normalization', 'Only backfill cl_collation_id ' .
            'field from cl_collation', false );
        $this->addOption( 'remote', 'Use Shellbox to calculate the new sort keys ' .
            'remotely.' );
        $this->addOption( 'dry-run', 'Don\'t actually change the collations, just ' .
            'compile statistics.' );
        $this->addOption( 'verbose-stats', 'Show more statistics.' );
    }
 
    private function init() {
        $services = $this->getServiceContainer();
        $this->namespaceInfo = $services->getNamespaceInfo();
        $this->collationNameStore = new NameTableStore(
            $this->getServiceContainer()->getDBLoadBalancer(),
            $this->getServiceContainer()->getMainWANObjectCache(),
            LoggerFactory::getInstance( 'SecondaryDataUpdate' ),
            'collation',
            'collation_id',
            'collation_name'
        );
 
        if ( $this->hasOption( 'target-collation' ) ) {
            $this->collationName = $this->getOption( 'target-collation' );
        } else {
            $this->collationName = $this->getConfig()->get( MainConfigNames::CategoryCollation );
        }
        if ( $this->hasOption( 'remote' ) ) {
            $realCollationName = 'remote-' . $this->collationName;
        } else {
            $realCollationName = $this->collationName;
        }
        $this->collation = $services->getCollationFactory()->makeCollation( $realCollationName );
 
        // Collation check: in some cases the constructor will work,
        // but this will raise an exception, breaking all category pages
        $this->collation->getSortKey( 'MediaWiki' );
 
        $this->force = $this->getOption( 'force' );
        $this->dryRun = $this->getOption( 'dry-run' );
        $this->verboseStats = $this->getOption( 'verbose-stats' );
        $this->dbw = $this->getDB( DB_PRIMARY );
        $this->dbr = $this->getReplicaDB();
        $this->table = $this->getOption( 'table', 'categorylinks' );
        $this->targetTable = $this->getOption( 'target-table' );
        $this->normalization = $this->getOption( 'only-migrate-normalization', false );
    }
 
    public function execute() {
        $this->init();
        $batchSize = $this->getBatchSize();
 
        if ( $this->normalization ) {
            $this->runNormalizationMigration();
            return;
        }
 
        if ( $this->targetTable ) {
            if ( !$this->dbw->tableExists( $this->targetTable, __METHOD__ ) ) {
                $this->output( "Creating table {$this->targetTable}\n" );
                $this->dbw->query(
                    'CREATE TABLE ' . $this->dbw->tableName( $this->targetTable ) .
                    ' LIKE ' . $this->dbw->tableName( $this->table ),
                    __METHOD__
                );
            }
        }
 
        $collationConds = [];
        if ( !$this->force && !$this->targetTable ) {
            if ( $this->hasOption( 'previous-collation' ) ) {
                $collationConds['collation_name'] = $this->getOption( 'previous-collation' );
            } else {
                $collationConds[] = $this->dbr->expr( 'collation_name', '!=', $this->collationName );
            }
        }
        $maxPageId = (int)$this->dbr->newSelectQueryBuilder()
            ->select( 'MAX(page_id)' )
            ->from( 'page' )
            ->caller( __METHOD__ )->fetchField();
        $batchValue = 0;
        do {
            $this->output( "Selecting next $batchSize pages from cl_from = $batchValue... " );
 
            // cl_type must be selected as a number for proper paging because
            // enums suck.
            if ( $this->dbw->getType() === 'mysql' ) {
                $clType = 'cl_type+0 AS "cl_type_numeric"';
            } else {
                $clType = 'cl_type';
            }
            $res = $this->dbw->newSelectQueryBuilder()
                ->select( [
                    'cl_from', 'cl_target_id', 'cl_sortkey_prefix', 'cl_sortkey', $clType,
                    'cl_timestamp', 'collation_name', 'page_namespace', 'page_title'
                ] )
                ->from( $this->table )
                ->join( 'collation', null, 'cl_collation_id = collation_id' )
                // per T58041
                ->straightJoin( 'page', null, 'cl_from = page_id' )
                ->where( $collationConds )
                ->andWhere(
                    $this->dbw->expr( 'cl_from', '>=', $batchValue )
                        ->and( 'cl_from', '<', $batchValue + $this->getBatchSize() )
                )
                ->orderBy( 'cl_from' )
                ->caller( __METHOD__ )->fetchResultSet();
            $this->output( "processing... " );
 
            if ( $res->numRows() ) {
                if ( $this->targetTable ) {
                    $this->copyBatch( $res );
                } else {
                    $this->updateBatch( $res );
                }
            }
            $batchValue += $this->getBatchSize();
 
            if ( $this->dryRun ) {
                $this->output( "{$this->numRowsProcessed} rows would be updated so far.\n" );
            } else {
                $this->output( "{$this->numRowsProcessed} done.\n" );
            }
        } while ( $maxPageId >= $batchValue );
 
        if ( !$this->dryRun ) {
            $this->output( "{$this->numRowsProcessed} rows processed\n" );
        }
 
        if ( $this->verboseStats ) {
            $this->output( "\n" );
            $this->showSortKeySizeHistogram();
        }
    }
 
    private function updateBatch( IResultWrapper $res ) {
        if ( !$this->dryRun ) {
            $this->beginTransactionRound( __METHOD__ );
        }
        foreach ( $res as $row ) {
            $title = Title::newFromRow( $row );
            if ( !$row->collation_name ) {
                # This is an old-style row, so the sortkey needs to be
                # converted.
                if ( $row->cl_sortkey === $title->getText()
                    || $row->cl_sortkey === $title->getPrefixedText()
                ) {
                    $prefix = '';
                } else {
                    # Custom sortkey, so use it as a prefix
                    $prefix = $row->cl_sortkey;
                }
            } else {
                $prefix = $row->cl_sortkey_prefix;
            }
            # cl_type will be wrong for lots of pages if cl_collation is 0,
            # so let's update it while we're here.
            $type = $this->namespaceInfo->getCategoryLinkType( $row->page_namespace );
            $newSortKey = $this->collation->getSortKey(
                $title->getCategorySortkey( $prefix ) );
            $this->updateSortKeySizeHistogram( $newSortKey );
            // Truncate to 230 bytes to avoid DB error
            $newSortKey = substr( $newSortKey, 0, 230 );
 
            if ( $this->dryRun ) {
                // Add 1 to the count if the sortkey was changed. (Note that this doesn't count changes in
                // other fields, if any, those usually only happen when upgrading old MediaWikis.)
                $this->numRowsProcessed += ( $row->cl_sortkey !== $newSortKey );
            } else {
                $collationId = $this->collationNameStore->acquireId( $this->collationName );
                $this->dbw->newUpdateQueryBuilder()
                    ->update( $this->table )
                    ->set( [
                        'cl_sortkey' => $newSortKey,
                        'cl_sortkey_prefix' => $prefix,
                        'cl_collation_id' => $collationId,
                        'cl_type' => $type,
                        'cl_timestamp = cl_timestamp',
                    ] )
                    ->where( [ 'cl_from' => $row->cl_from, 'cl_target_id' => $row->cl_target_id ] )
                    ->caller( __METHOD__ )
                    ->execute();
                $this->numRowsProcessed++;
            }
        }
        if ( !$this->dryRun ) {
            $this->commitTransactionRound( __METHOD__ );
        }
    }
 
    private function copyBatch( IResultWrapper $res ) {
        $sortKeyInputs = [];
        foreach ( $res as $row ) {
            $title = Title::newFromRow( $row );
            $sortKeyInputs[] = $title->getCategorySortkey( $row->cl_sortkey_prefix );
        }
        $sortKeys = $this->collation->getSortKeys( $sortKeyInputs );
        $rowsToInsert = [];
        foreach ( $res as $i => $row ) {
            if ( !isset( $sortKeys[$i] ) ) {
                throw new RuntimeException( 'Unable to get sort key' );
            }
            $newSortKey = $sortKeys[$i];
            $this->updateSortKeySizeHistogram( $newSortKey );
            // Truncate to 230 bytes to avoid DB error
            $newSortKey = substr( $newSortKey, 0, 230 );
            $type = $this->namespaceInfo->getCategoryLinkType( $row->page_namespace );
            $collationId = $this->collationNameStore->acquireId( $this->collationName );
            $rowsToInsert[] = [
                'cl_from' => $row->cl_from,
                'cl_target_id' => $row->cl_target_id,
                'cl_sortkey' => $newSortKey,
                'cl_sortkey_prefix' => $row->cl_sortkey_prefix,
                'cl_collation_id' => $collationId,
                'cl_type' => $type,
                'cl_timestamp' => $row->cl_timestamp
            ];
        }
        if ( $this->dryRun ) {
            $this->numRowsProcessed += count( $rowsToInsert );
        } else {
            $this->beginTransactionRound( __METHOD__ );
            $this->dbw->newInsertQueryBuilder()
                ->insertInto( $this->targetTable )
                ->ignore()
                ->rows( $rowsToInsert )
                ->caller( __METHOD__ )->execute();
            $this->numRowsProcessed += $this->dbw->affectedRows();
            $this->commitTransactionRound( __METHOD__ );
        }
    }
 
    private function updateSortKeySizeHistogram( string $key ) {
        if ( !$this->verboseStats ) {
            return;
        }
        $length = strlen( $key );
        if ( !isset( $this->sizeHistogram[$length] ) ) {
            $this->sizeHistogram[$length] = 0;
        }
        $this->sizeHistogram[$length]++;
    }
 
    private function showSortKeySizeHistogram() {
        if ( !$this->sizeHistogram ) {
            return;
        }
        $maxLength = max( array_keys( $this->sizeHistogram ) );
        if ( $maxLength === 0 ) {
            return;
        }
        $numBins = 20;
        $coarseHistogram = array_fill( 0, $numBins, 0 );
        $coarseBoundaries = [];
        $boundary = 0;
        for ( $i = 0; $i < $numBins - 1; $i++ ) {
            $boundary += $maxLength / $numBins;
            $coarseBoundaries[$i] = round( $boundary );
        }
        $coarseBoundaries[$numBins - 1] = $maxLength + 1;
        $raw = '';
        for ( $i = 0; $i <= $maxLength; $i++ ) {
            if ( $raw !== '' ) {
                $raw .= ', ';
            }
            $val = $this->sizeHistogram[$i] ?? 0;
            for ( $coarseIndex = 0; $coarseIndex < $numBins - 1; $coarseIndex++ ) {
                if ( $coarseBoundaries[$coarseIndex] > $i ) {
                    $coarseHistogram[$coarseIndex] += $val;
                    break;
                }
            }
            if ( $coarseIndex === ( $numBins - 1 ) ) {
                $coarseHistogram[$coarseIndex] += $val;
            }
            $raw .= $val;
        }
 
        $this->output( "Sort key size histogram\nRaw data: $raw\n\n" );
 
        $maxBinVal = max( $coarseHistogram );
        $scale = (int)( 60 / $maxBinVal );
        $prevBoundary = 0;
        for ( $coarseIndex = 0; $coarseIndex < $numBins; $coarseIndex++ ) {
            $val = $coarseHistogram[$coarseIndex] ?? 0;
            $boundary = $coarseBoundaries[$coarseIndex];
            $this->output(
                sprintf( "%-10s %-10d |%s\n",
                    $prevBoundary . '-' . ( $boundary - 1 ) . ': ',
                    $val,
                    str_repeat( '*', $scale * $val )
                )
            );
            $prevBoundary = $boundary;
        }
    }
 
    private function runNormalizationMigration() {
        if ( !$this->dbw->fieldExists( $this->table, 'cl_collation', __METHOD__ ) ) {
            $this->output( "The cl_collation column appears to already be normalized. Skipping.\n" );
            return;
        }
        if ( !$this->dbw->fieldExists( $this->table, 'cl_collation_id', __METHOD__ ) ) {
            $this->output( "The cl_collation_id column doesn't exist. Run update.php to create it.\n" );
            return;
        }
        if ( !$this->dbw->tableExists( 'collation', __METHOD__ ) ) {
            $this->output( "The collation table doesn't exist. Run update.php to create it.\n" );
            return;
        }
 
        $maxPageId = (int)$this->dbr->newSelectQueryBuilder()
            ->select( 'MAX(page_id)' )
            ->from( 'page' )
            ->caller( __METHOD__ )->fetchField();
        $batchValue = 0;
        $batchSize = $this->getBatchSize();
 
        do {
            $this->output( "Selecting next $batchSize pages from cl_from = $batchValue... " );
 
            $res = $this->dbw->newSelectQueryBuilder()
                ->select( [ 'cl_collation' ] )
                ->distinct()
                ->from( $this->table )
                ->where( [ 'cl_collation_id' => 0 ] )
                ->andWhere(
                    $this->dbw->expr( 'cl_from', '>=', $batchValue )
                        ->and( 'cl_from', '<', $batchValue + $this->getBatchSize() )
                )
                ->caller( __METHOD__ )->fetchResultSet();
            $this->output( "processing... " );
 
            if ( $res->numRows() && !$this->dryRun ) {
                foreach ( $res as $row ) {
                    $collationName = $row->cl_collation;
                    $collationId = $this->collationNameStore->acquireId( $collationName );
                    $this->dbw->newUpdateQueryBuilder()
                        ->update( $this->table )
                        ->set( [ 'cl_collation_id' => $collationId ] )
                        ->where( [ 'cl_collation' => $collationName ] )
                        ->andWhere(
                            $this->dbw->expr( 'cl_from', '>=', $batchValue )
                                ->and( 'cl_from', '<', $batchValue + $this->getBatchSize() )
                        )
                        ->caller( __METHOD__ )->execute();
                    $this->numRowsProcessed += $this->dbw->affectedRows();
                }
 
                $this->waitForReplication();
            }
            $batchValue += $this->getBatchSize();
 
            $this->output( "{$this->numRowsProcessed} done.\n" );
        } while ( $maxPageId >= $batchValue );
 
        if ( !$this->dryRun ) {
            $this->output( "{$this->numRowsProcessed} rows processed\n" );
        }
    }
}
 
// @codeCoverageIgnoreStart
$maintClass = UpdateCollation::class;
require_once RUN_MAINTENANCE_IF_MAIN;
// @codeCoverageIgnoreEnd