Starlit\Db\Monolog\DbHandler

Complexity

Total Complexity

17

Size/Duplication

Total Lines	175
Duplicated Lines	0 %

Coupling/Cohesion

Components	2
Dependencies	4

Test Coverage

Coverage

94.44%

Importance

Changes

0

10 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	15	1
A	processRecord()	0	16	5
A	write()	0	12	1
A	clean()	0	10	4
A	getChannelEntriesCount()	0	7	1
A	deleteXOldestChannelEntries()	0	11	1
A	clear()	0	4	1
A	getDefaultFormatter()	0	4	1
A	setCleanDivisor()	0	4	1
A	setCleanProbability()	0	4	1

<?php
/**
 * Starlit Db.
 *
 * @copyright Copyright (c) 2016 Starweb AB
 * @license   BSD 3-Clause
 */

namespace Starlit\Db\Monolog;

use Monolog\Handler\AbstractProcessingHandler;
use Monolog\Logger;
use Starlit\Db\Db;

/**
 * Monolog handler to log to a database table.
 *
 * Use a table structure like this for compatibility:
 *
 * CREATE TABLE `log` (
 *   `log_entry_id` BIGINT(11) UNSIGNED NOT NULL AUTO_INCREMENT,
 *   `time` TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
 *   `channel` VARCHAR(64) NOT NULL,
 *   `level` VARCHAR(10) NOT NULL,
 *   `message` TEXT NOT NULL,
 *   PRIMARY KEY (`log_entry_id`),
 *   KEY `time` (`time`),
 *   KEY `channel` (`channel`),
 *   KEY `level` (`level`)
 * );
 *
 * @author Andreas Nilsson <http://github.com/jandreasn>
 */
class DbHandler extends AbstractProcessingHandler
{
    /**
     * @var Db
     */
    protected $db;

    /**
     * @var string
     */
    protected $table;

    /**
     * @var array
     */
    protected $additionalFields;

    /**
     * @var int
     */
    protected $maxEntries;

    /**
     * @var int
     */
    protected $cleanDivisor = 100;

    /**
     * Default probability of table being cleaned (1/100 = 1%).
     *
     * @var int
     */
    protected $cleanProbability = 1;

    /**
     * @param Db        $db
     * @param int       $maxEntries
     * @param array     $additionalFields
     * @param int       $level
     * @param bool      $bubble
     * @param string    $table
     */
    public function __construct(
        Db $db,
        $maxEntries = null,
        array $additionalFields = [],
        $level = Logger::DEBUG,
        $bubble = true,
        $table = 'log'
    ) {
        parent::__construct($level, $bubble);

        $this->db = $db;
        $this->maxEntries = $maxEntries;
        $this->additionalFields = $additionalFields;
        $this->table = $table;
    }

    /**
     * {@inheritdoc}
     */
    protected function processRecord(array $record)
    {
        $record = parent::processRecord($record);

        $record['additionalFieldsData'] = [];
        foreach ($this->additionalFields as $field) {
            foreach (['context', 'extra'] as $sourceKey) {
                if (isset($record[$sourceKey]) && array_key_exists($field, $record[$sourceKey])) {
                    $record['additionalFieldsData'][$field] = $record[$sourceKey][$field];
                    unset($record[$sourceKey][$field]);
                }
            }
        }

        return $record;
    }

    /**
     * {@inheritdoc}
     */
    protected function write(array $record)
    {
        $dbData = [
            'channel' => $record['channel'],
            'level'   => $record['level_name'],
            'message' => $record['formatted'],
        ] + $record['additionalFieldsData'];

        $this->db->insert($this->table, $dbData);

        $this->clean($record['channel']);
    }

    /**
     * @param string $channel
     */
    protected function clean($channel)
    {
        if ($this->maxEntries && mt_rand(1, $this->cleanDivisor) <= $this->cleanProbability) {
            $currentCount = $this->getChannelEntriesCount($channel);
            if ($currentCount > $this->maxEntries) {
                $entriesToDelete = $currentCount - $this->maxEntries;
                $this->deleteXOldestChannelEntries($channel, $entriesToDelete);
            }
        }
    }

    /**
     * @param string $channel
     * @return int
     */
    protected function getChannelEntriesCount($channel)
    {
        return $this->db->fetchValue(
            'SELECT COUNT(*) FROM `' . $this->table . '` WHERE `channel` = ?',
            [$channel]
        );
    }

    /**
     * @param $channel
     * @param $entriesToDelete
     */
    protected function deleteXOldestChannelEntries($channel, $entriesToDelete)
    {
        $this->db->exec(
            sprintf(
                'DELETE FROM `%s` WHERE `channel` = ?  ORDER BY `time` ASC LIMIT %d',
                $this->table,
                (int) $entriesToDelete
            ),
            [$channel]
        );
    }

    /**
     * @param Logger $logger
     */
    public function clear(Logger $logger)
    {
        $this->db->exec(sprintf('DELETE FROM `%s` WHERE `channel` = ?', $this->table), [$logger->getName()]);
    }

    /**
     * {@inheritdoc}
     */
    protected function getDefaultFormatter()
    {
        return new DbFormatter();
    }

    /**
     * @see setCleanProbability()
     * @param int $cleanDivisor
     */
    public function setCleanDivisor($cleanDivisor)
    {
        $this->cleanDivisor = $cleanDivisor;
    }

    /**
     * Sets the probability that log table is cleaned on log write.
     *
     * The clean probability together with clean divisor is used to calculate the probability.
     * With a clean probability of 1 and a divisor of 100, there's a 1% chance (1/100) the table
     * will be cleaned.
     *
     * @param int $cleanProbability
     */
    public function setCleanProbability($cleanProbability)
    {
        $this->cleanProbability = $cleanProbability;
    }
}