Add API for configuring custom ObservationRegistry (#2747)

Fixes #2719
reactor · Mar 29, 2023 · bfb341d · bfb341d
1 parent 4effedc
commit bfb341d
Showing 1 changed file with 14 additions and 2 deletions.
diff --git a/reactor-netty-core/src/main/java/reactor/netty/Metrics.java b/reactor-netty-core/src/main/java/reactor/netty/Metrics.java
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2019-2022 VMware, Inc. or its affiliates, All Rights Reserved.
+ * Copyright (c) 2019-2023 VMware, Inc. or its affiliates, All Rights Reserved.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -39,7 +39,7 @@
 public class Metrics {
 	public static final MeterRegistry REGISTRY = io.micrometer.core.instrument.Metrics.globalRegistry;
 	public static final String OBSERVATION_KEY = "micrometer.observation";
-	public static final ObservationRegistry OBSERVATION_REGISTRY = ObservationRegistry.create();
+	public static ObservationRegistry OBSERVATION_REGISTRY = ObservationRegistry.create();
 	static {
 		OBSERVATION_REGISTRY.observationConfig().observationHandler(
 				new ObservationHandler.FirstMatchingCompositeObservationHandler(
@@ -313,6 +313,18 @@ public static String formatSocketAddress(@Nullable SocketAddress socketAddress)
 		return null;
 	}
 
+	/**
+	 * Set the {@link ObservationRegistry} to use in Reactor Netty for tracing related purposes.
+	 *
+	 * @return the previously configured registry.
+	 * @since 1.1.6
+	 */
+	public static ObservationRegistry observationRegistry(ObservationRegistry observationRegistry) {
+		ObservationRegistry previous = OBSERVATION_REGISTRY;
+		OBSERVATION_REGISTRY = observationRegistry;
+		return previous;
+	}
+
 	public static Context updateContext(Context context, Object observation) {
 		return context.hasKey(OBSERVATION_KEY) ? context : context.put(OBSERVATION_KEY, observation);
 	}