package org.jgroups.tests;
import org.jgroups.ReceiverAdapter;
import java.io.IOException;
import java.net.DatagramPacket;
import java.net.DatagramSocket;
import java.net.InetAddress;
/**
* Class that measure RTT between a client and server using datagram sockets
* @author Bela Ban
* @version $Id: RoundTripUdp.java,v 1.2 2006/08/05 19:30:59 belaban Exp $
*/
public class RoundTripUdp extends ReceiverAdapter {
DatagramSocket sock;
InetAddress host;
int port=7500;
int num=1000;
int msg_size=10;
boolean server=false;
final byte[] RSP_BUF=new byte[]{1}; // 1=response
int num_responses=0;
private void start(boolean server, int num, int msg_size, InetAddress host, int port) throws IOException {
this.server=server;
this.num=num;
this.msg_size=msg_size;
this.host=host;
this.port=port;
if(server) {
sock=new DatagramSocket(port, host);
System.out.println("server started (ctrl-c to kill)");
while(true) {
byte[] buf=new byte[65000];
DatagramPacket packet;
packet=new DatagramPacket(buf, 0, buf.length);
sock.receive(packet);
packet=new DatagramPacket(RSP_BUF, 0, RSP_BUF.length, packet.getAddress(), packet.getPort());
sock.send(packet); // send the response
}
}
else {
sock=new DatagramSocket();
System.out.println("sending " + num + " requests");
sendRequests();
}
sock.close();
}
private void sendRequests() {
byte[] buf=new byte[msg_size];
long start, stop, total;
double requests_per_sec;
double ms_per_req;
int print=num / 10;
num_responses=0;
for(int i=0; i < buf.length; i++) {
buf[i]=0; // 0=request
}
start=System.currentTimeMillis();
for(int i=0; i < num; i++) {
DatagramPacket packet=new DatagramPacket(buf, 0, buf.length, host, port);
try {
sock.send(packet);
byte[] response=new byte[1];
DatagramPacket rsp_packet=new DatagramPacket(response, 0, response.length);
sock.receive(rsp_packet);
num_responses++;
if(num_responses >= num) {
System.out.println("received all responses (" + num_responses + ")");
break;
}
if(num_responses % print == 0) {
System.out.println("- received " + num_responses);
}
}
catch(Exception e) {
e.printStackTrace();
}
}
stop=System.currentTimeMillis();
total=stop-start;
requests_per_sec=num / (total / 1000.0);
ms_per_req=total / (double)num;
System.out.println("Took " + total + "ms for " + num + " requests: " + requests_per_sec +
" requests/sec, " + ms_per_req + " ms/request");
}
public static void main(String[] args) throws IOException {
boolean server=false;
int num=100;
int msg_size=10; // 10 bytes
InetAddress host=null;
int port=7500;
for(int i=0; i < args.length; i++) {
if(args[i].equals("-num")) {
num=Integer.parseInt(args[++i]);
continue;
}
if(args[i].equals("-server")) {
server=true;
continue;
}
if(args[i].equals("-size")) {
msg_size=Integer.parseInt(args[++i]);
continue;
}
if(args[i].equals("-host")) {
host=InetAddress.getByName(args[++i]);
continue;
}
if(args[i].equals("-port")) {
port=Integer.parseInt(args[++i]);
continue;
}
RoundTripUdp.help();
return;
}
if(host == null)
host=InetAddress.getLocalHost();
new RoundTripUdp().start(server, num, msg_size, host, port);
}
private static void help() {
System.out.println("RoundTrip [-server] [-num <number of messages>] " +
"[-size <size of each message (in bytes)>] [-host <host>] [-port <port>]");
}
}