Skip to content

Commit

Permalink
Fix broken content negotiation for Prometheus with OpenMetrics
Browse files Browse the repository at this point in the history
Update Prometheus `TextOutputFormat` so that OpenMetrics is used in
preference to text output when an appropriate accept header is found.

If the accept header contains `*/*` or is missing then the text format
will be used.

See gh-28130
  • Loading branch information
wilkinsona committed Oct 5, 2021
1 parent d8141e6 commit 437a160
Show file tree
Hide file tree
Showing 2 changed files with 19 additions and 6 deletions.
Expand Up @@ -36,25 +36,30 @@
public enum TextOutputFormat implements Producible<TextOutputFormat> {

/**
* OpenMetrics text version 1.0.0.
* Prometheus text version 0.0.4.
*/
CONTENT_TYPE_OPENMETRICS_100(TextFormat.CONTENT_TYPE_OPENMETRICS_100) {
CONTENT_TYPE_004(TextFormat.CONTENT_TYPE_004) {

@Override
void write(Writer writer, Enumeration<MetricFamilySamples> samples) throws IOException {
TextFormat.writeOpenMetrics100(writer, samples);
TextFormat.write004(writer, samples);
}

@Override
public boolean isDefault() {
return true;
}

},

/**
* Prometheus text version 0.0.4.
* OpenMetrics text version 1.0.0.
*/
CONTENT_TYPE_004(TextFormat.CONTENT_TYPE_004) {
CONTENT_TYPE_OPENMETRICS_100(TextFormat.CONTENT_TYPE_OPENMETRICS_100) {

@Override
void write(Writer writer, Enumeration<MetricFamilySamples> samples) throws IOException {
TextFormat.write004(writer, samples);
TextFormat.writeOpenMetrics100(writer, samples);
}

};
Expand Down
Expand Up @@ -55,6 +55,14 @@ void scrapeCanProduceOpenMetrics100(WebTestClient client) {
.contains("counter1_total").contains("counter2_total").contains("counter3_total"));
}

@WebEndpointTest
void scrapePrefersToProduceOpenMetrics100(WebTestClient client) {
MediaType openMetrics = MediaType.parseMediaType(TextFormat.CONTENT_TYPE_OPENMETRICS_100);
MediaType textPlain = MediaType.parseMediaType(TextFormat.CONTENT_TYPE_004);
client.get().uri("/actuator/prometheus").accept(openMetrics, textPlain).exchange().expectStatus().isOk()
.expectHeader().contentType(openMetrics);
}

@WebEndpointTest
void scrapeWithIncludedNames(WebTestClient client) {
client.get().uri("/actuator/prometheus?includedNames=counter1_total,counter2_total").exchange().expectStatus()
Expand Down

0 comments on commit 437a160

Please sign in to comment.