前段时间做了一个服务器端接口，是附带文件上传的；后来我们要对这个接口进行压力测试；
其实很多现成的方式可以做压力测试，但是附带文件的的压力测试缺不怎么符合我的需求，jmeter是可以做附带文件上传的压力测试的，只是它是图形界面，而我目前的需求是要在测试机器上面去跑测试，而测试服务器是不能带图形界面的，所以jmeter的方案否决掉；
apache ab test，也是一个压力测试的好工具，只是研究了好久老搞不掂怎么做附带文件上传的压力测试（备注：在本文的最后我附带一下我研究的结果，说多了都是泪）
好了，现在我说下我自己的这个测试工具：
它依赖于赖于httpclient相关的包，包括：commons-codec-1.6.jar、commons-logging-1.1.3.jar、fluent-hc-4.3.4.jar、httpclient-4.3.4.jar、httpclient-cache-4.3.4.jar、httpcore-4.3.2.jar、httpmime-4.3.4.jar、httpmime-4.3.4.jar；
大家可以到apache的官方网站：http://hc.apache.org/downloads.cgi 去下载相关的包；
| import java.io.File; import java.io.IOException; import java.util.concurrent.ExecutorService; import java.util.concurrent.Executors; import java.util.concurrent.atomic.AtomicInteger; import org.apache.http.HttpEntity; import org.apache.http.client.methods.CloseableHttpResponse; import org.apache.http.client.methods.HttpPost; import org.apache.http.entity.ContentType; import org.apache.http.entity.mime.MultipartEntityBuilder; import org.apache.http.entity.mime.content.FileBody; import org.apache.http.entity.mime.content.StringBody; import org.apache.http.impl.client.CloseableHttpClient; import org.apache.http.impl.client.HttpClients; import org.apache.http.util.EntityUtils; public class ClientMultipartFormPostTest { private static ExecutorService pool = Executors.newFixedThreadPool(300); public static void main(String[] args) throws Exception { final String path = args[0];//文件地址 final String url = args[1]; //调用的URL final int i_len = Integer.parseInt(args[2]);//线程总数 final int j_len = Integer.parseInt(args[3]);//每个线程的请求数（暂时没用到） final AtomicInteger c = new AtomicInteger(0); final long s = System.currentTimeMillis(); for (int i = 0; i < i_len; i++) { new Thread(new Runnable() { @Override public void run() { while (true) { try { upLoadLogMultiThread(url,path); int cc = c.addAndGet(1); if (cc % 1000 == 0) { System.out.println(String.format("c: %d, t: %d", cc, (System.currentTimeMillis() - s))); } } catch (IOException e) { // TODO Auto-generated catch block e.printStackTrace(); } } } }).start(); } } public static void upLoadLogMultiThread(String url,String path) throws IOException{ CloseableHttpClient httpclient = HttpClients.createDefault(); try { HttpPost httppost = new HttpPost(url); FileBody bin = new FileBody(new File(path)); StringBody comment = new StringBody("A binary file of some kind", ContentType.TEXT_PLAIN); HttpEntity reqEntity = MultipartEntityBuilder.create() .addPart("bin", bin) .addPart("comment", comment) .build(); httppost.setEntity(reqEntity); CloseableHttpResponse response = httpclient.execute(httppost); try { HttpEntity resEntity = response.getEntity(); EntityUtils.consume(resEntity); } finally { response.close(); } } finally { httpclient.close(); } } } |
