Adapt to observation changes in Framework
See spring-projects/spring-framework#29388
This commit is contained in:
parent
05c5b1225c
commit
0acaa0e460
|
|
@ -56,16 +56,16 @@ class ClientObservationConventionAdapter implements ClientRequestObservationConv
|
|||
@Override
|
||||
public KeyValues getLowCardinalityKeyValues(ClientRequestObservationContext context) {
|
||||
mutateClientRequest(context);
|
||||
Iterable<Tag> tags = this.tagsProvider.tags(context.getCarrier(), context.getResponse(), context.getError());
|
||||
Iterable<Tag> tags = this.tagsProvider.tags(context.getRequest(), context.getResponse(), context.getError());
|
||||
return KeyValues.of(tags, Tag::getKey, Tag::getValue);
|
||||
}
|
||||
|
||||
private void mutateClientRequest(ClientRequestObservationContext context) {
|
||||
// WebClientExchangeTagsProvider relies on a request attribute to get the URI
|
||||
// template, we need to adapt to that.
|
||||
ClientRequest clientRequest = ClientRequest.from(context.getCarrier())
|
||||
ClientRequest clientRequest = ClientRequest.from(context.getRequest())
|
||||
.attribute(URI_TEMPLATE_ATTRIBUTE, context.getUriTemplate()).build();
|
||||
context.setCarrier(clientRequest);
|
||||
context.setRequest(clientRequest);
|
||||
}
|
||||
|
||||
@Override
|
||||
|
|
|
|||
|
|
@ -45,7 +45,7 @@ class ClientObservationConventionAdapterTests {
|
|||
private ClientObservationConventionAdapter convention = new ClientObservationConventionAdapter(TEST_METRIC_NAME,
|
||||
new DefaultWebClientExchangeTagsProvider());
|
||||
|
||||
private ClientRequest request = ClientRequest.create(HttpMethod.GET, URI.create("/resource/test")).build();
|
||||
private ClientRequest.Builder requestBuilder = ClientRequest.create(HttpMethod.GET, URI.create("/resource/test"));
|
||||
|
||||
private ClientResponse response = ClientResponse.create(HttpStatus.OK).body("foo").build();
|
||||
|
||||
|
|
@ -54,7 +54,8 @@ class ClientObservationConventionAdapterTests {
|
|||
@BeforeEach
|
||||
void setup() {
|
||||
this.context = new ClientRequestObservationContext();
|
||||
this.context.setCarrier(this.request);
|
||||
this.context.setCarrier(this.requestBuilder);
|
||||
this.context.setRequest(this.requestBuilder.build());
|
||||
this.context.setResponse(this.response);
|
||||
this.context.setUriTemplate("/resource/{name}");
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in New Issue