PoolWorkArticleView.php

Go to the documentation of this file.

<?php
namespace MediaWiki\PoolCounter;
 
use MediaWiki\Logger\Spi as LoggerSpi;
use MediaWiki\MediaWikiServices;
use MediaWiki\Page\ParserOutputAccess;
use MediaWiki\Parser\ParserOutput;
use MediaWiki\Revision\RevisionRecord;
use MediaWiki\Revision\RevisionRenderer;
use MediaWiki\Revision\SlotRecord;
use MediaWiki\Status\Status;
use MediaWiki\WikiMap\WikiMap;
use ParserOptions;
 
class PoolWorkArticleView extends PoolCounterWork {
    protected $parserOptions;
    protected $revision;
    private $renderer;
    protected $loggerSpi;
 
    public function __construct(
        string $workKey,
        RevisionRecord $revision,
        ParserOptions $parserOptions,
        RevisionRenderer $revisionRenderer,
        LoggerSpi $loggerSpi
    ) {
        parent::__construct( 'ArticleView', $workKey );
        $this->revision = $revision;
        $this->parserOptions = $parserOptions;
        $this->renderer = $revisionRenderer;
        $this->loggerSpi = $loggerSpi;
    }
 
    public function doWork() {
        return $this->renderRevision();
    }
 
    public function renderRevision(
        ?ParserOutput $previousOutput = null,
        bool $doSample = false,
        string $sourceLabel = ''
    ): Status {
        $renderedRevision = $this->renderer->getRenderedRevision(
            $this->revision,
            $this->parserOptions,
            null,
            [
                'audience' => RevisionRecord::RAW,
                'previous-output' => $previousOutput,
            ]
        );
 
        $parserOutput = $renderedRevision->getRevisionParserOutput();
 
        if ( $doSample ) {
            $stats = MediaWikiServices::getInstance()->getStatsFactory();
            $content = $this->revision->getContent( SlotRecord::MAIN );
            $labels = [
                'source' => $sourceLabel,
                'type' => $previousOutput === null ? 'full' : 'selective',
                'reason' => $this->parserOptions->getRenderReason(),
                'parser' => $this->parserOptions->getUseParsoid() ? 'parsoid' : 'legacy',
                'opportunistic' => 'false',
                'wiki' => WikiMap::getCurrentWikiId(),
                'model' => $content ? $content->getModel() : 'unknown',
            ];
            $totalStat = $stats->getCounter( 'ParserCache_selective_total' );
            $timeStat = $stats->getCounter( 'ParserCache_selective_cpu_seconds' );
            foreach ( $labels as $key => $value ) {
                $totalStat->setLabel( $key, $value );
                $timeStat->setLabel( $key, $value );
            }
            $totalStat->increment();
            $timeStat->incrementBy( $parserOutput->getTimeProfile( 'cpu' ) );
        }
 
        return Status::newGood( $parserOutput );
    }
 
    public function error( $status ) {
        return $status;
    }
 
}
 
class_alias( PoolWorkArticleView::class, 'PoolWorkArticleView' );