mediawiki-extensions-Echo/maintenance/backfillUnreadWikis.php

<?php

$IP = getenv( 'MW_INSTALL_PATH' );
if ( $IP === false ) {
	$IP = __DIR__ . '/../../..';
}
require_once ( "$IP/maintenance/Maintenance.php" );

class BackfillUnreadWikis extends Maintenance {
	public function __construct() {
		parent::__construct();

		$this->mDescription = "Backfill echo_unread_wikis table and recache notification counts for all users";

		$this->setBatchSize( 300 );
	}

	public function execute() {
		global $wgEchoSharedTrackingCluster;

		$dbr = wfGetDB( DB_SLAVE );
		$iterator = new BatchRowIterator( $dbr, 'user', 'user_id', $this->mBatchSize );
		$iterator->setFetchColumns( User::selectFields() );

		$processed = 0;
		foreach ( $iterator as $batch ) {
			foreach ( $batch as $row ) {
				$user = User::newFromRow( $row );

				$notifUser = MWEchoNotifUser::newFromUser( $user );
				$notifUser->resetNotificationCount( DB_SLAVE, false );
			}

			$processed += count( $batch );
			$this->output( "Updated $processed users.\n" );
			wfWaitForSlaves( false, false, $wgEchoSharedTrackingCluster );
		}
	}
}

$maintClass = "BackfillUnreadWikis";
require_once ( DO_MAINTENANCE );