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

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();

}

}

}