mediawiki-extensions-Echo/maintenance/backfillUnreadWikis.php
Petr Pchelko a10b0b07c8 Use CentralIdLookupFactory and pass UserIdentity
Change-Id: I44144df7cf244eb867c1b261c10cc29b020f8409
2021-07-21 19:23:42 -07:00

94 lines
2.9 KiB
PHP

<?php
use MediaWiki\MediaWikiServices;
$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->addDescription( "Backfill echo_unread_wikis table" );
$this->addOption( 'rebuild', 'Only recompute already-existing rows' );
$this->setBatchSize( 300 );
$this->requireExtension( 'Echo' );
}
public function execute() {
$dbFactory = MWEchoDbFactory::newFromDefault();
$lookup = MediaWikiServices::getInstance()->getCentralIdLookup();
$rebuild = $this->hasOption( 'rebuild' );
if ( $rebuild ) {
$iterator = new BatchRowIterator(
$dbFactory->getSharedDb( DB_REPLICA ),
'echo_unread_wikis',
'euw_user',
$this->getBatchSize()
);
$iterator->addConditions( [ 'euw_wiki' => wfWikiID() ] );
} else {
$userQuery = User::getQueryInfo();
$iterator = new BatchRowIterator(
wfGetDB( DB_REPLICA ), $userQuery['tables'], 'user_id', $this->getBatchSize()
);
$iterator->setFetchColumns( $userQuery['fields'] );
$iterator->addJoinConditions( $userQuery['joins'] );
}
$iterator->setCaller( __METHOD__ );
$processed = 0;
$userFactory = MediaWikiServices::getInstance()->getUserFactory();
foreach ( $iterator as $batch ) {
foreach ( $batch as $row ) {
if ( $rebuild ) {
$userIdentity = $lookup->localUserFromCentralId(
$row->euw_user,
CentralIdLookup::AUDIENCE_RAW
);
if ( !$userIdentity ) {
continue;
}
$user = $userFactory->newFromUserIdentity( $userIdentity );
} else {
$user = User::newFromRow( $row );
}
$notifUser = MWEchoNotifUser::newFromUser( $user );
$uw = EchoUnreadWikis::newFromUser( $user );
if ( $uw ) {
$alertCount = $notifUser->getNotificationCount( EchoAttributeManager::ALERT, false );
$alertUnread = $notifUser->getLastUnreadNotificationTime( EchoAttributeManager::ALERT, false );
$msgCount = $notifUser->getNotificationCount( EchoAttributeManager::MESSAGE, false );
$msgUnread = $notifUser->getLastUnreadNotificationTime( EchoAttributeManager::MESSAGE, false );
if ( ( $alertCount !== 0 && $alertUnread === false ) ||
( $msgCount !== 0 && $msgUnread === false )
) {
// If there are alerts, there should be an alert timestamp (same for messages).
// Otherwise, there is a race condition between the two values, indicating there's already
// just been an updateCount call, so we can skip this user.
continue;
}
$uw->updateCount( wfWikiID(), $alertCount, $alertUnread, $msgCount, $msgUnread );
}
}
$processed += count( $batch );
$this->output( "Updated $processed users.\n" );
$dbFactory->waitForReplicas();
}
}
}
$maintClass = BackfillUnreadWikis::class;
require_once RUN_MAINTENANCE_IF_MAIN;