UserEditTracker.php

Go to the documentation of this file.

<?php
 
namespace MediaWiki\User;
 
use DBAccessObjectUtils;
use IDBAccessObject;
use InvalidArgumentException;
use JobQueueGroup;
use MediaWiki\Deferred\DeferredUpdates;
use MediaWiki\Deferred\UserEditCountUpdate;
use UserEditCountInitJob;
use Wikimedia\Rdbms\IConnectionProvider;
use Wikimedia\Timestamp\ConvertibleTimestamp;
 
class UserEditTracker {
 
    private const FIRST_EDIT = 1;
    private const LATEST_EDIT = 2;
 
    private ActorMigration $actorMigration;
    private IConnectionProvider $dbProvider;
    private JobQueueGroup $jobQueueGroup;
 
    private $userEditCountCache = [];
 
    public function __construct(
        ActorMigration $actorMigration,
        IConnectionProvider $dbProvider,
        JobQueueGroup $jobQueueGroup
    ) {
        $this->actorMigration = $actorMigration;
        $this->dbProvider = $dbProvider;
        $this->jobQueueGroup = $jobQueueGroup;
    }
 
    public function getUserEditCount( UserIdentity $user ): ?int {
        $userId = $user->getId();
        if ( !$userId ) {
            return null;
        }
 
        $cacheKey = 'u' . $userId;
        if ( isset( $this->userEditCountCache[ $cacheKey ] ) ) {
            return $this->userEditCountCache[ $cacheKey ];
        }
 
        $count = $this->dbProvider->getReplicaDatabase()->newSelectQueryBuilder()
            ->select( 'user_editcount' )
            ->from( 'user' )
            ->where( [ 'user_id' => $userId ] )
            ->caller( __METHOD__ )->fetchField();
 
        if ( $count === null ) {
            // it has not been initialized. do so.
            $count = $this->initializeUserEditCount( $user );
        }
 
        $this->userEditCountCache[ $cacheKey ] = $count;
        return $count;
    }
 
    public function initializeUserEditCount( UserIdentity $user ): int {
        $dbr = $this->dbProvider->getReplicaDatabase();
        $actorWhere = $this->actorMigration->getWhere( $dbr, 'rev_user', $user );
 
        $count = (int)$dbr->selectField(
            [ 'revision' ] + $actorWhere['tables'],
            'COUNT(*)',
            [ $actorWhere['conds'] ],
            __METHOD__,
            [],
            $actorWhere['joins']
        );
 
        // Defer updating the edit count via a job (T259719)
        $this->jobQueueGroup->push( new UserEditCountInitJob( [
            'userId' => $user->getId(),
            'editCount' => $count,
        ] ) );
 
        return $count;
    }
 
    public function incrementUserEditCount( UserIdentity $user ) {
        if ( !$user->getId() ) {
            // Can't store editcount without user row (i.e. unregistered)
            return;
        }
 
        DeferredUpdates::addUpdate(
            new UserEditCountUpdate( $user, 1 ),
            DeferredUpdates::POSTSEND
        );
    }
 
    public function getFirstEditTimestamp( UserIdentity $user, int $flags = IDBAccessObject::READ_NORMAL ) {
        return $this->getUserEditTimestamp( $user, self::FIRST_EDIT, $flags );
    }
 
    public function getLatestEditTimestamp( UserIdentity $user, int $flags = IDBAccessObject::READ_NORMAL ) {
        return $this->getUserEditTimestamp( $user, self::LATEST_EDIT, $flags );
    }
 
    private function getUserEditTimestamp( UserIdentity $user, int $type, int $flags = IDBAccessObject::READ_NORMAL ) {
        if ( !$user->getId() ) {
            return false;
        }
        [ $index ] = DBAccessObjectUtils::getDBOptions( $flags );
        $db = DBAccessObjectUtils::getDBFromIndex( $this->dbProvider, $index );
 
        $actorWhere = $this->actorMigration->getWhere( $db, 'rev_user', $user );
 
        $sortOrder = ( $type === self::FIRST_EDIT ) ? 'ASC' : 'DESC';
        $time = $db->selectField(
            [ 'revision' ] + $actorWhere['tables'],
            'rev_timestamp',
            [ $actorWhere['conds'] ],
            __METHOD__,
            [ 'ORDER BY' => "rev_timestamp $sortOrder" ],
            $actorWhere['joins']
        );
 
        if ( !$time ) {
            return false; // no edits
        }
 
        return ConvertibleTimestamp::convert( TS_MW, $time );
    }
 
    public function clearUserEditCache( UserIdentity $user ) {
        $userId = $user->getId();
        if ( !$userId ) {
            return;
        }
 
        $cacheKey = 'u' . $userId;
        unset( $this->userEditCountCache[ $cacheKey ] );
    }
 
    public function setCachedUserEditCount( UserIdentity $user, int $editCount ) {
        $userId = $user->getId();
        if ( !$userId ) {
            throw new InvalidArgumentException( __METHOD__ . ' with an anonymous user' );
        }
 
        $cacheKey = 'u' . $userId;
        $this->userEditCountCache[ $cacheKey ] = $editCount;
    }
 
}