KafkaHandler.php

Go to the documentation of this file.

<?php
namespace MediaWiki\Logger\Monolog;

use Kafka\MetaDataFromKafka;
use Kafka\Produce;
use MediaWiki\Logger\LoggerFactory;
use Monolog\Handler\AbstractProcessingHandler;
use Monolog\Logger;
use Psr\Log\LoggerInterface;

class KafkaHandler extends AbstractProcessingHandler {
    protected $produce;

    protected $options;

    protected $partitions = [];

    private static $defaultOptions = [
        'alias' => [], // map from monolog channel to kafka topic
        'swallowExceptions' => false, // swallow exceptions sending records
        'logExceptions' => null, // A PSR3 logger to inform about errors
    ];

    public function __construct(
        Produce $produce, array $options, $level = Logger::DEBUG, $bubble = true
    ) {
        parent::__construct( $level, $bubble );
        $this->produce = $produce;
        $this->options = array_merge( self::$defaultOptions, $options );
    }

    public static function factory(
        $kafkaServers, array $options = [], $level = Logger::DEBUG, $bubble = true
    ) {
        $metadata = new MetaDataFromKafka( $kafkaServers );
        $produce = new Produce( $metadata );

        if ( isset( $options['sendTimeout'] ) ) {
            $timeOut = $options['sendTimeout'];
            $produce->getClient()->setStreamOption( 'SendTimeoutSec', 0 );
            $produce->getClient()->setStreamOption( 'SendTimeoutUSec',
                intval( $timeOut * 1000000 )
            );
        }
        if ( isset( $options['recvTimeout'] ) ) {
            $timeOut = $options['recvTimeout'];
            $produce->getClient()->setStreamOption( 'RecvTimeoutSec', 0 );
            $produce->getClient()->setStreamOption( 'RecvTimeoutUSec',
                intval( $timeOut * 1000000 )
            );
        }
        if ( isset( $options['logExceptions'] ) && is_string( $options['logExceptions'] ) ) {
            $options['logExceptions'] = LoggerFactory::getInstance( $options['logExceptions'] );
        }

        return new self( $produce, $options, $level, $bubble );
    }

    protected function write( array $record ) {
        if ( $record['formatted'] !== null ) {
            $this->addMessages( $record['channel'], [ $record['formatted'] ] );
            $this->send();
        }
    }

    public function handleBatch( array $batch ) {
        $channels = [];
        foreach ( $batch as $record ) {
            if ( $record['level'] < $this->level ) {
                continue;
            }
            $channels[$record['channel']][] = $this->processRecord( $record );
        }

        $formatter = $this->getFormatter();
        foreach ( $channels as $channel => $records ) {
            $messages = [];
            foreach ( $records as $idx => $record ) {
                $message = $formatter->format( $record );
                if ( $message !== null ) {
                    $messages[] = $message;
                }
            }
            if ( $messages ) {
                $this->addMessages( $channel, $messages );
            }
        }

        $this->send();
    }

    protected function send() {
        try {
            $this->produce->send();
        } catch ( \Kafka\Exception $e ) {
            $ignore = $this->warning(
                'Error sending records to kafka: {exception}',
                [ 'exception' => $e ] );
            if ( !$ignore ) {
                throw $e;
            }
        }
    }

    protected function getRandomPartition( $topic ) {
        if ( !array_key_exists( $topic, $this->partitions ) ) {
            try {
                $partitions = $this->produce->getAvailablePartitions( $topic );
            } catch ( \Kafka\Exception $e ) {
                $ignore = $this->warning(
                    'Error getting metadata for kafka topic {topic}: {exception}',
                    [ 'topic' => $topic, 'exception' => $e ] );
                if ( $ignore ) {
                    return null;
                }
                throw $e;
            }
            if ( $partitions ) {
                $key = array_rand( $partitions );
                $this->partitions[$topic] = $partitions[$key];
            } else {
                $details = $this->produce->getClient()->getTopicDetail( $topic );
                $ignore = $this->warning(
                    'No partitions available for kafka topic {topic}',
                    [ 'topic' => $topic, 'kafka' => $details ]
                );
                if ( !$ignore ) {
                    throw new \RuntimeException( "No partitions available for kafka topic $topic" );
                }
                $this->partitions[$topic] = null;
            }
        }
        return $this->partitions[$topic];
    }

    protected function addMessages( $channel, array $records ) {
        if ( isset( $this->options['alias'][$channel] ) ) {
            $topic = $this->options['alias'][$channel];
        } else {
            $topic = "monolog_$channel";
        }
        $partition = $this->getRandomPartition( $topic );
        if ( $partition !== null ) {
            $this->produce->setMessages( $topic, $partition, $records );
        }
    }

    protected function warning( $message, array $context = [] ) {
        if ( $this->options['logExceptions'] instanceof LoggerInterface ) {
            $this->options['logExceptions']->warning( $message, $context );
        }
        return $this->options['swallowExceptions'];
    }
}