mediawiki-extensions-Echo/maintenance/updateEchoSchemaForSuppression.php

<?php
/**
 * Update event_page_id in echo_event based on event_page_title and
 * event_page_namespace
 *
 * @ingroup Maintenance
 */

use MediaWiki\Extension\Notifications\DbFactory;
use MediaWiki\Extension\Notifications\SuppressionRowUpdateGenerator;

require_once getenv( 'MW_INSTALL_PATH' ) !== false
	? getenv( 'MW_INSTALL_PATH' ) . '/maintenance/Maintenance.php'
	: __DIR__ . '/../../../maintenance/Maintenance.php';

/**
 * Maintenance script that populates the event_page_id column of echo_event
 *
 * @ingroup Maintenance
 */
class UpdateEchoSchemaForSuppression extends LoggedUpdateMaintenance {

	/**
	 * @var string The table to update
	 */
	protected $table = 'echo_event';

	/**
	 * @var string The primary key column of the table to update
	 */
	protected $idField = 'event_id';

	public function __construct() {
		parent::__construct();
		$this->setBatchSize( 500 );
		$this->requireExtension( 'Echo' );
	}

	public function getUpdateKey() {
		return __CLASS__;
	}

	public function doDBUpdates() {
		global $wgEchoCluster;
		$lbFactory = DbFactory::newFromDefault();

		$dbr = $lbFactory->getEchoDb( DB_REPLICA );
		$dbw = $lbFactory->getEchoDb( DB_PRIMARY );
		'@phan-var \Wikimedia\Rdbms\IMaintainableDatabase $dbw';

		if ( !$dbw->fieldExists( 'echo_event', 'event_page_title', __METHOD__ ) ) {
			$this->output( "No event_page_title field, skipping migration from event_page_title to event_page_id\n" );
			return true;
		}

		$reader = new BatchRowIterator( $dbr, $this->table, $this->idField, $this->getBatchSize() );
		$reader->addConditions( [
			"event_page_title IS NOT NULL",
			"event_page_id" => null,
		] );
		$reader->setFetchColumns( [ 'event_page_namespace', 'event_page_title', 'event_extra', 'event_type' ] );
		$reader->setCaller( __METHOD__ );

		$writer = new BatchRowWriter( $dbw, $this->table, $wgEchoCluster );
		$writer->setCaller( __METHOD__ );

		$updater = new BatchRowUpdate(
			$reader,
			$writer,
			new SuppressionRowUpdateGenerator
		);
		$updater->setOutput( function ( $text ) {
			$this->output( $text );
		} );
		$updater->execute();
		return true;
	}
}

$maintClass = UpdateEchoSchemaForSuppression::class;
require_once RUN_MAINTENANCE_IF_MAIN;