Browse Source

Make `MongoHandlerContext` public to leverage contextual information to customize `Observation`s.

Closes #4622
pull/4655/head
Mark Paluch 2 years ago
parent
commit
2b6d07ab7f
No known key found for this signature in database
GPG Key ID: 55BC6374BAA9D973
  1. 15
      spring-data-mongodb/src/main/java/org/springframework/data/mongodb/observability/MongoHandlerContext.java

15
spring-data-mongodb/src/main/java/org/springframework/data/mongodb/observability/MongoHandlerContext.java

@ -15,12 +15,17 @@ @@ -15,12 +15,17 @@
*/
package org.springframework.data.mongodb.observability;
import io.micrometer.observation.Observation;
import io.micrometer.observation.transport.Kind;
import io.micrometer.observation.transport.SenderContext;
import java.util.Arrays;
import java.util.LinkedHashSet;
import java.util.Set;
import org.bson.BsonDocument;
import org.bson.BsonValue;
import org.springframework.lang.Nullable;
import com.mongodb.ConnectionString;
@ -29,10 +34,6 @@ import com.mongodb.event.CommandFailedEvent; @@ -29,10 +34,6 @@ import com.mongodb.event.CommandFailedEvent;
import com.mongodb.event.CommandStartedEvent;
import com.mongodb.event.CommandSucceededEvent;
import io.micrometer.observation.Observation;
import io.micrometer.observation.transport.Kind;
import io.micrometer.observation.transport.SenderContext;
/**
* A {@link Observation.Context} that contains MongoDB events.
*
@ -41,7 +42,7 @@ import io.micrometer.observation.transport.SenderContext; @@ -41,7 +42,7 @@ import io.micrometer.observation.transport.SenderContext;
* @author Mark Paluch
* @since 4.0
*/
class MongoHandlerContext extends SenderContext<Object> {
public class MongoHandlerContext extends SenderContext<Object> {
/**
* @see <a href=
@ -96,11 +97,11 @@ class MongoHandlerContext extends SenderContext<Object> { @@ -96,11 +97,11 @@ class MongoHandlerContext extends SenderContext<Object> {
return connectionString;
}
public void setCommandSucceededEvent(CommandSucceededEvent commandSucceededEvent) {
void setCommandSucceededEvent(CommandSucceededEvent commandSucceededEvent) {
this.commandSucceededEvent = commandSucceededEvent;
}
public void setCommandFailedEvent(CommandFailedEvent commandFailedEvent) {
void setCommandFailedEvent(CommandFailedEvent commandFailedEvent) {
this.commandFailedEvent = commandFailedEvent;
}

Loading…
Cancel
Save