Simple container now adds ClassLoaderIntercepter to services.

forge · Jan 23, 2014 · 8abf050 · 8abf050
1 parent 8b6f751
commit 8abf050
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/...c/main/java/org/jboss/forge/furnace/container/simple/impl/SimpleExportedInstanceImpl.java b/...c/main/java/org/jboss/forge/furnace/container/simple/impl/SimpleExportedInstanceImpl.java
@@ -11,6 +11,8 @@
 import org.jboss.forge.furnace.Furnace;
 import org.jboss.forge.furnace.addons.Addon;
 import org.jboss.forge.furnace.exception.ContainerException;
+import org.jboss.forge.furnace.proxy.ClassLoaderInterceptor;
+import org.jboss.forge.furnace.proxy.Proxies;
 import org.jboss.forge.furnace.spi.ExportedInstance;
 
 /**
@@ -33,22 +35,25 @@ public SimpleExportedInstanceImpl(Furnace furnace, Addon addon, Class<T> clazz)
    @Override
    public T get()
    {
+      T delegate = null;
       try
       {
          try
          {
             Constructor<T> constructor = type.getConstructor(Furnace.class);
-            return constructor.newInstance(furnace);
+            delegate = constructor.newInstance(furnace);
          }
          catch (NoSuchMethodException e)
          {
-            return type.newInstance();
+            delegate = type.newInstance();
          }
       }
       catch (Exception e)
       {
          throw new ContainerException("Could not create instance of [" + type.getName() + "] through reflection.", e);
       }
+      return Proxies.enhance(addon.getClassLoader(), delegate, new ClassLoaderInterceptor(addon.getClassLoader(),
+               delegate));
    }
 
    @Override