mirror of
https://gerrit.wikimedia.org/r/mediawiki/extensions/Echo
synced 2024-11-24 07:54:13 +00:00
8823aa15cd
Phan says:
maintenance/backfillUnreadWikis.php:51 PhanTypeMismatchArgumentNullable Argument 1 ($userIdentity) is $lookup->localUserFromCentralId($row->euw_user, CentralIdLookup::AUDIENCE_RAW) of type ?\MediaWiki\DAO\WikiAwareEntity|?\MediaWiki\User\UserIdentity but \MediaWiki\User\UserFactory::newFromUserIdentity() takes \MediaWiki\User\UserIdentity defined at ../../includes/user/UserFactory.php:166 (expected type to be non-nullable)
Follow-up to 64531785a6.
Change-Id: I261e87d16bc23e9f1b5d3385553c5899632c9127
91 lines
2.8 KiB
PHP
91 lines
2.8 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 ) {
|
|
$userIdent = $lookup->localUserFromCentralId( $row->euw_user, CentralIdLookup::AUDIENCE_RAW );
|
|
if ( !$userIdent ) {
|
|
continue;
|
|
}
|
|
$user = $userFactory->newFromUserIdentity( $userIdent );
|
|
} 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;
|