第 7 周作业一

package com.rudy;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.ArrayList;
/*
    用你熟悉的编程语言写一个 web 性能压测工具，输入参数：URL，请求总次数，并发数。
    输出参数：平均响应时间，95% 响应时间。
    用这个测试工具以 10 并发、100 次请求压测 www.baidu.com
 */
public class PerformancePressureTool implements Runnable {
    private ArrayList<Long> longs = null;
    private String urlStr;              //请求Url
    private int requestCount;           //请求数
    private int concurrentCount;        //并发数
    private int receiveCount;           //到达数
    private int needReciveCount;        //需要到达数
    public PerformancePressureTool(String urlStr, int requestCount, int concurrentCount, int needReciveCount) {
        if (longs == null) {
            longs = new ArrayList<>();
        }
        this.urlStr = urlStr;
        this.requestCount = requestCount;
        this.concurrentCount = concurrentCount;
        this.needReciveCount = needReciveCount;
    }
    public void onCallback(long time) {
        longs.add(time);
        receiveCount++;
        if (receiveCount == needReciveCount) {
            long sum = 0;
            for (int i = 0; i < longs.size() - 1; i++) {
                sum += longs.get(i);
            }
            System.out.println("请求到达数:" + receiveCount + " 响应时间:" + sum / receiveCount);
        }
    }
    //获取平均时间
    public void avgResponTime() {
        for (int i = 0; i < concurrentCount; i++) {
            Thread thread = new Thread(this);
            thread.start();
        }
    }
    /**
     * 获取Url响应时间，下载对应url的内容
     *
     * @return
     */
    public void getResponTime() {
        HttpURLConnection urlConnection = null;
        InputStream openStream = null;
        long start = System.currentTimeMillis();
        long end = 0;
        try {
            URL url = new URL(urlStr);
            urlConnection = (HttpURLConnection) url.openConnection();
            urlConnection.connect();
            openStream = url.openStream();
            StringBuilder builder = new StringBuilder();
            BufferedReader reader = new BufferedReader(new InputStreamReader(openStream));
            String tmp = null;
            while ((tmp = reader.readLine()) != null) {
                builder.append(tmp + "\n");
            }
            end = System.currentTimeMillis();
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            if (urlConnection != null) {
                try {
                    urlConnection.disconnect();
                } catch (Exception e) {
                    e.printStackTrace();
                }
            }
            if (openStream != null) {
                try {
                    openStream.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
                onCallback(end - start);
            }
        }
    }
    @Override
    public void run() {
        for (int i = 0; i < requestCount; i++) {
            getResponTime();
        }
    }
    public static void main(String[] args) {
        PerformancePressureTool performancePressureTool = new PerformancePressureTool("http://www.baidu.com", 10, 10, 95);
        performancePressureTool.avgResponTime();
    }
}