includes/debug/logger/MonologSpi.php

   1 <?php
   2 /**
   3  * This program is free software; you can redistribute it and/or modify
   4  * it under the terms of the GNU General Public License as published by
   5  * the Free Software Foundation; either version 2 of the License, or
   6  * (at your option) any later version.
   7  *
   8  * This program is distributed in the hope that it will be useful,
   9  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  11  * GNU General Public License for more details.
  12  *
  13  * You should have received a copy of the GNU General Public License along
  14  * with this program; if not, write to the Free Software Foundation, Inc.,
  15  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  16  * http://www.gnu.org/copyleft/gpl.html
  17  *
  18  * @file
  19  */
  20
  21 namespace MediaWiki\Logger;
  22
  23 use MediaWiki\Logger\Monolog\BufferHandler;
  24 use Monolog\Logger;
  25 use ObjectFactory;
  26
  27 /**
  28  * LoggerFactory service provider that creates loggers implemented by
  29  * Monolog.
  30  *
  31  * Configured using an array of configuration data with the keys 'loggers',
  32  * 'processors', 'handlers' and 'formatters'.
  33  *
  34  * The ['loggers']['\@default'] configuration will be used to create loggers
  35  * for any channel that isn't explicitly named in the 'loggers' configuration
  36  * section.
  37  *
  38  * Configuration will most typically be provided in the $wgMWLoggerDefaultSpi
  39  * global configuration variable used by LoggerFactory to construct its
  40  * default SPI provider:
  41  * @code
  42  * $wgMWLoggerDefaultSpi = array(
  43  *   'class' => '\\MediaWiki\\Logger\\MonologSpi',
  44  *   'args' => array( array(
  45  *       'loggers' => array(
  46  *           '@default' => array(
  47  *               'processors' => array( 'wiki', 'psr', 'pid', 'uid', 'web' ),
  48  *               'handlers'   => array( 'stream' ),
  49  *           ),
  50  *           'runJobs' => array(
  51  *               'processors' => array( 'wiki', 'psr', 'pid' ),
  52  *               'handlers'   => array( 'stream' ),
  53  *           )
  54  *       ),
  55  *       'processors' => array(
  56  *           'wiki' => array(
  57  *               'class' => '\\MediaWiki\\Logger\\Monolog\\WikiProcessor',
  58  *           ),
  59  *           'psr' => array(
  60  *               'class' => '\\Monolog\\Processor\\PsrLogMessageProcessor',
  61  *           ),
  62  *           'pid' => array(
  63  *               'class' => '\\Monolog\\Processor\\ProcessIdProcessor',
  64  *           ),
  65  *           'uid' => array(
  66  *               'class' => '\\Monolog\\Processor\\UidProcessor',
  67  *           ),
  68  *           'web' => array(
  69  *               'class' => '\\Monolog\\Processor\\WebProcessor',
  70  *           ),
  71  *       ),
  72  *       'handlers' => array(
  73  *           'stream' => array(
  74  *               'class'     => '\\Monolog\\Handler\\StreamHandler',
  75  *               'args'      => array( 'path/to/your.log' ),
  76  *               'formatter' => 'line',
  77  *           ),
  78  *           'redis' => array(
  79  *               'class'     => '\\Monolog\\Handler\\RedisHandler',
  80  *               'args'      => array( function() {
  81  *                       $redis = new Redis();
  82  *                       $redis->connect( '127.0.0.1', 6379 );
  83  *                       return $redis;
  84  *                   },
  85  *                   'logstash'
  86  *               ),
  87  *               'formatter' => 'logstash',
  88  *               'buffer' => true,
  89  *           ),
  90  *           'udp2log' => array(
  91  *               'class' => '\\MediaWiki\\Logger\\Monolog\\LegacyHandler',
  92  *               'args' => array(
  93  *                   'udp://127.0.0.1:8420/mediawiki
  94  *               ),
  95  *               'formatter' => 'line',
  96  *           ),
  97  *       ),
  98  *       'formatters' => array(
  99  *           'line' => array(
 100  *               'class' => '\\Monolog\\Formatter\\LineFormatter',
 101  *            ),
 102  *            'logstash' => array(
 103  *                'class' => '\\Monolog\\Formatter\\LogstashFormatter',
 104  *                'args'  => array( 'mediawiki', php_uname( 'n' ), null, '', 1 ),
 105  *            ),
 106  *       ),
 107  *   ) ),
 108  * );
 109  * @endcode
 110  *
 111  * @see https://github.com/Seldaek/monolog
 112  * @since 1.25
 113  * @author Bryan Davis <bd808@wikimedia.org>
 114  * @copyright © 2014 Bryan Davis and Wikimedia Foundation.
 115  */
 116 class MonologSpi implements Spi {
 117
 118         /**
 119          * @var array $singletons
 120          */
 121         protected $singletons;
 122
 123         /**
 124          * Configuration for creating new loggers.
 125          * @var array $config
 126          */
 127         protected $config;
 128
 129         /**
 130          * @param array $config Configuration data.
 131          */
 132         public function __construct( array $config ) {
 133                 $this->config = [];
 134                 $this->mergeConfig( $config );
 135         }
 136
 137         /**
 138          * Merge additional configuration data into the configuration.
 139          *
 140          * @since 1.26
 141          * @param array $config Configuration data.
 142          */
 143         public function mergeConfig( array $config ) {
 144                 foreach ( $config as $key => $value ) {
 145                         if ( isset( $this->config[$key] ) ) {
 146                                 $this->config[$key] = array_merge( $this->config[$key], $value );
 147                         } else {
 148                                 $this->config[$key] = $value;
 149                         }
 150                 }
 151                 $this->reset();
 152         }
 153
 154         /**
 155          * Reset internal caches.
 156          *
 157          * This is public for use in unit tests. Under normal operation there should
 158          * be no need to flush the caches.
 159          */
 160         public function reset() {
 161                 $this->singletons = [
 162                         'loggers'    => [],
 163                         'handlers'   => [],
 164                         'formatters' => [],
 165                         'processors' => [],
 166                 ];
 167         }
 168
 169         /**
 170          * Get a logger instance.
 171          *
 172          * Creates and caches a logger instance based on configuration found in the
 173          * $wgMWLoggerMonologSpiConfig global. Subsequent request for the same channel
 174          * name will return the cached instance.
 175          *
 176          * @param string $channel Logging channel
 177          * @return \Psr\Log\LoggerInterface Logger instance
 178          */
 179         public function getLogger( $channel ) {
 180                 if ( !isset( $this->singletons['loggers'][$channel] ) ) {
 181                         // Fallback to using the '@default' configuration if an explict
 182                         // configuration for the requested channel isn't found.
 183                         $spec = isset( $this->config['loggers'][$channel] ) ?
 184                                 $this->config['loggers'][$channel] :
 185                                 $this->config['loggers']['@default'];
 186
 187                         $monolog = $this->createLogger( $channel, $spec );
 188                         $this->singletons['loggers'][$channel] = $monolog;
 189                 }
 190
 191                 return $this->singletons['loggers'][$channel];
 192         }
 193
 194         /**
 195          * Create a logger.
 196          * @param string $channel Logger channel
 197          * @param array $spec Configuration
 198          * @return \Monolog\Logger
 199          */
 200         protected function createLogger( $channel, $spec ) {
 201                 $obj = new Logger( $channel );
 202
 203                 if ( isset( $spec['calls'] ) ) {
 204                         foreach ( $spec['calls'] as $method => $margs ) {
 205                                 call_user_func_array( [ $obj, $method ], $margs );
 206                         }
 207                 }
 208
 209                 if ( isset( $spec['processors'] ) ) {
 210                         foreach ( $spec['processors'] as $processor ) {
 211                                 $obj->pushProcessor( $this->getProcessor( $processor ) );
 212                         }
 213                 }
 214
 215                 if ( isset( $spec['handlers'] ) ) {
 216                         foreach ( $spec['handlers'] as $handler ) {
 217                                 $obj->pushHandler( $this->getHandler( $handler ) );
 218                         }
 219                 }
 220                 return $obj;
 221         }
 222
 223         /**
 224          * Create or return cached processor.
 225          * @param string $name Processor name
 226          * @return callable
 227          */
 228         public function getProcessor( $name ) {
 229                 if ( !isset( $this->singletons['processors'][$name] ) ) {
 230                         $spec = $this->config['processors'][$name];
 231                         $processor = ObjectFactory::getObjectFromSpec( $spec );
 232                         $this->singletons['processors'][$name] = $processor;
 233                 }
 234                 return $this->singletons['processors'][$name];
 235         }
 236
 237         /**
 238          * Create or return cached handler.
 239          * @param string $name Processor name
 240          * @return \Monolog\Handler\HandlerInterface
 241          */
 242         public function getHandler( $name ) {
 243                 if ( !isset( $this->singletons['handlers'][$name] ) ) {
 244                         $spec = $this->config['handlers'][$name];
 245                         $handler = ObjectFactory::getObjectFromSpec( $spec );
 246                         if ( isset( $spec['formatter'] ) ) {
 247                                 $handler->setFormatter(
 248                                         $this->getFormatter( $spec['formatter'] )
 249                                 );
 250                         }
 251                         if ( isset( $spec['buffer'] ) && $spec['buffer'] ) {
 252                                 $handler = new BufferHandler( $handler );
 253                         }
 254                         $this->singletons['handlers'][$name] = $handler;
 255                 }
 256                 return $this->singletons['handlers'][$name];
 257         }
 258
 259         /**
 260          * Create or return cached formatter.
 261          * @param string $name Formatter name
 262          * @return \Monolog\Formatter\FormatterInterface
 263          */
 264         public function getFormatter( $name ) {
 265                 if ( !isset( $this->singletons['formatters'][$name] ) ) {
 266                         $spec = $this->config['formatters'][$name];
 267                         $formatter = ObjectFactory::getObjectFromSpec( $spec );
 268                         $this->singletons['formatters'][$name] = $formatter;
 269                 }
 270                 return $this->singletons['formatters'][$name];
 271         }
 272 }