Class: Riemann::Tools::DirSpace

Inherits:
Object
  • Object
show all
Includes:
Riemann::Tools
Defined in:
lib/riemann/tools/dir_space.rb

Constant Summary

Constants included from Riemann::Tools

VERSION

Instance Attribute Summary

Attributes included from Riemann::Tools

#argv

Instance Method Summary collapse

Methods included from Riemann::Tools

#attributes, #endpoint_name, included, #options, #report, #riemann, #run

Constructor Details

#initialize ⇒ DirSpace 



17
18
19
20
21
22
23
24
25
 
# File 'lib/riemann/tools/dir_space.rb', line 17

def initialize
  super

  @dir = opts.fetch(:directory)
  @service_prefix = opts.fetch(:service_prefix)
  @warning = opts.fetch(:warning, nil)
  @critical = opts.fetch(:critical, nil)
  @alert_on_missing = opts.fetch(:alert_on_missing)
end

Instance Method Details

#state(metric) ⇒ Object 



47
48
49
50
51
52
53
54
55
 
# File 'lib/riemann/tools/dir_space.rb', line 47

def state(metric)
  if @critical && metric > @critical
    'critical'
  elsif @warning && metric > @warning
    'warning'
  else
    'ok'
  end
end

#tick ⇒ Object 



27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
 
# File 'lib/riemann/tools/dir_space.rb', line 27

def tick
  if Dir.exist?(@dir)
    metric = `du '#{@dir}'`.lines.to_a.last.split("\t")[0].to_i
    report(
      service: "#{@service_prefix} #{@dir}",
      metric: metric,
      state: state(metric),
      tags: ['dir_space'],
    )
  elsif @alert_on_missing
    report(
      service: "#{@service_prefix} #{@dir} missing",
      description: "#{@service_prefix} #{@dir} does not exist",
      metric: metric,
      state: 'critical',
      tags: ['dir_space'],
    )
  end
end