Author: ron.sigal(a)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(); } }