/*
* Copyright (c) 2008-2017 the original author or authors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.cometd.oort;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.CyclicBarrier;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.concurrent.atomic.AtomicReference;
import org.cometd.bayeux.Channel;
import org.cometd.bayeux.Message;
import org.cometd.bayeux.client.ClientSessionChannel;
import org.cometd.bayeux.server.LocalSession;
import org.cometd.bayeux.server.ServerMessage;
import org.cometd.bayeux.server.ServerSession;
import org.cometd.bayeux.server.ServerTransport;
import org.cometd.client.BayeuxClient;
import org.cometd.client.transport.LongPollingTransport;
import org.cometd.server.AbstractService;
import org.cometd.server.ServerSessionImpl;
import org.eclipse.jetty.client.HttpClient;
import org.eclipse.jetty.server.NetworkConnector;
import org.eclipse.jetty.server.Server;
import org.junit.After;
import org.junit.Assert;
import org.junit.Test;
public class SetiTest extends OortTest {
private List<Seti> setis = new ArrayList<>();
public SetiTest(String serverTransport) {
super(serverTransport);
}
protected Seti startSeti(Oort oort) throws Exception {
Seti seti = new Seti(oort);
seti.start();
setis.add(seti);
return seti;
}
@After
public void stopSetis() throws Exception {
for (int i = setis.size() - 1; i >= 0; --i) {
stopSeti(setis.get(i));
}
}
protected void stopSeti(Seti seti) throws Exception {
seti.stop();
}
@Test
public void testAssociateAndSendMessage() throws Exception {
Server server1 = startServer(0);
Oort oort1 = startOort(server1);
Server server2 = startServer(0);
Oort oort2 = startOort(server2);
CountDownLatch latch = new CountDownLatch(1);
oort2.addCometListener(new CometJoinedListener(latch));
OortComet oortComet12 = oort1.observeComet(oort2.getURL());
Assert.assertTrue(oortComet12.waitFor(5000, BayeuxClient.State.CONNECTED));
Assert.assertTrue(latch.await(5, TimeUnit.SECONDS));
OortComet oortComet21 = oort2.findComet(oort1.getURL());
Assert.assertTrue(oortComet21.waitFor(5000, BayeuxClient.State.CONNECTED));
Seti seti1 = startSeti(oort1);
Seti seti2 = startSeti(oort2);
CountDownLatch presenceLatch = new CountDownLatch(4);
UserPresentListener presenceListener = new UserPresentListener(presenceLatch);
seti1.addPresenceListener(presenceListener);
seti2.addPresenceListener(presenceListener);
new SetiService(seti1);
new SetiService(seti2);
BayeuxClient client1 = startClient(oort1, null);
Assert.assertTrue(client1.waitFor(5000, BayeuxClient.State.CONNECTED));
BayeuxClient client2 = startClient(oort2, null);
Assert.assertTrue(client2.waitFor(5000, BayeuxClient.State.CONNECTED));
LatchListener publishLatch = new LatchListener();
String loginChannelName = "/service/login";
Map<String, Object> login1 = new HashMap<>();
login1.put("user", "user1");
ClientSessionChannel loginChannel1 = client1.getChannel(loginChannelName);
loginChannel1.addListener(publishLatch);
loginChannel1.publish(login1);
Assert.assertTrue(publishLatch.await(5, TimeUnit.SECONDS));
publishLatch.reset(1);
Map<String, Object> login2 = new HashMap<>();
login2.put("user", "user2");
ClientSessionChannel loginChannel2 = client2.getChannel(loginChannelName);
loginChannel2.addListener(publishLatch);
loginChannel2.publish(login2);
Assert.assertTrue(publishLatch.await(5, TimeUnit.SECONDS));
Assert.assertTrue(presenceLatch.await(5, TimeUnit.SECONDS));
String channel = "/service/forward";
final CountDownLatch messageLatch = new CountDownLatch(1);
client2.getChannel(channel).addListener(new ClientSessionChannel.MessageListener() {
@Override
public void onMessage(ClientSessionChannel channel, Message message) {
messageLatch.countDown();
}
});
Map<String, Object> data1 = new HashMap<>();
data1.put("peer", "user2");
client1.getChannel(channel).publish(data1);
Assert.assertTrue(messageLatch.await(5, TimeUnit.SECONDS));
}
@Test
public void testDisassociate() throws Exception {
Server server1 = startServer(0);
Oort oort1 = startOort(server1);
Server server2 = startServer(0);
Oort oort2 = startOort(server2);
CountDownLatch latch = new CountDownLatch(1);
oort2.addCometListener(new CometJoinedListener(latch));
OortComet oortComet12 = oort1.observeComet(oort2.getURL());
Assert.assertTrue(oortComet12.waitFor(5000, BayeuxClient.State.CONNECTED));
Assert.assertTrue(latch.await(5, TimeUnit.SECONDS));
OortComet oortComet21 = oort2.findComet(oort1.getURL());
Assert.assertTrue(oortComet21.waitFor(5000, BayeuxClient.State.CONNECTED));
Seti seti1 = startSeti(oort1);
Seti seti2 = startSeti(oort2);
CountDownLatch presenceLatch = new CountDownLatch(4);
UserPresentListener presenceListener = new UserPresentListener(presenceLatch);
seti1.addPresenceListener(presenceListener);
seti2.addPresenceListener(presenceListener);
new SetiService(seti1);
new SetiService(seti2);
BayeuxClient client1 = startClient(oort1, null);
Assert.assertTrue(client1.waitFor(5000, BayeuxClient.State.CONNECTED));
BayeuxClient client2 = startClient(oort2, null);
Assert.assertTrue(client2.waitFor(5000, BayeuxClient.State.CONNECTED));
Map<String, Object> login1 = new HashMap<>();
login1.put("user", "user1");
client1.getChannel("/service/login").publish(login1);
Map<String, Object> login2 = new HashMap<>();
login2.put("user", "user2");
client2.getChannel("/service/login").publish(login2);
Assert.assertTrue(presenceLatch.await(5, TimeUnit.SECONDS));
CountDownLatch absenceLatch = new CountDownLatch(1);
UserAbsentListener absenceListener = new UserAbsentListener(absenceLatch);
seti1.addPresenceListener(absenceListener);
// Disassociate
Map<String, Object> logout2 = new HashMap<>();
logout2.put("user", "user2");
client2.getChannel("/service/logout").publish(logout2);
Assert.assertTrue(absenceLatch.await(5, TimeUnit.SECONDS));
String channel = "/service/forward";
final CountDownLatch messageLatch = new CountDownLatch(1);
client2.getChannel(channel).addListener(new ClientSessionChannel.MessageListener() {
@Override
public void onMessage(ClientSessionChannel channel, Message message) {
messageLatch.countDown();
}
});
Map<String, Object> data1 = new HashMap<>();
data1.put("peer", "user2");
client1.getChannel(channel).publish(data1);
// User2 has been disassociated, must not receive the message
Assert.assertFalse(messageLatch.await(1, TimeUnit.SECONDS));
}
@Test
public void testAutomaticDisassociation() throws Exception {
Server server1 = startServer(0);
Oort oort1 = startOort(server1);
Server server2 = startServer(0);
Oort oort2 = startOort(server2);
CountDownLatch latch = new CountDownLatch(1);
oort2.addCometListener(new CometJoinedListener(latch));
OortComet oortComet12 = oort1.observeComet(oort2.getURL());
Assert.assertTrue(oortComet12.waitFor(5000, BayeuxClient.State.CONNECTED));
Assert.assertTrue(latch.await(5, TimeUnit.SECONDS));
OortComet oortComet21 = oort2.findComet(oort1.getURL());
Assert.assertTrue(oortComet21.waitFor(5000, BayeuxClient.State.CONNECTED));
Seti seti1 = startSeti(oort1);
Seti seti2 = startSeti(oort2);
CountDownLatch presenceLatch = new CountDownLatch(4);
UserPresentListener presenceListener = new UserPresentListener(presenceLatch);
seti1.addPresenceListener(presenceListener);
seti2.addPresenceListener(presenceListener);
new SetiService(seti1);
new SetiService(seti2);
BayeuxClient client1 = startClient(oort1, null);
Assert.assertTrue(client1.waitFor(5000, BayeuxClient.State.CONNECTED));
Map<String, Object> login1 = new HashMap<>();
login1.put("user", "user1");
client1.getChannel("/service/login").publish(login1);
final AtomicReference<String> session2 = new AtomicReference<>();
HttpClient httpClient = new HttpClient();
httpClient.start();
BayeuxClient client2 = new BayeuxClient(oort2.getURL(), new LongPollingTransport(null, httpClient)) {
@Override
protected void processConnect(Message.Mutable connect) {
// Send the login message, so Seti can associate this user
Map<String, Object> login2 = new HashMap<>();
login2.put("user", "user2");
getChannel("/service/login").publish(login2);
// Modify the advice so that it does not reconnect,
// simulating that the client is gone so the server expires it
session2.set(getId());
connect.setSuccessful(false);
connect.getAdvice(true).put(Message.RECONNECT_FIELD, Message.RECONNECT_NONE_VALUE);
super.processConnect(connect);
}
};
client2.handshake();
Assert.assertTrue(client2.waitFor(5000, BayeuxClient.State.DISCONNECTED));
httpClient.stop();
Assert.assertTrue(presenceLatch.await(5, TimeUnit.SECONDS));
CountDownLatch absenceLatch = new CountDownLatch(1);
seti1.addPresenceListener(new UserAbsentListener(absenceLatch));
// Wait for the server to expire client2 and for Seti to disassociate it
final CountDownLatch removedLatch = new CountDownLatch(1);
oort2.getBayeuxServer().getSession(session2.get()).addListener(new ServerSession.RemoveListener() {
@Override
public void removed(ServerSession session, boolean timeout) {
removedLatch.countDown();
}
});
long maxTimeout = ((ServerTransport)oort2.getBayeuxServer().getTransport("websocket")).getMaxInterval();
Assert.assertTrue(removedLatch.await(maxTimeout + 5000, TimeUnit.MILLISECONDS));
Assert.assertTrue(absenceLatch.await(5, TimeUnit.SECONDS));
Assert.assertFalse(seti2.isAssociated("user2"));
}
@Test
public void testAssociationWithMultipleSessions() throws Exception {
Server server1 = startServer(0);
Oort oort1 = startOort(server1);
Server server2 = startServer(0);
Oort oort2 = startOort(server2);
Server server3 = startServer(0);
Oort oort3 = startOort(server3);
CountDownLatch latch = new CountDownLatch(6);
CometJoinedListener listener1 = new CometJoinedListener(latch);
oort1.addCometListener(listener1);
oort2.addCometListener(listener1);
oort3.addCometListener(listener1);
OortComet oortCometAB = oort1.observeComet(oort2.getURL());
Assert.assertTrue(oortCometAB.waitFor(5000, BayeuxClient.State.CONNECTED));
OortComet oortCometAC = oort1.observeComet(oort3.getURL());
Assert.assertTrue(oortCometAC.waitFor(5000, BayeuxClient.State.CONNECTED));
Assert.assertTrue(latch.await(5, TimeUnit.SECONDS));
OortComet oortCometBA = oort2.findComet(oort1.getURL());
Assert.assertTrue(oortCometBA.waitFor(5000, BayeuxClient.State.CONNECTED));
OortComet oortCometBC = oort2.findComet(oort3.getURL());
Assert.assertTrue(oortCometBC.waitFor(5000, BayeuxClient.State.CONNECTED));
OortComet oortCometCA = oort3.findComet(oort1.getURL());
Assert.assertTrue(oortCometCA.waitFor(5000, BayeuxClient.State.CONNECTED));
OortComet oortCometCB = oort3.findComet(oort2.getURL());
Assert.assertTrue(oortCometCB.waitFor(5000, BayeuxClient.State.CONNECTED));
Seti seti1 = startSeti(oort1);
Seti seti2 = startSeti(oort2);
Seti seti3 = startSeti(oort3);
CountDownLatch presenceLatch = new CountDownLatch(6);
UserPresentListener presenceListener = new UserPresentListener(presenceLatch);
seti1.addPresenceListener(presenceListener);
seti2.addPresenceListener(presenceListener);
seti3.addPresenceListener(presenceListener);
new SetiService(seti1);
new SetiService(seti2);
new SetiService(seti3);
BayeuxClient client1A = startClient(oort1, null);
Assert.assertTrue(client1A.waitFor(5000, BayeuxClient.State.CONNECTED));
BayeuxClient client1B = startClient(oort1, null);
Assert.assertTrue(client1B.waitFor(5000, BayeuxClient.State.CONNECTED));
BayeuxClient client1C = startClient(oort2, null);
Assert.assertTrue(client1C.waitFor(5000, BayeuxClient.State.CONNECTED));
BayeuxClient client3 = startClient(oort3, null);
Assert.assertTrue(client3.waitFor(5000, BayeuxClient.State.CONNECTED));
LatchListener publishLatch = new LatchListener();
Map<String, Object> login1A = new HashMap<>();
login1A.put("user", "user1");
ClientSessionChannel loginChannel1A = client1A.getChannel("/service/login");
loginChannel1A.addListener(publishLatch);
loginChannel1A.publish(login1A);
Assert.assertTrue(publishLatch.await(5, TimeUnit.SECONDS));
// Login the same user to the same server with a different client
publishLatch.reset(1);
Map<String, Object> login1B = new HashMap<>();
login1B.put("user", "user1");
ClientSessionChannel loginChannel1B = client1B.getChannel("/service/login");
loginChannel1B.addListener(publishLatch);
loginChannel1B.publish(login1B);
Assert.assertTrue(publishLatch.await(5, TimeUnit.SECONDS));
// Login the same user to another server with a different client
publishLatch.reset(1);
Map<String, Object> login1C = new HashMap<>();
login1C.put("user", "user1");
ClientSessionChannel loginChannel1C = client1C.getChannel("/service/login");
loginChannel1C.addListener(publishLatch);
loginChannel1C.publish(login1C);
Assert.assertTrue(publishLatch.await(5, TimeUnit.SECONDS));
publishLatch.reset(1);
Map<String, Object> login2 = new HashMap<>();
login2.put("user", "user2");
ClientSessionChannel loginChannel2 = client3.getChannel("/service/login");
loginChannel2.addListener(publishLatch);
loginChannel2.publish(login2);
Assert.assertTrue(publishLatch.await(5, TimeUnit.SECONDS));
Assert.assertTrue(presenceLatch.await(5, TimeUnit.SECONDS));
// Send a message from client3: client1A, client1B and client1C must receive it
String channel = "/service/forward";
final LatchListener messageLatch = new LatchListener(3);
final AtomicInteger counter = new AtomicInteger();
client1A.getChannel(channel).addListener(new ClientSessionChannel.MessageListener() {
@Override
public void onMessage(ClientSessionChannel channel, Message message) {
counter.incrementAndGet();
messageLatch.countDown();
}
});
client1B.getChannel(channel).addListener(new ClientSessionChannel.MessageListener() {
@Override
public void onMessage(ClientSessionChannel channel, Message message) {
counter.incrementAndGet();
messageLatch.countDown();
}
});
client1C.getChannel(channel).addListener(new ClientSessionChannel.MessageListener() {
@Override
public void onMessage(ClientSessionChannel channel, Message message) {
counter.incrementAndGet();
messageLatch.countDown();
}
});
Map<String, Object> data = new HashMap<>();
data.put("peer", "user1");
client3.getChannel(channel).publish(data);
Assert.assertTrue(messageLatch.await(5, TimeUnit.SECONDS));
// Wait a bit more to collect other messages that may be delivered wrongly
Thread.sleep(1000);
// Be sure exactly 3 have been delivered
Assert.assertEquals(3, counter.get());
// Disassociate client1A
publishLatch.reset(1);
Map<String, Object> logout = new HashMap<>();
logout.put("user", "user1");
ClientSessionChannel logoutChannel1A = client1A.getChannel("/service/logout");
logoutChannel1A.addListener(publishLatch);
logoutChannel1A.publish(logout);
Assert.assertTrue(publishLatch.await(5, TimeUnit.SECONDS));
// Send again the message from client3, now only client1B and client1C must get it
counter.set(0);
messageLatch.reset(2);
client3.getChannel(channel).publish(data);
Assert.assertTrue(messageLatch.await(5, TimeUnit.SECONDS));
// Wait a bit more to collect other messages that may be delivered wrongly
Thread.sleep(1000);
// Be sure exactly 2 have been delivered
Assert.assertEquals(2, counter.get());
}
@Test
public void testIsPresent() throws Exception {
Server server1 = startServer(0);
Oort oort1 = startOort(server1);
Server server2 = startServer(0);
Oort oort2 = startOort(server2);
CountDownLatch latch = new CountDownLatch(1);
oort2.addCometListener(new CometJoinedListener(latch));
OortComet oortComet12 = oort1.observeComet(oort2.getURL());
Assert.assertTrue(oortComet12.waitFor(5000, BayeuxClient.State.CONNECTED));
Assert.assertTrue(latch.await(5, TimeUnit.SECONDS));
OortComet oortComet21 = oort2.findComet(oort1.getURL());
Assert.assertTrue(oortComet21.waitFor(5000, BayeuxClient.State.CONNECTED));
Seti seti1 = startSeti(oort1);
Seti seti2 = startSeti(oort2);
new SetiService(seti1);
new SetiService(seti2);
BayeuxClient client1 = startClient(oort1, null);
Assert.assertTrue(client1.waitFor(5000, BayeuxClient.State.CONNECTED));
final CountDownLatch presenceOnLatch = new CountDownLatch(1);
final CountDownLatch presenceOffLatch = new CountDownLatch(1);
Seti.PresenceListener listener = new Seti.PresenceListener() {
@Override
public void presenceAdded(Event event) {
presenceOnLatch.countDown();
}
@Override
public void presenceRemoved(Event event) {
presenceOffLatch.countDown();
}
};
seti2.addPresenceListener(listener);
LatchListener publishLatch = new LatchListener();
Map<String, Object> login1 = new HashMap<>();
String userId = "user1";
login1.put("user", userId);
ClientSessionChannel loginChannel1 = client1.getChannel("/service/login");
loginChannel1.addListener(publishLatch);
loginChannel1.publish(login1);
Assert.assertTrue(publishLatch.await(5, TimeUnit.SECONDS));
Assert.assertTrue(presenceOnLatch.await(5, TimeUnit.SECONDS));
Assert.assertTrue(seti1.isAssociated(userId));
Assert.assertTrue(seti1.isPresent(userId));
Assert.assertTrue(seti2.isPresent(userId));
publishLatch.reset(1);
Map<String, Object> logout1 = new HashMap<>();
logout1.put("user", userId);
ClientSessionChannel logoutChannel1 = client1.getChannel("/service/logout");
logoutChannel1.addListener(publishLatch);
logoutChannel1.publish(logout1);
Assert.assertTrue(publishLatch.await(5, TimeUnit.SECONDS));
Assert.assertTrue(presenceOffLatch.await(5, TimeUnit.SECONDS));
Assert.assertFalse(seti1.isAssociated(userId));
Assert.assertFalse(seti1.isPresent(userId));
Assert.assertFalse(seti2.isPresent(userId));
seti2.removePresenceListener(listener);
}
@Test
public void testIsPresentWhenNodeJoins() throws Exception {
Server server1 = startServer(0);
Oort oort1 = startOort(server1);
Seti seti1 = startSeti(oort1);
new SetiService(seti1);
BayeuxClient client1 = startClient(oort1, null);
Assert.assertTrue(client1.waitFor(5000, BayeuxClient.State.CONNECTED));
Map<String, Object> login1 = new HashMap<>();
String userId = "user1";
login1.put("user", userId);
ClientSessionChannel loginChannel1 = client1.getChannel("/service/login");
final CountDownLatch publishLatch = new CountDownLatch(1);
loginChannel1.publish(login1, new ClientSessionChannel.MessageListener() {
@Override
public void onMessage(ClientSessionChannel channel, Message message) {
publishLatch.countDown();
}
});
Assert.assertTrue(publishLatch.await(5, TimeUnit.SECONDS));
// Now user1 is associated on node1, start node2
Server server2 = startServer(0);
Oort oort2 = startOort(server2);
CountDownLatch latch = new CountDownLatch(1);
oort2.addCometListener(new CometJoinedListener(latch));
OortComet oortComet12 = oort1.observeComet(oort2.getURL());
Assert.assertTrue(oortComet12.waitFor(5000, BayeuxClient.State.CONNECTED));
Assert.assertTrue(latch.await(5, TimeUnit.SECONDS));
OortComet oortComet21 = oort2.findComet(oort1.getURL());
Assert.assertTrue(oortComet21.waitFor(5000, BayeuxClient.State.CONNECTED));
Seti seti2 = startSeti(oort2);
// Wait for the Seti state to broadcast
Thread.sleep(1000);
Assert.assertTrue(seti2.isPresent(userId));
// Stop Seti1
final CountDownLatch presenceOffLatch = new CountDownLatch(1);
seti2.addPresenceListener(new Seti.PresenceListener.Adapter() {
@Override
public void presenceRemoved(Event event) {
presenceOffLatch.countDown();
}
});
stopSeti(seti1);
Assert.assertTrue(presenceOffLatch.await(5, TimeUnit.SECONDS));
}
@Test
public void testPresenceFiresEventLocally() throws Exception {
Server server1 = startServer(0);
Oort oort1 = startOort(server1);
Server server2 = startServer(0);
Oort oort2 = startOort(server2);
CountDownLatch latch = new CountDownLatch(1);
oort2.addCometListener(new CometJoinedListener(latch));
OortComet oortComet12 = oort1.observeComet(oort2.getURL());
Assert.assertTrue(oortComet12.waitFor(5000, BayeuxClient.State.CONNECTED));
Assert.assertTrue(latch.await(5, TimeUnit.SECONDS));
OortComet oortComet21 = oort2.findComet(oort1.getURL());
Assert.assertTrue(oortComet21.waitFor(5000, BayeuxClient.State.CONNECTED));
Seti seti1 = startSeti(oort1);
Seti seti2 = startSeti(oort2);
new SetiService(seti1);
new SetiService(seti2);
BayeuxClient client1 = startClient(oort1, null);
Assert.assertTrue(client1.waitFor(5000, BayeuxClient.State.CONNECTED));
BayeuxClient client2 = startClient(oort2, null);
Assert.assertTrue(client2.waitFor(5000, BayeuxClient.State.CONNECTED));
final CountDownLatch localPresenceOnLatch = new CountDownLatch(1);
final CountDownLatch remotePresenceOnLatch = new CountDownLatch(1);
final CountDownLatch localPresenceOffLatch = new CountDownLatch(1);
final CountDownLatch remotePresenceOffLatch = new CountDownLatch(1);
Seti.PresenceListener listener = new Seti.PresenceListener() {
@Override
public void presenceAdded(Event event) {
if (event.isLocal()) {
localPresenceOnLatch.countDown();
} else {
remotePresenceOnLatch.countDown();
}
}
@Override
public void presenceRemoved(Event event) {
if (event.isLocal()) {
localPresenceOffLatch.countDown();
} else {
remotePresenceOffLatch.countDown();
}
}
};
seti2.addPresenceListener(listener);
// Login user1
final CountDownLatch loginLatch1 = new CountDownLatch(1);
Map<String, Object> login1 = new HashMap<>();
String userId1 = "user1";
login1.put("user", userId1);
ClientSessionChannel loginChannel1 = client1.getChannel("/service/login");
loginChannel1.publish(login1, new ClientSessionChannel.MessageListener() {
@Override
public void onMessage(ClientSessionChannel channel, Message message) {
loginLatch1.countDown();
}
});
Assert.assertTrue(loginLatch1.await(5, TimeUnit.SECONDS));
Assert.assertTrue(remotePresenceOnLatch.await(5, TimeUnit.SECONDS));
// Login user2
final CountDownLatch loginLatch2 = new CountDownLatch(1);
Map<String, Object> login2 = new HashMap<>();
String userId2 = "user2";
login2.put("user", userId2);
ClientSessionChannel loginChannel2 = client2.getChannel("/service/login");
loginChannel2.publish(login2, new ClientSessionChannel.MessageListener() {
@Override
public void onMessage(ClientSessionChannel channel, Message message) {
loginLatch2.countDown();
}
});
Assert.assertTrue(loginLatch2.await(5, TimeUnit.SECONDS));
Assert.assertTrue(localPresenceOnLatch.await(5, TimeUnit.SECONDS));
// Logout user2
final CountDownLatch logoutLatch2 = new CountDownLatch(1);
Map<String, Object> logout2 = new HashMap<>();
logout2.put("user", userId2);
ClientSessionChannel logoutChannel2 = client2.getChannel("/service/logout");
logoutChannel2.publish(logout2, new ClientSessionChannel.MessageListener() {
@Override
public void onMessage(ClientSessionChannel channel, Message message) {
logoutLatch2.countDown();
}
});
Assert.assertTrue(logoutLatch2.await(5, TimeUnit.SECONDS));
Assert.assertTrue(localPresenceOffLatch.await(5, TimeUnit.SECONDS));
// Logout user1
final CountDownLatch logoutLatch1 = new CountDownLatch(1);
Map<String, Object> logout1 = new HashMap<>();
logout1.put("user", userId1);
ClientSessionChannel logoutChannel1 = client1.getChannel("/service/logout");
logoutChannel1.publish(logout1, new ClientSessionChannel.MessageListener() {
@Override
public void onMessage(ClientSessionChannel channel, Message message) {
logoutLatch1.countDown();
}
});
Assert.assertTrue(logoutLatch1.await(5, TimeUnit.SECONDS));
Assert.assertTrue(remotePresenceOffLatch.await(5, TimeUnit.SECONDS));
seti2.removePresenceListener(listener);
}
@Test
public void testStopRemovesAssociationsAndPresences() throws Exception {
Server server1 = startServer(0);
Oort oort1 = startOort(server1);
Server server2 = startServer(0);
Oort oort2 = startOort(server2);
CountDownLatch latch = new CountDownLatch(1);
oort2.addCometListener(new CometJoinedListener(latch));
OortComet oortComet12 = oort1.observeComet(oort2.getURL());
Assert.assertTrue(oortComet12.waitFor(5000, BayeuxClient.State.CONNECTED));
Assert.assertTrue(latch.await(5, TimeUnit.SECONDS));
OortComet oortComet21 = oort2.findComet(oort1.getURL());
Assert.assertTrue(oortComet21.waitFor(5000, BayeuxClient.State.CONNECTED));
Seti seti1 = startSeti(oort1);
Seti seti2 = startSeti(oort2);
new SetiService(seti1);
new SetiService(seti2);
BayeuxClient client1 = startClient(oort1, null);
Assert.assertTrue(client1.waitFor(5000, BayeuxClient.State.CONNECTED));
BayeuxClient client2 = startClient(oort2, null);
Assert.assertTrue(client2.waitFor(5000, BayeuxClient.State.CONNECTED));
final CountDownLatch presenceAddedLatch = new CountDownLatch(4);
seti1.addPresenceListener(new UserPresentListener(presenceAddedLatch));
seti2.addPresenceListener(new UserPresentListener(presenceAddedLatch));
// Login user1
final CountDownLatch loginLatch1 = new CountDownLatch(1);
Map<String, Object> login1 = new HashMap<>();
String userId1 = "user1";
login1.put("user", userId1);
ClientSessionChannel loginChannel1 = client1.getChannel("/service/login");
loginChannel1.publish(login1, new ClientSessionChannel.MessageListener() {
@Override
public void onMessage(ClientSessionChannel channel, Message message) {
loginLatch1.countDown();
}
});
Assert.assertTrue(loginLatch1.await(5, TimeUnit.SECONDS));
// Login user2
final CountDownLatch loginLatch2 = new CountDownLatch(1);
Map<String, Object> login2 = new HashMap<>();
String userId2 = "user2";
login2.put("user", userId2);
ClientSessionChannel loginChannel2 = client2.getChannel("/service/login");
loginChannel2.publish(login2, new ClientSessionChannel.MessageListener() {
@Override
public void onMessage(ClientSessionChannel channel, Message message) {
loginLatch2.countDown();
}
});
Assert.assertTrue(loginLatch2.await(5, TimeUnit.SECONDS));
// Make sure all Setis see all users
Assert.assertTrue(presenceAddedLatch.await(5, TimeUnit.SECONDS));
final CountDownLatch presenceRemovedLatch = new CountDownLatch(1);
seti2.addPresenceListener(new UserAbsentListener(presenceRemovedLatch));
// Stop Seti1
stopSeti(seti1);
Assert.assertTrue(presenceRemovedLatch.await(5, TimeUnit.SECONDS));
// Make sure Seti1 is cleared
Assert.assertFalse(seti1.isAssociated(userId1));
Assert.assertFalse(seti1.isPresent(userId2));
// Make sure user1 is gone from Seti2
Assert.assertTrue(seti2.isAssociated(userId2));
Assert.assertFalse(seti2.isPresent(userId1));
}
@Test
public void testNetworkDisconnectAndReconnect() throws Exception {
Server server1 = startServer(0);
Oort oort1 = startOort(server1);
Server server2 = startServer(0);
Oort oort2 = startOort(server2);
CountDownLatch latch = new CountDownLatch(1);
oort2.addCometListener(new CometJoinedListener(latch));
OortComet oortComet12 = oort1.observeComet(oort2.getURL());
Assert.assertTrue(oortComet12.waitFor(5000, BayeuxClient.State.CONNECTED));
Assert.assertTrue(latch.await(5, TimeUnit.SECONDS));
OortComet oortComet21 = oort2.findComet(oort1.getURL());
Assert.assertTrue(oortComet21.waitFor(5000, BayeuxClient.State.CONNECTED));
Seti seti1 = startSeti(oort1);
Seti seti2 = startSeti(oort2);
new SetiService(seti1);
new SetiService(seti2);
BayeuxClient client1 = startClient(oort1, null);
Assert.assertTrue(client1.waitFor(5000, BayeuxClient.State.CONNECTED));
BayeuxClient client2 = startClient(oort2, null);
Assert.assertTrue(client2.waitFor(5000, BayeuxClient.State.CONNECTED));
CountDownLatch presenceAddedLatch = new CountDownLatch(4);
seti1.addPresenceListener(new UserPresentListener(presenceAddedLatch));
seti2.addPresenceListener(new UserPresentListener(presenceAddedLatch));
// Login user1
final CountDownLatch loginLatch1 = new CountDownLatch(1);
Map<String, Object> login1 = new HashMap<>();
String userId1 = "user1";
login1.put("user", userId1);
ClientSessionChannel loginChannel1 = client1.getChannel("/service/login");
loginChannel1.publish(login1, new ClientSessionChannel.MessageListener() {
@Override
public void onMessage(ClientSessionChannel channel, Message message) {
loginLatch1.countDown();
}
});
Assert.assertTrue(loginLatch1.await(5, TimeUnit.SECONDS));
// Login user2
final CountDownLatch loginLatch2 = new CountDownLatch(1);
Map<String, Object> login2 = new HashMap<>();
String userId2 = "user2";
login2.put("user", userId2);
ClientSessionChannel loginChannel2 = client2.getChannel("/service/login");
loginChannel2.publish(login2, new ClientSessionChannel.MessageListener() {
@Override
public void onMessage(ClientSessionChannel channel, Message message) {
loginLatch2.countDown();
}
});
Assert.assertTrue(loginLatch2.await(5, TimeUnit.SECONDS));
// Make sure all Setis see all users
Assert.assertTrue(presenceAddedLatch.await(5, TimeUnit.SECONDS));
final CountDownLatch presenceRemovedLatch = new CountDownLatch(2);
seti1.addPresenceListener(new UserAbsentListener(presenceRemovedLatch));
seti2.addPresenceListener(new UserAbsentListener(presenceRemovedLatch));
// Simulate network crash
oortComet12.disconnect();
oortComet12.waitFor(5000, BayeuxClient.State.DISCONNECTED);
// The other OortComet is automatically disconnected
oortComet21.waitFor(5000, BayeuxClient.State.DISCONNECTED);
Assert.assertTrue(presenceRemovedLatch.await(5, TimeUnit.SECONDS));
// Make sure user1 is gone from Seti2
Assert.assertTrue(seti1.isAssociated(userId1));
Assert.assertFalse(seti2.isPresent(userId1));
// Make sure user2 is gone from Seti1
Assert.assertTrue(seti2.isAssociated(userId2));
Assert.assertFalse(seti2.isPresent(userId1));
Assert.assertEquals(1, seti2.getAssociationCount(userId2));
// Simulate network is up again
presenceAddedLatch = new CountDownLatch(2);
seti1.addPresenceListener(new UserPresentListener(presenceAddedLatch));
seti2.addPresenceListener(new UserPresentListener(presenceAddedLatch));
latch = new CountDownLatch(1);
oort2.addCometListener(new CometJoinedListener(latch));
oortComet12 = oort1.observeComet(oort2.getURL());
Assert.assertTrue(oortComet12.waitFor(5000, BayeuxClient.State.CONNECTED));
Assert.assertTrue(latch.await(5, TimeUnit.SECONDS));
oortComet21 = oort2.findComet(oort1.getURL());
Assert.assertTrue(oortComet21.waitFor(5000, BayeuxClient.State.CONNECTED));
Assert.assertTrue(presenceAddedLatch.await(5, TimeUnit.SECONDS));
Assert.assertTrue(seti1.isAssociated(userId1));
Assert.assertTrue(seti1.isPresent(userId2));
Assert.assertTrue(seti2.isAssociated(userId2));
Assert.assertTrue(seti2.isPresent(userId1));
Set<String> userIds = seti1.getUserIds();
Assert.assertEquals(2, userIds.size());
Assert.assertTrue(userIds.contains(userId1));
Assert.assertTrue(userIds.contains(userId2));
Assert.assertEquals(1, seti1.getAssociationCount(userId1));
Assert.assertEquals(0, seti1.getAssociationCount(userId2));
Assert.assertEquals(1, seti1.getPresenceCount(userId1));
Assert.assertEquals(1, seti1.getPresenceCount(userId2));
}
@Test
public void testMultipleServerCrashes() throws Exception {
Server server1 = startServer(0);
Oort oort1 = startOort(server1);
Server server2 = startServer(0);
Oort oort2 = startOort(server2);
CountDownLatch oortLatch = new CountDownLatch(1);
oort2.addCometListener(new CometJoinedListener(oortLatch));
OortComet oortComet12 = oort1.observeComet(oort2.getURL());
Assert.assertTrue(oortComet12.waitFor(5000, BayeuxClient.State.CONNECTED));
Assert.assertTrue(oortLatch.await(5, TimeUnit.SECONDS));
OortComet oortComet21 = oort2.findComet(oort1.getURL());
Assert.assertTrue(oortComet21.waitFor(5000, BayeuxClient.State.CONNECTED));
Seti seti1 = startSeti(oort1);
Seti seti2 = startSeti(oort2);
new SetiService(seti1);
new SetiService(seti2);
BayeuxClient client1 = startClient(oort1, null);
Assert.assertTrue(client1.waitFor(5000, BayeuxClient.State.CONNECTED));
CountDownLatch presenceAddedLatch = new CountDownLatch(2);
seti1.addPresenceListener(new UserPresentListener(presenceAddedLatch));
seti2.addPresenceListener(new UserPresentListener(presenceAddedLatch));
// Login user1
final CountDownLatch loginLatch1 = new CountDownLatch(1);
Map<String, Object> login1 = new HashMap<>();
String userId1 = "user1";
login1.put("user", userId1);
ClientSessionChannel loginChannel1 = client1.getChannel("/service/login");
loginChannel1.publish(login1, new ClientSessionChannel.MessageListener() {
@Override
public void onMessage(ClientSessionChannel channel, Message message) {
loginLatch1.countDown();
}
});
Assert.assertTrue(loginLatch1.await(5, TimeUnit.SECONDS));
Assert.assertTrue(presenceAddedLatch.await(5, TimeUnit.SECONDS));
int switches = 2;
for (int i = 0; i < switches; ++i) {
// Simulate network crash
oortComet12.disconnect();
oortComet12.waitFor(5000, BayeuxClient.State.DISCONNECTED);
// The other OortComet is automatically disconnected
oortComet21.waitFor(5000, BayeuxClient.State.DISCONNECTED);
// Stop node1
int port1 = ((NetworkConnector)server1.getConnectors()[0]).getLocalPort();
stopSeti(seti1);
stopOort(oort1);
stopServer(server1);
// Disconnect user and login it to node2
client1.disconnect();
Assert.assertTrue(client1.waitFor(5000, BayeuxClient.State.DISCONNECTED));
client1 = startClient(oort2, null);
final CountDownLatch loginLatch2 = new CountDownLatch(1);
loginChannel1 = client1.getChannel("/service/login");
loginChannel1.publish(login1, new ClientSessionChannel.MessageListener() {
@Override
public void onMessage(ClientSessionChannel channel, Message message) {
loginLatch2.countDown();
}
});
Assert.assertTrue(loginLatch2.await(5, TimeUnit.SECONDS));
// Bring node1 back online
server1 = startServer(port1);
oort1 = startOort(server1);
oortLatch = new CountDownLatch(1);
oort2.addCometListener(new CometJoinedListener(oortLatch));
oortComet12 = oort1.observeComet(oort2.getURL());
Assert.assertTrue(oortComet12.waitFor(5000, BayeuxClient.State.CONNECTED));
Assert.assertTrue(oortLatch.await(5, TimeUnit.SECONDS));
oortComet21 = oort2.findComet(oort1.getURL());
Assert.assertTrue(oortComet21.waitFor(5000, BayeuxClient.State.CONNECTED));
seti1 = startSeti(oort1);
new SetiService(seti1);
// Wait for cloud/seti notifications to happen
Thread.sleep(1000);
System.err.println(seti1.dump());
System.err.println(seti2.dump());
Assert.assertFalse(seti1.isAssociated(userId1));
Assert.assertTrue(seti1.isPresent(userId1));
Assert.assertTrue(seti2.isAssociated(userId1));
Assert.assertTrue(seti2.isPresent(userId1));
// Simulate network crash
oortComet12.disconnect();
oortComet12.waitFor(5000, BayeuxClient.State.DISCONNECTED);
// The other OortComet is automatically disconnected
oortComet21.waitFor(5000, BayeuxClient.State.DISCONNECTED);
// Stop node2
int port2 = ((NetworkConnector)server2.getConnectors()[0]).getLocalPort();
stopSeti(seti2);
stopOort(oort2);
stopServer(server2);
// Disconnect user and login it to node1
client1.disconnect();
Assert.assertTrue(client1.waitFor(5000, BayeuxClient.State.DISCONNECTED));
client1 = startClient(oort1, null);
final CountDownLatch loginLatch3 = new CountDownLatch(1);
loginChannel1 = client1.getChannel("/service/login");
loginChannel1.publish(login1, new ClientSessionChannel.MessageListener() {
@Override
public void onMessage(ClientSessionChannel channel, Message message) {
loginLatch3.countDown();
}
});
Assert.assertTrue(loginLatch3.await(5, TimeUnit.SECONDS));
// Bring node2 back online
server2 = startServer(port2);
oort2 = startOort(server2);
oortLatch = new CountDownLatch(1);
oort1.addCometListener(new CometJoinedListener(oortLatch));
oortComet21 = oort2.observeComet(oort1.getURL());
Assert.assertTrue(oortComet21.waitFor(5000, BayeuxClient.State.CONNECTED));
Assert.assertTrue(oortLatch.await(5, TimeUnit.SECONDS));
oortComet12 = oort1.findComet(oort2.getURL());
Assert.assertTrue(oortComet12.waitFor(5000, BayeuxClient.State.CONNECTED));
seti2 = startSeti(oort2);
new SetiService(seti2);
// Wait for cloud/seti notifications to happen
Thread.sleep(1000);
System.err.println(seti1.dump());
System.err.println(seti2.dump());
Assert.assertTrue(seti1.isAssociated(userId1));
Assert.assertTrue(seti1.isPresent(userId1));
Assert.assertFalse(seti2.isAssociated(userId1));
Assert.assertTrue(seti2.isPresent(userId1));
}
}
@Test
public void testMessageToObservedChannelIsForwarded() throws Exception {
testForwardBehaviour(true);
}
@Test
public void testMessageToNonObservedChannelIsNotForwarded() throws Exception {
testForwardBehaviour(false);
}
private void testForwardBehaviour(boolean forward) throws Exception {
Server server1 = startServer(0);
Oort oort1 = startOort(server1);
Server server2 = startServer(0);
Oort oort2 = startOort(server2);
CountDownLatch latch = new CountDownLatch(1);
oort2.addCometListener(new CometJoinedListener(latch));
OortComet oortComet12 = oort1.observeComet(oort2.getURL());
Assert.assertTrue(oortComet12.waitFor(5000, BayeuxClient.State.CONNECTED));
Assert.assertTrue(latch.await(5, TimeUnit.SECONDS));
OortComet oortComet21 = oort2.findComet(oort1.getURL());
Assert.assertTrue(oortComet21.waitFor(5000, BayeuxClient.State.CONNECTED));
final Seti seti1 = startSeti(oort1);
Seti seti2 = startSeti(oort2);
new SetiService(seti1);
new SetiService(seti2);
BayeuxClient client1 = startClient(oort1, null);
Assert.assertTrue(client1.waitFor(5000, BayeuxClient.State.CONNECTED));
BayeuxClient client2 = startClient(oort2, null);
Assert.assertTrue(client2.waitFor(5000, BayeuxClient.State.CONNECTED));
CountDownLatch presenceAddedLatch = new CountDownLatch(4);
seti1.addPresenceListener(new UserPresentListener(presenceAddedLatch));
seti2.addPresenceListener(new UserPresentListener(presenceAddedLatch));
// Login user1
final CountDownLatch loginLatch1 = new CountDownLatch(1);
Map<String, Object> login1 = new HashMap<String, Object>();
String userId1 = "user1";
login1.put("user", userId1);
ClientSessionChannel loginChannel1 = client1.getChannel("/service/login");
loginChannel1.publish(login1, new ClientSessionChannel.MessageListener() {
@Override
public void onMessage(ClientSessionChannel channel, Message message) {
loginLatch1.countDown();
}
});
Assert.assertTrue(loginLatch1.await(5, TimeUnit.SECONDS));
// Login user2
final CountDownLatch loginLatch2 = new CountDownLatch(1);
Map<String, Object> login2 = new HashMap<String, Object>();
String userId2 = "user2";
login2.put("user", userId2);
ClientSessionChannel loginChannel2 = client2.getChannel("/service/login");
loginChannel2.publish(login2, new ClientSessionChannel.MessageListener() {
@Override
public void onMessage(ClientSessionChannel channel, Message message) {
loginLatch2.countDown();
}
});
Assert.assertTrue(loginLatch2.await(5, TimeUnit.SECONDS));
// Make sure all Setis see all users
Assert.assertTrue(presenceAddedLatch.await(5, TimeUnit.SECONDS));
// Setup test: register a service for the service channel
// that broadcasts to another channel that is not observed
final String serviceChannel = "/service/foo";
final String broadcastChannel = "/foo";
if (forward) {
oort2.observeChannel(broadcastChannel);
// Give some time for the subscribe to happen
Thread.sleep(1000);
}
new BroadcastService(seti1, serviceChannel, broadcastChannel);
// Subscribe user2
LatchListener subscribeListener = new LatchListener(1);
final CountDownLatch messageLatch = new CountDownLatch(1);
client2.getChannel(Channel.META_SUBSCRIBE).addListener(subscribeListener);
client2.getChannel(broadcastChannel).subscribe(new ClientSessionChannel.MessageListener() {
@Override
public void onMessage(ClientSessionChannel channel, Message message) {
messageLatch.countDown();
}
});
Assert.assertTrue(subscribeListener.await(5, TimeUnit.SECONDS));
client1.getChannel(serviceChannel).publish("data1");
Assert.assertEquals(forward, messageLatch.await(1, TimeUnit.SECONDS));
}
public static class BroadcastService extends AbstractService {
private final String broadcastChannel;
public BroadcastService(Seti seti, String channel, String broadcastChannel) {
super(seti.getOort().getBayeuxServer(), seti.getId());
this.broadcastChannel = broadcastChannel;
addService(channel, "process");
}
public void process(ServerSession session, ServerMessage message) {
getLocalSession().getChannel(broadcastChannel).publish("data2");
}
}
public static class SetiService extends AbstractService {
private final Seti seti;
private SetiService(Seti seti) {
super(seti.getOort().getBayeuxServer(), seti.getId());
this.seti = seti;
addService("/service/login", "login");
addService("/service/logout", "logout");
addService("/service/forward", "forward");
}
public void login(ServerSession session, ServerMessage message) {
Map<String, Object> data = message.getDataAsMap();
String user = (String)data.get("user");
seti.associate(user, session);
}
public void logout(ServerSession session, ServerMessage message) {
Map<String, Object> data = message.getDataAsMap();
String user = (String)data.get("user");
seti.disassociate(user, session);
}
public void forward(ServerSession session, ServerMessage message) {
Map<String, Object> data = message.getDataAsMap();
String peer = (String)data.get("peer");
seti.sendMessage(peer, message.getChannel(), data);
}
}
@Test
public void testConcurrent() throws Exception {
Server server1 = startServer(0);
final Oort oort1 = startOort(server1);
Server server2 = startServer(0);
Oort oort2 = startOort(server2);
CountDownLatch latch = new CountDownLatch(1);
oort2.addCometListener(new CometJoinedListener(latch));
OortComet oortComet12 = oort1.observeComet(oort2.getURL());
Assert.assertTrue(oortComet12.waitFor(5000, BayeuxClient.State.CONNECTED));
Assert.assertTrue(latch.await(5, TimeUnit.SECONDS));
OortComet oortComet21 = oort2.findComet(oort1.getURL());
Assert.assertTrue(oortComet21.waitFor(5000, BayeuxClient.State.CONNECTED));
final Seti seti1 = startSeti(oort1);
Seti seti2 = startSeti(oort2);
int threads = 64;
final int iterations = 32;
final CyclicBarrier barrier = new CyclicBarrier(threads + 1);
CountDownLatch presentLatch = new CountDownLatch(threads * iterations);
seti2.addPresenceListener(new UserPresentListener(presentLatch));
CountDownLatch absentLatch = new CountDownLatch(threads * iterations);
seti2.addPresenceListener(new UserAbsentListener(absentLatch));
for (int i = 0; i < threads; ++i) {
final int index = i;
new Thread(new Runnable() {
@Override
public void run() {
try {
Map<String, ServerSession> sessions = new HashMap<>();
barrier.await();
for (int j = 0; j < iterations; ++j) {
String key = String.valueOf(index * iterations + j);
LocalSession localSession = oort1.getBayeuxServer().newLocalSession(key);
localSession.handshake();
ServerSession session = localSession.getServerSession();
sessions.put(key, session);
seti1.associate(key, session);
}
barrier.await();
for (Map.Entry<String, ServerSession> entry : sessions.entrySet()) {
seti1.disassociate(entry.getKey(), entry.getValue());
}
} catch (Throwable x) {
x.printStackTrace();
}
}
}).start();
}
// Wait for all threads to be ready.
barrier.await();
// Wait for all threads to finish associations.
Assert.assertTrue(presentLatch.await(30, TimeUnit.SECONDS));
// The 2 Setis should be in sync.
Assert.assertEquals(seti1.getUserIds(), seti2.getUserIds());
// Start disassociations.
barrier.await();
// Wait for all threads to finish disassociations.
Assert.assertTrue(absentLatch.await(30, TimeUnit.SECONDS));
// The 2 Setis should be empty.
Assert.assertEquals(0, seti1.getAssociatedUserIds().size());
Assert.assertEquals(0, seti1.getUserIds().size());
Assert.assertEquals(0, seti2.getUserIds().size());
}
@Test
public void testDisassociationRemovesListeners() throws Exception {
Server server1 = startServer(0);
Oort oort1 = startOort(server1);
Seti seti1 = startSeti(oort1);
String user = "user";
LocalSession localSession = oort1.getBayeuxServer().newLocalSession(user);
localSession.handshake();
ServerSession session = localSession.getServerSession();
seti1.associate(user, session);
seti1.disassociate(user, session);
Assert.assertEquals(0, ((ServerSessionImpl)session).getListeners().size());
}
private static class UserPresentListener extends Seti.PresenceListener.Adapter {
private final CountDownLatch latch;
private UserPresentListener(CountDownLatch latch) {
this.latch = latch;
}
@Override
public void presenceAdded(Event event) {
latch.countDown();
}
}
private static class UserAbsentListener extends Seti.PresenceListener.Adapter {
private final CountDownLatch latch;
private UserAbsentListener(CountDownLatch latch) {
this.latch = latch;
}
@Override
public void presenceRemoved(Event event) {
latch.countDown();
}
}
}