RebuildMessageGroupStatsJob.php

<?php
declare( strict_types = 1 );
 
namespace MediaWiki\Extension\Translate\Statistics;
 
use GenericParameterJob;
use InvalidArgumentException;
use MediaWiki\Extension\Translate\Jobs\GenericTranslateJob;
use MediaWiki\Extension\Translate\MessageGroupProcessing\MessageGroups;
use MediaWiki\MediaWikiServices;
 
class RebuildMessageGroupStatsJob extends GenericTranslateJob implements GenericParameterJob {
    public const GROUP_ID = 'groupid';
    public const LANGUAGE_CODE = 'languagecode';
    public const CLEAR_GROUPS = 'cleargroups';
    public const REFRESH = 'purge';
    protected $removeDuplicates = true;
 
    public static function newJob( array $params ): self {
        return new self( $params );
    }
 
    public static function newRefreshGroupsJob( array $messageGroupIds ): self {
        return new self( [ self::CLEAR_GROUPS => $messageGroupIds ] );
    }
 
    public function __construct( array $params = [] ) {
        parent::__construct( 'RebuildMessageGroupStatsJob', $params );
    }
 
    public function run(): bool {
        $lb = MediaWikiServices::getInstance()->getDBLoadBalancerFactory();
        if ( !$lb->waitForReplication() ) {
            $this->logWarning( 'Continuing despite replication lag' );
        }
 
        $params = $this->params;
        $flags = 0;
 
        // Sanity check that this is run via JobQueue. Immediate writes are only safe when they
        // are run in isolation, e.g. as a separate job in the JobQueue.
        if ( defined( 'MEDIAWIKI_JOB_RUNNER' ) ) {
            $flags |= MessageGroupStats::FLAG_IMMEDIATE_WRITES;
        }
 
        // This is to make sure the priority value is not read from the process cache.
        // There is still a possibility that, due to replication lag, an old value is read.
        MessageGroups::singleton()->clearProcessCache();
 
        if ( isset( $params[self::REFRESH] ) && $params[self::REFRESH] ) {
            $flags |= MessageGroupStats::FLAG_NO_CACHE;
        }
 
        if ( isset( $params[self::GROUP_ID] ) && isset( $params[self::LANGUAGE_CODE] ) ) {
            MessageGroupStats::forItem( $params[self::GROUP_ID], $params[self::LANGUAGE_CODE], $flags );
        } elseif ( isset( $params[self::GROUP_ID] ) ) {
            MessageGroupStats::forGroup( $params[self::GROUP_ID], $flags );
        } elseif ( isset( $params[self::CLEAR_GROUPS] ) ) {
            MessageGroupStats::clearGroup( $params[self::CLEAR_GROUPS], $flags );
        } elseif ( isset( $params[self::LANGUAGE_CODE] ) ) {
            MessageGroupStats::forLanguage( $params[self::LANGUAGE_CODE], $flags );
        } else {
            throw new InvalidArgumentException( 'No groupid or languagecode or cleargroup provided' );
        }
 
        return true;
    }
}