1
2
3
4
5
6
7
8
9
10
11
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
|
#include "server/metrics.h"
#include <event2/buffer.h>
#include <event2/http.h>
#include <cstring>
#include <sstream>
#include "monitoring/prometheus/exporter.h"
using std::ostringstream;
using std::strncmp;
namespace cert_trans {
namespace {
const char kPrometheusProtoContentType[] =
"application/vnd.google.protobuf;"
"proto=io.prometheus.client.MetricFamily;encoding=delimited";
const size_t kPrometheusProtoContentTypeLen =
std::strlen(kPrometheusProtoContentType);
} // namespace
void ExportPrometheusMetrics(evhttp_request* req) {
if (evhttp_request_get_command(req) != EVHTTP_REQ_GET) {
evhttp_send_reply(req, HTTP_BADMETHOD, /*reason*/ nullptr,
/*databuf*/ nullptr);
return;
}
ostringstream oss;
const char* req_accept(
evhttp_find_header(evhttp_request_get_input_headers(req), "Accept"));
if (req_accept &&
std::strncmp(req_accept, kPrometheusProtoContentType,
kPrometheusProtoContentTypeLen) == 0) {
evhttp_add_header(evhttp_request_get_output_headers(req), "Content-Type",
kPrometheusProtoContentType);
ExportMetricsToPrometheus(&oss);
} else {
evhttp_add_header(evhttp_request_get_output_headers(req), "Content-Type",
"text/html");
ExportMetricsToHtml(&oss);
}
evbuffer_add(evhttp_request_get_output_buffer(req), oss.str().data(),
oss.str().size());
evhttp_send_reply(req, HTTP_OK, /*reason*/ nullptr, /*databuf*/ nullptr);
}
} // namespace cert_trans
|
