JBoss Remoting SVN: r5945 - remoting2/branches/2.2/src/tests/org/jboss/test/remoting/lease.
by jboss-remoting-commits@lists.jboss.org
Author: ron.sigal(a)jboss.com
Date: 2010-07-31 09:44:54 -0400 (Sat, 31 Jul 2010)
New Revision: 5945
Added:
remoting2/branches/2.2/src/tests/org/jboss/test/remoting/lease/LeaseCreationTimeoutTestCase.java
Log:
JBREM-1238: New unit tests.
Added: remoting2/branches/2.2/src/tests/org/jboss/test/remoting/lease/LeaseCreationTimeoutTestCase.java
===================================================================
--- remoting2/branches/2.2/src/tests/org/jboss/test/remoting/lease/LeaseCreationTimeoutTestCase.java (rev 0)
+++ remoting2/branches/2.2/src/tests/org/jboss/test/remoting/lease/LeaseCreationTimeoutTestCase.java 2010-07-31 13:44:54 UTC (rev 5945)
@@ -0,0 +1,784 @@
+/*
+ * JBoss, Home of Professional Open Source
+ * Copyright 2009, JBoss Inc., 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.lease;
+
+import java.io.ByteArrayOutputStream;
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.OutputStream;
+import java.io.PrintStream;
+import java.net.InetAddress;
+import java.net.ServerSocket;
+import java.net.Socket;
+import java.util.HashMap;
+import java.util.Map;
+
+import javax.management.MBeanServer;
+import javax.net.ServerSocketFactory;
+
+import junit.framework.TestCase;
+
+import org.apache.log4j.ConsoleAppender;
+import org.apache.log4j.Level;
+import org.apache.log4j.PatternLayout;
+import org.jboss.logging.Logger;
+import org.jboss.logging.Log4jLoggerPlugin;
+import org.jboss.logging.XLevel;
+import org.jboss.remoting.CannotConnectException;
+import org.jboss.remoting.Client;
+import org.jboss.remoting.ConnectionListener;
+import org.jboss.remoting.InvocationRequest;
+import org.jboss.remoting.InvokerLocator;
+import org.jboss.remoting.Remoting;
+import org.jboss.remoting.ServerInvocationHandler;
+import org.jboss.remoting.ServerInvoker;
+import org.jboss.remoting.callback.InvokerCallbackHandler;
+import org.jboss.remoting.transport.Connector;
+import org.jboss.remoting.transport.PortUtil;
+
+
+/**
+ * Unit test for JBREM-1238.
+ *
+ * @author <a href="ron.sigal(a)jboss.com">Ron Sigal</a>
+ * @version
+ * <p>
+ * Copyright July 28, 2010
+ * </p>
+ */
+public class LeaseCreationTimeoutTestCase extends TestCase
+{
+ private static Logger log = Logger.getLogger(LeaseCreationTimeoutTestCase.class);
+
+ protected static int breakServer;
+ protected static int counter;
+
+ protected String host;
+ protected int port;
+ protected String locatorURI;
+ protected InvokerLocator serverLocator;
+ protected Connector connector;
+ protected TestInvocationHandler invocationHandler;
+ protected PrintStream originalPrintStream;
+ protected ByteArrayOutputStream baos;
+
+
+ public void setUp() throws Exception
+ {
+ baos = new ByteArrayOutputStream();
+ originalPrintStream = System.out;
+ PrintStream ps = new PrintStream(baos, true);
+ System.setOut(ps);
+
+ String pattern = "[%d{ABSOLUTE}] [%t] %5p (%F:%L) - %m%n";
+ PatternLayout layout = new PatternLayout(pattern);
+ ConsoleAppender consoleAppender = new ConsoleAppender(layout);
+ consoleAppender.setTarget(ConsoleAppender.SYSTEM_OUT);
+
+ Log4jLoggerPlugin loggerPlugin = (Log4jLoggerPlugin) Logger.getLogger("org.jboss.remoting").getLoggerPlugin();
+ org.apache.log4j.Logger logger = loggerPlugin.getLogger();
+ logger.setLevel(XLevel.TRACE);
+ logger.addAppender(consoleAppender);
+ loggerPlugin = (Log4jLoggerPlugin) Logger.getLogger("org.jboss.test.remoting").getLoggerPlugin();
+ logger = loggerPlugin.getLogger();
+ logger.setLevel(XLevel.TRACE);
+ logger.addAppender(consoleAppender);
+
+ breakServer = -1;
+ counter = 0;
+ }
+
+
+ public void tearDown()
+ {
+ }
+
+
+ /**
+ * Verifies that default value of leaseCreationTimeout is set properly.
+ */
+ public void testLeaseCreationTimeoutDefault() throws Throwable
+ {
+ log.info("entering " + getName());
+ Client client = null;
+
+ try
+ {
+ // Start server.
+ setupServer(-1);
+
+ // Create client.
+ InvokerLocator clientLocator = new InvokerLocator(locatorURI);
+ HashMap clientConfig = new HashMap();
+ clientConfig.put(InvokerLocator.FORCE_REMOTE, "true");
+ clientConfig.put(Client.ENABLE_LEASE, "true");
+ addExtraClientConfig(clientConfig);
+ client = new Client(clientLocator, clientConfig);
+ client.connect();
+ log.info("client is connected");
+
+ // Test connection.
+ assertEquals("abc", client.invoke("abc"));
+ log.info("connection is good");
+
+ String s = baos.toString();
+ assertTrue(s.indexOf("setting timeout to 30000 for this invocation") > -1);
+ assertTrue(s.indexOf("starting lease timer with ping period of") > -1);
+ log.info(getName() + " PASSES");
+ }
+ finally
+ {
+ client.disconnect();
+ shutdownServer();
+ originalPrintStream.flush();
+ System.setOut(originalPrintStream);
+ System.out.println("starting to write log");
+ String s = baos.toString();
+ System.out.println(s);
+ }
+ }
+
+
+ /**
+ * Verifies that value of leaseCreationTimeout is set properly from configuration map.
+ */
+ public void testLeaseCreationTimeoutConfig() throws Throwable
+ {
+ log.info("entering " + getName());
+ Client client = null;
+
+ try
+ {
+ // Start server.
+ setupServer(-1);
+
+ // Create client.
+ InvokerLocator clientLocator = new InvokerLocator(locatorURI);
+ HashMap clientConfig = new HashMap();
+ clientConfig.put(InvokerLocator.FORCE_REMOTE, "true");
+ clientConfig.put(Client.ENABLE_LEASE, "true");
+ clientConfig.put(Remoting.LEASE_CREATION_TIMEOUT, "12345");
+ clientConfig.put(ServerInvoker.TIMEOUT, "0");
+ addExtraClientConfig(clientConfig);
+ client = new Client(clientLocator, clientConfig);
+ client.connect();
+ log.info("client is connected");
+
+ // Test connection.
+ assertEquals("abc", client.invoke("abc"));
+ log.info("connection is good");
+
+ String s = baos.toString();
+ assertTrue(s.indexOf("setting timeout to 12345 for this invocation") > -1);
+ assertTrue(s.indexOf("starting lease timer with ping period of") > -1);
+ log.info(getName() + " PASSES");
+ }
+ finally
+ {
+ client.disconnect();
+ shutdownServer();
+ originalPrintStream.flush();
+ System.setOut(originalPrintStream);
+ System.out.println("starting to write log");
+ String s = baos.toString();
+ System.out.println(s);
+ }
+ }
+
+
+ /**
+ * Verifies that MicroRemotingClientInvoker.establishLease() is terminated according to
+ * the value of Remoting.LEASE_CREATION_TIMEOUT so that another new connection can
+ * set up a lease.
+ */
+ public void testLeaseCreationWithBrokenEstablishLease() throws Throwable
+ {
+ log.info("entering " + getName());
+ Client client1 = null;
+ Client client2 = null;
+
+ try
+ {
+ // Start server.
+ setupServer(1);
+
+ // Create client.
+ InvokerLocator clientLocator = new InvokerLocator(locatorURI);
+ HashMap clientConfig = new HashMap();
+ clientConfig.put(InvokerLocator.FORCE_REMOTE, "true");
+ clientConfig.put(ServerInvoker.TIMEOUT, "0");
+ clientConfig.put(Client.ENABLE_LEASE, "true");
+ clientConfig.put(Remoting.LEASE_CREATION_TIMEOUT, "5000");
+ addExtraClientConfig(clientConfig);
+ client1 = new Client(clientLocator, clientConfig);
+
+ // Should not connect.
+ BrokenEstablisthLeaseTestThread btt = new BrokenEstablisthLeaseTestThread(client1);
+ btt.run();
+
+ // Should connect.
+ Thread.sleep(10000);
+ client2 = new Client(clientLocator, clientConfig);
+ WorkingTestThread ttt = new WorkingTestThread(client2);
+ ttt.run();
+
+ // Test lease creation.
+ Thread.sleep(5000);
+ assertTrue(btt.ok);
+ assertTrue(ttt.ok);
+
+ log.info(getName() + " PASSES");
+ }
+ finally
+ {
+ client1.disconnect();
+ client2.disconnect();
+ shutdownServer();
+ originalPrintStream.flush();
+ System.setOut(originalPrintStream);
+ System.out.println("starting to write log");
+ String s = baos.toString();
+ System.out.println(s);
+ }
+ }
+
+
+ /**
+ * Verifies that MicroRemotingClientInvoker.terminateLease() is terminated according to
+ * the value of Remoting.LEASE_CREATION_TIMEOUT so that another new connection can
+ * set up a lease. In this test the hang up is in LeasePinger.removeClient().
+ */
+ public void testLeaseTerminationWithBrokenRemoveClient() throws Throwable
+ {
+ log.info("entering " + getName());
+ Client client1 = null;
+ Client client2 = null;
+
+ try
+ {
+ // Start server.
+ setupServer(4);
+
+ // Create client.
+ InvokerLocator clientLocator = new InvokerLocator(locatorURI);
+ HashMap clientConfig = new HashMap();
+ clientConfig.put(InvokerLocator.FORCE_REMOTE, "true");
+ clientConfig.put(ServerInvoker.TIMEOUT, "0");
+ clientConfig.put(Client.ENABLE_LEASE, "true");
+ clientConfig.put(Remoting.LEASE_CREATION_TIMEOUT, "5000");
+ addExtraClientConfig(clientConfig);
+ client1 = new Client(clientLocator, clientConfig);
+
+ // Should not terminate properly in call to LeasePinger.removeClient().
+ BrokenTerminateLeaseRemoveClientTestThread btt = new BrokenTerminateLeaseRemoveClientTestThread(client1, baos);
+ btt.start();
+
+ // Should connect.
+ Thread.sleep(10000);
+ client2 = new Client(clientLocator, clientConfig);
+ WorkingTestThread ttt = new WorkingTestThread(client2);
+ ttt.start();
+
+ // Test lease creation.
+ Thread.sleep(5000);
+ assertTrue(btt.ok);
+ assertTrue(ttt.ok);
+
+ log.info(getName() + " PASSES");
+ }
+ finally
+ {
+ shutdownServer();
+ originalPrintStream.flush();
+ System.setOut(originalPrintStream);
+ System.out.println("starting to write log");
+ String s = baos.toString();
+ System.out.println(s);
+ }
+ }
+
+
+ /**
+ * Verifies that MicroRemotingClientInvoker.terminateLease() is terminated according to
+ * the value of Remoting.LEASE_CREATION_TIMEOUT so that another new connection can
+ * set up a lease. In this test the hang up is in LeasePinger.stopPing().
+ */
+ public void testLeaseTerminationWithBrokenStopPing() throws Throwable
+ {
+ log.info("entering " + getName());
+
+ try
+ {
+ // Start server.
+ setupServer(5);
+
+ // Create client.
+ InvokerLocator clientLocator = new InvokerLocator(locatorURI);
+ HashMap clientConfig = new HashMap();
+ clientConfig.put(InvokerLocator.FORCE_REMOTE, "true");
+ clientConfig.put(ServerInvoker.TIMEOUT, "0");
+ clientConfig.put(Client.ENABLE_LEASE, "true");
+ clientConfig.put(Remoting.LEASE_CREATION_TIMEOUT, "5000");
+ addExtraClientConfig(clientConfig);
+ Client client1 = new Client(clientLocator, clientConfig);
+
+ // Should not terminate properly in call to LeasePinger.stopPing().
+ BrokenTerminateLeaseStopPingTestThread btt = new BrokenTerminateLeaseStopPingTestThread(client1, baos);
+ btt.start();
+
+ // Should connect.
+ Thread.sleep(10000);
+ Client client2 = new Client(clientLocator, clientConfig);
+ WorkingTestThread ttt = new WorkingTestThread(client2);
+ ttt.start();
+
+ // Test lease creation.
+ Thread.sleep(5000);
+ assertTrue(btt.ok);
+ assertTrue(ttt.ok);
+
+ shutdownServer();
+ log.info(getName() + " PASSES");
+
+ }
+ finally
+ {
+ originalPrintStream.flush();
+ System.setOut(originalPrintStream);
+ System.out.println("starting to write log");
+ String s = baos.toString();
+ System.out.println(s);
+ }
+ }
+
+
+ protected String getTransport()
+ {
+ return "socket";
+ }
+
+
+ protected void addExtraClientConfig(Map config) {}
+ protected void addExtraServerConfig(Map config) {}
+
+
+ protected void setupServer(int breakServer) throws Exception
+ {
+ host = InetAddress.getLocalHost().getHostAddress();
+ port = PortUtil.findFreePort(host);
+ locatorURI = getTransport() + "://" + host + ":" + port;
+ String metadata = System.getProperty("remoting.metadata");
+ if (metadata != null)
+ {
+ locatorURI += "/?" + metadata;
+ }
+ serverLocator = new InvokerLocator(locatorURI);
+ log.info("Starting remoting server with locator uri of: " + locatorURI);
+ HashMap config = new HashMap();
+ config.put(InvokerLocator.FORCE_REMOTE, "true");
+ config.put(Remoting.USE_CLIENT_CONNECTION_IDENTITY, "true");
+ config.put("clientLeasePeriod", "10000");
+ if (breakServer > -1)
+ {
+ LeaseCreationTimeoutTestCase.breakServer = breakServer;
+ config.put(Remoting.CUSTOM_SERVER_SOCKET_FACTORY, new TestServerSocketFactory());
+ }
+ addExtraServerConfig(config);
+ connector = new Connector(serverLocator, config);
+ connector.create();
+ invocationHandler = new TestInvocationHandler();
+ connector.addInvocationHandler("test", invocationHandler);
+ connector.addConnectionListener(new TestConnectionListener());
+ connector.start();
+ }
+
+
+ protected void shutdownServer() throws Exception
+ {
+ if (connector != null)
+ connector.stop();
+ }
+
+
+ static class TestInvocationHandler implements ServerInvocationHandler
+ {
+ public void addListener(InvokerCallbackHandler callbackHandler) {}
+ public Object invoke(final InvocationRequest invocation) throws Throwable
+ {
+ return invocation.getParameter();
+ }
+ public void removeListener(InvokerCallbackHandler callbackHandler) {}
+ public void setMBeanServer(MBeanServer server) {}
+ public void setInvoker(ServerInvoker invoker) {}
+ }
+
+
+ static class TestConnectionListener implements ConnectionListener
+ {
+ public void handleConnectionException(Throwable throwable, Client client)
+ {
+ }
+ }
+
+
+ static public class TestServerSocketFactory extends ServerSocketFactory
+ {
+ public TestServerSocketFactory()
+ {
+ }
+ public ServerSocket createServerSocket() throws IOException
+ {
+ ServerSocket ss = new TestServerSocket();
+ log.info(this + " returning: " + ss);
+ return ss;
+ }
+ public ServerSocket createServerSocket(int port) throws IOException
+ {
+ ServerSocket ss = new TestServerSocket(port);
+ log.info(this + " returning: " + ss);
+ return ss;
+ }
+
+ public ServerSocket createServerSocket(int port, int backlog) throws IOException
+ {
+ ServerSocket ss = new TestServerSocket(port, backlog);
+ log.info(this + " returning: " + ss);
+ return ss;
+ }
+
+ public ServerSocket createServerSocket(int port, int backlog, InetAddress ifAddress) throws IOException
+ {
+ ServerSocket ss = new TestServerSocket(port, backlog, ifAddress);
+ log.info(this + " returning: " + ss);
+ return ss;
+ }
+ }
+
+
+ static class TestServerSocket extends ServerSocket
+ {
+ public TestServerSocket() throws IOException
+ {
+ super();
+ }
+ public TestServerSocket(int port) throws IOException
+ {
+ super(port);
+ }
+ public TestServerSocket(int port, int backlog) throws IOException
+ {
+ super(port, backlog);
+ }
+ public TestServerSocket(int port, int backlog, InetAddress bindAddr) throws IOException
+ {
+ super(port, backlog, bindAddr);
+ }
+ public Socket accept() throws IOException
+ {
+ Socket s = super.accept();
+ TestSocket ts = new TestSocket(s);
+ log.info(this + " returning: " + ts);
+ return ts;
+ }
+ public String toString()
+ {
+ return "TestServerSocket[" + getLocalPort() + "]";
+ }
+ }
+
+
+ static class TestSocket extends Socket
+ {
+ Socket s;
+
+ public TestSocket(Socket s)
+ {
+ this.s = s;
+ }
+ public void close() throws IOException
+ {
+ s.close();
+ }
+ public InputStream getInputStream() throws IOException
+ {
+ return s.getInputStream();
+ }
+ public OutputStream getOutputStream() throws IOException
+ {
+ return new TestOutputStream(s.getOutputStream());
+ }
+ public InetAddress getInetAddress()
+ {
+ return s.getInetAddress();
+ }
+ public String toString()
+ {
+ return "TestSocket[" + s.getLocalPort() + "->" + s.getPort() + "]";
+ }
+ }
+
+
+ static class TestOutputStream extends OutputStream
+ {
+ OutputStream os;
+ boolean closed;
+ boolean doThrow = true;
+
+ public TestOutputStream(OutputStream os)
+ {
+ this.os = os;
+ }
+ public void close()throws IOException
+ {
+ closed = true;
+ super.close();
+ log.info(this + " closed");
+ }
+ public void write(int b) throws IOException
+ {
+ System.out.print("b: " + b);
+ if (closed)
+ {
+ log.info("TestOutputStream closed, cannot write");
+ throw new IOException("closed");
+ }
+ if (doThrow && counter++ == breakServer)
+ {
+ log.info("sleeping");
+ while (true)
+ {
+ try
+ {
+ Thread.sleep(30000);
+ }
+ catch (InterruptedException e)
+ {
+ }
+ }
+ }
+ os.write(b);
+ }
+ public void write(byte b[], int off, int len) throws IOException
+ {
+ System.out.print("b: ");
+ for (int i = 0; i < len; i++)
+ {
+ System.out.print(b[i] + " ");
+ }
+ System.out.println("");
+ if (closed)
+ {
+ log.info("TestOutputStream closed, cannot write");
+ throw new IOException("closed");
+ }
+ log.info("TestOutputStream: counter = " + counter + ", broken = " + breakServer);
+ if (counter++ == breakServer)
+ {
+ log.info(this + " sleeping");
+ try
+ {
+ Thread.sleep(20000);
+ }
+ catch (InterruptedException e)
+ {
+ }
+ log.info(this + " woke up");
+ }
+ if (closed)
+ {
+ log.info("TestOutputStream closed, cannot write");
+ throw new IOException("closed");
+ }
+ try
+ {
+ log.info(this + " calling write()");
+ doThrow = false;
+ os.write(b, off, len);
+ os.flush();
+ doThrow = true;
+ log.info(this + " back from write()");
+ }
+ catch (IOException e)
+ {
+ log.info("exception: ", e);
+ throw e;
+ }
+ }
+ public void flush() throws IOException
+ {
+ os.flush();
+ }
+ }
+
+
+ static class BrokenEstablisthLeaseTestThread extends Thread
+ {
+ public boolean ok;
+ Client client;
+
+ public BrokenEstablisthLeaseTestThread(Client client)
+ {
+ this.client = client;
+ setName("BrokenEstablisthLeaseTestThread[" + client + "]");
+ }
+ public void run()
+ {
+ try
+ {
+ log.info(this + " STARTING");
+ log.info(this + " trying to connect " + client);
+ client.connect();
+ }
+ catch (CannotConnectException e)
+ {
+ if ("Error setting up client lease upon performing connect.".equals(e.getMessage()))
+ {
+ ok = true;
+ log.info(this + " got expected exception");
+ }
+ else
+ {
+ log.info(this + " got expected exception with unexpected message: " + e.getMessage());
+ }
+ }
+ catch (Exception e)
+ {
+ log.error(this + " got unexpected exception: " + e);
+ }
+ }
+ public String toString()
+ {
+ return getName();
+ }
+ }
+
+
+ static class BrokenTerminateLeaseRemoveClientTestThread extends Thread
+ {
+ public boolean ok;
+ Client client;
+ ByteArrayOutputStream baos;
+
+ public BrokenTerminateLeaseRemoveClientTestThread(Client client, ByteArrayOutputStream baos)
+ {
+ this.client = client;
+ this.baos = baos;
+ setName("BrokenTerminateLeaseRemoveClientTestThread[" + client + "]");
+ }
+ public void run()
+ {
+ try
+ {
+ log.info(this + " STARTING");
+ log.info(this + " trying to connect " + client);
+ client.connect();
+ assertEquals("def", client.invoke("def"));
+ client.disconnect();
+ String s = baos.toString();
+ assertTrue(s.indexOf("Terminating individual lease for session id") > -1);
+ assertTrue(s.indexOf("failed sending disconnect for client lease") > -1);
+ ok = true;
+ }
+ catch (Throwable t)
+ {
+ log.error(this + " got unexpected exception: " + t);
+ }
+ }
+ public String toString()
+ {
+ return getName();
+ }
+ }
+
+
+ static class BrokenTerminateLeaseStopPingTestThread extends Thread
+ {
+ public boolean ok;
+ Client client;
+ ByteArrayOutputStream baos;
+
+ public BrokenTerminateLeaseStopPingTestThread(Client client, ByteArrayOutputStream baos)
+ {
+ this.client = client;
+ this.baos = baos;
+ setName("BrokenTerminateLeaseStopPingTestThread[" + client + "]");
+ }
+ public void run()
+ {
+ try
+ {
+ log.info(this + " STARTING");
+ log.info(this + " trying to connect " + client);
+ client.connect();
+ assertEquals("def", client.invoke("def"));
+ client.disconnect();
+ String s = baos.toString();
+ assertTrue(s.indexOf("Terminating individual lease for session id") > -1);
+ assertFalse(s.indexOf("failed sending disconnect for client lease") > -1);
+ assertTrue(s.indexOf(": disconnectTimeout: 5000") > -1);
+ assertTrue(s.indexOf("error shutting down lease pinger") > -1);
+ ok = true;
+ }
+ catch (Throwable t)
+ {
+ log.error(this + " got unexpected exception: " + t);
+ }
+ }
+ public String toString()
+ {
+ return getName();
+ }
+ }
+
+
+ static class WorkingTestThread extends Thread
+ {
+ public boolean ok;
+ Client client;
+
+ public WorkingTestThread(Client client)
+ {
+ this.client = client;
+ setName("WorkingTestThread[" + client + "]");
+ }
+ public void run()
+ {
+ log.info(this + " STARTING");
+ try
+ {
+ log.info(this + " trying to connect " + client);
+ client.connect();
+ assertEquals("abc", client.invoke("abc"));
+ log.info(this + " connection is good");
+ ok = true;
+ }
+ catch (Throwable t)
+ {
+ log.error(this + " got unexpected exception: " + t);
+ }
+ }
+ public String toString()
+ {
+ return getName();
+ }
+ }
+}
\ No newline at end of file
14 years, 3 months
JBoss Remoting SVN: r5944 - remoting2/branches/2.2/src/main/org/jboss/remoting.
by jboss-remoting-commits@lists.jboss.org
Author: ron.sigal(a)jboss.com
Date: 2010-07-31 09:44:08 -0400 (Sat, 31 Jul 2010)
New Revision: 5944
Modified:
remoting2/branches/2.2/src/main/org/jboss/remoting/Lease.java
Log:
JBREM-1238: Fixed mangled log statement.
Modified: remoting2/branches/2.2/src/main/org/jboss/remoting/Lease.java
===================================================================
--- remoting2/branches/2.2/src/main/org/jboss/remoting/Lease.java 2010-07-31 13:41:57 UTC (rev 5943)
+++ remoting2/branches/2.2/src/main/org/jboss/remoting/Lease.java 2010-07-31 13:44:08 UTC (rev 5944)
@@ -291,7 +291,7 @@
notifier.connectionTerminated(locatorURL, clientHolder.getSessionId(), clientHolder.getConfig());
if(isTraceEnabled)
{
- log.trace(this + " Notified connection lif (isTraceEnabled) log.tracef lease termination due to disconnect from client (client session id = " + clientHolder.getSessionId());
+ log.trace(this + " Notified connection listener of lease termination due to disconnect from client (client session id = " + clientHolder.getSessionId());
}
}
}
14 years, 3 months
JBoss Remoting SVN: r5943 - remoting2/branches/2.2/src/main/org/jboss/remoting.
by jboss-remoting-commits@lists.jboss.org
Author: ron.sigal(a)jboss.com
Date: 2010-07-31 09:41:57 -0400 (Sat, 31 Jul 2010)
New Revision: 5943
Modified:
remoting2/branches/2.2/src/main/org/jboss/remoting/MicroRemoteClientInvoker.java
Log:
JBREM-1238: Added use of leaseCreationTimeout.
Modified: remoting2/branches/2.2/src/main/org/jboss/remoting/MicroRemoteClientInvoker.java
===================================================================
--- remoting2/branches/2.2/src/main/org/jboss/remoting/MicroRemoteClientInvoker.java 2010-07-31 13:41:18 UTC (rev 5942)
+++ remoting2/branches/2.2/src/main/org/jboss/remoting/MicroRemoteClientInvoker.java 2010-07-31 13:41:57 UTC (rev 5943)
@@ -32,6 +32,7 @@
public abstract class MicroRemoteClientInvoker extends AbstractInvoker implements ClientInvoker
{
private static final Logger log = Logger.getLogger(MicroRemoteClientInvoker.class);
+ private static final String LEASE_CREATION_TIMEOUT_DEFAULT = "30000";
private boolean trace = log.isTraceEnabled();
protected boolean connected = false;
@@ -44,6 +45,7 @@
private boolean parentFirstClassLoading = true;
private boolean changeInvalidStateToCannotConnect = false;
private boolean useCurrentThreadClassLoader;
+ private String leaseCreationTimeout = LEASE_CREATION_TIMEOUT_DEFAULT;
public MicroRemoteClientInvoker(InvokerLocator locator)
{
@@ -343,6 +345,16 @@
return this.invokerSessionID;
}
+ public String getLeaseCreationTimeout()
+ {
+ return leaseCreationTimeout;
+ }
+
+ public void setLeaseCreationTimeout(String leaseCreationTimeout)
+ {
+ this.leaseCreationTimeout = leaseCreationTimeout;
+ }
+
public void terminateLease(String sessionId, int disconnectTimeout)
{
terminateLease(sessionId, disconnectTimeout, null);
@@ -360,6 +372,10 @@
if (trace) log.trace(this + " entering terminateLease() for " + leasePinger);
if(leasePinger != null)
{
+ if (disconnectTimeout == Client.DEFAULT_DISCONNECT_TIMEOUT)
+ {
+ disconnectTimeout = Integer.parseInt(leaseCreationTimeout);
+ }
leasePinger.setDisconnectTimeout(disconnectTimeout);
if (sessionId == null)
@@ -468,6 +484,7 @@
Map requestMap = new HashMap();
requestMap.put(LeasePinger.LEASE_PINGER_ID, leasePingerId);
requestMap.put(LeasePinger.TIME_STAMP, Long.toString(System.currentTimeMillis()));
+ requestMap.put(ServerInvoker.TIMEOUT, leaseCreationTimeout);
if (trace) log.trace(this + " initiating lease for leasePingerId " + leasePingerId);
InvocationRequest ir = new InvocationRequest(invokerSessionID, null, "$PING$", requestMap, new HashMap(), null);
14 years, 3 months
JBoss Remoting SVN: r5942 - remoting2/branches/2.2/src/main/org/jboss/remoting.
by jboss-remoting-commits@lists.jboss.org
Author: ron.sigal(a)jboss.com
Date: 2010-07-31 09:41:18 -0400 (Sat, 31 Jul 2010)
New Revision: 5942
Modified:
remoting2/branches/2.2/src/main/org/jboss/remoting/Remoting.java
Log:
JBREM-1238: Added LEASE_CREATION_TIMEOUT.
Modified: remoting2/branches/2.2/src/main/org/jboss/remoting/Remoting.java
===================================================================
--- remoting2/branches/2.2/src/main/org/jboss/remoting/Remoting.java 2010-07-31 13:10:48 UTC (rev 5941)
+++ remoting2/branches/2.2/src/main/org/jboss/remoting/Remoting.java 2010-07-31 13:41:18 UTC (rev 5942)
@@ -106,4 +106,9 @@
* will be taken only from the InvokerLocator.
*/
public static final String PASS_CONFIG_MAP_TO_MARSHAL_FACTORY = "passConfigMapToMarshalFactory";
+
+ /**
+ * A flag for determining the timeout to use for the lease creation invocation.
+ */
+ public static final String LEASE_CREATION_TIMEOUT= "leaseCreationTimeout";
}
14 years, 3 months
JBoss Remoting SVN: r5941 - remoting2/branches/2.x.
by jboss-remoting-commits@lists.jboss.org
Author: ron.sigal(a)jboss.com
Date: 2010-07-31 09:10:48 -0400 (Sat, 31 Jul 2010)
New Revision: 5941
Modified:
remoting2/branches/2.x/build.xml
Log:
JBREM-1238: Excluded LeaseCreationTimeoutTestCase when jdk is 1.4.
Modified: remoting2/branches/2.x/build.xml
===================================================================
--- remoting2/branches/2.x/build.xml 2010-07-30 19:37:15 UTC (rev 5940)
+++ remoting2/branches/2.x/build.xml 2010-07-31 13:10:48 UTC (rev 5941)
@@ -529,6 +529,7 @@
<exclude name="org/jboss/test/remoting/**/http/**" if="isJDK4"/>
<exclude name="org/jboss/test/remoting/**/https/**" if="isJDK4"/>
<exclude name="org/jboss/test/remoting/**/servlet/**" if="isJDK4"/>
+ <exclude name="org/jboss/test/remoting/lease/LeaseCreationTimeoutTestCase*" if="isJDK4"/>
<classpath refid="tests.classpath"/>
</javac>
14 years, 3 months
JBoss Remoting SVN: r5940 - remoting2/branches/2.2/src/tests/org/jboss/test/remoting/transport/socket.
by jboss-remoting-commits@lists.jboss.org
Author: ron.sigal(a)jboss.com
Date: 2010-07-30 15:37:15 -0400 (Fri, 30 Jul 2010)
New Revision: 5940
Added:
remoting2/branches/2.2/src/tests/org/jboss/test/remoting/transport/socket/IdleTimerTaskNotifyTestCase.java
Log:
JBREM-1230: New unit test.
Added: remoting2/branches/2.2/src/tests/org/jboss/test/remoting/transport/socket/IdleTimerTaskNotifyTestCase.java
===================================================================
--- remoting2/branches/2.2/src/tests/org/jboss/test/remoting/transport/socket/IdleTimerTaskNotifyTestCase.java (rev 0)
+++ remoting2/branches/2.2/src/tests/org/jboss/test/remoting/transport/socket/IdleTimerTaskNotifyTestCase.java 2010-07-30 19:37:15 UTC (rev 5940)
@@ -0,0 +1,260 @@
+/*
+ * JBoss, Home of Professional Open Source.
+ * Copyright 2009, Red Hat Middleware LLC, and individual contributors
+ * as indicated by the @author tags. See the copyright.txt file 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.transport.socket;
+
+import java.net.InetAddress;
+import java.util.HashMap;
+import java.util.Map;
+
+import javax.management.MBeanServer;
+
+import junit.framework.TestCase;
+
+import org.apache.log4j.ConsoleAppender;
+import org.apache.log4j.Logger;
+import org.apache.log4j.PatternLayout;
+import org.jboss.logging.XLevel;
+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.InvokerCallbackHandler;
+import org.jboss.remoting.transport.Connector;
+import org.jboss.remoting.transport.PortUtil;
+import org.jboss.remoting.transport.socket.SocketServerInvoker;
+
+/**
+ * Unit test for JBREM-1230.
+ *
+ * @author <a href="ron.sigal(a)jboss.com">Ron Sigal</a>
+ * @version $Revision: 1.1 $
+ * <p>
+ * Copyright Jul 30, 2010
+ */
+public class IdleTimerTaskNotifyTestCase extends TestCase
+{
+ private static Logger log = Logger.getLogger(IdleTimerTaskNotifyTestCase.class);
+
+ private static boolean firstTime = true;
+ private static String DELAY = "delay";
+ private static String MAX_POOL_SIZE = "2";
+ private static String IDLE_TIMEOUT = "10";
+
+ protected String host;
+ protected int port;
+ protected String locatorURI;
+ protected InvokerLocator serverLocator;
+ protected Connector connector;
+ protected TestInvocationHandler invocationHandler;
+
+
+ public void setUp() throws Exception
+ {
+ if (firstTime)
+ {
+ firstTime = false;
+ Logger.getLogger("org.jboss.remoting").setLevel(XLevel.TRACE);
+ Logger.getLogger("org.jboss.test.remoting").setLevel(XLevel.INFO);
+ String pattern = "[%d{ABSOLUTE}] [%t] %5p (%F:%L) - %m%n";
+ PatternLayout layout = new PatternLayout(pattern);
+ ConsoleAppender consoleAppender = new ConsoleAppender(layout);
+ Logger.getRootLogger().addAppender(consoleAppender);
+ }
+ }
+
+
+ public void tearDown()
+ {
+ }
+
+
+ public void testIdleTimerTaskNotify() throws Throwable
+ {
+ log.info("entering " + getName());
+
+ // Start server and verify key parameters.
+ setupServer();
+ SocketServerInvoker serverInvoker = (SocketServerInvoker) connector.getServerInvoker();
+ assertEquals(MAX_POOL_SIZE, Integer.toString(serverInvoker.getMaxPoolSize()));
+ assertEquals(IDLE_TIMEOUT, Integer.toString(serverInvoker.getIdleTimeout()));
+
+ // Set up invocations.
+ InvokerLocator clientLocator = new InvokerLocator(locatorURI);
+ HashMap clientConfig = new HashMap();
+ clientConfig.put(InvokerLocator.FORCE_REMOTE, "true");
+ addExtraClientConfig(clientConfig);
+
+ TestThread tt1 = new TestThread("1", new Integer(25000), clientLocator, clientConfig);
+ TestThread tt2 = new TestThread("2", new Integer(5000), clientLocator, clientConfig);
+ TestThread tt3 = new TestThread("3", new Integer(0), clientLocator, clientConfig);
+ tt1.start();
+ Thread.sleep(1000);
+ tt2.start();
+ Thread.sleep(1000);
+ tt3.start();
+ Thread.sleep(10000);
+
+ // At this point, SocketServerInvoker has created a socket for the invocation from tt3,
+ // has called clientpool.evict(), and is waiting on WorkerThread#0 to shut down.
+ // * WorkerThread#0 is still running the invocation from tt1.
+ // * WorkerThread#1 has completed the invocation from tt2 and is waiting for another one.
+ // * The invocation from tt2 is still waiting for a thread.
+ assertFalse(tt1.ok);
+ assertTrue(tt2.ok);
+ assertFalse(tt3.ok);
+ Thread.sleep(15000);
+
+ // At this point, WorkerThread#0 is still executing the invocation from tt1, but IdleTimerTask
+ // has detected that WorkerThread#1 has been idle. It shuts down WorkerThread#1 and calls
+ // clientpool.notify(). SocketServerInvoker wakes up and creates WorkerThread#2, which processes
+ // the invocation from tt3.
+ assertTrue(tt3.ok);
+ Thread.sleep(5000);
+
+ // At this point, the invocation from tt1 has completed.
+ assertTrue(tt1.ok);
+
+ shutdownServer();
+ log.info(getName() + " PASSES");
+ }
+
+
+ protected String getTransport()
+ {
+ return "socket";
+ }
+
+
+ protected void addExtraClientConfig(Map config) {}
+ protected void addExtraServerConfig(Map config) {}
+
+
+ protected void setupServer() throws Exception
+ {
+ host = InetAddress.getLocalHost().getHostAddress();
+ port = PortUtil.findFreePort(host);
+ locatorURI = getTransport() + "://" + host + ":" + port + "/?";
+ String metadata = System.getProperty("remoting.metadata");
+ if (metadata != null)
+ {
+ locatorURI += "&" + metadata;
+ }
+ serverLocator = new InvokerLocator(locatorURI);
+ log.info("Starting remoting server with locator uri of: " + locatorURI);
+ HashMap config = new HashMap();
+ config.put(InvokerLocator.FORCE_REMOTE, "true");
+ config.put("maxPoolSize", MAX_POOL_SIZE);
+ config.put("idleTimeout", IDLE_TIMEOUT);
+ addExtraServerConfig(config);
+ connector = new Connector(serverLocator, config);
+ connector.create();
+ invocationHandler = new TestInvocationHandler();
+ connector.addInvocationHandler("test", invocationHandler);
+ connector.start();
+ }
+
+
+ protected void shutdownServer() throws Exception
+ {
+ if (connector != null)
+ connector.stop();
+ }
+
+
+ static class TestInvocationHandler implements ServerInvocationHandler
+ {
+ public void addListener(InvokerCallbackHandler callbackHandler) {}
+ public Object invoke(final InvocationRequest invocation) throws Throwable
+ {
+ Map metadata = invocation.getRequestPayload();
+ int delay = 5000;
+ if (metadata != null)
+ {
+ Object o = metadata.get(DELAY);
+ if (o != null && o instanceof Integer)
+ {
+ delay = ((Integer)o).intValue();
+ }
+ }
+ log.info(this + " sleeping for " + delay + " ms");
+ Thread.sleep(delay);
+ log.info(this + " returned from sleep");
+ return invocation.getParameter();
+ }
+ public void removeListener(InvokerCallbackHandler callbackHandler) {}
+ public void setMBeanServer(MBeanServer server) {}
+ public void setInvoker(ServerInvoker invoker) {}
+ }
+
+
+ static class TestThread extends Thread
+ {
+ public boolean ok;
+ String id;
+ Integer delay;
+ InvokerLocator locator;
+ Map config;
+
+ public TestThread(String id, Integer delay, InvokerLocator locator, Map config)
+ {
+ this.id = id;
+ this.delay = delay;
+ this.locator = locator;
+ this.config = config;
+ setName("TestThread[" + id + "]");
+ }
+
+ public void run()
+ {
+ Client client = null;
+
+ try
+ {
+ client = new Client(locator, config);
+ client.connect();
+ log.info(this + " client connected");
+ Map metadata = new HashMap();
+ metadata.put(DELAY, delay);
+ if (id.equals(client.invoke(id, metadata)))
+ {
+ ok = true;
+ log.info(this + " invocation successful");
+ }
+ }
+ catch (Throwable e)
+ {
+ log.error("invocation on " + client + " failed", e);
+ }
+ finally
+ {
+ client.disconnect();
+ }
+ }
+
+ public String toString()
+ {
+ return getName();
+ }
+ }
+}
\ No newline at end of file
Property changes on: remoting2/branches/2.2/src/tests/org/jboss/test/remoting/transport/socket/IdleTimerTaskNotifyTestCase.java
___________________________________________________________________
Name: svn:executable
+ *
14 years, 3 months
JBoss Remoting SVN: r5939 - remoting2/branches/2.2/src/main/org/jboss/remoting/transport/socket.
by jboss-remoting-commits@lists.jboss.org
Author: ron.sigal(a)jboss.com
Date: 2010-07-30 15:36:16 -0400 (Fri, 30 Jul 2010)
New Revision: 5939
Modified:
remoting2/branches/2.2/src/main/org/jboss/remoting/transport/socket/SocketServerInvoker.java
Log:
JBREM-1230: IdleTimerTask calls clientpool.notify() after shutting down a ServerThread.
Modified: remoting2/branches/2.2/src/main/org/jboss/remoting/transport/socket/SocketServerInvoker.java
===================================================================
--- remoting2/branches/2.2/src/main/org/jboss/remoting/transport/socket/SocketServerInvoker.java 2010-07-29 20:14:46 UTC (rev 5938)
+++ remoting2/branches/2.2/src/main/org/jboss/remoting/transport/socket/SocketServerInvoker.java 2010-07-30 19:36:16 UTC (rev 5939)
@@ -801,6 +801,7 @@
synchronized (clientpool)
{
clientpool.remove(svrThread);
+ clientpool.notify();
}
svrThread.shutdown();
}
14 years, 3 months
JBoss Remoting SVN: r5938 - remoting2/branches/2.x/src/tests/org/jboss/test/remoting/lease.
by jboss-remoting-commits@lists.jboss.org
Author: ron.sigal(a)jboss.com
Date: 2010-07-29 16:14:46 -0400 (Thu, 29 Jul 2010)
New Revision: 5938
Added:
remoting2/branches/2.x/src/tests/org/jboss/test/remoting/lease/LeaseCreationTimeoutTestCase.java
Log:
JBREM-1238: New unit tests.
Added: remoting2/branches/2.x/src/tests/org/jboss/test/remoting/lease/LeaseCreationTimeoutTestCase.java
===================================================================
--- remoting2/branches/2.x/src/tests/org/jboss/test/remoting/lease/LeaseCreationTimeoutTestCase.java (rev 0)
+++ remoting2/branches/2.x/src/tests/org/jboss/test/remoting/lease/LeaseCreationTimeoutTestCase.java 2010-07-29 20:14:46 UTC (rev 5938)
@@ -0,0 +1,783 @@
+/*
+ * JBoss, Home of Professional Open Source
+ * Copyright 2009, JBoss Inc., 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.lease;
+
+import java.io.ByteArrayOutputStream;
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.OutputStream;
+import java.io.PrintStream;
+import java.net.InetAddress;
+import java.net.ServerSocket;
+import java.net.Socket;
+import java.util.HashMap;
+import java.util.Map;
+
+import javax.management.MBeanServer;
+import javax.net.ServerSocketFactory;
+
+import junit.framework.TestCase;
+
+import org.apache.log4j.ConsoleAppender;
+import org.apache.log4j.Level;
+import org.apache.log4j.PatternLayout;
+import org.jboss.logging.Logger;
+import org.jboss.logging.log4j.Log4jLoggerPlugin;
+import org.jboss.remoting.CannotConnectException;
+import org.jboss.remoting.Client;
+import org.jboss.remoting.ConnectionListener;
+import org.jboss.remoting.InvocationRequest;
+import org.jboss.remoting.InvokerLocator;
+import org.jboss.remoting.Remoting;
+import org.jboss.remoting.ServerInvocationHandler;
+import org.jboss.remoting.ServerInvoker;
+import org.jboss.remoting.callback.InvokerCallbackHandler;
+import org.jboss.remoting.transport.Connector;
+import org.jboss.remoting.transport.PortUtil;
+
+
+/**
+ * Unit test for JBREM-1238.
+ *
+ * @author <a href="ron.sigal(a)jboss.com">Ron Sigal</a>
+ * @version
+ * <p>
+ * Copyright July 28, 2010
+ * </p>
+ */
+public class LeaseCreationTimeoutTestCase extends TestCase
+{
+ private static Logger log = Logger.getLogger(LeaseCreationTimeoutTestCase.class);
+
+ protected static int breakServer;
+ protected static int counter;
+
+ protected String host;
+ protected int port;
+ protected String locatorURI;
+ protected InvokerLocator serverLocator;
+ protected Connector connector;
+ protected TestInvocationHandler invocationHandler;
+ protected PrintStream originalPrintStream;
+ protected ByteArrayOutputStream baos;
+
+
+ public void setUp() throws Exception
+ {
+ baos = new ByteArrayOutputStream();
+ originalPrintStream = System.out;
+ PrintStream ps = new PrintStream(baos, true);
+ System.setOut(ps);
+
+ String pattern = "[%d{ABSOLUTE}] [%t] %5p (%F:%L) - %m%n";
+ PatternLayout layout = new PatternLayout(pattern);
+ ConsoleAppender consoleAppender = new ConsoleAppender(layout);
+ consoleAppender.setTarget(ConsoleAppender.SYSTEM_OUT);
+
+ Log4jLoggerPlugin loggerPlugin = (Log4jLoggerPlugin) Logger.getLogger("org.jboss.remoting").getLoggerPlugin();
+ org.apache.log4j.Logger logger = loggerPlugin.getLogger();
+ logger.setLevel(Level.TRACE);
+ logger.addAppender(consoleAppender);
+ loggerPlugin = (Log4jLoggerPlugin) Logger.getLogger("org.jboss.test.remoting").getLoggerPlugin();
+ logger = loggerPlugin.getLogger();
+ logger.setLevel(Level.TRACE);
+ logger.addAppender(consoleAppender);
+
+ breakServer = -1;
+ counter = 0;
+ }
+
+
+ public void tearDown()
+ {
+ }
+
+
+ /**
+ * Verifies that default value of leaseCreationTimeout is set properly.
+ */
+ public void testLeaseCreationTimeoutDefault() throws Throwable
+ {
+ log.info("entering " + getName());
+ Client client = null;
+
+ try
+ {
+ // Start server.
+ setupServer(-1);
+
+ // Create client.
+ InvokerLocator clientLocator = new InvokerLocator(locatorURI);
+ HashMap clientConfig = new HashMap();
+ clientConfig.put(InvokerLocator.FORCE_REMOTE, "true");
+ clientConfig.put(Client.ENABLE_LEASE, "true");
+ addExtraClientConfig(clientConfig);
+ client = new Client(clientLocator, clientConfig);
+ client.connect();
+ log.info("client is connected");
+
+ // Test connection.
+ assertEquals("abc", client.invoke("abc"));
+ log.info("connection is good");
+
+ String s = baos.toString();
+ assertTrue(s.indexOf("setting timeout to 30000 for this invocation") > -1);
+ assertTrue(s.indexOf("starting lease timer with ping period of") > -1);
+ log.info(getName() + " PASSES");
+ }
+ finally
+ {
+ client.disconnect();
+ shutdownServer();
+ originalPrintStream.flush();
+ System.setOut(originalPrintStream);
+ System.out.println("starting to write log");
+ String s = baos.toString();
+ System.out.println(s);
+ }
+ }
+
+
+ /**
+ * Verifies that value of leaseCreationTimeout is set properly from configuration map.
+ */
+ public void testLeaseCreationTimeoutConfig() throws Throwable
+ {
+ log.info("entering " + getName());
+ Client client = null;
+
+ try
+ {
+ // Start server.
+ setupServer(-1);
+
+ // Create client.
+ InvokerLocator clientLocator = new InvokerLocator(locatorURI);
+ HashMap clientConfig = new HashMap();
+ clientConfig.put(InvokerLocator.FORCE_REMOTE, "true");
+ clientConfig.put(Client.ENABLE_LEASE, "true");
+ clientConfig.put(Remoting.LEASE_CREATION_TIMEOUT, "12345");
+ clientConfig.put(ServerInvoker.TIMEOUT, "0");
+ addExtraClientConfig(clientConfig);
+ client = new Client(clientLocator, clientConfig);
+ client.connect();
+ log.info("client is connected");
+
+ // Test connection.
+ assertEquals("abc", client.invoke("abc"));
+ log.info("connection is good");
+
+ String s = baos.toString();
+ assertTrue(s.indexOf("setting timeout to 12345 for this invocation") > -1);
+ assertTrue(s.indexOf("starting lease timer with ping period of") > -1);
+ log.info(getName() + " PASSES");
+ }
+ finally
+ {
+ client.disconnect();
+ shutdownServer();
+ originalPrintStream.flush();
+ System.setOut(originalPrintStream);
+ System.out.println("starting to write log");
+ String s = baos.toString();
+ System.out.println(s);
+ }
+ }
+
+
+ /**
+ * Verifies that MicroRemotingClientInvoker.establishLease() is terminated according to
+ * the value of Remoting.LEASE_CREATION_TIMEOUT so that another new connection can
+ * set up a lease.
+ */
+ public void testLeaseCreationWithBrokenEstablishLease() throws Throwable
+ {
+ log.info("entering " + getName());
+ Client client1 = null;
+ Client client2 = null;
+
+ try
+ {
+ // Start server.
+ setupServer(1);
+
+ // Create client.
+ InvokerLocator clientLocator = new InvokerLocator(locatorURI);
+ HashMap clientConfig = new HashMap();
+ clientConfig.put(InvokerLocator.FORCE_REMOTE, "true");
+ clientConfig.put(ServerInvoker.TIMEOUT, "0");
+ clientConfig.put(Client.ENABLE_LEASE, "true");
+ clientConfig.put(Remoting.LEASE_CREATION_TIMEOUT, "5000");
+ addExtraClientConfig(clientConfig);
+ client1 = new Client(clientLocator, clientConfig);
+
+ // Should not connect.
+ BrokenEstablisthLeaseTestThread btt = new BrokenEstablisthLeaseTestThread(client1);
+ btt.run();
+
+ // Should connect.
+ Thread.sleep(10000);
+ client2 = new Client(clientLocator, clientConfig);
+ WorkingTestThread ttt = new WorkingTestThread(client2);
+ ttt.run();
+
+ // Test lease creation.
+ Thread.sleep(5000);
+ assertTrue(btt.ok);
+ assertTrue(ttt.ok);
+
+ log.info(getName() + " PASSES");
+ }
+ finally
+ {
+ client1.disconnect();
+ client2.disconnect();
+ shutdownServer();
+ originalPrintStream.flush();
+ System.setOut(originalPrintStream);
+ System.out.println("starting to write log");
+ String s = baos.toString();
+ System.out.println(s);
+ }
+ }
+
+
+ /**
+ * Verifies that MicroRemotingClientInvoker.terminateLease() is terminated according to
+ * the value of Remoting.LEASE_CREATION_TIMEOUT so that another new connection can
+ * set up a lease. In this test the hang up is in LeasePinger.removeClient().
+ */
+ public void testLeaseTerminationWithBrokenRemoveClient() throws Throwable
+ {
+ log.info("entering " + getName());
+ Client client1 = null;
+ Client client2 = null;
+
+ try
+ {
+ // Start server.
+ setupServer(4);
+
+ // Create client.
+ InvokerLocator clientLocator = new InvokerLocator(locatorURI);
+ HashMap clientConfig = new HashMap();
+ clientConfig.put(InvokerLocator.FORCE_REMOTE, "true");
+ clientConfig.put(ServerInvoker.TIMEOUT, "0");
+ clientConfig.put(Client.ENABLE_LEASE, "true");
+ clientConfig.put(Remoting.LEASE_CREATION_TIMEOUT, "5000");
+ addExtraClientConfig(clientConfig);
+ client1 = new Client(clientLocator, clientConfig);
+
+ // Should not terminate properly in call to LeasePinger.removeClient().
+ BrokenTerminateLeaseRemoveClientTestThread btt = new BrokenTerminateLeaseRemoveClientTestThread(client1, baos);
+ btt.start();
+
+ // Should connect.
+ Thread.sleep(10000);
+ client2 = new Client(clientLocator, clientConfig);
+ WorkingTestThread ttt = new WorkingTestThread(client2);
+ ttt.start();
+
+ // Test lease creation.
+ Thread.sleep(5000);
+ assertTrue(btt.ok);
+ assertTrue(ttt.ok);
+
+ log.info(getName() + " PASSES");
+ }
+ finally
+ {
+ shutdownServer();
+ originalPrintStream.flush();
+ System.setOut(originalPrintStream);
+ System.out.println("starting to write log");
+ String s = baos.toString();
+ System.out.println(s);
+ }
+ }
+
+
+ /**
+ * Verifies that MicroRemotingClientInvoker.terminateLease() is terminated according to
+ * the value of Remoting.LEASE_CREATION_TIMEOUT so that another new connection can
+ * set up a lease. In this test the hang up is in LeasePinger.stopPing().
+ */
+ public void testLeaseTerminationWithBrokenStopPing() throws Throwable
+ {
+ log.info("entering " + getName());
+
+ try
+ {
+ // Start server.
+ setupServer(5);
+
+ // Create client.
+ InvokerLocator clientLocator = new InvokerLocator(locatorURI);
+ HashMap clientConfig = new HashMap();
+ clientConfig.put(InvokerLocator.FORCE_REMOTE, "true");
+ clientConfig.put(ServerInvoker.TIMEOUT, "0");
+ clientConfig.put(Client.ENABLE_LEASE, "true");
+ clientConfig.put(Remoting.LEASE_CREATION_TIMEOUT, "5000");
+ addExtraClientConfig(clientConfig);
+ Client client1 = new Client(clientLocator, clientConfig);
+
+ // Should not terminate properly in call to LeasePinger.stopPing().
+ BrokenTerminateLeaseStopPingTestThread btt = new BrokenTerminateLeaseStopPingTestThread(client1, baos);
+ btt.start();
+
+ // Should connect.
+ Thread.sleep(10000);
+ Client client2 = new Client(clientLocator, clientConfig);
+ WorkingTestThread ttt = new WorkingTestThread(client2);
+ ttt.start();
+
+ // Test lease creation.
+ Thread.sleep(5000);
+ assertTrue(btt.ok);
+ assertTrue(ttt.ok);
+
+ shutdownServer();
+ log.info(getName() + " PASSES");
+
+ }
+ finally
+ {
+ originalPrintStream.flush();
+ System.setOut(originalPrintStream);
+ System.out.println("starting to write log");
+ String s = baos.toString();
+ System.out.println(s);
+ }
+ }
+
+
+ protected String getTransport()
+ {
+ return "socket";
+ }
+
+
+ protected void addExtraClientConfig(Map config) {}
+ protected void addExtraServerConfig(Map config) {}
+
+
+ protected void setupServer(int breakServer) throws Exception
+ {
+ host = InetAddress.getLocalHost().getHostAddress();
+ port = PortUtil.findFreePort(host);
+ locatorURI = getTransport() + "://" + host + ":" + port;
+ String metadata = System.getProperty("remoting.metadata");
+ if (metadata != null)
+ {
+ locatorURI += "/?" + metadata;
+ }
+ serverLocator = new InvokerLocator(locatorURI);
+ log.info("Starting remoting server with locator uri of: " + locatorURI);
+ HashMap config = new HashMap();
+ config.put(InvokerLocator.FORCE_REMOTE, "true");
+ config.put(Remoting.USE_CLIENT_CONNECTION_IDENTITY, "true");
+ config.put("clientLeasePeriod", "10000");
+ if (breakServer > -1)
+ {
+ LeaseCreationTimeoutTestCase.breakServer = breakServer;
+ config.put(Remoting.CUSTOM_SERVER_SOCKET_FACTORY, new TestServerSocketFactory());
+ }
+ addExtraServerConfig(config);
+ connector = new Connector(serverLocator, config);
+ connector.create();
+ invocationHandler = new TestInvocationHandler();
+ connector.addInvocationHandler("test", invocationHandler);
+ connector.addConnectionListener(new TestConnectionListener());
+ connector.start();
+ }
+
+
+ protected void shutdownServer() throws Exception
+ {
+ if (connector != null)
+ connector.stop();
+ }
+
+
+ static class TestInvocationHandler implements ServerInvocationHandler
+ {
+ public void addListener(InvokerCallbackHandler callbackHandler) {}
+ public Object invoke(final InvocationRequest invocation) throws Throwable
+ {
+ return invocation.getParameter();
+ }
+ public void removeListener(InvokerCallbackHandler callbackHandler) {}
+ public void setMBeanServer(MBeanServer server) {}
+ public void setInvoker(ServerInvoker invoker) {}
+ }
+
+
+ static class TestConnectionListener implements ConnectionListener
+ {
+ public void handleConnectionException(Throwable throwable, Client client)
+ {
+ }
+ }
+
+
+ static public class TestServerSocketFactory extends ServerSocketFactory
+ {
+ public TestServerSocketFactory()
+ {
+ }
+ public ServerSocket createServerSocket() throws IOException
+ {
+ ServerSocket ss = new TestServerSocket();
+ log.info(this + " returning: " + ss);
+ return ss;
+ }
+ public ServerSocket createServerSocket(int port) throws IOException
+ {
+ ServerSocket ss = new TestServerSocket(port);
+ log.info(this + " returning: " + ss);
+ return ss;
+ }
+
+ public ServerSocket createServerSocket(int port, int backlog) throws IOException
+ {
+ ServerSocket ss = new TestServerSocket(port, backlog);
+ log.info(this + " returning: " + ss);
+ return ss;
+ }
+
+ public ServerSocket createServerSocket(int port, int backlog, InetAddress ifAddress) throws IOException
+ {
+ ServerSocket ss = new TestServerSocket(port, backlog, ifAddress);
+ log.info(this + " returning: " + ss);
+ return ss;
+ }
+ }
+
+
+ static class TestServerSocket extends ServerSocket
+ {
+ public TestServerSocket() throws IOException
+ {
+ super();
+ }
+ public TestServerSocket(int port) throws IOException
+ {
+ super(port);
+ }
+ public TestServerSocket(int port, int backlog) throws IOException
+ {
+ super(port, backlog);
+ }
+ public TestServerSocket(int port, int backlog, InetAddress bindAddr) throws IOException
+ {
+ super(port, backlog, bindAddr);
+ }
+ public Socket accept() throws IOException
+ {
+ Socket s = super.accept();
+ TestSocket ts = new TestSocket(s);
+ log.info(this + " returning: " + ts);
+ return ts;
+ }
+ public String toString()
+ {
+ return "TestServerSocket[" + getLocalPort() + "]";
+ }
+ }
+
+
+ static class TestSocket extends Socket
+ {
+ Socket s;
+
+ public TestSocket(Socket s)
+ {
+ this.s = s;
+ }
+ public void close() throws IOException
+ {
+ s.close();
+ }
+ public InputStream getInputStream() throws IOException
+ {
+ return s.getInputStream();
+ }
+ public OutputStream getOutputStream() throws IOException
+ {
+ return new TestOutputStream(s.getOutputStream());
+ }
+ public InetAddress getInetAddress()
+ {
+ return s.getInetAddress();
+ }
+ public String toString()
+ {
+ return "TestSocket[" + s.getLocalPort() + "->" + s.getPort() + "]";
+ }
+ }
+
+
+ static class TestOutputStream extends OutputStream
+ {
+ OutputStream os;
+ boolean closed;
+ boolean doThrow = true;
+
+ public TestOutputStream(OutputStream os)
+ {
+ this.os = os;
+ }
+ public void close()throws IOException
+ {
+ closed = true;
+ super.close();
+ log.info(this + " closed");
+ }
+ public void write(int b) throws IOException
+ {
+ System.out.print("b: " + b);
+ if (closed)
+ {
+ log.info("TestOutputStream closed, cannot write");
+ throw new IOException("closed");
+ }
+ if (doThrow && counter++ == breakServer)
+ {
+ log.info("sleeping");
+ while (true)
+ {
+ try
+ {
+ Thread.sleep(30000);
+ }
+ catch (InterruptedException e)
+ {
+ }
+ }
+ }
+ os.write(b);
+ }
+ public void write(byte b[], int off, int len) throws IOException
+ {
+ System.out.print("b: ");
+ for (int i = 0; i < len; i++)
+ {
+ System.out.print(b[i] + " ");
+ }
+ System.out.println("");
+ if (closed)
+ {
+ log.info("TestOutputStream closed, cannot write");
+ throw new IOException("closed");
+ }
+ log.info("TestOutputStream: counter = " + counter + ", broken = " + breakServer);
+ if (counter++ == breakServer)
+ {
+ log.info(this + " sleeping");
+ try
+ {
+ Thread.sleep(20000);
+ }
+ catch (InterruptedException e)
+ {
+ }
+ log.info(this + " woke up");
+ }
+ if (closed)
+ {
+ log.info("TestOutputStream closed, cannot write");
+ throw new IOException("closed");
+ }
+ try
+ {
+ log.info(this + " calling write()");
+ doThrow = false;
+ os.write(b, off, len);
+ os.flush();
+ doThrow = true;
+ log.info(this + " back from write()");
+ }
+ catch (IOException e)
+ {
+ log.info("exception: ", e);
+ throw e;
+ }
+ }
+ public void flush() throws IOException
+ {
+ os.flush();
+ }
+ }
+
+
+ static class BrokenEstablisthLeaseTestThread extends Thread
+ {
+ public boolean ok;
+ Client client;
+
+ public BrokenEstablisthLeaseTestThread(Client client)
+ {
+ this.client = client;
+ setName("BrokenEstablisthLeaseTestThread[" + client + "]");
+ }
+ public void run()
+ {
+ try
+ {
+ log.info(this + " STARTING");
+ log.info(this + " trying to connect " + client);
+ client.connect();
+ }
+ catch (CannotConnectException e)
+ {
+ if ("Error setting up client lease upon performing connect.".equals(e.getMessage()))
+ {
+ ok = true;
+ log.info(this + " got expected exception");
+ }
+ else
+ {
+ log.info(this + " got expected exception with unexpected message: " + e.getMessage());
+ }
+ }
+ catch (Exception e)
+ {
+ log.error(this + " got unexpected exception: " + e);
+ }
+ }
+ public String toString()
+ {
+ return getName();
+ }
+ }
+
+
+ static class BrokenTerminateLeaseRemoveClientTestThread extends Thread
+ {
+ public boolean ok;
+ Client client;
+ ByteArrayOutputStream baos;
+
+ public BrokenTerminateLeaseRemoveClientTestThread(Client client, ByteArrayOutputStream baos)
+ {
+ this.client = client;
+ this.baos = baos;
+ setName("BrokenTerminateLeaseRemoveClientTestThread[" + client + "]");
+ }
+ public void run()
+ {
+ try
+ {
+ log.info(this + " STARTING");
+ log.info(this + " trying to connect " + client);
+ client.connect();
+ assertEquals("def", client.invoke("def"));
+ client.disconnect();
+ String s = baos.toString();
+ assertTrue(s.indexOf("Terminating individual lease for session id") > -1);
+ assertTrue(s.indexOf("failed sending disconnect for client lease") > -1);
+ ok = true;
+ }
+ catch (Throwable t)
+ {
+ log.error(this + " got unexpected exception: " + t);
+ }
+ }
+ public String toString()
+ {
+ return getName();
+ }
+ }
+
+
+ static class BrokenTerminateLeaseStopPingTestThread extends Thread
+ {
+ public boolean ok;
+ Client client;
+ ByteArrayOutputStream baos;
+
+ public BrokenTerminateLeaseStopPingTestThread(Client client, ByteArrayOutputStream baos)
+ {
+ this.client = client;
+ this.baos = baos;
+ setName("BrokenTerminateLeaseStopPingTestThread[" + client + "]");
+ }
+ public void run()
+ {
+ try
+ {
+ log.info(this + " STARTING");
+ log.info(this + " trying to connect " + client);
+ client.connect();
+ assertEquals("def", client.invoke("def"));
+ client.disconnect();
+ String s = baos.toString();
+ assertTrue(s.indexOf("Terminating individual lease for session id") > -1);
+ assertFalse(s.indexOf("failed sending disconnect for client lease") > -1);
+ assertTrue(s.indexOf(": disconnectTimeout: 5000") > -1);
+ assertTrue(s.indexOf("error shutting down lease pinger") > -1);
+ ok = true;
+ }
+ catch (Throwable t)
+ {
+ log.error(this + " got unexpected exception: " + t);
+ }
+ }
+ public String toString()
+ {
+ return getName();
+ }
+ }
+
+
+ static class WorkingTestThread extends Thread
+ {
+ public boolean ok;
+ Client client;
+
+ public WorkingTestThread(Client client)
+ {
+ this.client = client;
+ setName("WorkingTestThread[" + client + "]");
+ }
+ public void run()
+ {
+ log.info(this + " STARTING");
+ try
+ {
+ log.info(this + " trying to connect " + client);
+ client.connect();
+ assertEquals("abc", client.invoke("abc"));
+ log.info(this + " connection is good");
+ ok = true;
+ }
+ catch (Throwable t)
+ {
+ log.error(this + " got unexpected exception: " + t);
+ }
+ }
+ public String toString()
+ {
+ return getName();
+ }
+ }
+}
\ No newline at end of file
14 years, 4 months
JBoss Remoting SVN: r5937 - remoting2/branches/2.x/src/main/org/jboss/remoting.
by jboss-remoting-commits@lists.jboss.org
Author: ron.sigal(a)jboss.com
Date: 2010-07-29 16:14:07 -0400 (Thu, 29 Jul 2010)
New Revision: 5937
Modified:
remoting2/branches/2.x/src/main/org/jboss/remoting/Lease.java
Log:
JBREM-1238: Fixed mangled log statement.
Modified: remoting2/branches/2.x/src/main/org/jboss/remoting/Lease.java
===================================================================
--- remoting2/branches/2.x/src/main/org/jboss/remoting/Lease.java 2010-07-29 20:13:05 UTC (rev 5936)
+++ remoting2/branches/2.x/src/main/org/jboss/remoting/Lease.java 2010-07-29 20:14:07 UTC (rev 5937)
@@ -291,7 +291,7 @@
notifier.connectionTerminated(locatorURL, clientHolder.getSessionId(), clientHolder.getConfig());
if(isTraceEnabled)
{
- log.trace(this + " Notified connection lif (isTraceEnabled) log.tracef lease termination due to disconnect from client (client session id = " + clientHolder.getSessionId());
+ log.trace(this + " Notified connection listener of lease termination due to disconnect from client (client session id = " + clientHolder.getSessionId());
}
}
}
14 years, 4 months
JBoss Remoting SVN: r5936 - remoting2/branches/2.x/src/main/org/jboss/remoting.
by jboss-remoting-commits@lists.jboss.org
Author: ron.sigal(a)jboss.com
Date: 2010-07-29 16:13:05 -0400 (Thu, 29 Jul 2010)
New Revision: 5936
Modified:
remoting2/branches/2.x/src/main/org/jboss/remoting/MicroRemoteClientInvoker.java
Log:
JBREM-1238: Added use of leaseCreationTimeout.
Modified: remoting2/branches/2.x/src/main/org/jboss/remoting/MicroRemoteClientInvoker.java
===================================================================
--- remoting2/branches/2.x/src/main/org/jboss/remoting/MicroRemoteClientInvoker.java 2010-07-29 20:12:38 UTC (rev 5935)
+++ remoting2/branches/2.x/src/main/org/jboss/remoting/MicroRemoteClientInvoker.java 2010-07-29 20:13:05 UTC (rev 5936)
@@ -40,6 +40,7 @@
public abstract class MicroRemoteClientInvoker extends AbstractInvoker implements ClientInvoker
{
private static final Logger log = Logger.getLogger(MicroRemoteClientInvoker.class);
+ private static final String LEASE_CREATION_TIMEOUT_DEFAULT = "30000";
private boolean trace = log.isTraceEnabled();
protected boolean connected = false;
@@ -51,6 +52,7 @@
private String invokerSessionID = new GUID().toString();
protected boolean parentFirstClassLoading = true;
private boolean changeInvalidStateToCannotConnect = false;
+ private String leaseCreationTimeout = LEASE_CREATION_TIMEOUT_DEFAULT;
protected List connectHomes;
@@ -388,6 +390,16 @@
return this.invokerSessionID;
}
+ public String getLeaseCreationTimeout()
+ {
+ return leaseCreationTimeout;
+ }
+
+ public void setLeaseCreationTimeout(String leaseCreationTimeout)
+ {
+ this.leaseCreationTimeout = leaseCreationTimeout;
+ }
+
public void terminateLease(String sessionId, int disconnectTimeout)
{
terminateLease(sessionId, disconnectTimeout, null);
@@ -405,6 +417,10 @@
if (trace) log.trace(this + " entering terminateLease() for " + leasePinger);
if(leasePinger != null)
{
+ if (disconnectTimeout == Client.DEFAULT_DISCONNECT_TIMEOUT)
+ {
+ disconnectTimeout = Integer.parseInt(leaseCreationTimeout);
+ }
leasePinger.setDisconnectTimeout(disconnectTimeout);
if (sessionId == null)
@@ -506,6 +522,7 @@
Map requestMap = new HashMap();
requestMap.put(LeasePinger.LEASE_PINGER_ID, leasePingerId);
requestMap.put(LeasePinger.TIME_STAMP, Long.toString(System.currentTimeMillis()));
+ requestMap.put(ServerInvoker.TIMEOUT, leaseCreationTimeout);
if (trace) log.trace(this + " initiating lease for leasePingerId " + leasePingerId);
InvocationRequest ir = new InvocationRequest(invokerSessionID, null, "$PING$", requestMap, new HashMap(), null);
@@ -619,6 +636,22 @@
String sflag = flag.toString();
changeInvalidStateToCannotConnect = Boolean.valueOf(sflag).booleanValue();
}
+
+ flag = configuration.get(Remoting.LEASE_CREATION_TIMEOUT);
+ if (flag != null)
+ {
+ String sflag = flag.toString();
+ try
+ {
+ Integer.valueOf(sflag);
+ leaseCreationTimeout = sflag;
+ }
+ catch (NumberFormatException e)
+ {
+ log.warn(this + " value of " + Remoting.LEASE_CREATION_TIMEOUT +
+ " should be in numeric form: " + sflag + ", using " + LEASE_CREATION_TIMEOUT_DEFAULT);
+ }
+ }
}
/**
14 years, 4 months