From 2b6d07ab7f3cf865d5af93e7a36798d5b74b8437 Mon Sep 17 00:00:00 2001
From: Mark Paluch <mark.paluch@broadcom.com>
Date: Tue, 23 Jan 2024 15:02:39 +0100
Subject: [PATCH] Make `MongoHandlerContext` public to leverage contextual
 information to customize `Observation`s.

Closes #4622
---
 .../observability/MongoHandlerContext.java        | 15 ++++++++-------
 1 file changed, 8 insertions(+), 7 deletions(-)

diff --git a/spring-data-mongodb/src/main/java/org/springframework/data/mongodb/observability/MongoHandlerContext.java b/spring-data-mongodb/src/main/java/org/springframework/data/mongodb/observability/MongoHandlerContext.java
index 468d23a28..889ec69a6 100644
--- a/spring-data-mongodb/src/main/java/org/springframework/data/mongodb/observability/MongoHandlerContext.java
+++ b/spring-data-mongodb/src/main/java/org/springframework/data/mongodb/observability/MongoHandlerContext.java
@@ -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;
 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;
  * @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> {
 		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;
 	}