Class: Betterlog::Logger

Inherits:
Logger
  • Object
show all
Includes:
ComplexConfig::Provider::Shortcuts, Enumerable
Defined in:
lib/betterlog/logger.rb

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(redis, shift_age = 0, shift_size = 1048576, name: nil, buffer_size: nil, **opts) ⇒ Logger

Returns a new instance of Logger.



11
12
13
14
15
16
17
18
19
20
21
 
# File 'lib/betterlog/logger.rb', line 11

def initialize(redis, shift_age = 0, shift_size = 1048576, name: nil, buffer_size: nil, **opts)
  @redis       = redis
  @fallback    = ::Logger.new(STDERR)
  if level = cc.log.level?
    self.level      = level
    @fallback.level = level
  end
  @name        = name || self.class.name
  @buffer_size = determine_buffer_size(buffer_size)
  super(nil, shift_age, shift_size, **opts)
end

Class Method Details

.for_redis_url(url, shift_age = 0, shift_size = 1048576, **opts) ⇒ Object 



23
24
25
26
27
28
 
# File 'lib/betterlog/logger.rb', line 23

def self.for_redis_url(url, shift_age = 0, shift_size = 1048576,  **opts)
  redis = Redis.new(url: url)
  redis.ping
  new(redis, shift_age, shift_size, **opts)
rescue Redis::CannotConnectError
end

Instance Method Details

#<<(msg) ⇒ Object 



75
76
77
78
79
 
# File 'lib/betterlog/logger.rb', line 75

def <<(msg)
  redis_write(msg)
rescue Redis::BaseConnectionError
  @fallback << msg
end

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



52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
 
# File 'lib/betterlog/logger.rb', line 52

def add(severity, message = nil, progname = nil)
  severity ||= UNKNOWN
  if severity < @level
    return true
  end
  if progname.nil?
    progname = @progname
  end
  if message.nil?
    if block_given?
      message = yield
    else
      message = progname
      progname = @progname
    end
  end
  redis_write(
    format_message(format_severity(severity), Time.now, progname, message))
  true
rescue Redis::BaseConnectionError
  @fallback.add(severity, message, progname)
end

#clear ⇒ Object 



81
82
83
84
 
# File 'lib/betterlog/logger.rb', line 81

def clear
  @redis.del @name
  self
end

#each(chunk_size: 100 * 1024, &block) ⇒ Object 



126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
 
# File 'lib/betterlog/logger.rb', line 126

def each(chunk_size: 100 * 1024, &block)
  chunk_size > 0 or raise ArgumentError, 'chunk_size > 0 required'
  Enumerator.new do |y|
    buffer = ''
    buffer.encode! 'ASCII-8BIT'
    each_chunk(chunk_size: chunk_size) do |chunk|
      buffer << chunk
      buffer.gsub!(/\A(.*?#$/)/n) do |line|
        y.yield(line)
        ''
      end
    end
    buffer.length > 0 and y.yield(buffer)
  end.each(&block)
end

#each_chunk(chunk_size: 100 * 1024, &block) ⇒ Object 



86
87
88
89
90
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
122
123
124
 
# File 'lib/betterlog/logger.rb', line 86

def each_chunk(chunk_size: 100 * 1024, &block)
  chunk_size > 0 or raise ArgumentError, 'chunk_size > 0 required'

  # Delete any remaining temporary keys if we were interrtupted earlier
  # (or in some other process.)
  @redis.scan_each(match: "#{@name}_*") do |key|
    @redis.del key
  rescue Redis::BaseConnectionError
  end

  @redis.exists?(@name) or return Enumerator.new {}

  Enumerator.new do |y|
    name_tmp = "#{@name}_#{rand}"
    @redis.rename @name, name_tmp

    s = 0
    e = @redis.strlen(name_tmp) - 1
    until s > e
      range = @redis.getrange(name_tmp, s, s + chunk_size - 1)
      range.force_encoding 'ASCII-8BIT'
      y.yield range
      s += chunk_size
    end

  ensure
    begin
      @redis.del name_tmp
    rescue Redis::BaseConnectionError
      # We have to delete this later if del command failed here,
      # see the beginning of this method.
    end
  end.each(&block)

rescue Redis::BaseConnectionError => e
  # Maybe it works again later, just log the error…
  @fallback.error(e)
  Enumerator.new {}
end