[jboss-remoting-commits] JBoss Remoting SVN: r4985 - remoting2/branches/2.x/src/main/org/jboss/remoting/callback.

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

Author: ron.sigal at jboss.com
Date: 2009-04-13 06:32:31 -0400 (Mon, 13 Apr 2009)
New Revision: 4985

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

Modified: remoting2/branches/2.x/src/main/org/jboss/remoting/callback/ServerInvokerCallbackHandler.java
===================================================================
--- remoting2/branches/2.x/src/main/org/jboss/remoting/callback/ServerInvokerCallbackHandler.java	2009-04-13 10:31:34 UTC (rev 4984)
+++ remoting2/branches/2.x/src/main/org/jboss/remoting/callback/ServerInvokerCallbackHandler.java	2009-04-13 10:32:31 UTC (rev 4985)
@@ -85,6 +85,7 @@
 
    private SerializableStore callbackStore = null;
    private CallbackErrorHandler callbackErrorHandler = null;
+   private ServerInvoker serverInvoker;
 
    /**
     * The map key to use when looking up any callback store that
@@ -170,6 +171,7 @@
 
    private void init(InvocationRequest invocation, ServerInvoker owner) throws Exception
    {
+      serverInvoker = owner;
       clientSessionId = invocation.getSessionId();
       sessionId = invocation.getSessionId();
       
@@ -1025,12 +1027,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();
       }
    }