jmeter之类的工具应该说是非常好用的了，但是自己写的工具会更加的灵活，想怎么玩就怎么玩。
要求
我们首先要看下请求参数：
url ： 我们要测试的URL
total 总共要发送多少次请求
per 每次提交多少次的并发
返回的参数：
好啦，清楚需要什么，要返回什么，我们就可以开始写代码了。
代码实现
返回结果的类 ， 我使用了lombok 生成get&set方法。
@Data
public class Result {
private Integer avgTime;
private Integer percentile95Time;
}
主要的逻辑代码如下：
public class RequestTool {
* 获取结果的类
*/
private Map<String,Long> resultMap = new ConcurrentHashMap<String, Long>() ;
private OkHttpClient client = new OkHttpClient.Builder().build();
private CountDownLatch countDownLatch ;
private int total;
* 并发测试请求
* @param url 请求URL
* @param total 请求总数
* @param per 并发数
* @return
*/
public Result request(final String url ,int total,int per) throws Exception {
ExecutorService executorService = Executors.newFixedThreadPool(per);
this.countDownLatch = new CountDownLatch(total);
this.total = total;
for (int i = 0; i < total; i++) {
final int requestNum = i+1;
executorService.execute(new Runnable() {
public void run() {
doRequest(requestNum ,url);
}
});
}
countDownLatch.await();
executorService.shutdown();
return calResult();
}
private Result calResult() {
List<Long> list = new ArrayList<Long>(resultMap.values());
Collections.sort(list);
System.out.println("request results : " + JSON.toJSONString(list));
int totalTime = 0;
int percentile95Time = 0;
int times = (int)(total * 0.95);
int size = list.size();
for (int i = 0; i < size; i++) {
Long item = list.get(i);
totalTime += item;
if(i < times ){
percentile95Time += item;
}
}
Result result = new Result();
result.setAvgTime( totalTime/ size);
result.setPercentile95Time( percentile95Time / times );
return result;
}
private void doRequest(int requestNum, String url) {
long start = System.currentTimeMillis();
try {
Request request = new Request.Builder().url(url).get().build();
Response response = client.newCall(request).execute();
if(response.code() != 200){
System.out.println(" requestNum = "+requestNum+", code = " + response.code());
}
} catch (IOException e) {
System.out.println(" reuqest error : " + e.getLocalizedMessage() );
}finally {
long end = System.currentTimeMillis() - start;
resultMap.put(String.valueOf(requestNum),end );
countDownLatch.countDown();
}
}
}
执行test
* 并发测试工具类
*/
public class ConcurrentHttpTest {
public static void main(String[] args) {
RequestTool requestTool = new RequestTool();
String url = "https://fanyi.baidu.com/#en/zh/per";
try {
Result result = requestTool.request( url, 100, 10);
System.out.println(" result : " + JSON.toJSONString(result));
} catch (Exception e) {
e.printStackTrace();
}
}
}
测试结果：
request results（每次请求的结果） : [200,203,213,217,218,237,250,252,280,282,298,314,317,325,345,353,354,359,361,363,366,367,368,369,375,376,376,390,397,398,404,404,407,428,431,437,437,439,440,444,449,451,451,455,458,465,466,475,477,477,480,487,494,494,495,502,503,503,517,523,532,539,540,543,546,547,555,560,568,570,570,587,597,599,609,611,615,623,631,634,644,661,664,673,678,686,689,694,705,736,759,765,777,816,870,873,922,1722,1945,2187]
result（计算平均响应时长） : {"avgTime":534,"percentile95Time":482}
maven依赖库
我的maven依赖：
<dependencies>
<dependency>
<groupId>com.squareup.okhttp3</groupId>
<artifactId>okhttp</artifactId>
<version>3.11.0</version>
</dependency>
<dependency>
<groupId>com.alibaba</groupId>
<artifactId>fastjson</artifactId>
<version>1.2.72</version>
</dependency>
<dependency>
<groupId>org.projectlombok</groupId>
<artifactId>lombok</artifactId>
<version>1.18.12</version>
</dependency>
</dependencies>
评论