Class: PrometheusExporter::Server::WebServer
- Inherits:
-
Object
- Object
- PrometheusExporter::Server::WebServer
- Defined in:
- lib/prometheus_exporter/server/web_server.rb
Instance Attribute Summary collapse
-
#collector ⇒ Object
readonly
Returns the value of attribute collector.
Instance Method Summary collapse
- #add_gauge(name, help, value) ⇒ Object
- #get_rss ⇒ Object
- #handle_metrics(req, res) ⇒ Object
-
#initialize(port:, collector: nil, verbose: false) ⇒ WebServer
constructor
A new instance of WebServer.
- #metrics ⇒ Object
- #start ⇒ Object
- #stop ⇒ Object
Constructor Details
#initialize(port:, collector: nil, verbose: false) ⇒ WebServer
Returns a new instance of WebServer.
12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 |
# File 'lib/prometheus_exporter/server/web_server.rb', line 12 def initialize(port: , collector: nil, verbose: false) @verbose = verbose @total_metrics = PrometheusExporter::Metric::Counter.new("total_collector_metrics", "total metrics processed by exporter web") @total_sessions = PrometheusExporter::Metric::Counter.new("total_collector_sessions", "total send_metric sessions processed by exporter web") @total_bad_metrics = PrometheusExporter::Metric::Counter.new("total_collector_bad_metrics", "total mis-handled metrics by collector") @total_metrics.observe(0) @total_sessions.observe(0) @total_bad_metrics.observe(0) access_log = [] if verbose access_log = [ [$stderr, WEBrick::AccessLog::COMMON_LOG_FORMAT], [$stderr, WEBrick::AccessLog::REFERER_LOG_FORMAT], ] logger = WEBrick::Log.new($stderr) end @server = WEBrick::HTTPServer.new( Port: port, Logger: logger, AccessLog: access_log ) @collector = collector || Collector.new @port = port @server.mount_proc '/' do |req, res| res['ContentType'] = 'text/plain; charset=utf-8' if req.path == '/metrics' res.status = 200 if req.header["accept-encoding"].to_s.include?("gzip") sio = StringIO.new collected_metrics = metrics begin writer = Zlib::GzipWriter.new(sio) writer.write(collected_metrics) ensure writer.close end res.body = sio.string res.header["content-encoding"] = "gzip" else res.body = metrics end elsif req.path == '/send-metrics' handle_metrics(req, res) else res.status = 404 res.body = "Not Found! The Prometheus Ruby Exporter only listens on /metrics and /send-metrics" end end end |
Instance Attribute Details
#collector ⇒ Object (readonly)
Returns the value of attribute collector.
10 11 12 |
# File 'lib/prometheus_exporter/server/web_server.rb', line 10 def collector @collector end |
Instance Method Details
#add_gauge(name, help, value) ⇒ Object
150 151 152 153 154 |
# File 'lib/prometheus_exporter/server/web_server.rb', line 150 def add_gauge(name, help, value) gauge = PrometheusExporter::Metric::Gauge.new(name, help) gauge.observe(value) gauge end |
#get_rss ⇒ Object
144 145 146 147 148 |
# File 'lib/prometheus_exporter/server/web_server.rb', line 144 def get_rss @pagesize ||= `getconf PAGESIZE`.to_i rescue 4096 @pid ||= Process.pid File.read("/proc/#{@pid}/statm").split(' ')[1].to_i * @pagesize rescue 0 end |
#handle_metrics(req, res) ⇒ Object
71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 |
# File 'lib/prometheus_exporter/server/web_server.rb', line 71 def handle_metrics(req, res) @total_sessions.observe req.body do |block| begin @total_metrics.observe @collector.process(block) rescue => e if @verbose STDERR.puts STDERR.puts e.inspect STDERR.puts e.backtrace STDERR.puts end @total_bad_metrics.observe res.body = "Bad Metrics #{e}" res.status = e.respond_to?(:status_code) ? e.status_code : 500 return end end res.body = "OK" res.status = 200 end |
#metrics ⇒ Object
109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 |
# File 'lib/prometheus_exporter/server/web_server.rb', line 109 def metrics metric_text = nil begin Timeout::timeout(2) do metric_text = @collector.prometheus_metrics_text end rescue Timeout::Error # we timed out ... bummer STDERR.puts "Generating Prometheus metrics text timed out" end metrics = [] metrics << add_gauge( "collector_working", "Is the master process collector able to collect metrics", metric_text && metric_text.length > 0 ? 1 : 0 ) metrics << add_gauge( "collector_rss", "total memory used by collector process", get_rss ) metrics << @total_metrics metrics << @total_sessions metrics << @total_bad_metrics <<~TEXT #{metrics.map(&:to_prometheus_text).join("\n\n")} #{metric_text} TEXT end |
#start ⇒ Object
95 96 97 98 99 100 101 102 103 |
# File 'lib/prometheus_exporter/server/web_server.rb', line 95 def start @runner ||= Thread.start do begin @server.start rescue => e STDERR.puts "Failed to start prometheus collector web on port #{@port}: #{e}" end end end |
#stop ⇒ Object
105 106 107 |
# File 'lib/prometheus_exporter/server/web_server.rb', line 105 def stop @server.shutdown end |