Class: Ocular::Logging::KafkaLogger

Inherits:
Object
  • Object
show all
Defined in:
lib/ocular/logging/kafka_logger.rb

Overview

Most of the Logger class is copied from the Ruby Logger class source code.

Defined Under Namespace

Classes: Formatter

Instance Method Summary collapse

Constructor Details

#initialize(settings = nil, kafka = nil) ⇒ KafkaLogger

Returns a new instance of KafkaLogger.



12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
 
# File 'lib/ocular/logging/kafka_logger.rb', line 12

def initialize(settings=nil, kafka=nil)
    @level = Severity::DEBUG
    @formatter = Formatter.new

    if kafka != nil
        @kafka = kafka
    else
        #logger = Logger.new(STDOUT)
        #logger.level = Logger::DEBUG
        #settings[:client][:logger] = logger
        @kafka = Kafka.new(settings[:client])
    end
    @settings = settings

    @producer = @kafka.producer

    @semaphore = Mutex.new
end

Instance Method Details

#add(severity, message = nil, run_id = nil, &block) ⇒ Object 



62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
 
# File 'lib/ocular/logging/kafka_logger.rb', line 62

def add(severity, message = nil, run_id = nil, &block)
    severity ||= Severity::UNKNOWN
    if severity < @level
        return true
    end

    if message.nil?
        if block_given?
            message = yield
        else
            message = progname
        end
    end

    begin
        @semaphore.synchronize do
            @producer.produce(@formatter.format_message(severity, Time.now, run_id, message), topic: @settings[:topic], partition_key: run_id)
            deliver_messages()
        end
    rescue StandardError => e
        STDERR.puts "Error on producing kafka message: #{e}"
        STDERR.puts "Message was #{message}, stacktrace: #{e.backtrace.join("\n")}"
    end
end

#debug(message = nil, &block) ⇒ Object Also known as: log 



37
38
39
 
# File 'lib/ocular/logging/kafka_logger.rb', line 37

def debug(message = nil, &block)
    add(Severity::DEBUG, message, @run_id, &block)
end

#deliver_messages ⇒ Object 



58
59
60
 
# File 'lib/ocular/logging/kafka_logger.rb', line 58

def deliver_messages
    @producer.deliver_messages
end

#error(message = nil, &block) ⇒ Object 



50
51
52
 
# File 'lib/ocular/logging/kafka_logger.rb', line 50

def error(message = nil, &block)
    add(Severity::ERROR, message, @run_id, &block)
end

#fatal(message = nil, &block) ⇒ Object 



54
55
56
 
# File 'lib/ocular/logging/kafka_logger.rb', line 54

def fatal(message = nil, &block)
    add(Severity::FATAL, message, @run_id, &block)
end

#info(message = nil, &block) ⇒ Object 



42
43
44
 
# File 'lib/ocular/logging/kafka_logger.rb', line 42

def info(message = nil, &block)
    add(Severity::INFO, message, @run_id, &block)
end

#log_cause(type, environment, run_id = nil) ⇒ Object 



99
100
101
102
103
104
105
106
107
108
109
110
 
# File 'lib/ocular/logging/kafka_logger.rb', line 99

def log_cause(type, environment, run_id = nil)
    begin
        @semaphore.synchronize do
            @producer.produce(@formatter.format_cause(type, environment, Time.now, run_id), topic: @settings[:topic], partition_key: run_id)
            deliver_messages()
        end
    rescue StandardError => e
        STDERR.puts "Error on producing kafka log_cause: #{e}"
        STDERR.puts "type: #{type}, stacktrace: #{e.backtrace.join("\n")}"                    
    end

end

#log_event(property, value, run_id = nil) ⇒ Object 



87
88
89
90
91
92
93
94
95
96
97
 
# File 'lib/ocular/logging/kafka_logger.rb', line 87

def log_event(property, value, run_id = nil)
    begin
        @semaphore.synchronize do
            @producer.produce(@formatter.format_event(property, value, Time.now, run_id), topic: @settings[:topic], partition_key: run_id)
            deliver_messages()
        end
    rescue StandardError => e
        STDERR.puts "Error on producing kafka log_event: #{e}"
        STDERR.puts "#{property} => #{value}, stacktrace: #{e.backtrace.join("\n")}"                    
    end
end

#log_timing(key, value, run_id = nil) ⇒ Object 



112
113
114
115
116
117
118
119
120
121
122
123
124
 
# File 'lib/ocular/logging/kafka_logger.rb', line 112

def log_timing(key, value, run_id = nil)
    begin
        @semaphore.synchronize do
            @producer.produce(@formatter.format_event("timing:" + key, value, Time.now, run_id), topic: @settings[:topic], partition_key: run_id)
            deliver_messages()
        end
    rescue StandardError => e
        STDERR.puts "Error on producing kafka log_timing: #{e}"
        STDERR.puts "Timing: #{key} => #{value}, stacktrace: #{e.backtrace.join("\n")}"

    end

end

#reconnect ⇒ Object 



31
32
33
34
35
 
# File 'lib/ocular/logging/kafka_logger.rb', line 31

def reconnect()
    @kafka = Kafka.new(@settings[:client])
    @producer = @kafka.producer
    @semaphore = Mutex.new
end

#warn(message = nil, &block) ⇒ Object 



46
47
48
 
# File 'lib/ocular/logging/kafka_logger.rb', line 46

def warn(message = nil, &block)
    add(Severity::WARN, message, @run_id, &block)
end