ParsoidCachePrewarmJob.php

Go to the documentation of this file.

<?php
use MediaWiki\Logger\LoggerFactory;
use MediaWiki\Page\PageLookup;
use MediaWiki\Page\PageRecord;
use MediaWiki\Parser\Parsoid\ParsoidOutputAccess;
use MediaWiki\Revision\RevisionLookup;
use MediaWiki\Revision\SlotRecord;
use Psr\Log\LoggerInterface;
 
class ParsoidCachePrewarmJob extends Job {
    private LoggerInterface $logger;
    private ParsoidOutputAccess $parsoidOutputAccess;
    private PageLookup $pageLookup;
    private RevisionLookup $revisionLookup;
 
    public function __construct(
        array $params,
        ParsoidOutputAccess $parsoidOutputAccess,
        PageLookup $pageLookup,
        RevisionLookup $revisionLookup
    ) {
        parent::__construct( 'parsoidCachePrewarm', $params );
 
        // TODO: find a way to inject the logger
        $this->logger = LoggerFactory::getInstance( 'ParsoidCachePrewarmJob' );
        $this->parsoidOutputAccess = $parsoidOutputAccess;
        $this->pageLookup = $pageLookup;
        $this->revisionLookup = $revisionLookup;
    }
 
    public static function newSpec(
        int $revisionId,
        $page,
        array $params = []
    ): JobSpecification {
        if ( !is_int( $page ) && !( $page instanceof PageRecord ) ) {
            throw new InvalidArgumentException(
                '$page must be a int or a PageRecord object.' );
        }
 
        $pageId = is_int( $page ) ? $page : $page->getId();
 
        $params += [ 'options' => 0 ];
 
        $opts = [];
        if ( $page instanceof PageRecord ) {
            $pageTouched = $page->getTouched();
            $params += self::newRootJobParams(
                "parsoidCachePrewarm:$pageId:$revisionId:$pageTouched:{$params['options']}"
            );
            $params += [ 'page_touched' => $pageTouched ];
            $opts = [ 'removeDuplicates' => true ];
        }
 
        return new JobSpecification(
            'parsoidCachePrewarm',
            [
                'revId' => $revisionId,
                'pageId' => $pageId,
            ] + $params,
            $opts
        );
    }
 
    private function doParsoidCacheUpdate() {
        $page = $this->pageLookup->getPageById( $this->params['pageId'] );
        $revId = $this->params['revId'];
 
        if ( $page === null ) {
            // This happens when the page got deleted in the meantime.
            $this->logger->info( "Page with ID {$this->params['pageId']} not found" );
            return;
        }
 
        if ( $page->getLatest() !== $revId ) {
            $this->logger->info(
                'ParsoidCachePrewarmJob: The ID of the new revision does not match the page\'s current revision ID'
            );
            return;
        }
 
        $rev = $this->revisionLookup->getRevisionById( $revId );
        if ( !$rev ) {
            return;
        }
 
        $parserOpts = ParserOptions::newFromAnon();
 
        $renderReason = $this->params['causeAction'] ?? $this->command;
        $parserOpts->setRenderReason( $renderReason );
 
        $mainSlot = $rev->getSlot( SlotRecord::MAIN );
        if ( !$this->parsoidOutputAccess->supportsContentModel( $mainSlot->getModel() ) ) {
            $this->logger->debug( __METHOD__ . ': Parsoid does not support content model ' . $mainSlot->getModel() );
            return;
        }
 
        $this->logger->debug( __METHOD__ . ': generating Parsoid output' );
 
        // We may get the OPT_FORCE_PARSE flag this way
        $options = $this->params['options'] ?? 0;
 
        // getParserOutput() will write to ParserCache.
        $status = $this->parsoidOutputAccess->getParserOutput(
            $page,
            $parserOpts,
            $rev,
            $options | ParsoidOutputAccess::OPT_LOG_LINT_DATA
        );
 
        if ( !$status->isOK() ) {
            $this->logger->error( __METHOD__ . ': Parsoid error', [
                'errors' => $status->getErrors(),
                'page' => $page->getDBkey(),
                'rev' => $rev->getId(),
            ] );
        }
    }
 
    public function run() {
        $this->doParsoidCacheUpdate();
 
        return true;
    }
}