Skip to content

Commit

Permalink
HBASE-27395 Adding description to Prometheus metrics (#4807)
Browse files Browse the repository at this point in the history
Signed-off-by: Tamas Payer <[email protected]>
Signed-off-by: Balazs Meszaros <[email protected]>
  • Loading branch information
lucakovacs authored Oct 7, 2022
1 parent 4b45256 commit 46d37a7
Show file tree
Hide file tree
Showing 2 changed files with 11 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -41,7 +41,7 @@ public class PrometheusHadoopServlet extends HttpServlet {

@Override
protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws IOException {
writeMetrics(resp.getWriter());
writeMetrics(resp.getWriter(), "true".equals(req.getParameter("description")));
}

static String toPrometheusName(String metricRecordName, String metricName) {
Expand All @@ -57,15 +57,21 @@ static String toPrometheusName(String metricRecordName, String metricName) {
*/
@RestrictedApi(explanation = "Should only be called in tests or self", link = "",
allowedOnPath = ".*/src/test/.*|.*/PrometheusHadoopServlet\\.java")
void writeMetrics(Writer writer) throws IOException {
void writeMetrics(Writer writer, boolean desc) throws IOException {
Collection<MetricsRecord> metricRecords = MetricsExportHelper.export();
for (MetricsRecord metricsRecord : metricRecords) {
for (AbstractMetric metrics : metricsRecord.metrics()) {
if (metrics.type() == MetricType.COUNTER || metrics.type() == MetricType.GAUGE) {

String key = toPrometheusName(metricsRecord.name(), metrics.name());

if (desc) {
String description = metrics.description();
if (!description.isEmpty()) writer.append("# HELP ").append(description).append('\n');
}

writer.append("# TYPE ").append(key).append(" ")
.append(metrics.type().toString().toLowerCase()).append("\n").append(key).append("{");
.append(metrics.type().toString().toLowerCase()).append('\n').append(key).append("{");

/* add tags */
String sep = "";
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -60,7 +60,8 @@ public void testPublish() throws IOException {

// WHEN
PrometheusHadoopServlet prom2Servlet = new PrometheusHadoopServlet();
prom2Servlet.writeMetrics(writer);
// Test with no description
prom2Servlet.writeMetrics(writer, false);

// THEN
String writtenMetrics = stream.toString(UTF_8.name());
Expand Down

0 comments on commit 46d37a7

Please sign in to comment.