Invoke CamelContextCustomizer as part of DefaultConfigurationConfigur…

…er::afterConfigure
apache · Mar 4, 2024 · 21597f0 · 21597f0
1 parent d492f09
commit 21597f0
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 0 deletions.
diff --git a/core/camel-main/src/main/java/org/apache/camel/main/DefaultConfigurationConfigurer.java b/core/camel-main/src/main/java/org/apache/camel/main/DefaultConfigurationConfigurer.java
@@ -46,6 +46,7 @@
 import org.apache.camel.spi.AsyncProcessorAwaitManager;
 import org.apache.camel.spi.BacklogDebugger;
 import org.apache.camel.spi.BeanIntrospection;
+import org.apache.camel.spi.CamelContextCustomizer;
 import org.apache.camel.spi.ClassResolver;
 import org.apache.camel.spi.CliConnectorFactory;
 import org.apache.camel.spi.CompileStrategy;
@@ -605,6 +606,12 @@ public static void afterConfigure(final CamelContext camelContext) throws Except
             vault.setHashicorpVaultConfiguration(hashicorp);
         }
         configureVault(camelContext);
+
+        // apply custom configurations if any
+        Set<CamelContextCustomizer> customizers = registry.findByType(CamelContextCustomizer.class);
+        if (!customizers.isEmpty()) {
+            customizers.forEach(c -> c.configure(camelContext));
+        }
     }
 
     /**

diff --git a/core/camel-main/src/test/java/org/apache/camel/main/MainCustomizerTest.java b/core/camel-main/src/test/java/org/apache/camel/main/MainCustomizerTest.java
@@ -18,12 +18,15 @@
 
 import org.apache.camel.BindToRegistry;
 import org.apache.camel.CamelConfiguration;
+import org.apache.camel.CamelContext;
 import org.apache.camel.component.log.LogComponent;
+import org.apache.camel.spi.CamelContextCustomizer;
 import org.apache.camel.spi.ComponentCustomizer;
 import org.apache.camel.support.CustomizersSupport;
 import org.apache.camel.support.processor.DefaultExchangeFormatter;
 import org.junit.jupiter.api.Test;
 
+import static org.junit.jupiter.api.Assertions.assertEquals;
 import static org.junit.jupiter.api.Assertions.assertFalse;
 import static org.junit.jupiter.api.Assertions.assertTrue;
 
@@ -76,6 +79,27 @@ public void testComponentCustomizerDisabledWithProperties() {
         }
     }
 
+    @Test
+    public void testContextCustomizer() {
+        Main main = new Main();
+
+        try {
+            main.configure().addConfiguration(MyConfiguration.class);
+            main.bind("name-customizer", new CamelContextCustomizer() {
+                @Override
+                public void configure(CamelContext camelContext) {
+                    camelContext.getCamelContextExtension().setName("customized-name");
+                }
+            });
+
+            main.start();
+
+            assertEquals("customized-name", main.getCamelContext().getName());
+        } finally {
+            main.stop();
+        }
+    }
+
     // ****************************
     //
     // Helpers