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

Loading…
Cancel
Save