在线网络速度测试JAVA程序(一):思路和控制台主程序【转】

  1. 云栖社区>
  2. 博客>
  3. 正文

在线网络速度测试JAVA程序(一):思路和控制台主程序【转】

华山青竹 2014-01-18 13:10:00 浏览318
展开阅读全文

来源:http://hancool.blog.51cto.com/1836252/1352228

事情的缘由

因上级公司的信息化主管部门经常被投诉说是各种业务应用系统反映系统使用慢的问题,而都把问题归结于网速不给力。而业务系统一部份是总公司一级部署的,其它很多都是各个业务部门各自为政建设的,不知道从哪里找来的各种小公司,搞的五法八门的技术路线,没得统一标准、技术平台,所以水平就参差不齐。人少的时候还好点,业务和使用人一多,系统慢得象蜗牛。而网络基础设施这几年来改造、提升得差不多了,都是百兆桌面、千兆骨干了。地市级公司内部虽然办公点分散,但都是使用电力专用光纤建设的千兆骨干的MPLSVPN技术,地市公司与网省公司是千兆或万兆专用光纤。因此其实网络基础设施还是绰绰有余。为了自证清白,上级主管部门就盟生了建一个分层分布式的网速测试系统的想法。你说网速慢,那你在客户端上打开网页,测试一下各级网速,就可能看看到底是网速慢,还是应用系统本身太垃圾。

 

思路

在网络上给出的参考方法,WEB版的大部份是客户端用JAVASCRIPT下载一个大文件,然后根据速度=总下载量/时间,得到网速;专业的开源测试工具是iperf,是c版本的,和业余的相比,专业的不只一点点。于是,了解了一下iperf的基本思路和方法,于是采用JAVA来写,利用socket直接实现通信,服务端和客户端配合,通过TCP通信的数据量和使用时间来计算网络速度,一方面效率高、减少其它因素(如HTTP协议、JAVASCRIPT加载与执行效率等)的影响;另一方面,利用applet可以实现B/S应用,客户端部署使用简单。程序代码参考了OSCHINA的JGroups项目的JPerf.java。项目名称为jperf,使用eclipse开发和测试,目前实现了基本功能(因目前只是进行验证,因此未进行完整的异常处理、验证等)。

  1 import java.io.*;
  2 import java.net.InetAddress;
  3 import java.net.ServerSocket;
  4 import java.net.Socket;
  5 import java.net.UnknownHostException;
  6 import java.text.NumberFormat;
  7 /***
  8  * Tool to measure TCP throughput, similar to iperf
  9  * @author Bela Ban
 10  * @version $Id: JPerf.java,v 1.4 2007/12/28 23:29:17 belaban Exp $
 11  */
 12 public class jperf implements Runnable{
 13     boolean client;
 14     boolean direction;//test direction:true-up test,false-down
 15     int num;
 16     InetAddress local_addr;
 17     int local_port;
 18     int remote_port;
 19     InetAddress remote_addr;
 20     int size;
 21     int receivebuf=200000;
 22     int sendbuf=200000;
 23     static  NumberFormat f;
 24     Socket client_sock;
 25                
 26     static {
 27         f=NumberFormat.getNumberInstance();
 28         f.setGroupingUsed(false);
 29         f.setMaximumFractionDigits(2);
 30     }
 31                 
 32     public jperf(){
 33     }
 34                 
 35     //实现Thread类的接口Runnable, 用于支持服务端连接的多线程
 36     @Override
 37     public void run() {
 38         // TODO Auto-generated method stub
 39         try {
 40             server_accept_data();
 41         } catch (IOException e) {
 42             // TODO Auto-generated catch block
 43             e.printStackTrace();
 44         } catch (ClassNotFoundException e) {
 45             // TODO Auto-generated catch block
 46             e.printStackTrace();
 47         }
 48                     
 49     }
 50     //用于applet和GUI调用
 51     public jperf(String remote_addr,int remote_port,int num,int size,boolean direction) throws UnknownHostException{
 52                      
 53          this.remote_addr=InetAddress.getByName(remote_addr);
 54          this.remote_port=remote_port;
 55          this.num=num;
 56          this.size=size;
 57          this.direction=direction;
 58      }
 59                 
 60     private void start(boolean client, boolean direction,int num, int size, String local_addr, int local_port,
 61                        String remote_addr, int remote_port,int receivebuf, int sendbuf) throws IOException, ClassNotFoundException {
 62         this.client=client;
 63         this.direction=direction;
 64         this.num=num;
 65         this.size=size;
 66         this.local_addr=InetAddress.getByName(local_addr);
 67         this.local_port=local_port;      
 68         this.remote_addr=InetAddress.getByName(remote_addr);
 69         this.remote_port=remote_port;
 70         this.receivebuf=receivebuf;
 71         this.sendbuf=sendbuf;
 72                     
 73         if(client) {
 74            client();
 75         }
 76         else {
 77             server();
 78         }
 79     }
 80     //客户端调用
 81     public String client() throws IOException{
 82                             
 83         System.out.println("-- creating socket to " + this.remote_addr + ":" + this.remote_port);
 84         client_sock=new Socket(this.remote_addr, remote_port);
 85                     
 86         String result="";
 87         if(sendExchangeData()==true){
 88             if(direction==true) result=sendData(num,size);
 89             else result=receiveData(num,size);
 90         }
 91         else{
 92             result="connect to server and exchange data fail!";
 93         }
 94         System.out.println(result);  
 95         client_sock.close();
 96                     
 97         return result;
 98     }
 99     //客户端向服务端发送测试 的数据参数
100     private boolean sendExchangeData() throws IOException{
101         boolean ret=true;
102                     
103         client_sock.setReceiveBufferSize(receivebuf);
104         client_sock.setSendBufferSize(sendbuf);
105         ObjectOutputStream write=new ObjectOutputStream(new BufferedOutputStream(client_sock.getOutputStream()));
106         write.writeObject(num);
107         write.flush();
108         write.writeObject(size);
109         write.flush();
110         write.writeObject(direction);
111         write.flush();
112                     
113         return ret;
114     }
115     //服务端调用
116     private void server() throws IOException, ClassNotFoundException
117     {              
118          ServerSocket srv_sock=new ServerSocket(local_port, 10, this.local_addr);
119          System.out.println("-- waiting for client on " + srv_sock.getLocalSocketAddress());
120          while(true){
121              //wait for a client connect
122              Socket client_sock=srv_sock.accept();
123              //start a new thread deal this connection:
124              jperf thread_client=new jperf();
125              thread_client.client_sock=client_sock;
126              thread_client.sendbuf=sendbuf;
127              thread_client.receivebuf=receivebuf;
128              //每一个客户端单独一个线程,支持多个客户端同时连接
129              Thread thread=new Thread(thread_client);
130              thread.start();
131          }
132                      
133     }
134     //服务器接收和发送测试 数据
135     private void server_accept_data() throws IOException, ClassNotFoundException{             
136         client_sock.setReceiveBufferSize(receivebuf);
137         client_sock.setSendBufferSize(sendbuf);
138         System.out.println("-- accepted data connection from " + client_sock.getRemoteSocketAddress());
139         ObjectInputStream in=new ObjectInputStream(new BufferedInputStream(client_sock.getInputStream()));
140                         
141         int num=0,size=0;
142         boolean direction=false;
143         num=(Integer)in.readObject();
144         size=(Integer)in.readObject();      
145         direction=(Boolean)in.readObject();
146                     
147         if(num>0 && size>0) {
148             String result;
149             if(direction==true) result=receiveData(num,size);
150             else result=sendData(num,size);
151                         
152             System.out.println(result);
153          }      
154          else{
155             System.out.println("-- invalid exchange data! ");
156          }
157          client_sock.close();
158     }
159     //发送数据,并计算测试结果
160     private String sendData(int num,int size) throws IOException
161     {     
162         System.out.println("-- sending data to "+client_sock.getRemoteSocketAddress().toString()+ " total "+num + " messages");
163         DataOutputStream out=new DataOutputStream(new BufferedOutputStream(client_sock.getOutputStream()));
164         byte[] buf=new byte[size];      
165         for(int i=0;i<buf.length;i++) buf[i]=(byte)(i%128);
166         long start=0, stop;
167         int cnt=1;
168         int incr=num/10;
169         start=System.currentTimeMillis();
170         for(int i=0; i < num; i++) {           
171             out.write(buf, 0, buf.length);
172             out.flush();
173             if(cnt % incr == 0)
174                 System.out.println("-- sent " + cnt + " messages");
175             cnt++;
176         }
177         stop=System.currentTimeMillis();
178         long diff=stop-start;
179         String result=report("发送报文至 "+client_sock.getRemoteSocketAddress().toString(),(long)num*(long)size,diff);      
180                     
181         return result;
182     }
183                 
184     //接收数据,并计算测试结果
185     private String receiveData(int num,int size) throws IOException{
186         System.out.println("-- accepted data from " + client_sock.getRemoteSocketAddress().toString()+" total "+num+" messages");
187         DataInputStream in=new DataInputStream(new BufferedInputStream(client_sock.getInputStream()));
188         byte[] buf=new byte[size];          
189         long counter=0;
190         int incr=num/10;
191         long start=0, stop;      
192         while(true) {
193             int len=in.read(buf, 0, buf.length);
194             if(len<=0) break;
195             if(start == 0)
196                 start=System.currentTimeMillis();
197             counter+=len;
198             if((counter/size) % incr == 0)
199                 System.out.println("-- received " + counter/size);
200         }          
201         stop=System.currentTimeMillis();
202         long diff=stop-start;
203         String result=report("接收报文来自 "+client_sock.getRemoteSocketAddress().toString(),counter,diff);
204                     
205         return result;
206     }
207     //计算测试结果 
208     private String report(String direction,long totalbyte, double diff)
209     {
210         StringBuilder sb=new StringBuilder();
211         double tbs=totalbyte/(1024*1024);
212         if(tbs<1000)  sb.append("\n"+direction+",测试数据总数" + f.format(tbs) + "Mbyte"+" ,用时 " + diff + "毫秒 ");
213         else{
214             tbs=tbs/1024;
215             sb.append("\n"+direction+",测试数据总数" + f.format(tbs)+ "Gbyte"+" ,用时 " + diff + "毫秒 ");
216         }
217          //tcp throughput:
218          double throughput=totalbyte / (diff / 1000.0) / 1024.0;
219          if(throughput < 1000)
220              sb.append("\n网络吞吐量: " + f.format(throughput) + "KB/秒");
221          else {
222              throughput/=1024.0;
223              sb.append("\n网络吞吐量: " + f.format(throughput) + "MB/秒");
224          }
225          //bandwidth
226          double bandwidth=totalbyte / (diff / 1000.0) / 1024.0*8;
227          if(bandwidth < 1000){
228              sb.append("\n网络带宽: " + f.format(bandwidth) + "Kb/秒");
229          }
230          else {
231             bandwidth/=1024.0;
232              if(bandwidth>1000){
233                 bandwidth/=1024;
234                 sb.append("\n网络带宽: " + f.format(bandwidth) + "Gb/秒");
235              }
236              else sb.append("\n网络带宽: " + f.format(bandwidth) + "Mb/秒");          
237          }
238          return sb.toString();
239     }
240     static void help() {
241         System.out.println("JPerf [-help] [-client] [-direction <up|down>] [-num <number of msgs] [-size <bytes>] [-local_addr <interface>] [-local_port <port]" +
242                 "[-remote_addr <IP addr>] [-remote_port <port>] [-receivebuf <bytes>] [-sendbuf <bytes>]");
243     }
244     //主程序执行入口
245     public static void main(String[] args) throws UnknownHostException {
246         boolean client=false;
247         boolean direction=false;//test direction:true-up test,false-down test
248         int num=10000;
249         int size=8192;
250         String local_addr=InetAddress.getLocalHost().getHostAddress();
251         String remote_addr=local_addr;
252         int local_port=5000;
253         int remote_port=5000;
254         int receivebuf=200000, sendbuf=200000;
255         for(int i=0; i < args.length; i++) {
256             if(args[i].equals("-client")) {
257                 client=true;
258                 continue;
259             }
260             if(args[i].equals("-num")) {
261                 num=Integer.parseInt(args[++i]);
262                 continue;
263             }
264             if(args[i].equals("-size")) {
265                 size=Integer.parseInt(args[++i]);
266                 continue;
267             }
268             if(args[i].equals("-local_addr")) {
269                 local_addr=args[++i];
270                 continue;
271             }
272             if(args[i].equals("-remote_addr")) {
273                 remote_addr=args[++i];
274                 continue;
275             }
276             if(args[i].equals("-local_port")) {
277                 local_port=Integer.parseInt(args[++i]);
278                 continue;
279             }
280             if(args[i].equals("-remote_port")) {
281                 remote_port=Integer.parseInt(args[++i]);
282                 continue;
283             }
284             if(args[i].equals("-receivebuf")) {
285                 receivebuf=Integer.parseInt(args[++i]);
286                 continue;
287             }
288             if(args[i].equals("-sendbuf")) {
289                 sendbuf=Integer.parseInt(args[++i]);
290                 continue;
291             }
292             if(args[i].equals("-direction")) {
293                  String value=args[++i];
294                  if(value.toLowerCase().equals("up")) direction=true;
295                  else direction=false;
296                              
297                  continue;
298             }
299             help();
300             return;
301         }
302         try {
303             new jperf().start(client, direction,num, size, local_addr, local_port, remote_addr, remote_port, receivebuf, sendbuf);
304         }
305         catch(IOException e) {
306             e.printStackTrace();
307         } catch (ClassNotFoundException e) {
308             e.printStackTrace();
309         }
310     } 
311 }
View Code

编译代码:javac jperf.java

在服务端执行:java jperf,启动后默认在5000端口上进行监听。

在客户端执行:java jperf -client -remote_addr x.x.x.x,默认参数为数据报文数10000,数据包大小8192(针对默认的WINDOWS系统的TCP窗口),端口5000,方向为测试下行。

Java jperf -client -remote_addr x.x.x.x -direction up测试上行。

经过测试,两台通过百兆口交换机连接的计算机,吞吐量为11.3MB/s,带宽为90.4Mb/s左右。

 

本文出自 “一个业余程序员的历程” 博客,请务必保留此出处http://hancool.blog.51cto.com/1836252/1352228

 

网友评论

登录后评论
0/500
评论
华山青竹
+ 关注