[jboss-cvs] JBossRemoting/src/tests/org/jboss/test/remoting/callback/multiple ...
Ron Sigal
ron_sigal at yahoo.com
Wed Dec 6 22:23:20 EST 2006
User: rsigal
Date: 06/12/06 22:23:20
Added: src/tests/org/jboss/test/remoting/callback/multiple Tag:
remoting_2_x CallbackTestCase.java
CallbackTestClient.java CallbackTestServer.java
Log:
BREM-622: New unit tests for multiple registrations of an InvokerCallbackHandler.
Revision Changes Path
No revision
No revision
1.1.2.2 +72 -0 JBossRemoting/src/tests/org/jboss/test/remoting/callback/multiple/CallbackTestCase.java
(In the diff below, changes in quantity of whitespace are not shown.)
Index: CallbackTestCase.java
===================================================================
RCS file: CallbackTestCase.java
diff -N CallbackTestCase.java
--- /dev/null 1 Jan 1970 00:00:00 -0000
+++ CallbackTestCase.java 7 Dec 2006 03:23:20 -0000 1.1.2.2
@@ -0,0 +1,72 @@
+/*
+* JBoss, a division of Red Hat
+* Copyright 2006, Red Hat Middleware, LLC, and individual contributors as indicated
+* by the @authors tag. See the copyright.txt in the distribution for a
+* full listing of individual contributors.
+*
+* This is free software; you can redistribute it and/or modify it
+* under the terms of the GNU Lesser General Public License as
+* published by the Free Software Foundation; either version 2.1 of
+* the License, or (at your option) any later version.
+*
+* This software is distributed in the hope that it will be useful,
+* but WITHOUT ANY WARRANTY; without even the implied warranty of
+* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+* Lesser General Public License for more details.
+*
+* You should have received a copy of the GNU Lesser General Public
+* License along with this software; if not, write to the Free
+* Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
+* 02110-1301 USA, or see the FSF site: http://www.fsf.org.
+*/
+package org.jboss.test.remoting.callback.multiple;
+
+import org.jboss.jrunit.harness.TestDriver;
+
+/**
+ * @author <a href="mailto:tom at jboss.org">Tom Elrod</a>
+ */
+public class CallbackTestCase extends TestDriver
+{
+ public void declareTestClasses()
+ {
+ addTestClasses(CallbackTestClient.class.getName(),
+ 1,
+ CallbackTestServer.class.getName());
+ }
+
+ /**
+ * How long to wait for test results to be returned from the client(s). If goes longer than the
+ * specified limit, will throw an exception and kill the running test cases. Default value is
+ * RESULTS_TIMEOUT.
+ *
+ * @return
+ */
+ protected long getResultsTimeout()
+ {
+ return 600000;
+ }
+
+ /**
+ * How long for the server test case to wait for tear down message. If exceeds timeout,
+ * will throw exception. The default value is TEARDOWN_TIMEOUT.
+ *
+ * @return
+ */
+ protected long getTearDownTimeout()
+ {
+ return 600000;
+ }
+
+ /**
+ * How long to allow each of the test cases to run their tests. If exceeds this timeout
+ * will throw exception and kill tests. The default value is RUN_TEST_TIMEOUT.
+ *
+ * @return
+ */
+ protected long getRunTestTimeout()
+ {
+ return 600000;
+ }
+
+}
\ No newline at end of file
1.1.2.2 +556 -0 JBossRemoting/src/tests/org/jboss/test/remoting/callback/multiple/CallbackTestClient.java
(In the diff below, changes in quantity of whitespace are not shown.)
Index: CallbackTestClient.java
===================================================================
RCS file: CallbackTestClient.java
diff -N CallbackTestClient.java
--- /dev/null 1 Jan 1970 00:00:00 -0000
+++ CallbackTestClient.java 7 Dec 2006 03:23:20 -0000 1.1.2.2
@@ -0,0 +1,556 @@
+package org.jboss.test.remoting.callback.multiple;
+
+import junit.framework.TestCase;
+import org.jboss.logging.Logger;
+import org.jboss.remoting.AbstractInvoker;
+import org.jboss.remoting.Client;
+import org.jboss.remoting.InvocationRequest;
+import org.jboss.remoting.InvokerLocator;
+import org.jboss.remoting.ServerInvocationHandler;
+import org.jboss.remoting.ServerInvoker;
+import org.jboss.remoting.callback.Callback;
+import org.jboss.remoting.callback.HandleCallbackException;
+import org.jboss.remoting.callback.InvokerCallbackHandler;
+import org.jboss.remoting.transport.ClientInvoker;
+import org.jboss.remoting.transport.Connector;
+
+import javax.management.MBeanServer;
+
+import java.lang.reflect.Field;
+import java.util.ArrayList;
+import java.util.Iterator;
+import java.util.List;
+import java.util.Map;
+
+/**
+ * These tests supplement the tests in org.jboss.test.remoting.callback.multiple.client
+ * and org.jboss.test.remoting.callback.multiple.server.
+ *
+ * @author <a href="ron.sigal at jboss.com">Ron Sigal</a>
+ * @version $Revision: 1.1.2.2 $
+ * <p>
+ * Copyright Dec 6, 2006
+ * </p>
+ */
+public class CallbackTestClient extends TestCase
+{
+ // Default locator values
+ private static String transport = "socket";
+ private static String host = "localhost";
+ private static int port = 5530;
+ private static int nextPort = port + 10;
+
+ private String locatorURI = transport + "://" + host + ":" + port;
+
+ private static final Logger log = Logger.getLogger(CallbackTestClient.class);
+
+ public static final String RESPONSE_VALUE = "This is the return to SampleInvocationHandler invocation";
+
+
+ public void testPush1Client1Connector2Handlers() throws Throwable
+ {
+ log.info("entering " + getName());
+ CallbackTestClient.CallbackHandler callbackHandler1 = new CallbackTestClient.CallbackHandler();
+ CallbackTestClient.CallbackHandler callbackHandler2 = new CallbackTestClient.CallbackHandler();
+ InvokerLocator locator = new InvokerLocator(locatorURI);
+ Client client = new Client(locator);
+ client.connect();
+ Connector connector = null;
+
+ try
+ {
+ String callbackLocatorURI = transport + "://" + host + ":" + nextPort++;
+ InvokerLocator callbackLocator = new InvokerLocator(callbackLocatorURI);
+ connector = setupServer(callbackLocator);
+ client.addListener(callbackHandler1, callbackLocator);
+ client.addListener(callbackHandler2, callbackLocator);
+ client.invoke("Do something", null);
+ client.invoke("Do something else", null);
+
+ // need to wait for brief moment so server can callback
+ Thread.sleep(5000);
+
+ int count = callbackHandler1.getCallbackCount();
+ assertEquals(2, count);
+ count = callbackHandler2.getCallbackCount();
+ assertEquals(2, count);
+ }
+ finally
+ {
+ if(client != null)
+ {
+ // remove callback handler from server
+ client.removeListener(callbackHandler1);
+ client.removeListener(callbackHandler2);
+ client.disconnect();
+ }
+ teardownServer(connector);
+ }
+ }
+
+ public void testPush1Client2Connectors1Handler() throws Throwable
+ {
+ log.info("entering " + getName());
+ CallbackTestClient.CallbackHandler callbackHandler = new CallbackTestClient.CallbackHandler();
+ InvokerLocator locator = new InvokerLocator(locatorURI);
+ Client client = new Client(locator);
+ client.connect();
+ Connector connector1 = null;
+ Connector connector2 = null;
+
+ try
+ {
+ String callbackLocatorURI1 = transport + "://" + host + ":" + nextPort++;
+ InvokerLocator callbackLocator1 = new InvokerLocator(callbackLocatorURI1);
+ connector1 = setupServer(callbackLocator1);
+ client.addListener(callbackHandler, callbackLocator1);
+ String callbackLocatorURI2 = transport + "://" + host + ":" + nextPort++;
+ InvokerLocator callbackLocator2 = new InvokerLocator(callbackLocatorURI2);
+ connector2 = setupServer(callbackLocator2);
+ client.addListener(callbackHandler, callbackLocator2);
+ client.invoke("Do something", null);
+ client.invoke("Do something else", null);
+
+ // need to wait for brief moment so server can callback
+ Thread.sleep(5000);
+
+ int count = callbackHandler.getCallbackCount();
+ assertEquals(4, count);
+ }
+ finally
+ {
+ if(client != null)
+ {
+ // remove callback handler from server
+ client.removeListener(callbackHandler);
+ client.disconnect();
+ }
+ teardownServer(connector1);
+ teardownServer(connector2);
+ }
+ }
+
+ public void testPush1Client2Connectors2Handlers() throws Throwable
+ {
+ log.info("entering " + getName());
+ CallbackTestClient.CallbackHandler callbackHandler1 = new CallbackTestClient.CallbackHandler();
+ CallbackTestClient.CallbackHandler callbackHandler2 = new CallbackTestClient.CallbackHandler();
+ InvokerLocator locator = new InvokerLocator(locatorURI);
+ Client client = new Client(locator);
+ client.connect();
+ Connector connector1 = null;
+ Connector connector2 = null;
+
+ try
+ {
+ String callbackLocatorURI1 = transport + "://" + host + ":" + nextPort++;
+ InvokerLocator callbackLocator1 = new InvokerLocator(callbackLocatorURI1);
+ connector1 = setupServer(callbackLocator1);
+ client.addListener(callbackHandler1, callbackLocator1);
+ String callbackLocatorURI2 = transport + "://" + host + ":" + nextPort++;
+ InvokerLocator callbackLocator2 = new InvokerLocator(callbackLocatorURI2);
+ connector2 = setupServer(callbackLocator2);
+ client.addListener(callbackHandler2, callbackLocator2);
+ client.invoke("Do something", null);
+ client.invoke("Do something else", null);
+
+ // need to wait for brief moment so server can callback
+ Thread.sleep(5000);
+
+ int count = callbackHandler1.getCallbackCount();
+ assertEquals(2, count);
+ count = callbackHandler2.getCallbackCount();
+ assertEquals(2, count);
+ }
+ finally
+ {
+ if(client != null)
+ {
+ // remove callback handler from server
+ client.removeListener(callbackHandler1);
+ client.removeListener(callbackHandler2);
+ client.disconnect();
+ }
+ teardownServer(connector1);
+ teardownServer(connector2);
+ }
+ }
+
+ public void testPush2Clients1Connector1Handler() throws Throwable
+ {
+ log.info("entering " + getName());
+ CallbackTestClient.CallbackHandler callbackHandler = new CallbackTestClient.CallbackHandler();
+ InvokerLocator locator = new InvokerLocator(locatorURI);
+ Client client1 = new Client(locator);
+ client1.connect();
+ Client client2 = new Client(locator);
+ client2.connect();
+ Connector connector = null;
+
+ try
+ {
+ String callbackLocatorURI = transport + "://" + host + ":" + nextPort++;
+ InvokerLocator callbackLocator = new InvokerLocator(callbackLocatorURI);
+ connector = setupServer(callbackLocator);
+ log.info("created callback server: " + connector.getLocator());
+ client1.addListener(callbackHandler, callbackLocator);
+ client2.addListener(callbackHandler, callbackLocator);
+ client1.invoke("Do something", null);
+ client2.invoke("Do something else", null);
+
+ // need to wait for brief moment so server can callback
+ Thread.sleep(5000);
+
+ // The CallbackHandler should be registered just once.
+ int count = callbackHandler.getCallbackCount();
+ assertEquals(2, count);
+ }
+ finally
+ {
+ if(client1 != null)
+ {
+ // remove callback handler from server
+ client1.removeListener(callbackHandler);
+ client1.disconnect();
+ }
+ teardownServer(connector);
+ }
+ }
+
+ public void testPush2Clients1Connector2Handlers() throws Throwable
+ {
+ log.info("entering " + getName());
+ CallbackTestClient.CallbackHandler callbackHandler1 = new CallbackTestClient.CallbackHandler();
+ CallbackTestClient.CallbackHandler callbackHandler2 = new CallbackTestClient.CallbackHandler();
+ InvokerLocator locator = new InvokerLocator(locatorURI);
+ Client client1 = new Client(locator);
+ client1.connect();
+ Client client2 = new Client(locator);
+ client2.connect();
+ Connector connector = null;
+
+ try
+ {
+ String callbackLocatorURI = transport + "://" + host + ":" + nextPort++;
+ InvokerLocator callbackLocator = new InvokerLocator(callbackLocatorURI);
+ connector = setupServer(callbackLocator);
+ client1.addListener(callbackHandler1, callbackLocator);
+ client2.addListener(callbackHandler2, callbackLocator);
+ client1.invoke("Do something", null);
+ client2.invoke("Do something else", null);
+
+ // need to wait for brief moment so server can callback
+ Thread.sleep(5000);
+
+ int count = callbackHandler1.getCallbackCount();
+ assertEquals(2, count);
+ count = callbackHandler1.getCallbackCount();
+ assertEquals(2, count);
+ }
+ finally
+ {
+ Field field = AbstractInvoker.class.getDeclaredField("localServerLocators");
+ field.setAccessible(true);
+
+ if(client1 != null)
+ {
+ ClientInvoker invoker = client1.getInvoker();
+ Map localServerLocators = (Map) field.get(invoker);
+ assertEquals(2, localServerLocators.size());
+ client1.removeListener(callbackHandler1);
+ assertEquals(1, localServerLocators.size());
+ client1.removeListener(callbackHandler2);
+ // client1 shouldn't be able to remove callbackHandler2.
+ assertEquals(1, localServerLocators.size());
+ client1.disconnect();
+ }
+ if (client2 != null)
+ {
+ ClientInvoker invoker = client2.getInvoker();
+ Map localServerLocators = (Map) field.get(invoker);
+ assertEquals(1, localServerLocators.size());
+ client2.removeListener(callbackHandler2);
+ assertEquals(0, localServerLocators.size());
+ client2.disconnect();
+ }
+ teardownServer(connector);
+ }
+ }
+
+ public void testPush2Clients2Connectors1Handler() throws Throwable
+ {
+ log.info("entering " + getName());
+ CallbackTestClient.CallbackHandler callbackHandler = new CallbackTestClient.CallbackHandler();
+ InvokerLocator locator = new InvokerLocator(locatorURI);
+ Client client1 = new Client(locator);
+ client1.connect();
+ Client client2 = new Client(locator);
+ client2.connect();
+ Connector connector1 = null;
+ Connector connector2 = null;
+
+ try
+ {
+ String callbackLocatorURI1 = transport + "://" + host + ":" + nextPort++;
+ InvokerLocator callbackLocator1 = new InvokerLocator(callbackLocatorURI1);
+ connector1 = setupServer(callbackLocator1);
+ String callbackLocatorURI2 = transport + "://" + host + ":" + nextPort++;
+ InvokerLocator callbackLocator2 = new InvokerLocator(callbackLocatorURI2);
+ connector2 = setupServer(callbackLocator2);
+ client1.addListener(callbackHandler, callbackLocator1);
+ client2.addListener(callbackHandler, callbackLocator2);
+ client1.invoke("Do something", null);
+ client2.invoke("Do something else", null);
+
+ // need to wait for brief moment so server can callback
+ Thread.sleep(5000);
+
+ int count = callbackHandler.getCallbackCount();
+ assertEquals(4, count);
+ }
+ finally
+ {
+ Field field = AbstractInvoker.class.getDeclaredField("localServerLocators");
+ field.setAccessible(true);
+
+ if(client1 != null)
+ {
+ ClientInvoker invoker = client1.getInvoker();
+ Map localServerLocators = (Map) field.get(invoker);
+ assertEquals(2, localServerLocators.size());
+ client1.removeListener(callbackHandler);
+ // callbackHandler should still be registered with connector2.
+ assertEquals(1, localServerLocators.size());
+ client1.disconnect();
+ }
+ if (client2 != null)
+ {
+ ClientInvoker invoker = client2.getInvoker();
+ Map localServerLocators = (Map) field.get(invoker);
+ assertEquals(1, localServerLocators.size());
+ client2.removeListener(callbackHandler);
+ assertEquals(0, localServerLocators.size());
+ client2.disconnect();
+ }
+ teardownServer(connector1);
+ teardownServer(connector2);
+ }
+ }
+
+ public void testPush2Clients2Connectors2Handlers() throws Throwable
+ {
+ log.info("entering " + getName());
+ CallbackTestClient.CallbackHandler callbackHandler1 = new CallbackTestClient.CallbackHandler();
+ CallbackTestClient.CallbackHandler callbackHandler2 = new CallbackTestClient.CallbackHandler();
+ InvokerLocator locator = new InvokerLocator(locatorURI);
+ Client client1 = new Client(locator);
+ client1.connect();
+ Client client2 = new Client(locator);
+ client2.connect();
+ Connector connector1 = null;
+ Connector connector2 = null;
+
+ try
+ {
+ String callbackLocatorURI1 = transport + "://" + host + ":" + nextPort++;
+ InvokerLocator callbackLocator1 = new InvokerLocator(callbackLocatorURI1);
+ connector1 = setupServer(callbackLocator1);
+ String callbackLocatorURI2 = transport + "://" + host + ":" + nextPort++;
+ InvokerLocator callbackLocator2 = new InvokerLocator(callbackLocatorURI2);
+ connector2 = setupServer(callbackLocator2);
+ client1.addListener(callbackHandler1, callbackLocator1);
+ client2.addListener(callbackHandler2, callbackLocator2);
+ client1.invoke("Do something", null);
+ client2.invoke("Do something else", null);
+
+ // need to wait for brief moment so server can callback
+ Thread.sleep(5000);
+
+ int count = callbackHandler1.getCallbackCount();
+ assertEquals(2, count);
+ count = callbackHandler1.getCallbackCount();
+ assertEquals(2, count);
+ }
+ finally
+ {
+ Field field = AbstractInvoker.class.getDeclaredField("localServerLocators");
+ field.setAccessible(true);
+
+ if(client1 != null)
+ {
+ ClientInvoker invoker = client1.getInvoker();
+ Map localServerLocators = (Map) field.get(invoker);
+ assertEquals(2, localServerLocators.size());
+ client1.removeListener(callbackHandler1);
+ assertEquals(1, localServerLocators.size());
+ client1.removeListener(callbackHandler2);
+ // client1 shouldn't be able to remove callbackHandler2.
+ assertEquals(1, localServerLocators.size());
+ client1.disconnect();
+ }
+ if (client2 != null)
+ {
+ ClientInvoker invoker = client2.getInvoker();
+ Map localServerLocators = (Map) field.get(invoker);
+ assertEquals(1, localServerLocators.size());
+ client2.removeListener(callbackHandler2);
+ assertEquals(0, localServerLocators.size());
+ client2.disconnect();
+ }
+ teardownServer(connector1);
+ teardownServer(connector2);
+ }
+ }
+
+
+ public Connector setupServer(InvokerLocator locator) throws Exception
+ {
+ log.info("Starting remoting server with locator uri of: " + locator);
+ try
+ {
+ Connector connector = new Connector();
+ connector.setInvokerLocator(locator.getLocatorURI());
+ connector.start();
+ log.info("started callback Connector: " + connector.getLocator());
+
+ CallbackTestClient.SampleInvocationHandler invocationHandler = new CallbackTestClient.SampleInvocationHandler();
+ // first parameter is sub-system name. can be any String value.
+ connector.addInvocationHandler("sample", invocationHandler);
+ return connector;
+ }
+ catch(Exception e)
+ {
+ log.error("Error starting callback server", e);
+ throw e;
+ }
+ }
+
+ public void teardownServer(Connector connector)
+ {
+ if(connector != null)
+ {
+ connector.stop();
+ connector.destroy();
+ }
+ }
+
+ /**
+ * Simple invocation handler implementation.
+ */
+ public static class SampleInvocationHandler implements ServerInvocationHandler
+ {
+
+ private List listeners = new ArrayList();
+
+
+ /**
+ * called to handle a specific invocation
+ *
+ * @param invocation
+ * @return
+ * @throws Throwable
+ */
+ public Object invoke(InvocationRequest invocation) throws Throwable
+ {
+ // Just going to return static string as this is just simple example code.
+
+ // Will also fire callback to listeners if they were to exist using
+ // simple invocation request.
+ Callback callback = new Callback("This is the payload of callback invocation.");
+ Iterator itr = listeners.iterator();
+ while(itr.hasNext())
+ {
+ InvokerCallbackHandler callbackHandler = (InvokerCallbackHandler) itr.next();
+ try
+ {
+ callbackHandler.handleCallback(callback);
+ }
+ catch(HandleCallbackException e)
+ {
+ e.printStackTrace();
+ }
+ }
+
+ return RESPONSE_VALUE;
+
+ }
+
+ /**
+ * Adds a callback handler that will listen for callbacks from
+ * the server invoker handler.
+ *
+ * @param callbackHandler
+ */
+ public void addListener(InvokerCallbackHandler callbackHandler)
+ {
+ listeners.add(callbackHandler);
+ }
+
+ /**
+ * Removes the callback handler that was listening for callbacks
+ * from the server invoker handler.
+ *
+ * @param callbackHandler
+ */
+ public void removeListener(InvokerCallbackHandler callbackHandler)
+ {
+ listeners.remove(callbackHandler);
+ }
+
+ /**
+ * set the mbean server that the handler can reference
+ *
+ * @param server
+ */
+ public void setMBeanServer(MBeanServer server)
+ {
+ // NO OP as do not need reference to MBeanServer for this handler
+ }
+
+ /**
+ * set the invoker that owns this handler
+ *
+ * @param invoker
+ */
+ public void setInvoker(ServerInvoker invoker)
+ {
+ // NO OP as do not need reference back to the server invoker
+ }
+
+ }
+
+
+ public class CallbackHandler implements InvokerCallbackHandler
+ {
+ private Callback callback;
+ private int callbackNumber = 0;
+
+ /**
+ * Will take the callback message and send back to client.
+ * If client locator is null, will store them till client polls to get them.
+ *
+ * @param callback
+ * @throws org.jboss.remoting.callback.HandleCallbackException
+ *
+ */
+ public void handleCallback(Callback callback) throws HandleCallbackException
+ {
+ callbackNumber++;
+ this.callback = callback;
+ System.out.println("Received callback value of: " + callback.getCallbackObject());
+ System.out.println("Received callback handle object of: " + callback.getCallbackHandleObject());
+ System.out.println("Received callback server invoker of: " + callback.getServerLocator());
+ }
+
+ public Callback getCallback()
+ {
+ return callback;
+ }
+
+ public int getCallbackCount()
+ {
+ return callbackNumber;
+ }
+ }
+
+}
1.1.2.2 +158 -0 JBossRemoting/src/tests/org/jboss/test/remoting/callback/multiple/CallbackTestServer.java
(In the diff below, changes in quantity of whitespace are not shown.)
Index: CallbackTestServer.java
===================================================================
RCS file: CallbackTestServer.java
diff -N CallbackTestServer.java
--- /dev/null 1 Jan 1970 00:00:00 -0000
+++ CallbackTestServer.java 7 Dec 2006 03:23:20 -0000 1.1.2.2
@@ -0,0 +1,158 @@
+package org.jboss.test.remoting.callback.multiple;
+
+import java.util.ArrayList;
+import java.util.Iterator;
+import java.util.List;
+
+import javax.management.MBeanServer;
+
+import org.jboss.jrunit.extensions.ServerTestCase;
+import org.jboss.remoting.InvocationRequest;
+import org.jboss.remoting.InvokerLocator;
+import org.jboss.remoting.ServerInvocationHandler;
+import org.jboss.remoting.ServerInvoker;
+import org.jboss.remoting.callback.Callback;
+import org.jboss.remoting.callback.InvokerCallbackHandler;
+import org.jboss.remoting.transport.Connector;
+
+/**
+ * Simple remoting server. Uses inner class SampleInvocationHandler
+ * as the invocation target handler class.
+ *
+ * @author <a href="mailto:telrod at e2technologies.net">Tom Elrod</a>
+ */
+public class CallbackTestServer extends ServerTestCase
+{
+ // Default locator values
+ private static String transport = "socket";
+ private static String host = "localhost";
+ private static int port = 5530;
+
+ private String locatorURI = transport + "://" + host + ":" + port;
+ private Connector connector;
+
+ // String to be returned from invocation handler upon client invocation calls.
+ public static final String RESPONSE_VALUE = "This is the return to SampleInvocationHandler invocation";
+ public static final String CALLBACK_VALUE = "This is the payload of callback invocation.";
+
+
+ public void setupServer() throws Exception
+ {
+ InvokerLocator locator = new InvokerLocator(locatorURI);
+ System.out.println("Starting remoting server with locator uri of: " + locatorURI);
+ connector = new Connector();
+ connector.setInvokerLocator(locator.getLocatorURI());
+ connector.start();
+
+ CallbackTestServer.SampleInvocationHandler invocationHandler = new CallbackTestServer.SampleInvocationHandler();
+ // first parameter is sub-system name. can be any String value.
+ connector.addInvocationHandler("sample", invocationHandler);
+ }
+
+ protected void setUp() throws Exception
+ {
+ setupServer();
+ }
+
+ protected void tearDown() throws Exception
+ {
+ if(connector != null)
+ {
+ connector.stop();
+ connector.destroy();
+ }
+ }
+
+ public static void main(String[] args)
+ {
+ CallbackTestServer server = new CallbackTestServer();
+ try
+ {
+ server.setUp();
+
+ Thread.sleep(600000);
+ }
+ catch (Exception e)
+ {
+ e.printStackTrace();
+ }
+ }
+
+ /**
+ * Simple invocation handler implementation.
+ */
+ public static class SampleInvocationHandler implements ServerInvocationHandler
+ {
+
+ private List listeners = new ArrayList();
+
+
+ /**
+ * called to handle a specific invocation
+ *
+ * @param invocation
+ * @return
+ * @throws Throwable
+ */
+ public Object invoke(InvocationRequest invocation) throws Throwable
+ {
+ // Just going to return static string as this is just simple example code.
+
+ // Will also fire callback to listeners if they were to exist using
+ // simple invocation request.
+ Callback callback = new Callback(CALLBACK_VALUE);
+ Iterator itr = listeners.iterator();
+ while(itr.hasNext())
+ {
+ InvokerCallbackHandler callbackHandler = (InvokerCallbackHandler) itr.next();
+ callbackHandler.handleCallback(callback);
+ }
+
+ return RESPONSE_VALUE;
+
+ }
+
+ /**
+ * Adds a callback handler that will listen for callbacks from
+ * the server invoker handler.
+ *
+ * @param callbackHandler
+ */
+ public void addListener(InvokerCallbackHandler callbackHandler)
+ {
+ listeners.add(callbackHandler);
+ }
+
+ /**
+ * Removes the callback handler that was listening for callbacks
+ * from the server invoker handler.
+ *
+ * @param callbackHandler
+ */
+ public void removeListener(InvokerCallbackHandler callbackHandler)
+ {
+ listeners.remove(callbackHandler);
+ }
+
+ /**
+ * set the mbean server that the handler can reference
+ *
+ * @param server
+ */
+ public void setMBeanServer(MBeanServer server)
+ {
+ // NO OP as do not need reference to MBeanServer for this handler
+ }
+
+ /**
+ * set the invoker that owns this handler
+ *
+ * @param invoker
+ */
+ public void setInvoker(ServerInvoker invoker)
+ {
+ // NO OP as do not need reference back to the server invoker
+ }
+
+ }
+}
More information about the jboss-cvs-commits
mailing list