updating ShutdownManager calls after bootique/bootique#346

bootique · Dec 6, 2023 · d33650d · d33650d
1 parent ac09df2
commit d33650d
Showing 1 changed file with 2 additions and 14 deletions.
diff --git a/src/main/java/io/bootique/curator/CuratorModule.java b/src/main/java/io/bootique/curator/CuratorModule.java
@@ -24,7 +24,6 @@
 import io.bootique.config.ConfigurationFactory;
 import io.bootique.di.Binder;
 import io.bootique.di.Provides;
-import io.bootique.log.BootLogger;
 import io.bootique.shutdown.ShutdownManager;
 import org.apache.curator.framework.CuratorFramework;
 
@@ -48,20 +47,9 @@ public void configure(Binder binder) {
 
     @Provides
     @Singleton
-    public CuratorFramework createCurator(
-            ConfigurationFactory configFactory,
-            BootLogger bootLogger,
-            ShutdownManager shutdownManager) {
-
+    public CuratorFramework createCurator(ConfigurationFactory configFactory, ShutdownManager shutdownManager) {
         CuratorFramework client = configFactory.config(CuratorFrameworkFactory.class, CONFIG_PREFIX).createZkClient();
-
-        shutdownManager.addShutdownHook(() -> {
-            bootLogger.trace(() -> "shutting down Curator...");
-            client.close();
-        });
-
         client.start();
-
-        return client;
+        return shutdownManager.onShutdown(client);
     }
 }