Sha256: 1554b4920f93f5a5b2b12ba549d22f2da3f8512f8a29149ce71a2e5c475a31bc
Contents?: true
Size: 1.66 KB
Versions: 13
Compression:
Stored size: 1.66 KB
Contents
# frozen_string_literal: true
require 'riemann/tools'
# Gets the number of files present on a directory and submits it to riemann
module Riemann
module Tools
class DirFilesCount
include Riemann::Tools
opt :directory, '', default: '/var/log'
opt :service_prefix, 'The first part of the service name, before the directory path', default: 'dir-files-count'
opt :warning, 'Dir files number warning threshold', type: Integer
opt :critical, 'Dir files number critical threshold', type: Integer
opt :alert_on_missing, 'Send a critical metric if the directory is missing?', default: true
def initialize
@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
def tick
if Dir.exist?(@dir)
metric = Dir.entries(@dir).size - 2
report(
service: "#{@service_prefix} #{@dir}",
metric: metric,
state: state(metric),
tags: ['dir_files_count'],
)
elsif @alert_on_missing
report(
service: "#{@service_prefix} #{@dir} missing",
description: "#{@service_prefix} #{@dir} does not exist",
metric: metric,
state: 'critical',
tags: ['dir_files_count'],
)
end
end
def state(metric)
if @critical && metric > @critical
'critical'
elsif @warning && metric > @warning
'warning'
else
'ok'
end
end
end
end
end
Version data entries
13 entries across 13 versions & 1 rubygems