243 lines
9.9 KiB
Java
243 lines
9.9 KiB
Java
|
/* Copyright (c) 2008, Nathan Sweet
|
||
|
* All rights reserved.
|
||
|
*
|
||
|
* Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following
|
||
|
* conditions are met:
|
||
|
*
|
||
|
* - Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer.
|
||
|
* - Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the following
|
||
|
* disclaimer in the documentation and/or other materials provided with the distribution.
|
||
|
* - Neither the name of Esoteric Software nor the names of its contributors may be used to endorse or promote products derived
|
||
|
* from this software without specific prior written permission.
|
||
|
*
|
||
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING,
|
||
|
* BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT
|
||
|
* SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
|
||
|
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
||
|
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
|
||
|
* NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||
|
*/
|
||
|
|
||
|
import static org.junit.Assert.assertEquals;
|
||
|
|
||
|
import java.io.IOException;
|
||
|
import java.util.ArrayList;
|
||
|
import java.util.List;
|
||
|
import java.util.Map;
|
||
|
import java.util.concurrent.ConcurrentHashMap;
|
||
|
import java.util.concurrent.atomic.AtomicInteger;
|
||
|
|
||
|
import org.junit.Test;
|
||
|
|
||
|
import dorkbox.network.connection.Connection;
|
||
|
import dorkbox.network.connection.Listener;
|
||
|
import dorkbox.network.connection.Listeners;
|
||
|
import dorkbox.network.serialization.Serialization;
|
||
|
import dorkbox.util.exceptions.SecurityException;
|
||
|
|
||
|
public
|
||
|
class MultipleThreadTest extends BaseTest {
|
||
|
private final Object lock = new Object();
|
||
|
private volatile boolean stillRunning = false;
|
||
|
|
||
|
private final Object finalRunLock = new Object();
|
||
|
private volatile boolean finalStillRunning = false;
|
||
|
|
||
|
private final int messageCount = 150;
|
||
|
private final int threadCount = 15;
|
||
|
private final int clientCount = 13;
|
||
|
|
||
|
private final List<Client> clients = new ArrayList<Client>(this.clientCount);
|
||
|
|
||
|
int perClientReceiveTotal = (this.messageCount * this.threadCount);
|
||
|
int serverReceiveTotal = perClientReceiveTotal * this.clientCount;
|
||
|
|
||
|
AtomicInteger sent = new AtomicInteger(0);
|
||
|
AtomicInteger totalClientReceived = new AtomicInteger(0);
|
||
|
AtomicInteger receivedServer = new AtomicInteger(1);
|
||
|
|
||
|
ConcurrentHashMap<Integer, DataClass> sentStringsToClientDebug = new ConcurrentHashMap<Integer, DataClass>();
|
||
|
|
||
|
@Test
|
||
|
public
|
||
|
void multipleThreads() throws SecurityException, IOException {
|
||
|
// our clients should receive messageCount * threadCount * clientCount TOTAL messages
|
||
|
final int totalClientReceivedCountExpected = this.clientCount * this.messageCount * this.threadCount;
|
||
|
final int totalServerReceivedCountExpected = this.clientCount * this.messageCount;
|
||
|
|
||
|
System.err.println("CLIENT RECEIVES: " + totalClientReceivedCountExpected);
|
||
|
System.err.println("SERVER RECEIVES: " + totalServerReceivedCountExpected);
|
||
|
|
||
|
|
||
|
Configuration configuration = new Configuration();
|
||
|
configuration.tcpPort = tcpPort;
|
||
|
configuration.host = host;
|
||
|
configuration.serialization.register(String[].class);
|
||
|
configuration.serialization.register(DataClass.class);
|
||
|
|
||
|
|
||
|
final Server server = new Server(configuration);
|
||
|
server.disableRemoteKeyValidation();
|
||
|
|
||
|
addEndPoint(server);
|
||
|
server.bind(false);
|
||
|
|
||
|
|
||
|
final Listeners listeners = server.listeners();
|
||
|
listeners.add(new Listener.OnConnected<Connection>() {
|
||
|
|
||
|
@Override
|
||
|
public
|
||
|
void connected(final Connection connection) {
|
||
|
System.err.println("Client connected to server.");
|
||
|
|
||
|
// kickoff however many threads we need, and send data to the client.
|
||
|
for (int i = 1; i <= MultipleThreadTest.this.threadCount; i++) {
|
||
|
final int index = i;
|
||
|
new Thread() {
|
||
|
@Override
|
||
|
public
|
||
|
void run() {
|
||
|
for (int i = 1; i <= MultipleThreadTest.this.messageCount; i++) {
|
||
|
int incrementAndGet = MultipleThreadTest.this.sent.getAndIncrement();
|
||
|
DataClass dataClass = new DataClass("Server -> client. Thread #" + index + " message# " + incrementAndGet,
|
||
|
incrementAndGet);
|
||
|
|
||
|
//System.err.println(dataClass.data);
|
||
|
MultipleThreadTest.this.sentStringsToClientDebug.put(incrementAndGet, dataClass);
|
||
|
connection.send()
|
||
|
.TCP(dataClass)
|
||
|
.flush();
|
||
|
}
|
||
|
|
||
|
}
|
||
|
}.start();
|
||
|
}
|
||
|
}
|
||
|
});
|
||
|
|
||
|
listeners.add(new Listener.OnMessageReceived<Connection, DataClass>() {
|
||
|
@Override
|
||
|
public
|
||
|
void received(Connection connection, DataClass object) {
|
||
|
int incrementAndGet = MultipleThreadTest.this.receivedServer.getAndIncrement();
|
||
|
//System.err.println("server #" + incrementAndGet);
|
||
|
|
||
|
|
||
|
if (incrementAndGet % MultipleThreadTest.this.messageCount == 0) {
|
||
|
System.err.println("Server receive DONE for client " + incrementAndGet);
|
||
|
|
||
|
stillRunning = false;
|
||
|
synchronized (MultipleThreadTest.this.lock) {
|
||
|
MultipleThreadTest.this.lock.notifyAll();
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if (incrementAndGet == totalServerReceivedCountExpected) {
|
||
|
System.err.println("Server DONE: " + incrementAndGet);
|
||
|
|
||
|
finalStillRunning = false;
|
||
|
synchronized (MultipleThreadTest.this.finalRunLock) {
|
||
|
MultipleThreadTest.this.finalRunLock.notifyAll();
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
});
|
||
|
|
||
|
// ----
|
||
|
finalStillRunning = true;
|
||
|
for (int i = 1; i <= this.clientCount; i++) {
|
||
|
final int index = i;
|
||
|
|
||
|
Client client = new Client(configuration);
|
||
|
this.clients.add(client);
|
||
|
|
||
|
addEndPoint(client);
|
||
|
client.listeners()
|
||
|
.add(new Listener.OnMessageReceived<Connection, DataClass>() {
|
||
|
final int clientIndex = index;
|
||
|
final AtomicInteger received = new AtomicInteger(1);
|
||
|
|
||
|
@Override
|
||
|
public
|
||
|
void received(Connection connection, DataClass object) {
|
||
|
totalClientReceived.getAndIncrement();
|
||
|
int clientLocalCounter = this.received.getAndIncrement();
|
||
|
MultipleThreadTest.this.sentStringsToClientDebug.remove(object.index);
|
||
|
|
||
|
//System.err.println(object.data);
|
||
|
// we finished!!
|
||
|
if (clientLocalCounter == perClientReceiveTotal) {
|
||
|
//System.err.println("Client #" + clientIndex + " received " + clientLocalCounter + " Sending back " +
|
||
|
// MultipleThreadTest.this.messageCount + " messages.");
|
||
|
|
||
|
// now spam back messages!
|
||
|
for (int i = 0; i < MultipleThreadTest.this.messageCount; i++) {
|
||
|
connection.send()
|
||
|
.TCP(new DataClass("Client #" + clientIndex + " -> Server message " + i, index));
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
});
|
||
|
|
||
|
|
||
|
stillRunning = true;
|
||
|
|
||
|
client.connect(5000);
|
||
|
|
||
|
while (stillRunning) {
|
||
|
synchronized (this.lock) {
|
||
|
try {
|
||
|
this.lock.wait(5 * 1000); // 5 secs
|
||
|
} catch (InterruptedException e) {
|
||
|
e.printStackTrace();
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
while (finalStillRunning) {
|
||
|
synchronized (this.finalRunLock) {
|
||
|
try {
|
||
|
this.finalRunLock.wait(5 * 1000); // 5 secs
|
||
|
} catch (InterruptedException e) {
|
||
|
e.printStackTrace();
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// CLIENT will wait until it's done connecting, but SERVER is async.
|
||
|
// the ONLY way to safely work in the server is with LISTENERS. Everything else can FAIL, because of it's async nature.
|
||
|
|
||
|
if (!this.sentStringsToClientDebug.isEmpty()) {
|
||
|
System.err.println("MISSED DATA: " + this.sentStringsToClientDebug.size());
|
||
|
for (Map.Entry<Integer, DataClass> i : this.sentStringsToClientDebug.entrySet()) {
|
||
|
System.err.println(i.getKey() + " : " + i.getValue().data);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
stopEndPoints();
|
||
|
assertEquals(totalClientReceivedCountExpected, totalClientReceived.get());
|
||
|
|
||
|
// offset by 1 since we start at 1
|
||
|
assertEquals(totalServerReceivedCountExpected, receivedServer.get()-1);
|
||
|
}
|
||
|
|
||
|
|
||
|
public static
|
||
|
class DataClass {
|
||
|
public String data;
|
||
|
public Integer index;
|
||
|
|
||
|
public
|
||
|
DataClass() {
|
||
|
}
|
||
|
|
||
|
public
|
||
|
DataClass(String data, Integer index) {
|
||
|
this.data = data;
|
||
|
this.index = index;
|
||
|
}
|
||
|
}
|
||
|
}
|