Class: MongoRequestLogger::Logger

Inherits:
ActiveSupport::Logger
  • Object
show all
Defined in:
lib/mongo_request_logger/logger.rb

Constant Summary collapse

LEVELS = 
{
    DEBUG => 'DEBUG',
    INFO => 'INFO',
    WARN => 'WARN',
    ERROR => 'ERROR',
    FATAL => 'FATAL',
    UNKNOWN => 'UNKNOWN',
}

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(adapter, file, level = DEBUG) ⇒ Logger

Returns a new instance of Logger.



8
9
10
11
12
13
 
# File 'lib/mongo_request_logger/logger.rb', line 8

def initialize(adapter, file, level=DEBUG)
  super(file, level)
  @adapter = adapter
  @mongo_record = {}  # This is required as a workaround for logging in the console
  create_indexes
end

Instance Attribute Details

#adapter ⇒ Object (readonly)

Returns the value of attribute adapter.



6
7
8
 
# File 'lib/mongo_request_logger/logger.rb', line 6

def adapter
  @adapter
end

Instance Method Details

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



91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
 
# File 'lib/mongo_request_logger/logger.rb', line 91

def add(severity, message = nil, progname = nil, &block)
  severity ||= UNKNOWN

  unless level > severity
    if message.nil?
      if block_given?
        message = yield
      else
        message = progname
      end
    end

    raise "FOO" if message.nil?

    time = Time.now.strftime "%H:%M:%S.%L"
    prefix = "#{time} [#{level_to_s(severity)}] "
    #if ActiveRecord::Base.colorize_logging
    #  # remove colorization done by rails and just save the actual message
    #  @mongo_record[:messages] << prefix + message.gsub(/(\e(\[([\d;]*[mz]?))?)?/, '').strip rescue nil
    #else
    @mongo_record[:messages] ||= []
    @mongo_record[:messages] << "#{prefix}#{message}"
    #end
  end

  # To do normal file logging as well, add a call to super here
  # We only log ERROR and above to the text file logs.
  if severity >= ERROR
    super
  end
end

#add_metadata(options = {}) ⇒ Object 



79
80
81
82
83
 
# File 'lib/mongo_request_logger/logger.rb', line 79

def add_metadata(options={})
  options.each_pair do |key, value|
    @mongo_record[key] = value
  end
end

#add_metadata_set(key, *values) ⇒ Object 



85
86
87
88
89
 
# File 'lib/mongo_request_logger/logger.rb', line 85

def add_metadata_set key, *values
  key = key.to_s
  @mongo_record[key] ||= Set.new
  @mongo_record[key] += values
end

#create_indexes ⇒ Object 



15
16
17
18
19
20
21
22
 
# File 'lib/mongo_request_logger/logger.rb', line 15

def create_indexes
  adapter.create_index '_id'
  adapter.create_index 'objects'
  adapter.create_index 'tags'
  adapter.create_index 'path'
  adapter.create_index 'user_id'
  adapter.create_index 'timestamp'
end

#level_to_s(level) ⇒ Object 



33
34
35
 
# File 'lib/mongo_request_logger/logger.rb', line 33

def level_to_s(level)
  LEVELS[level]
end

#log_request(options = {}) ⇒ Object 



54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
 
# File 'lib/mongo_request_logger/logger.rb', line 54

def log_request(options={})
  @mongo_record = options.merge({
                                    :messages => [],
                                    :timestamp => Time.now.utc,
                                    :pid => Process.pid,
                                })

  begin
    start_time = Time.now
    return yield
  rescue Exception => e
    # The backtrace is not useful metadata, and we want to write it to the text log as well.
    exception e
    @mongo_record[:exception] = {:class => e.class.to_s, :message => e.message}
    raise
  ensure
    # Benchmark.measure doesn't handle exceptions well, so we measure the time ourselves.
    end_time = Time.now
    runtime = end_time.to_f - start_time.to_f
    @mongo_record[:runtime] = (runtime.real * 1000).ceil

    do_log @mongo_record
  end
end

#sanitize(obj) ⇒ Object 



37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
 
# File 'lib/mongo_request_logger/logger.rb', line 37

def sanitize obj
  if obj.is_a? Hash
    result = {}
    obj.each do |k, v|
      # We replace all non-alphanumeric characters with underscores
      result[k.to_s.gsub(/\W/, "_")] = sanitize(v)
    end
    result
  elsif obj.is_a? Array or obj.is_a? Set
    obj.map {|o| sanitize(o)}
  elsif obj.is_a? Time or obj.is_a? String or obj.is_a? Integer
    obj
  else
    obj.to_s
  end
end