第七周 - 作业

/**
     * Web 性能压测工具，
     *  输入参数：URL，请求总次数，并发数。
     *  输出参数：平均响应时间，95% 响应时间。
     *
     * @param url
     * @param requestCount
     * @return
     */
    public double testUrl(String url, int requestCount) {
        try {
            if (url == null || "".equals(url)) {
                return 0;
            }
            long totleTime = 0;
            FutureTask[] tasks = new FutureTask[requestCount];
            for (int i = 0; i < requestCount; i++) {
                FutureTask futureTask = new FutureTask(new Callable() {
                    @Override
                    public Object call() throws Exception {
                        long time = requestUrl(url);
                        return time;
                    }
                });
                tasks[i] = futureTask;
            }
            for (int i = 0; i < requestCount; i++) {
                tasks[i].run();
            }
            int count = 0;
            for (int i = 0; i < requestCount; i++) {
                long time = (long) tasks[i].get();
                if (time != -1) {
                    totleTime += time;
                    count++;
                } else {
                    System.out.println("==========请求出错了");
                }
            }
            return totleTime / count;
        } catch (Exception ex) {
            ex.printStackTrace();
        }
        return -100;
    }
    private long requestUrl(String httpurl) {
        long startTime = System.currentTimeMillis();
        HttpURLConnection connection = null;
        try {
            // 创建远程url连接对象
            URL url = new URL(httpurl);
            // 通过远程url连接对象打开一个连接，强转成httpURLConnection类
            connection = (HttpURLConnection) url.openConnection();
            // 设置连接方式：get
            connection.setRequestMethod("GET");
            // 设置连接主机服务器的超时时间：15000毫秒
            connection.setConnectTimeout(15000);
            // 设置读取远程返回的数据时间：60000毫秒
            connection.setReadTimeout(60000);
            // 发送请求
            connection.connect();
            // 通过connection连接，获取输入流
            if (connection.getResponseCode() == 200) {
                long endTime = System.currentTimeMillis();
                return endTime - startTime;
            }
        } catch (Exception ex) {
            ex.printStackTrace();
        } finally {
            connection.disconnect();
        }
        return -1;
    }