自己实现的附带文件的压力测试方法

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]; //调用的URLfinal 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() {@Overridepublic 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 blocke.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();}}}

前段时间做了一个服务器端接口，是附带文件上传的；后来我们要对这个接口进行压力测试；　　其实很多现成的方式可以做压力测试，但是附带文件的的压力测试缺不怎么符合我的需求，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  去下载相关的包；