Java模拟并发操作进行压力测试代码

来源:互联网 发布:mac电脑的照片怎么删除 编辑:程序博客网 时间:2024/05/29 04:13
import java.io.BufferedReader;   import java.io.File;   import java.io.FileInputStream;   import java.io.InputStreamReader;   import java.io.PrintWriter;   import java.net.HttpURLConnection;   import java.net.URL;   import java.util.HashMap;   import java.util.Map;   import java.util.concurrent.ExecutorService;   import java.util.concurrent.Executors;   import java.util.concurrent.Semaphore;   public class ConcurrentTest {   private static int thread_num = 200;   private static int client_num = 460;   private static Map keywordMap = new HashMap();   static {   try {   InputStreamReader isr = new InputStreamReader(new FileInputStream(   new File("clicks.txt")), "GBK");   BufferedReader buffer = new BufferedReader(isr);   String line = "";   while ((line = buffer.readLine()) != null) {   keywordMap.put(line.substring(0, line.lastIndexOf(":")), "");   }   } catch (Exception e) {   e.printStackTrace();   }   }   public static void main(String[] args) {   int size = keywordMap.size();   // TODO Auto-generated method stub   ExecutorService exec = Executors.newCachedThreadPool();   // 50个线程可以同时访问   final Semaphore semp = new Semaphore(thread_num);   // 模拟2000个客户端访问   for (int index = 0; index < client_num; index++) {   final int NO = index;   Runnable run = new Runnable() {   public void run() {   try {   // 获取许可semp.acquire();   System.out.println("Thread:" + NO);   String host = "http://10.99.23.42:7001/KMQueryCenter/query.do?";   String para = "method=getQueryResult&pageNum=1&pageSize=5&"   + "queryKeyWord="   + getRandomSearchKey(NO)   + "&questionID=-1&questionIdPath=-1&searchType=1"   + "&proLine=&proSeries=&proType=" + NO;   System.out.println(host + para);   URL url = new URL(host);// 此处填写供测试的url   HttpURLConnection connection = (HttpURLConnection) url   .openConnection();   // connection.setRequestMethod("POST");   // connection.setRequestProperty("Proxy-Connection",   // "Keep-Alive");   connection.setDoOutput(true);   connection.setDoInput(true);   PrintWriter out = new PrintWriter(connection   .getOutputStream());   out.print(para);   out.flush();   out.close();   BufferedReader in = new BufferedReader(   new InputStreamReader(connection   .getInputStream()));   String line = "";   String result = "";   while ((line = in.readLine()) != null) {   result += line;   }   // System.out.println(result);   // Thread.sleep((long) (Math.random()) * 1000);   // 释放   System.out.println("第:" + NO + " 个");   semp.release();   } catch (Exception e) {   e.printStackTrace();   }   }   };   exec.execute(run);   }   // 退出线程池   exec.shutdown();   }   private static String getRandomSearchKey(final int no) {   String ret = "";   int size = keywordMap.size();   // int wanna = (int) (Math.random()) * (size - 1);   ret = (keywordMap.entrySet().toArray())[no].toString();   ret = ret.substring(0, ret.lastIndexOf("="));   System.out.println("\t" + ret);   return ret;   }   }

1 0