lib/presque/worker/Role/Logger.pm


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76

package presque::worker::Role::Logger;

use Moose::Role;
use Log::Dispatch;
use Log::Dispatch::Screen;

has logger => (
    is      => 'rw',
    isa     => 'Object',
    lazy    => 1,
    default => sub {
        my $self    = shift;
        my $context = $self->context;
        my $logger  = Log::Dispatch->new();
        if (my $log_conf = $context->{logger}) {
            foreach my $log_type (keys %{$log_conf}) {
                my $log_engine = $self->_load_log_engine($log_type);
                my $_logger =
                  $log_engine->new(%{$self->context->{logger}->{$log_type}});
                $logger->add($_logger);
            }
        }
        else {
            $logger->add(
                Log::Dispatch::Screen->new(
                    name      => 'screen',
                    min_level => 'debug',
                    newline   => 1,
                )
            );
        }
        $logger;
    }
);

sub _load_log_engine {
    my ($self, $engine) = @_;
    my $log_engine = "Log::Dispatch::" . ucfirst($engine);
    Class::MOP::load_class($log_engine);
    $log_engine;
}

before start => sub {
    my $self = shift;
    $self->log( 'start to listen for ' . $self->queue_name );
};

before work => sub {
    my $self = shift;
    $self->log( 'start to work', 'debug' );
};

before _shutdown => sub {
    my $self = shift;
    $self->log( 'shutting down' );
};

before _graceful_shutdown => sub {
    my $self = shift;
    $self->log( 'kill child' );
};

before _kill_child => sub {
    my $self = shift;
    $self->log( 'shutting down gracefuly' );
};

sub log {
    my( $self, $msg, $level ) = @_;
    $self->logger->log(
        message => '['.$self->worker_id.'] '.$msg,
        level   => $level || 'info',
    );
}

1;