Permalink
Fetching contributors…
Cannot retrieve contributors at this time
executable file 144 lines (124 sloc) 3.56 KB
#! /usr/bin/env ruby
#
# proc-status-metrics
#
# DESCRIPTION:
# For all processes owned by a user AND/OR matching a provided process
# name substring, return selected memory metrics from /proc/[PID]/status
#
# OUTPUT:
# metric data
#
# PLATFORMS:
# Linux
#
# DEPENDENCIES:
# gem: sensu-plugin
#
# USAGE:
#
# NOTES:
# - This check names the metrics after the full process name
# string as found in /proc/[PID]/cmdline, so data will be lost if
# you have multiple processes with identical cmdlines.
# - A process that changes its cmdline but still matches the search
# query will be duplicated as multiple discovered processes.
# - We make some assumptions in parsing /proc/[PID]/status, this will
# mostly only work with the memory-related stats.
#
# LICENSE:
# Copyright (c) 2014 Cozy Services Ltd. opensource@cozy.co
# Matt Greensmith mgreensmith@cozy.co
# Released under the same terms as Sensu (the MIT license); see LICENSE
# for details.
#
require 'sensu-plugin/metric/cli'
require 'socket'
#
# String
# monkeypatch for helping validate PID strings
#
class String
# if its and integer, set it to a string
def integer?
to_i.to_s == self
end
end
#
# Proc Status
#
# /proc/[PID]/status memory metrics plugin
#
class ProcStatus < Sensu::Plugin::Metric::CLI::Graphite
option :user,
description: 'Query processes owned by a user',
short: '-u USER',
long: '--user USER'
option :processname,
description: 'Process name substring to match against, not a regex.',
short: '-p PROCESSNAME',
long: '--process-name PROCESSNAME'
option :scheme,
description: 'Metric naming scheme',
long: '--scheme SCHEME',
default: "#{Socket.gethostname}.proc"
option :metrics,
description: 'Memory metrics to collect from /proc/[PID]/status, comma-separated',
short: '-m METRICS',
long: '--metrics METRICS',
default: 'VmSize,VmRSS,VmSwap'
# Build search command
#
def pgrep_command
pgrep_command = 'pgrep '
pgrep_command << "-u #{config[:user]} " if config[:user]
pgrep_command << "-f #{config[:processname]} " if config[:processname]
pgrep_command << '2<&1'
end
# Acquire process_pids
#
# @param pgrep_output [String]
#
def acquire_valid_pids(pgrep_output)
res = pgrep_output.split("\n").map(&:strip)
pids = res.select(&:integer?)
pids
end
# Acquire the sate for the supplied PID
#
# @param pid [String]
#
def acquire_stats_for_pid(pid)
return nil unless ::File.exist?(::File.join('/proc', pid, 'cmdline'))
cmdline_raw = `cat /proc/#{pid}/cmdline`
cmdline = cmdline_raw.strip.gsub(/[^[:alnum:]]/, '_')
metric_names = config[:metrics].split(',')
proc_status_lines = `cat /proc/#{pid}/status`.split("\n")
out = { cmdline.to_s => {} }
metric_names.each do |m|
line = proc_status_lines.find { |x| /^#{m}/.match(x) }
val = line ? line.split("\t")[1].to_i : nil
out[cmdline.to_s][m] = val
end
out
end
# Main functino
#
def run
raise 'You must supply -u USER or -p PROCESSNAME' unless config[:user] || config[:processname]
metrics = {}
pgrep_output = `#{pgrep_command}`
pids = acquire_valid_pids(pgrep_output)
pids.each do |p|
data = acquire_stats_for_pid(p)
metrics.merge!(data) unless data.nil?
end
timestamp = Time.now.to_i
metrics.each do |proc_name, stats|
stats.each do |stat_name, value|
output [config[:scheme], proc_name, stat_name].join('.'), value, timestamp
end
end
ok
end
end