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) ⇒ WebServer
constructor
A new instance of WebServer.
- #metrics ⇒ Object
- #start ⇒ Object
- #stop ⇒ Object
Constructor Details
#initialize(port:, collector: nil) ⇒ 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 |
# File 'lib/prometheus_exporter/server/web_server.rb', line 12 def initialize(port: , collector: nil) @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) @server = WEBrick::HTTPServer.new( Port: port, AccessLog: [], Logger: WEBrick::Log.new("/dev/null") ) @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 Discourse plugin 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
133 134 135 136 137 |
# File 'lib/prometheus_exporter/server/web_server.rb', line 133 def add_gauge(name, help, value) gauge = PrometheusExporter::Metric::Gauge.new(name, help) gauge.observe(value) gauge end |
#get_rss ⇒ Object
127 128 129 130 131 |
# File 'lib/prometheus_exporter/server/web_server.rb', line 127 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
60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 |
# File 'lib/prometheus_exporter/server/web_server.rb', line 60 def handle_metrics(req, res) @total_sessions.observe req.body do |block| begin @total_metrics.observe @collector.process(block) rescue => e @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
92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 |
# File 'lib/prometheus_exporter/server/web_server.rb', line 92 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
78 79 80 81 82 83 84 85 86 |
# File 'lib/prometheus_exporter/server/web_server.rb', line 78 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
88 89 90 |
# File 'lib/prometheus_exporter/server/web_server.rb', line 88 def stop @server.shutdown end |