Class: GRI::TextWriter

Inherits:
Writer show all
Defined in:
lib/gri/writer.rb

Constant Summary

Constants inherited from Writer

Writer::TYPES

Instance Attribute Summary

Attributes inherited from Writer

#loop

Class Method Summary collapse

Instance Method Summary collapse

Methods inherited from Writer

create

Constructor Details

#initialize(options = {}) ⇒ TextWriter

Returns a new instance of TextWriter.



119
120
121
122
123
 
# File 'lib/gri/writer.rb', line 119

def initialize options={}
  @options = options
  @dir = @options[:tra_dir]
  Dir.mkdir @dir unless File.exist? @dir
end

Class Method Details

.expire(dir, day) ⇒ Object 



158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
 
# File 'lib/gri/writer.rb', line 158

def self.expire dir, day
  run_expire_path = dir + '/.run_expire'
  now = Time.now
  if File.exist? run_expire_path
    last_run = nil
    open(run_expire_path) {|f| last_run = f.read}
    last_run = Time.at(last_run.to_i)
    return if (now - last_run < 24*3600) and !$debug
  end
  if (day = day.to_i) > 0
    t = (now - day * 24 * 3600)
    expire_date = "%04d%02d%02d" % [t.year, t.mon, t.day]
    begin
      Dir.glob(dir + '/*') {|hdir|
        next if File.symlink? hdir
        Dir.glob(hdir + '/*') {|path|
          if File.directory? path
            Dir.glob(path + '/*') {|path2|
              if (basename = File.basename path2) < expire_date
                File.unlink path2
              end
            }
          end
        }
      }
      open(run_expire_path, 'w') {|f| f.print now.to_i.to_s}
    rescue SystemCallError
      Log.error "ERROR: #{$!}"
    end
  end
end

Instance Method Details

#finalize ⇒ Object 



149
150
 
# File 'lib/gri/writer.rb', line 149

def finalize
end

#purge_logs ⇒ Object 



152
153
154
155
156
 
# File 'lib/gri/writer.rb', line 152

def purge_logs
  if (day = @options[:tra_expire_day].to_i) > 0
    self.class.expire @dir, day
  end
end

#write(records) ⇒ Object 



125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
 
# File 'lib/gri/writer.rb', line 125

def write records
  now = Time.now
  date = "%04d%02d%02d" % [now.year, now.mon, now.day]
  time = now.to_i
  @ios = {}
  for record in records
    #record['_time'] = time
    if (host = record['_host']) and (key = record['_key'])
      hdir = "#{@dir}/#{host}"
      Dir.mkdir hdir unless File.exist? hdir
      data_name, index = key.scan(/\A([^_\d]*)(_?.*)/).first
      interval = record['_interval']
      kdir = "#{hdir}/#{data_name}_#{interval}"
      Dir.mkdir kdir unless File.exist? kdir
      path = kdir + '/' + date
      unless io = @ios[path]
        io = @ios[path] = File.open(path, 'a')
      end
      io.puts LTSV.serialize(record)
    end
  end
  @ios.each {|k, io| io.close}
end