wiki.techinc.nl/includes/jobqueue/utils/PurgeJobUtils.php
Bartosz Dziewoński 4aebcfe097 Change more uses of getDBLoadBalancerFactory() to getConnectionProvider()
Some less trivial cases. Also update variable names.

This doesn't really change anything, but I hope it helps promote
getConnectionProvider() as the common way to do this.

Follow-up to 8604c384f6.

Change-Id: I6657d783375fac5c7fa856b884ff1fb09285e94c
2024-01-23 16:34:45 +01:00

84 lines
2.8 KiB
PHP

<?php
/**
* Base code for update jobs that put some secondary data extracted
* from article content into the database.
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License along
* with this program; if not, write to the Free Software Foundation, Inc.,
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
* http://www.gnu.org/copyleft/gpl.html
*
* @file
*/
use MediaWiki\Deferred\AutoCommitUpdate;
use MediaWiki\Deferred\DeferredUpdates;
use MediaWiki\MainConfigNames;
use MediaWiki\MediaWikiServices;
use Wikimedia\Rdbms\IDatabase;
class PurgeJobUtils {
/**
* Invalidate the cache of a list of pages from a single namespace.
* This is intended for use by subclasses.
*
* @param IDatabase $dbw
* @param int $namespace Namespace number
* @param string[] $dbkeys
*/
public static function invalidatePages( IDatabase $dbw, $namespace, array $dbkeys ) {
if ( $dbkeys === [] ) {
return;
}
$fname = __METHOD__;
DeferredUpdates::addUpdate( new AutoCommitUpdate(
$dbw,
__METHOD__,
static function () use ( $dbw, $namespace, $dbkeys, $fname ) {
$services = MediaWikiServices::getInstance();
$dbProvider = $services->getConnectionProvider();
// Determine which pages need to be updated.
// This is necessary to prevent the job queue from smashing the DB with
// large numbers of concurrent invalidations of the same page.
$now = $dbw->timestamp();
$ids = $dbw->newSelectQueryBuilder()
->select( 'page_id' )
->from( 'page' )
->where( [ 'page_namespace' => $namespace ] )
->andWhere( [ 'page_title' => $dbkeys ] )
->andWhere( $dbw->expr( 'page_touched', '<', $now ) )
->caller( $fname )->fetchFieldValues();
if ( !$ids ) {
return;
}
$batchSize =
$services->getMainConfig()->get( MainConfigNames::UpdateRowsPerQuery );
$ticket = $dbProvider->getEmptyTransactionTicket( $fname );
$idBatches = array_chunk( $ids, $batchSize );
foreach ( $idBatches as $idBatch ) {
$dbw->newUpdateQueryBuilder()
->update( 'page' )
->set( [ 'page_touched' => $now ] )
->where( [ 'page_id' => $idBatch ] )
->andWhere( $dbw->expr( 'page_touched', '<', $now ) ) // handle races
->caller( $fname )->execute();
if ( count( $idBatches ) > 1 ) {
$dbProvider->commitAndWaitForReplication( $fname, $ticket );
}
}
}
) );
}
}