feat: inject UnleashSubscriber to Unleash bean-definition (#34)

Unleash · Dec 3, 2024 · aebc91a · aebc91a
1 parent 5fab8f8
commit aebc91a
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/...ash-autoconfigure/src/main/java/org/unleash/features/config/UnleashAutoConfiguration.java b/...ash-autoconfigure/src/main/java/org/unleash/features/config/UnleashAutoConfiguration.java
@@ -4,6 +4,8 @@
 import io.getunleash.Unleash;
 import io.getunleash.UnleashContext;
 import io.getunleash.UnleashContextProvider;
+import io.getunleash.event.UnleashSubscriber;
+import io.getunleash.event.NoOpSubscriber;
 import io.getunleash.repository.OkHttpFeatureFetcher;
 import io.getunleash.strategy.Strategy;
 import io.getunleash.util.UnleashConfig;
@@ -44,7 +46,13 @@ public UnleashContextProvider unleashContextProvider(final UnleashProperties unl
     }
 
     @Bean
-    public Unleash unleash(final UnleashProperties unleashProperties, UnleashContextProvider unleashContextProvider) {
+    @ConditionalOnMissingBean
+    public UnleashSubscriber unleashSubscriber() {
+        return new NoOpSubscriber();
+    }
+
+    @Bean
+    public Unleash unleash(final UnleashProperties unleashProperties, UnleashContextProvider unleashContextProvider, UnleashSubscriber unleashSubscriber) {
         final var provider = getUnleashContextProviderWithThreadLocalSupport(unleashContextProvider);
         final var builder = UnleashConfig
                 .builder()
@@ -60,6 +68,7 @@ public Unleash unleash(final UnleashProperties unleashProperties, UnleashContext
                 .sendMetricsReadTimeout(unleashProperties.getSendMetricsReadTimeout())
                 .customHttpHeader("Authorization", unleashProperties.getApiToken())
                 .projectName(unleashProperties.getProjectName())
+                .subscriber(unleashSubscriber)
                 .synchronousFetchOnInitialisation(unleashProperties.isSynchronousFetchOnInitialisation())
                 .instanceId(StringUtils.hasText(unleashProperties.getInstanceId()) ? unleashProperties.getInstanceId() :
                         UUID.randomUUID().toString());