From ba4b97368bca3d6cb2dd88853757f491a31284e4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?St=C3=A9phane=20=C3=89pardaud?= <stef@epardaud.fr>
Date: Mon, 8 Apr 2024 15:17:29 +0200
Subject: [PATCH] Make a custom ContextManagerProvider with a single manager

Because Quarkus doesn't care about one-per CL and Franz said this is a
bottleneck for some reason
---
 .../QuarkusContextManagerProvider.java        | 42 +++++++++++++++++++
 .../SmallRyeContextPropagationRecorder.java   |  2 +-
 2 files changed, 43 insertions(+), 1 deletion(-)
 create mode 100644 extensions/smallrye-context-propagation/runtime/src/main/java/io/quarkus/smallrye/context/runtime/QuarkusContextManagerProvider.java

diff --git a/extensions/smallrye-context-propagation/runtime/src/main/java/io/quarkus/smallrye/context/runtime/QuarkusContextManagerProvider.java b/extensions/smallrye-context-propagation/runtime/src/main/java/io/quarkus/smallrye/context/runtime/QuarkusContextManagerProvider.java
new file mode 100644
index 0000000000000..36b220416e8b6
--- /dev/null
+++ b/extensions/smallrye-context-propagation/runtime/src/main/java/io/quarkus/smallrye/context/runtime/QuarkusContextManagerProvider.java
@@ -0,0 +1,42 @@
+package io.quarkus.smallrye.context.runtime;
+
+import org.eclipse.microprofile.context.spi.ContextManager;
+
+import io.smallrye.context.SmallRyeContextManager;
+import io.smallrye.context.SmallRyeContextManagerProvider;
+
+/**
+ * Quarkus doesn't need one manager per CL, we only have the one
+ */
+public class QuarkusContextManagerProvider extends SmallRyeContextManagerProvider {
+
+	private SmallRyeContextManager contextManager;
+
+	@Override
+	public SmallRyeContextManager getContextManager(ClassLoader classLoader) {
+		return contextManager;
+	}
+
+	@Override
+	public SmallRyeContextManager getContextManager() {
+		return contextManager;
+	}
+
+	@Override
+	public ContextManager findContextManager(ClassLoader classLoader) {
+		return contextManager;
+	}
+
+	@Override
+	public void registerContextManager(ContextManager manager, ClassLoader classLoader) {
+		if (manager instanceof SmallRyeContextManager == false) {
+			throw new IllegalArgumentException("Only instances of SmallRyeContextManager are supported: " + manager);
+		}
+		contextManager = (SmallRyeContextManager) manager;
+	}
+
+	@Override
+	public void releaseContextManager(ContextManager manager) {
+		contextManager = null;
+	}
+}
diff --git a/extensions/smallrye-context-propagation/runtime/src/main/java/io/quarkus/smallrye/context/runtime/SmallRyeContextPropagationRecorder.java b/extensions/smallrye-context-propagation/runtime/src/main/java/io/quarkus/smallrye/context/runtime/SmallRyeContextPropagationRecorder.java
index b1a40ce4a00c7..7d5dbd189b157 100644
--- a/extensions/smallrye-context-propagation/runtime/src/main/java/io/quarkus/smallrye/context/runtime/SmallRyeContextPropagationRecorder.java
+++ b/extensions/smallrye-context-propagation/runtime/src/main/java/io/quarkus/smallrye/context/runtime/SmallRyeContextPropagationRecorder.java
@@ -31,7 +31,7 @@ public void configureStaticInit(List<ThreadContextProvider> discoveredProviders,
         // build the manager at static init time
         // in the live-reload mode, the provider instance may be already set in the previous start
         if (ContextManagerProvider.INSTANCE.get() == null) {
-            ContextManagerProvider contextManagerProvider = new SmallRyeContextManagerProvider();
+            ContextManagerProvider contextManagerProvider = new QuarkusContextManagerProvider();
             ContextManagerProvider.register(contextManagerProvider);
         }