32 lines
969 B
Ruby
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
|
|
}
|