mirror of
https://gerrit.wikimedia.org/r/mediawiki/extensions/Echo
synced 2024-09-25 03:09:37 +00:00
d0714b2928
The $cached and $dbSource parameters are now unused, so remove them.
This affects get{Notification,Alert,Message}Count and
getLastUnread{Notification,Alert,Message}Time.
There are some callers in other extensions and in skins, but none of
them pass any parameters (except one, which I fixed in Ice42930280da).
Change-Id: If6f10c4f163ecb1def5a150656a60d1ab5f44d52
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( 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->waitForSlaves();
|
|
}
|
|
}
|
|
}
|
|
|
|
$maintClass = "BackfillUnreadWikis";
|
|
require_once RUN_MAINTENANCE_IF_MAIN;
|