use of org.jivesoftware.smackx.packet.Time in project Openfire by igniterealtime.
the class ThrottleTestWriter method main.
/**
* Starts the throttle test write client.
*
* @param args application arguments.
*/
public static void main(String[] args) {
if (args.length != 3) {
System.out.println("Usage: java ThrottleTestWriter [server] [username] [password]");
System.exit(0);
}
String server = args[0];
String username = args[1];
String password = args[2];
try {
// Connect to the server, without TLS encryption.
ConnectionConfiguration config = new ConnectionConfiguration(server);
config.setSecurityMode(ConnectionConfiguration.SecurityMode.disabled);
final XMPPConnection con = new XMPPConnection(config);
System.out.print("Connecting to " + server + "... ");
con.connect();
con.login(username, password, "writer");
System.out.print("success.");
System.out.println("");
// Get the "real" server address.
server = con.getServiceName();
String writerAddress = username + "@" + server + "/writer";
final String readerAddress = username + "@" + server + "/reader";
System.out.println("Registered as " + writerAddress);
// Look for the reader process.
System.out.print("Looking for " + readerAddress + "...");
while (true) {
IQ testIQ = new Time();
testIQ.setType(IQ.Type.GET);
testIQ.setTo(readerAddress);
PacketCollector collector = con.createPacketCollector(new PacketIDFilter(testIQ.getPacketID()));
con.sendPacket(testIQ);
// Wait 5 seconds.
long start = System.currentTimeMillis();
Packet result = collector.nextResult(5000);
collector.cancel();
// If we got a result, continue.
if (result != null && result.getError() == null) {
System.out.println(" found reader. Starting packet flood.");
break;
}
System.out.print(".");
long end = System.currentTimeMillis();
if (end - start < 5000) {
try {
Thread.sleep(5000 - (end - start));
} catch (Exception e) {
// ignore.
}
}
}
// Create a process to log how many packets we're writing out.
Runnable statsRunnable = new Runnable() {
public void run() {
while (!done) {
try {
Thread.sleep(5000);
} catch (Exception e) {
/* ignore */
}
int count = packetCount.getAndSet(0);
System.out.println("Packets per second: " + (count / 5));
}
}
};
Thread statsThread = new Thread(statsRunnable);
statsThread.setDaemon(true);
statsThread.start();
// Now start flooding packets.
Message testMessage = new Message(readerAddress);
testMessage.setBody("aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa");
while (!done) {
con.sendPacket(testMessage);
packetCount.getAndIncrement();
}
} catch (Exception e) {
System.out.println("\nError: " + e.getMessage());
e.printStackTrace();
}
}
use of org.jivesoftware.smackx.packet.Time in project Openfire by igniterealtime.
the class ThrottleTestReader method main.
/**
* Starts the throttle test reader client.
*
* @param args application arguments.
*/
public static void main(String[] args) {
if (args.length != 3) {
System.out.println("Usage: java ThrottleTestReader [server] [username] [password]");
System.exit(0);
}
String server = args[0];
String username = args[1];
String password = args[2];
try {
// Connect to the server, without TLS encryption.
ConnectionConfiguration config = new ConnectionConfiguration(server);
config.setSecurityMode(ConnectionConfiguration.SecurityMode.disabled);
final XMPPConnection con = new XMPPConnection(config);
System.out.print("Connecting to " + server + "... ");
con.connect();
con.login(username, password, "reader");
System.out.print("success.");
System.out.println("");
// Get the "real" server address.
server = con.getServiceName();
final String writerAddress = username + "@" + server + "/writer";
String readerAddress = username + "@" + server + "/reader";
System.out.println("Registered as " + readerAddress);
// Look for the reader process.
System.out.print("Waiting for " + writerAddress + "...");
PacketCollector collector = con.createPacketCollector(new AndFilter(new FromMatchesFilter(writerAddress), new PacketTypeFilter(Time.class)));
Time timeRequest = (Time) collector.nextResult();
Time timeReply = new Time(Calendar.getInstance());
timeReply.setPacketID(timeRequest.getPacketID());
timeReply.setType(IQ.Type.RESULT);
timeReply.setTo(timeRequest.getFrom());
con.sendPacket(timeReply);
System.out.println(" found writer. Now in reading mode.");
// Track how many packets we've read.
con.addPacketListener(new PacketListener() {
public void processPacket(Packet packet) {
packetCount.getAndIncrement();
}
}, new PacketTypeFilter(Message.class));
while (!done) {
Thread.sleep(5000);
int count = packetCount.getAndSet(0);
System.out.println("Packets per second: " + (count / 5));
}
// Sleep while we're reading packets.
Thread.sleep(Integer.MAX_VALUE);
} catch (Exception e) {
System.out.println("\nError: " + e.getMessage());
}
}
Aggregations