架构师训练营第七周作业

import lombok.AllArgsConstructor;
import org.apache.http.client.HttpClient;
import org.apache.http.client.config.RequestConfig;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.impl.client.HttpClients;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Date;
import java.util.List;
import java.util.concurrent.*;
/**
 * Created by leo on 2020/11/8.
 */
public class Main {
    public static void main(String[] args) throws InterruptedException {
        final String url = args[0];
        Integer totalCount = Integer.parseInt(args[1]);
        Integer concurrency = Integer.parseInt(args[2]);
        final Integer loopCount = totalCount / concurrency;
        final List<Long> costTimeList = Collections.synchronizedList(new ArrayList<Long>());
        final CountDownLatch countDownLatch = new CountDownLatch(concurrency);
        ExecutorService executorService = Executors.newFixedThreadPool(concurrency);
        for (int i = 0; i < concurrency; i++) {
            executorService.execute(new PressTask(url, loopCount, costTimeList, countDownLatch));
        }
        countDownLatch.await();
        Collections.sort(costTimeList);
        Long totalCostTime = 0L;
        for (Long costTime : costTimeList) {
            totalCostTime += costTime;
        }
        System.out.println("平均响应时间：" + totalCostTime / costTimeList.size() + "ms");
        System.out.println("95%响应时间：" + costTimeList.get(totalCount - ((Double) (totalCount * 0.95)).intValue()) + "ms");
        System.exit(0);
    }
}
@AllArgsConstructor
class PressTask implements Runnable {
    private String url;
    private Integer loopCount;
    private List<Long> costTimeList;
    private CountDownLatch countDownLatch;
    private static HttpClient httpClient;
    static {
        RequestConfig config = RequestConfig.custom()
                .setSocketTimeout(5000)
                .setConnectTimeout(5000)
                .setConnectionRequestTimeout(5000).build();
        httpClient = HttpClients.custom().setDefaultRequestConfig(config)
                .setMaxConnTotal(100)
                .setMaxConnPerRoute(100).build();
    }
    @Override
    public void run() {
        for (int i = 0; i < loopCount; i++) {
            try {
                Long costTime = doRequest(url);
                costTimeList.add(costTime);
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
        countDownLatch.countDown();
    }
    public static Long doRequest(String url) throws IOException {
        HttpGet httpGet = new HttpGet(url);
        Date startTime = new Date();
        httpClient.execute(httpGet);
        Date endTime = new Date();
        long costTime = endTime.getTime() - startTime.getTime();
        return endTime.getTime() - startTime.getTime();
    }
}