mirror of
https://gerrit.wikimedia.org/r/mediawiki/extensions/AbuseFilter.git
synced 2024-11-24 06:03:49 +00:00
bd84a6514c
This requires 1.42 for some new names Changes to the use statements done automatically via script Addition of missing use statements and changes to docs done manually Change-Id: Ic1e2c9a0c891382744e4792bba1effece48e53f3
64 lines
1.6 KiB
PHP
64 lines
1.6 KiB
PHP
<?php
|
|
|
|
namespace MediaWiki\Extension\AbuseFilter\Watcher;
|
|
|
|
use MediaWiki\Deferred\DeferredUpdates;
|
|
use MediaWiki\Extension\AbuseFilter\CentralDBManager;
|
|
use Wikimedia\Rdbms\IDatabase;
|
|
use Wikimedia\Rdbms\LBFactory;
|
|
|
|
/**
|
|
* Watcher that updates hit counts of filters
|
|
*/
|
|
class UpdateHitCountWatcher implements Watcher {
|
|
public const SERVICE_NAME = 'AbuseFilterUpdateHitCountWatcher';
|
|
|
|
/** @var LBFactory */
|
|
private $lbFactory;
|
|
|
|
/** @var CentralDBManager */
|
|
private $centralDBManager;
|
|
|
|
/**
|
|
* @param LBFactory $lbFactory
|
|
* @param CentralDBManager $centralDBManager
|
|
*/
|
|
public function __construct(
|
|
LBFactory $lbFactory,
|
|
CentralDBManager $centralDBManager
|
|
) {
|
|
$this->lbFactory = $lbFactory;
|
|
$this->centralDBManager = $centralDBManager;
|
|
}
|
|
|
|
/**
|
|
* @inheritDoc
|
|
*/
|
|
public function run( array $localFilters, array $globalFilters, string $group ): void {
|
|
// Run in a DeferredUpdate to avoid primary database queries on raw/view requests (T274455)
|
|
DeferredUpdates::addCallableUpdate( function () use ( $localFilters, $globalFilters ) {
|
|
if ( $localFilters ) {
|
|
$this->updateHitCounts( $this->lbFactory->getPrimaryDatabase(), $localFilters );
|
|
}
|
|
|
|
if ( $globalFilters ) {
|
|
$fdb = $this->centralDBManager->getConnection( DB_PRIMARY );
|
|
$this->updateHitCounts( $fdb, $globalFilters );
|
|
}
|
|
} );
|
|
}
|
|
|
|
/**
|
|
* @param IDatabase $dbw
|
|
* @param array $loggedFilters
|
|
*/
|
|
private function updateHitCounts( IDatabase $dbw, array $loggedFilters ): void {
|
|
$dbw->update(
|
|
'abuse_filter',
|
|
[ 'af_hit_count=af_hit_count+1' ],
|
|
[ 'af_id' => $loggedFilters ],
|
|
__METHOD__
|
|
);
|
|
}
|
|
}
|