Updated for new ForgeProxy API

forge · Jan 8, 2014 · 170c2c6 · 170c2c6
1 parent 63efbe4
commit 170c2c6
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/.../main/java/org/jboss/forge/furnace/container/cdi/services/ExportedInstanceLazyLoader.java b/.../main/java/org/jboss/forge/furnace/container/cdi/services/ExportedInstanceLazyLoader.java
@@ -45,7 +45,10 @@ public Object invoke(Object self, Method thisMethod, Method proceed, Object[] ar
          if (thisMethod.getDeclaringClass().getName().equals(ForgeProxy.class.getName()))
          {
             // Must call from "this." or method call is not properly processed by this class.
-            return this.getDelegate();
+            if (thisMethod.getName().equals("getDelegate"))
+               return this.getDelegate();
+            else if (thisMethod.getName().equals("getHandler"))
+               return this.getHandler();
          }
       }
       catch (Exception e)
@@ -114,4 +117,10 @@ public Object getDelegate() throws Exception
       return delegate;
    }
 
+   @Override
+   public Object getHandler() throws Exception
+   {
+      return this;
+   }
+
 }