dnsbl_exporter/config.ru

32 lines
969 B
Ruby

# vim: set noet sw=2 ts=2 sts=2:
require 'rack'
require_relative 'dnsbl_exporter'
DnsblCollector.load_config './config.yml'
run lambda {|env|
begin
req = Rack::Request.new env
case req.path
when '/metrics'
collector = DnsblCollector.new
target =
begin
target = IPAddr.new req.params['target']
rescue IPAddr::AddressFamilyError
return [400, {"Content-Type" => "text/plain"}, ["Valid target-IP-Address expected.\n"]]
end
collector.collect target
[200, {"Content-Type" => "text/plain"}, [
Prometheus::Client::Formats::Text.marshal( collector.registry),
Prometheus::Client::Formats::Text.marshal( DnsblCollector.registry),
]]
else
[404, {"Content-Type" => "text/plain"}, ["Not found.\nYou want to try /metrics?\n"]]
end
rescue
STDERR.puts "#$! (#{$!.class})", *$!.backtrace
[500, {"Content-Type" => "text/plain"}, ["Server-error.\n"]]
end
}