架构师训练营 1 期 - 第七周作业（vaik）

package com.vaik.pressuretest;
import lombok.Data;import java.util.List;
@Datapublic class Task {    //请求的URL    private String url;    //开启线程数    private int threadNumber;    //总请求数量    private int requestNumber;    //已完成请求数    private int completedNumber;    //请求结果集    private List<TaskResult> results;}

package com.vaik.pressuretest;import lombok.Data;
@Datapublic class TaskResult {    //开始时间戳    private long startTime;    //结束时间戳    private long endTime;    //消耗时间长    private long consumeTime;    //任务号    private long taskNumber;    //执行任务线程    private String runThreadName;    //执行结果    private String msg;}

package com.vaik.pressuretest;
public class TaskRunner implements Runnable {    private static final Object obj = new Object();    private Task task;    public TaskRunner(Task task){        this.task = task;    }
    @Override    public void run() {        while (!isEnd()){            synchronized (obj){                task.setCompletedNumber(task.getCompletedNumber()+1);                //System.out.println("setCompletedNumber "+task.getCompletedNumber());            }            TaskResult taskResult = new TaskResult();            long startTime = System.currentTimeMillis();            taskResult.setStartTime(startTime);            //System.out.println("start HttpRequestUtil");            String res = HttpRequestUtil.get(task.getUrl());            //System.out.println("HttpRequestUtil response:" + res);            if(res.isEmpty()){                taskResult.setMsg(Thread.currentThread().getName()+" has exception, taskNumber"+(task.getCompletedNumber()+1));            }else{                taskResult.setMsg(Thread.currentThread().getName()+ " success taskNumber:"+(task.getCompletedNumber()+1));            }            long endTime = System.currentTimeMillis();            taskResult.setEndTime(endTime);            taskResult.setConsumeTime(endTime-startTime);            taskResult.setTaskNumber(task.getCompletedNumber()+1);            taskResult.setRunThreadName(Thread.currentThread().getName());            task.getResults().add(taskResult);            //System.out.println("end HttpRequestUtil");        }    }
    private boolean isEnd(){        synchronized (obj) {            boolean bl = false;            if (task.getCompletedNumber() >= task.getRequestNumber()) {                bl = true;            }            return bl;        }    }}

package com.vaik.pressuretest;
import java.util.ArrayList;import java.util.Comparator;import java.util.List;
public class SimplePressureTestDemo {    public static void main(String[] args) throws InterruptedException {        if(args.length == 3) {            String url = args[0];            int threadNumber = Integer.parseInt(args[1]);            int requestNumber = Integer.parseInt(args[2]);            com.vaik.pressuretest.Task task = new com.vaik.pressuretest.Task();            task.setUrl(url);            task.setThreadNumber(threadNumber);            task.setRequestNumber(requestNumber);            task.setCompletedNumber(0);            List<TaskResult> list = new ArrayList<>();            task.setResults(list);            executeTask(task);            while(task.getCompletedNumber()<task.getRequestNumber()){                Thread.sleep(1000);                printResult(task);            }
        }else{            System.out.println("参数个数不正确，正确格式 url concurrentNumber requestNumber");        }    }
    public static void executeTask(Task task){        System.out.println("Start executeTask");        for(int i =0;i<task.getThreadNumber();i++){            TaskRunner taskRunner = new TaskRunner(task);            Thread thread = new Thread(taskRunner);            thread.setName("Thread#"+i);            thread.start();        }    }
    public static void printResult(Task task){        List<TaskResult> results = task.getResults();        if(results.size()>0) {            results.sort(Comparator.comparing(TaskResult::getConsumeTime));            double avg = results.stream().mapToDouble(TaskResult::getConsumeTime).average().getAsDouble();            System.out.println("总需发送请求：" + task.getRequestNumber() + " 实际完成请求：" + task.getCompletedNumber());            System.out.println("平均响应时间：" + avg+"毫秒");            if (results.size() > 95)                System.out.println("95%响应时间：" + results.get(95).getConsumeTime()+"毫秒");        }    }}

package com.vaik.pressuretest;import org.apache.http.client.config.RequestConfig;import org.apache.http.client.methods.CloseableHttpResponse;import org.apache.http.client.methods.HttpGet;import org.apache.http.client.methods.HttpPost;import org.apache.http.entity.StringEntity;import org.apache.http.impl.client.CloseableHttpClient;import org.apache.http.impl.client.HttpClients;import org.apache.http.impl.conn.PoolingHttpClientConnectionManager;
import java.io.BufferedReader;import java.io.IOException;import java.io.InputStreamReader;import java.nio.charset.Charset;
/** * @author wanghang * @Description * @date: 2018/04/07 17:49 */public class HttpRequestUtil {    private static CloseableHttpClient httpClient;
    static {        PoolingHttpClientConnectionManager cm = new PoolingHttpClientConnectionManager();        cm.setMaxTotal(100);        cm.setDefaultMaxPerRoute(20);        cm.setDefaultMaxPerRoute(50);        httpClient = HttpClients.custom().setConnectionManager(cm).build();    }
    public static String get(String url) {        CloseableHttpResponse response = null;        BufferedReader in = null;        String result = "";        try {            HttpGet httpGet = new HttpGet(url);            RequestConfig requestConfig = RequestConfig.custom().setConnectTimeout(30000).setConnectionRequestTimeout(30000).setSocketTimeout(30000).build();            httpGet.setConfig(requestConfig);            httpGet.setConfig(requestConfig);            httpGet.addHeader("Content-type", "application/json; charset=utf-8");            httpGet.setHeader("Accept", "application/json");            response = httpClient.execute(httpGet);            in = new BufferedReader(new InputStreamReader(response.getEntity().getContent()));            StringBuffer sb = new StringBuffer("");            String line = "";            String NL = System.getProperty("line.separator");            while ((line = in.readLine()) != null) {                sb.append(line + NL);            }            in.close();            result = sb.toString();        } catch (IOException e) {            e.printStackTrace();        } finally {            try {                if (null != response) {                    response.close();                }            } catch (IOException e) {                e.printStackTrace();            }        }        return result;    }
    public static String post(String url, String jsonString) {        CloseableHttpResponse response = null;        BufferedReader in = null;        String result = "";        try {            HttpPost httpPost = new HttpPost(url);            RequestConfig requestConfig = RequestConfig.custom().setConnectTimeout(30000).setConnectionRequestTimeout(30000).setSocketTimeout(30000).build();            httpPost.setConfig(requestConfig);            httpPost.setConfig(requestConfig);            httpPost.addHeader("Content-type", "application/json; charset=utf-8");            httpPost.setHeader("Accept", "application/json");            httpPost.setEntity(new StringEntity(jsonString, Charset.forName("UTF-8")));            response = httpClient.execute(httpPost);            in = new BufferedReader(new InputStreamReader(response.getEntity().getContent()));            StringBuffer sb = new StringBuffer("");            String line = "";            String NL = System.getProperty("line.separator");            while ((line = in.readLine()) != null) {                sb.append(line + NL);            }            in.close();            result = sb.toString();        } catch (IOException e) {            e.printStackTrace();        } finally {            try {                if (null != response) {                    response.close();                }            } catch (IOException e) {                e.printStackTrace();            }        }        return result;    }}