Java模拟并发操作进行服务器压力测试

使用线程池，实现多个用户并发访问的操作

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.PrintWriter; 
import java.net.HttpURLConnection; 
import java.net.URL; 
import java.net.URLConnection;
import java.util.Map; 
import java.util.Random;
import java.util.TreeMap;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.ExecutorService; 
import java.util.concurrent.Executors; 
import java.util.concurrent.ThreadPoolExecutor;

import org.json.JSONArray;
import org.json.JSONObject;
import org.tfelab.conn.ConnectionManager;

import com.ibm.icu.text.NumberFormat;
public class MultiThreadProxyHubApiTest {
	
	static int count = 0;
	//总访问量是client_num，并发量是thread_num
	int thread_num = 10;
	int client_num = 1000;
	
	float avg_exec_time = 0;
	float sum_exec_time = 0;
	long first_exec_time = Long.MAX_VALUE;
	long last_done_time = Long.MIN_VALUE;
	float total_exec_time = 0;
	
	String url = "";
	String postData = "";
	
	public MultiThreadProxyHubApiTest(int thread_num, int client_num, String url, String postData){
		
		this.thread_num = thread_num;
		this.client_num = client_num;
		this.url = url;
		this.postData = postData;
	}
	
	
	public void run() {
		
		final MultiThreadProxyHubApiTest currentObj = this;
		
		final ConcurrentHashMap<Integer, ThreadRecord> records = new ConcurrentHashMap<Integer, ThreadRecord>();

		// 建立ExecutorService线程池
		ExecutorService exec = Executors.newFixedThreadPool(thread_num);
		// thread_num个线程可以同时访问 
		// 模拟client_num个客户端访问
		
		final CountDownLatch doneSignal = new CountDownLatch(client_num);
		
		for (int i = 0; i < client_num; i++) {
			
			Runnable run = new Runnable() {
				
				public void run() {
					
					int index = getIndex();
					long st = System.currentTimeMillis();
					
					try {
						
						//这里引用一些开源包，看者可自行修改
						URLConnection urlconn = ConnectionManager.openUrlConnection(currentObj.url, currentObj.postData);
						urlconn.setReadTimeout(50000);
						BufferedReader reader = new BufferedReader(new InputStreamReader(urlconn.getInputStream(),"UTF-8"));
						
						
						String src = "";
						String line = null; 
						while ((line = reader.readLine()) != null) {
							src = src + line;
						}
						//System.out.println(src);

					} catch (Exception e) { 
						e.printStackTrace(); 
					}
					
					records.put(index, new ThreadRecord(st, System.currentTimeMillis()));
					doneSignal.countDown();//每调用一次countDown()方法，计数器减1
				} 
			}; 
			exec.execute(run); 
		}
		
		try {
			//计数器大于0 时，await()方法会阻塞程序继续执行
			doneSignal.await();
		} catch (InterruptedException e) {
			e.printStackTrace();
		}
		
		/**
		 * 获取每个线程的开始时间和结束时间
		 */
		for(int i : records.keySet()){
			ThreadRecord r = records.get(i);
			sum_exec_time += ((double)(r.et - r.st))/1000;
			
			if(r.st < first_exec_time){
				first_exec_time = r.st;
			}
			if(r.et > last_done_time){
				this.last_done_time = r.et;
			}
		}
		
		this.avg_exec_time = this.sum_exec_time / records.size();
		this.total_exec_time = ((float)(this.last_done_time - this.first_exec_time)) / 1000;
		NumberFormat nf=NumberFormat.getNumberInstance();
		nf.setMaximumFractionDigits(4);
		
		
		System.out.println("======================================================");
		System.out.println("Thread Num: " + thread_num + ", Client Count: "+ client_num +".");
		System.out.println("Avg Exec Time:   " + nf.format(this.avg_exec_time) + " s");
		System.out.println("Total Exec Time: " + nf.format(this.total_exec_time) + " s");
		System.out.println("Throughput:      " + nf.format(this.client_num /this.total_exec_time)+ " /s");
	}
	
	public static int getIndex(){
		return ++count;
	}
	
	public static void main(String[] args) {
		//总访问量和并发量两重循环，依次增大访问
		for(int j=500; j<600; j+= 100){
			for(int i=10; i<100; i+=10){
				//要测试的URL
				String url="http://www.baidu.com/";
				new MultiThreadProxyHubApiTest(i, j, url, "").run();
			}
		}
		System.out.println("finished!");
	}
}

class ThreadRecord {
	long st;
	long et;
	public ThreadRecord(long st, long et){
		this.st = st;
		this.et = et;
	}
}

测试结果样例：

======================================================
Thread Num: 10, Client Count: 500.
Avg Exec Time:   0.0508 s
Total Exec Time: 2.585 s
Throughput:      193.4236 /s
======================================================
Thread Num: 20, Client Count: 500.
Avg Exec Time:   0.0607 s
Total Exec Time: 2.007 s
Throughput:      249.1281 /s
======================================================
Thread Num: 30, Client Count: 500.
Avg Exec Time:   0.1218 s
Total Exec Time: 4.277 s
Throughput:      116.9044 /s