Fix incorrect warnings

Netflix · Aug 17, 2016 · ed7708d · ed7708d
1 parent e5e5f65
commit ed7708d
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/archaius2-guice/src/main/java/com/netflix/archaius/guice/ConfigurationInjectingListener.java b/archaius2-guice/src/main/java/com/netflix/archaius/guice/ConfigurationInjectingListener.java
@@ -53,18 +53,18 @@ CascadeStrategy getCascadeStrategy() {
 
     @Override
     public <T> void onProvision(ProvisionInvocation<T> provision) {
-        if (injector == null) {
-            LOG.warn("Can't inject configuration until ConfigurationInjectingListener has been initialized");
-            return;
-        }
-
         Class<?> clazz = provision.getBinding().getKey().getTypeLiteral().getRawType();
 
         //
         // Configuration Loading
         //
         final ConfigurationSource source = clazz.getDeclaredAnnotation(ConfigurationSource.class);
         if (source != null) {
+            if (injector == null) {
+                LOG.warn("Can't inject configuration into {} until ConfigurationInjectingListener has been initialized", clazz.getName());
+                return;
+            }
+
             CascadeStrategy strategy = source.cascading() != ConfigurationSource.NullCascadeStrategy.class
                     ? injector.getInstance(source.cascading()) : getCascadeStrategy();
 
@@ -87,6 +87,11 @@ public <T> void onProvision(ProvisionInvocation<T> provision) {
         //
         Configuration configAnnot = clazz.getAnnotation(Configuration.class);
         if (configAnnot != null) {
+            if (injector == null) {
+                LOG.warn("Can't inject configuration into {} until ConfigurationInjectingListener has been initialized", clazz.getName());
+                return;
+            }
+
             try {
                 mapper.mapConfig(provision.provision(), config, new IoCContainer() {
                     @Override