98 lines
		
	
	
		
			2.8 KiB
		
	
	
	
		
			PHP
		
	
	
	
		
		
			
		
	
	
			98 lines
		
	
	
		
			2.8 KiB
		
	
	
	
		
			PHP
		
	
	
	
|  | <?php | ||
|  | 
 | ||
|  | /* | ||
|  |  * This file is part of the Monolog package. | ||
|  |  * | ||
|  |  * (c) Jordi Boggiano <j.boggiano@seld.be> | ||
|  |  * | ||
|  |  * For the full copyright and license information, please view the LICENSE | ||
|  |  * file that was distributed with this source code. | ||
|  |  */ | ||
|  | 
 | ||
|  | namespace Monolog\Handler; | ||
|  | 
 | ||
|  | use Monolog\Formatter\LineFormatter; | ||
|  | use Monolog\Logger; | ||
|  | 
 | ||
|  | /** | ||
|  |  * Logs to a Redis key using rpush | ||
|  |  * | ||
|  |  * usage example: | ||
|  |  * | ||
|  |  *   $log = new Logger('application'); | ||
|  |  *   $redis = new RedisHandler(new Predis\Client("tcp://localhost:6379"), "logs", "prod"); | ||
|  |  *   $log->pushHandler($redis); | ||
|  |  * | ||
|  |  * @author Thomas Tourlourat <thomas@tourlourat.com> | ||
|  |  */ | ||
|  | class RedisHandler extends AbstractProcessingHandler | ||
|  | { | ||
|  |     private $redisClient; | ||
|  |     private $redisKey; | ||
|  |     protected $capSize; | ||
|  | 
 | ||
|  |     /** | ||
|  |      * @param \Predis\Client|\Redis $redis   The redis instance | ||
|  |      * @param string                $key     The key name to push records to | ||
|  |      * @param int                   $level   The minimum logging level at which this handler will be triggered | ||
|  |      * @param bool                  $bubble  Whether the messages that are handled can bubble up the stack or not | ||
|  |      * @param int                   $capSize Number of entries to limit list size to | ||
|  |      */ | ||
|  |     public function __construct($redis, $key, $level = Logger::DEBUG, $bubble = true, $capSize = false) | ||
|  |     { | ||
|  |         if (!(($redis instanceof \Predis\Client) || ($redis instanceof \Redis))) { | ||
|  |             throw new \InvalidArgumentException('Predis\Client or Redis instance required'); | ||
|  |         } | ||
|  | 
 | ||
|  |         $this->redisClient = $redis; | ||
|  |         $this->redisKey = $key; | ||
|  |         $this->capSize = $capSize; | ||
|  | 
 | ||
|  |         parent::__construct($level, $bubble); | ||
|  |     } | ||
|  | 
 | ||
|  |     /** | ||
|  |      * {@inheritDoc} | ||
|  |      */ | ||
|  |     protected function write(array $record) | ||
|  |     { | ||
|  |         if ($this->capSize) { | ||
|  |             $this->writeCapped($record); | ||
|  |         } else { | ||
|  |             $this->redisClient->rpush($this->redisKey, $record["formatted"]); | ||
|  |         } | ||
|  |     } | ||
|  | 
 | ||
|  |     /** | ||
|  |      * Write and cap the collection | ||
|  |      * Writes the record to the redis list and caps its | ||
|  |      * | ||
|  |      * @param  array $record associative record array | ||
|  |      * @return void | ||
|  |      */ | ||
|  |     protected function writeCapped(array $record) | ||
|  |     { | ||
|  |         if ($this->redisClient instanceof \Redis) { | ||
|  |             $this->redisClient->multi() | ||
|  |                 ->rpush($this->redisKey, $record["formatted"]) | ||
|  |                 ->ltrim($this->redisKey, -$this->capSize, -1) | ||
|  |                 ->exec(); | ||
|  |         } else { | ||
|  |             $redisKey = $this->redisKey; | ||
|  |             $capSize = $this->capSize; | ||
|  |             $this->redisClient->transaction(function ($tx) use ($record, $redisKey, $capSize) { | ||
|  |                 $tx->rpush($redisKey, $record["formatted"]); | ||
|  |                 $tx->ltrim($redisKey, -$capSize, -1); | ||
|  |             }); | ||
|  |         } | ||
|  |     } | ||
|  | 
 | ||
|  |     /** | ||
|  |      * {@inheritDoc} | ||
|  |      */ | ||
|  |     protected function getDefaultFormatter() | ||
|  |     { | ||
|  |         return new LineFormatter(); | ||
|  |     } | ||
|  | } |