updating ShutdownManager calls after bootique/bootique#346

bootique · Dec 6, 2023 · decbd10 · decbd10
1 parent 2c9f7b0
commit decbd10
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/bootique-cxf-core/src/main/java/io/bootique/cxf/BusProvider.java b/bootique-cxf-core/src/main/java/io/bootique/cxf/BusProvider.java
@@ -15,9 +15,9 @@
 import org.apache.cxf.interceptor.Interceptor;
 import org.apache.cxf.message.Message;
 
-import java.util.Set;
 import javax.inject.Inject;
 import javax.inject.Provider;
+import java.util.Set;
 
 public class BusProvider implements Provider<Bus> {
 
@@ -56,7 +56,6 @@ public BusProvider(
     public Bus get() {
         ExtensionManagerBus bus = new ExtensionManagerBus();
 
-
         ConfiguredBeanLocator originalLocator = bus.getExtension(ConfiguredBeanLocator.class);
 
         MultisourceBeanLocator multisourceBeanLocator = new MultisourceBeanLocator(beanLocator, originalLocator);
@@ -75,9 +74,6 @@ public Bus get() {
 
         BusFactory.possiblySetDefaultBus(bus);
 
-        shutdownManager.addShutdownHook(bus::shutdown);
-
-
-        return bus;
+        return shutdownManager.onShutdown(bus, b -> b.shutdown());
     }
 }