[jboss-remoting-commits] JBoss Remoting SVN: r5061 - remoting2/branches/2.2/src/main/org/jboss/remoting/callback.

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

Author: ron.sigal at jboss.com
Date: 2009-04-18 02:34:10 -0400 (Sat, 18 Apr 2009)
New Revision: 5061

Modified:
   remoting2/branches/2.2/src/main/org/jboss/remoting/callback/ServerInvokerCallbackHandler.java
Log:
JBREM-1119: Added shutdown() method.

Modified: remoting2/branches/2.2/src/main/org/jboss/remoting/callback/ServerInvokerCallbackHandler.java
===================================================================
--- remoting2/branches/2.2/src/main/org/jboss/remoting/callback/ServerInvokerCallbackHandler.java	2009-04-18 06:33:42 UTC (rev 5060)
+++ remoting2/branches/2.2/src/main/org/jboss/remoting/callback/ServerInvokerCallbackHandler.java	2009-04-18 06:34:10 UTC (rev 5061)
@@ -81,6 +81,7 @@
 
    private SerializableStore callbackStore = null;
    private CallbackErrorHandler callbackErrorHandler = null;
+   private ServerInvoker serverInvoker;
 
    /**
     * The map key to use when looking up any callback store that
@@ -166,6 +167,7 @@
 
    private void init(InvocationRequest invocation, ServerInvoker owner) throws Exception
    {
+      serverInvoker = owner;
       clientSessionId = invocation.getSessionId();
       sessionId = invocation.getSessionId();
       
@@ -1020,12 +1022,18 @@
       }
    }
 
+   public void shutdown()
+   {
+      serverInvoker.shutdownCallbackHandler(this, invocation);
+      destroy();
+      log.debug(this + " shut down");
+   }
+   
    public void handleConnectionException(Throwable throwable, Client client)
    {
       if (clientSessionId.equals(client.getSessionId()))
       {
-         destroy();
-         log.debug(this + " shut down");
+         shutdown();
       }
    }