public void testSend() throws Exception {
final Acceptor acceptor = new Acceptor("localhost", 0, 1);
final FanOutServerSender serverSender =
new FanOutServerSender(acceptor, ConnectionType.AGENT, 3);
final Socket[] socket = new Socket[5];
for (int i = 0; i < socket.length; ++i) {
socket[i] =
new Connector(
InetAddress.getByName(null).getHostName(),
acceptor.getPort(),
ConnectionType.AGENT)
.connect();
}
// Sleep until we've accepted all connections. Give up after a few
// seconds.
final ResourcePool socketSet = acceptor.getSocketSet(ConnectionType.AGENT);
for (int i = 0; socketSet.countActive() != 5 && i < 10; ++i) {
Thread.sleep(i * i * 10);
}
final SimpleMessage message1 = new SimpleMessage();
final SimpleMessage message2 = new SimpleMessage();
serverSender.send(message1);
serverSender.send(message2);
for (int i = 0; i < socket.length; ++i) {
final InputStream socketInput = socket[i].getInputStream();
final Message m1 = readMessage(socketInput);
final Message m2 = readMessage(socketInput);
assertEquals(message1, m1);
assertEquals(message2, m2);
assertEquals(0, socketInput.available());
socket[i].close();
}
serverSender.shutdown();
acceptor.shutdown();
}
public void testShutdown() throws Exception {
final Acceptor acceptor = new Acceptor("localhost", 0, 1);
final FanOutServerSender serverSender =
new FanOutServerSender(acceptor, ConnectionType.AGENT, 3);
assertEquals(1, acceptor.getThreadGroup().activeCount());
final Socket socket =
new Connector(
InetAddress.getByName(null).getHostName(), acceptor.getPort(), ConnectionType.AGENT)
.connect();
// Sleep until we've accepted the connection. Give up after a few
// seconds.
final ResourcePool socketSet = acceptor.getSocketSet(ConnectionType.AGENT);
for (int i = 0; socketSet.countActive() != 1 && i < 10; ++i) {
Thread.sleep(i * i * 10);
}
final Message message = new SimpleMessage();
serverSender.send(message);
final InputStream socketStream = socket.getInputStream();
final Object o1 = readMessage(socketStream);
assertNotNull(o1);
serverSender.shutdown();
try {
serverSender.send(message);
fail("Expected CommunicationException");
} catch (CommunicationException e) {
}
try {
final Object o2 = readMessage(socketStream);
assertTrue(o2 instanceof CloseCommunicationMessage);
} catch (StreamCorruptedException e) {
// Occasionally this occurs because the connection is shutdown.
// Whatever.
}
acceptor.shutdown();
}
/**
* Registers a new {@link Acceptor} from which the <code>ServerReceiver</code> should process
* messages. Actively polls connections of the given types for messages, deserialises them, and
* queues them for retrieval using {@link #waitForMessage()}.
*
* <p>A single <code>ServerReceiver</code> can listen to messages from multiple {@link Acceptor}s.
* You can register the same {@link Acceptor} with multiple <code>ServerReceiver</code>s, but then
* there is no way of controlling which receiver will receive messages from a given {@link
* Acceptor}.
*
* @param acceptor Acceptor.
* @param connectionTypes Type of connections to listen for.
* @param numberOfThreads How many threads to dedicate to processing the Acceptor. The threads
* this method spawns just read, deserialise, and queue. Set <code>numberOfThreads</code> to
* the number of concurrent streams you expect to be able to read.
* @param idleThreadPollDelay Time in milliseconds that an idle thread should sleep if there are
* no sockets to process.
* @exception CommunicationException If this <code>ServerReceiver</code> has been shutdown.
*/
public void receiveFrom(
Acceptor acceptor,
ConnectionType[] connectionTypes,
int numberOfThreads,
final int idleThreadPollDelay)
throws CommunicationException {
if (connectionTypes.length == 0) {
// Nothing to do.
return;
}
final ResourcePool[] acceptedSocketSets = new ResourcePool[connectionTypes.length];
for (int i = 0; i < connectionTypes.length; ++i) {
acceptedSocketSets[i] = acceptor.getSocketSet(connectionTypes[i]);
}
final ThreadPool.InterruptibleRunnableFactory runnableFactory =
new ThreadPool.InterruptibleRunnableFactory() {
public InterruptibleRunnable create() {
return new ServerReceiverRunnable(
new CombinedResourcePool(acceptedSocketSets), idleThreadPollDelay);
}
};
final ThreadPool threadPool =
new ThreadPool(
"ServerReceiver (" + acceptor.getPort() + ", " + Arrays.asList(connectionTypes) + ")",
numberOfThreads,
runnableFactory);
synchronized (this) {
try {
m_messageQueue.checkIfShutdown();
} catch (ShutdownException e) {
throw new CommunicationException("Shut down", e);
}
m_threadPools.add(threadPool);
}
threadPool.start();
}
public void testSendAddressedMessage() throws Exception {
final Acceptor acceptor = new Acceptor("localhost", 0, 1);
final FanOutServerSender serverSender =
new FanOutServerSender(acceptor, ConnectionType.AGENT, 3);
final Socket[] socket = new Socket[5];
for (int i = 0; i < socket.length; ++i) {
socket[i] =
new Connector(
InetAddress.getByName(null).getHostName(),
acceptor.getPort(),
ConnectionType.AGENT)
.connect(new StubAddress(new Integer(i)));
}
// Sleep until we've accepted all connections. Give up after a few
// seconds.
final ResourcePool socketSet = acceptor.getSocketSet(ConnectionType.AGENT);
for (int i = 0; socketSet.countActive() != 5 && i < 10; ++i) {
Thread.sleep(i * i * 10);
}
final SimpleMessage message1 = new SimpleMessage();
final SimpleMessage message2 = new SimpleMessage();
serverSender.send(new StubAddress(new Integer(1)), message1);
serverSender.send(new StubAddress(new Integer(2)), message2);
for (int i = 0; i < socket.length; ++i) {
final InputStream socketInput = socket[i].getInputStream();
if (i == 1) {
final Message m = readMessage(socketInput);
assertEquals(message1, m);
} else if (i == 2) {
final Message m = readMessage(socketInput);
assertEquals(message2, m);
}
assertEquals(0, socketInput.available());
socket[i].close();
}
serverSender.shutdown();
try {
serverSender.send(
new Address() {
public boolean includes(Address address) {
return false;
}
},
message1);
fail("Expected CommunicationException");
} catch (CommunicationException e) {
}
acceptor.shutdown();
}