mirror of
https://gerrit.wikimedia.org/r/mediawiki/extensions/Echo
synced 2024-12-01 02:46:46 +00:00
b2b6d8b420
As suggested in 26505b170adb24a6ae68945920db322c9382e470 for better readability. Also the sniff now knows about the maintenance script Change-Id: If3ab122c1fc12fcfbce777881ca21ecd99fb2a70
78 lines
2.6 KiB
PHP
78 lines
2.6 KiB
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";
|
|
$this->addOption( 'rebuild', 'Only recompute already-existing rows' );
|
|
$this->setBatchSize( 300 );
|
|
$this->requireExtension( 'Echo' );
|
|
}
|
|
|
|
public function execute() {
|
|
$dbFactory = MWEchoDbFactory::newFromDefault();
|
|
$lookup = CentralIdLookup::factory();
|
|
|
|
$rebuild = $this->hasOption( 'rebuild' );
|
|
if ( $rebuild ) {
|
|
$iterator = new BatchRowIterator( $dbFactory->getSharedDb( DB_REPLICA ), 'echo_unread_wikis', 'euw_user', $this->mBatchSize );
|
|
$iterator->addConditions( [ 'euw_wiki' => wfWikiID() ] );
|
|
} else {
|
|
$userQuery = User::getQueryInfo();
|
|
$iterator = new BatchRowIterator(
|
|
wfGetDB( DB_REPLICA ), $userQuery['tables'], 'user_id', $this->mBatchSize
|
|
);
|
|
$iterator->setFetchColumns( $userQuery['fields'] );
|
|
$iterator->addJoinConditions( $userQuery['joins'] );
|
|
}
|
|
|
|
$processed = 0;
|
|
foreach ( $iterator as $batch ) {
|
|
foreach ( $batch as $row ) {
|
|
if ( $rebuild ) {
|
|
$user = $lookup->localUserFromCentralId( $row->euw_user, CentralIdLookup::AUDIENCE_RAW );
|
|
} else {
|
|
$user = User::newFromRow( $row );
|
|
}
|
|
if ( !$user ) {
|
|
continue;
|
|
}
|
|
|
|
$notifUser = MWEchoNotifUser::newFromUser( $user );
|
|
$uw = EchoUnreadWikis::newFromUser( $user );
|
|
if ( $uw ) {
|
|
$alertCount = $notifUser->getNotificationCount( true, DB_REPLICA, EchoAttributeManager::ALERT, false );
|
|
$alertUnread = $notifUser->getLastUnreadNotificationTime( true, DB_REPLICA, EchoAttributeManager::ALERT, false );
|
|
|
|
$msgCount = $notifUser->getNotificationCount( true, DB_REPLICA, EchoAttributeManager::MESSAGE, false );
|
|
$msgUnread = $notifUser->getLastUnreadNotificationTime( true, DB_REPLICA, 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->waitForSlaves();
|
|
}
|
|
}
|
|
}
|
|
|
|
$maintClass = "BackfillUnreadWikis";
|
|
require_once RUN_MAINTENANCE_IF_MAIN;
|