mirror of
https://gerrit.wikimedia.org/r/mediawiki/extensions/Echo
synced 2024-11-28 09:40:41 +00:00
fe17b49fcd
Also test it more meaningfully by setting up a mock database and asserting that the right DELETE queries are issued. Change-Id: Id39723b92118e98d9c9f0cd7381e9396dce67c17
68 lines
1.9 KiB
PHP
68 lines
1.9 KiB
PHP
<?php
|
|
|
|
/**
|
|
* This job is created when sending notifications to the target users. The purpose
|
|
* of this job is to delete older notifications when the number of notifications a
|
|
* user has is more than $wgEchoMaxUpdateCount, it does not make sense to have tons
|
|
* of notifications in the history while users wouldn't bother to click 'load more'
|
|
* like 100 times to see them. What we gain from this is we could run expensive
|
|
* queries otherwise that would requires adding index and data denormalization.
|
|
*
|
|
* The initial job contains multiple users, which will in turn have individual jobs
|
|
* queued for them.
|
|
*/
|
|
class EchoNotificationDeleteJob extends Job {
|
|
|
|
/**
|
|
* UserIds to be processed
|
|
* @var int[]
|
|
*/
|
|
protected $userIds = [];
|
|
|
|
/**
|
|
* @param Title $title
|
|
* @param array $params
|
|
*/
|
|
public function __construct( $title, $params ) {
|
|
parent::__construct( __CLASS__, $title, $params );
|
|
$this->userIds = $params['userIds'];
|
|
}
|
|
|
|
/**
|
|
* Run the job of finding & deleting older notifications
|
|
* @return true
|
|
*/
|
|
public function run() {
|
|
global $wgEchoMaxUpdateCount;
|
|
if ( count( $this->userIds ) > 1 ) {
|
|
// If there are multiple users, queue a single job for each one
|
|
$jobs = [];
|
|
foreach ( $this->userIds as $userId ) {
|
|
$jobs[] = new EchoNotificationDeleteJob( $this->title, [ 'userIds' => [ $userId ] ] );
|
|
}
|
|
JobQueueGroup::singleton()->push( $jobs );
|
|
|
|
return true;
|
|
}
|
|
|
|
$notifMapper = new EchoNotificationMapper();
|
|
$targetMapper = new EchoTargetPageMapper();
|
|
|
|
// Back-compat for older jobs which used array( $userId => $userId );
|
|
$userIds = array_values( $this->userIds );
|
|
$userId = $userIds[0];
|
|
$user = User::newFromId( $userId );
|
|
$notif = $notifMapper->fetchByUserOffset( $user, $wgEchoMaxUpdateCount );
|
|
if ( $notif ) {
|
|
$notifMapper->deleteByUserEventOffset(
|
|
$user, $notif->getEvent()->getId()
|
|
);
|
|
$notifUser = MWEchoNotifUser::newFromUser( $user );
|
|
$notifUser->resetNotificationCount();
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
}
|