wikimedia/mediawiki-extensions-Wikibase

<?php

declare( strict_types = 1 );

namespace Wikibase\Repo\ParserOutput;

use InvalidArgumentException;
use Liuggio\StatsdClient\Factory\StatsdDataFactoryInterface;
use MediaWiki\Parser\ParserOutput;
use Wikibase\Lib\Store\EntityRevision;

/**
 * @license GPL-2.0-or-later
 */
class StatsdTimeRecordingEntityParserOutputGenerator implements EntityParserOutputGenerator {

    /** @var EntityParserOutputGenerator */
    private $inner;
    /** @var StatsdDataFactoryInterface */
    private $stats;
    /** @var string */
    private $timingPrefix;

    /**
     * @param EntityParserOutputGenerator $inner
     * @param StatsdDataFactoryInterface $stats
     * @param string $timingPrefix Resulting metric will be: $timingPrefix.getParserOutput.<html/nohtml>><entitytype>
     */
    public function __construct(
        EntityParserOutputGenerator $inner,
        StatsdDataFactoryInterface $stats,
        string $timingPrefix
    ) {
        $this->inner = $inner;
        $this->stats = $stats;
        $this->timingPrefix = $timingPrefix;
    }

    /**
     * Creates the parser output for the given entity.
     *
     * @param EntityRevision $entityRevision
     * @param bool $generateHtml
     *
     * @throws InvalidArgumentException
     * @return ParserOutput
     */
    public function getParserOutput(
        EntityRevision $entityRevision,
        $generateHtml = true
    ) {
        $start = microtime( true );
        $po = $this->inner->getParserOutput( $entityRevision, $generateHtml );
        $end = microtime( true );

        $htmlMetricPart = $generateHtml ? 'html' : 'nohtml';
        $this->stats->timing(
            "{$this->timingPrefix}.getParserOutput.{$htmlMetricPart}.{$entityRevision->getEntity()->getType()}",
            ( $end - $start ) * 1000
        );

        return $po;
    }

}