[jboss-cvs] JBossAS SVN: r74135 - projects/ejb3/trunk/core/src/main/java/org/jboss/ejb3/session.

[jboss-cvs-commits at lists.jboss.org]

Author: ALRubinger
Date: 2008-06-05 07:05:53 -0400 (Thu, 05 Jun 2008)
New Revision: 74135

Modified:
   projects/ejb3/trunk/core/src/main/java/org/jboss/ejb3/session/SessionSpecContainer.java
Log:
[EJBTHREE-1394] Refactoring alongside changes to EJB3 Proxy

Modified: projects/ejb3/trunk/core/src/main/java/org/jboss/ejb3/session/SessionSpecContainer.java
===================================================================
--- projects/ejb3/trunk/core/src/main/java/org/jboss/ejb3/session/SessionSpecContainer.java	2008-06-05 10:54:47 UTC (rev 74134)
+++ projects/ejb3/trunk/core/src/main/java/org/jboss/ejb3/session/SessionSpecContainer.java	2008-06-05 11:05:53 UTC (rev 74135)
@@ -24,7 +24,7 @@
 import org.jboss.ejb3.common.registrar.spi.NotBoundException;
 import org.jboss.ejb3.interceptors.container.ContainerMethodInvocation;
 import org.jboss.ejb3.proxy.container.InvokableContext;
-import org.jboss.ejb3.proxy.handler.session.stateful.StatefulProxyInvocationHandler;
+import org.jboss.ejb3.proxy.handler.session.stateful.StatefulProxyInvocationHandlerBase;
 import org.jboss.ejb3.proxy.jndiregistrar.JndiSessionRegistrarBase;
 import org.jboss.ejb3.proxy.lang.SerializableMethod;
 import org.jboss.ejb3.stateful.StatefulContainerInvocation;
@@ -176,9 +176,9 @@
          //TODO Use Polymorphism to have sessions only in StatefulContainer
          InvocationHandler handler = Proxy.getInvocationHandler(proxy);
          Object sessionId = null;
-         if (handler instanceof StatefulProxyInvocationHandler)
+         if (handler instanceof StatefulProxyInvocationHandlerBase)
          {
-            sessionId = ((StatefulProxyInvocationHandler) handler).getSessionId();
+            sessionId = ((StatefulProxyInvocationHandlerBase) handler).getSessionId();
          }
 
          if (unadvisedMethod != null && isHomeMethod(method))