Pnlinh\QueryLogger\QueryLoggerServiceProvider

Complexity

Total Complexity

10

Size/Duplication

Total Lines	102
Duplicated Lines	0 %

Importance

Changes	3
Bugs	1	Features	0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	writeMongoDbLog()	0	9	1
A	boot()	0	18	3
A	writeMySqlLog()	0	22	3
A	listenQueryLogging()	0	13	3

<?php

namespace Pnlinh\QueryLogger;

use DateTime;
use Illuminate\Database\Events\QueryExecuted;
use Illuminate\Http\Request;
use Illuminate\Log\LogManager;
use Illuminate\Support\Carbon;
use Illuminate\Support\ServiceProvider;

class QueryLoggerServiceProvider extends ServiceProvider
{
    public const MYSQL = 'mysql';
    public const SQLITE = 'sqlite';
    public const MONGODB = 'mongodb';

    /**
     * Bootstrap services.
     *
     * @param Request $request
     *
     * @throws \Illuminate\Contracts\Container\BindingResolutionException
     *
     * @return void
     */
    public function boot(Request $request)
    {
        if ($this->app->runningInConsole()) {
            $this->publishes([
                __DIR__.'/../config/query-logger.php' => config_path('query-logger.php'),
            ], 'query-logger');
        }

        if (!config('query-logger.enabled')) {
            return;
        }

        $logger = app('log');

        $logger->info('STARTING REQUEST');
        $logger->info(sprintf('[%s] -> %s', $request->method(), $request->path()));

        $this->listenQueryLogging($logger);
    }

    /**
     * Listen connection event.
     *
     * @param LogManager $logger
     */
    protected function listenQueryLogging(LogManager $logger)
    {
        app('db')->connection()->enableQueryLog();

        app('db')->listen(function (QueryExecuted $query) use ($logger) {
            $driver = $query->connection->getDriverName();

            if (in_array($driver, [static::SQLITE, static::MYSQL])) {
                $this->writeMySqlLog($query, $logger);
            }

            if ($driver === static::MONGODB) {
                $this->writeMongoDbLog($query, $logger);
            }
        });
    }

    /**
     * Write mysql log.
     *
     * @param QueryExecuted $query
     * @param LogManager    $logger
     */
    protected function writeMySqlLog(QueryExecuted $query, LogManager $logger)
    {
        foreach ($query->bindings as &$binding) {
            if ($binding instanceof DateTime) {
                $binding = Carbon::instance($binding)->toDateTimeString();
            }
        }

        $tempStr = sprintf('@%s@', date('Y'));

        $sql = $query->sql;
        $sql = str_replace(['%', '?'], [$tempStr, '"%s"'], $sql);
        $sql = vsprintf(str_replace('?', '"%s"', $sql), $query->bindings);
        $sql = str_replace($tempStr, '%', $sql);

        $messages = [
            '[SQL]',
            sprintf('[%s]', strtoupper($query->connectionName)),
            sprintf('[%s ms]', str_pad($query->time, 4, 0)),
            $sql,
        ];
        $logger->info(implode(' ', $messages));
    }

    /**
     * Write mongodb log.
     *
     * @param QueryExecuted $query
     * @param LogManager    $logger
     */
    protected function writeMongoDbLog(QueryExecuted $query, LogManager $logger)
    {
        $messages = [
            '[MONGODB]',
            sprintf('[%s]', strtoupper($query->connectionName)),
            sprintf('[%s ms]', str_pad($query->time, 4, 0)),
            $query->sql,
        ];
        $logger->info(implode(' ', $messages));
    }
}