第七周作业

package com.zhong.cli.service;
import com.zhong.cli.enums.RequestType;import com.zhong.cli.util.HttpClient;import org.springframework.beans.factory.annotation.Autowired;import org.springframework.stereotype.Component;
import java.net.URISyntaxException;import java.util.ArrayList;import java.util.HashMap;import java.util.List;import java.util.Map;import java.util.concurrent.*;import java.util.concurrent.atomic.AtomicReference;
/** * @Author Administrator * @Date 2020/11/8 8:43 * @Version 1.0 */@Componentpublic class PressureTestService {
    @Autowired    HttpClient httpClient;    List<String> execResult = new CopyOnWriteArrayList<>();    CountDownLatch latch = null;    List<Long> innerTimes = new ArrayList<>();
    public String request(String urlAddr,  Integer concurrentNum, Map<String, String> params) throws URISyntaxException, InterruptedException, ExecutionException {        String result = "主线程执行结束";        latch = new CountDownLatch(concurrentNum );
        BlockingQueue<Runnable> workQueue = new ArrayBlockingQueue<Runnable>(concurrentNum);
        ThreadPoolExecutor executor = new ThreadPoolExecutor(concurrentNum, concurrentNum, 1000, TimeUnit.MILLISECONDS, workQueue);
        for (int i = 0; i < concurrentNum; i++) {            executor.submit(new Task(urlAddr, new HashMap<>()));        }        latch.await();        for (String str : execResult) {            result = result.concat(str);        }        Long avgTime ;        Long totalTime = 0L;        //计算平均时间        for(Long time : innerTimes){            totalTime =totalTime+ time;        }        avgTime = totalTime/innerTimes.size();
        return "平均时间："+ avgTime +"毫秒";    }

    private class Task implements Callable<String> {
        private String urlAddr;
        private Map<String, String> hashmap;
        public Task(String urlAddr, Map<String, String> hashmap) {            this.urlAddr = urlAddr;            this.hashmap = hashmap;        }
        @Override        public String call() throws Exception {            Long beginTime = System.currentTimeMillis();            String tempresult = httpClient.requestURI(urlAddr, hashmap, RequestType.GET);            execResult.add(tempresult);            if(null != latch){                latch.countDown();            }            Long endTime = System.currentTimeMillis();            innerTimes.add(endTime-beginTime);            return tempresult;        }    }
}

package com.zhong.cli.util;
import com.zhong.cli.enums.RequestType;import org.apache.http.HttpEntity;import org.apache.http.client.methods.CloseableHttpResponse;import org.apache.http.client.methods.HttpGet;import org.apache.http.client.utils.URIBuilder;import org.apache.http.impl.client.CloseableHttpClient;import org.apache.http.impl.client.HttpClients;import org.apache.http.util.EntityUtils;import org.springframework.stereotype.Component;
import java.io.IOException;import java.net.URI;import java.util.Map;
/** * @Author Administrator * @Date 2020/11/8 8:55 * @Version 1.0 */@Componentpublic class HttpClient {

    public String requestURI(String urlAddr, Map<String, String> params, RequestType requestType) {        System.out.println("execute requestURI");        String result = "";        if (requestType.name().equals(RequestType.GET.name())) {            result = getResult(result, urlAddr);        }        return result;    }
    private String getResult(String result, String uri) {        CloseableHttpClient client = HttpClients.createDefault();        HttpGet httpGet = new HttpGet(uri);        try {            CloseableHttpResponse response = client.execute(httpGet);            int iGetResultCode = response.getStatusLine().getStatusCode();            HttpEntity entity = response.getEntity();            if (iGetResultCode >= 200 && iGetResultCode < 303) {                result = EntityUtils.toString(entity);            } else if (iGetResultCode >= 400 && iGetResultCode < 500) {                result = "请求的目标地址不存在：" + iGetResultCode;            } else {                result = "请求错误：" + iGetResultCode;            }        } catch (Exception e) {            e.printStackTrace();        }        return result;    }
    public URI getURL(String urlAddr, Map<String, String> params) {        try {            URIBuilder builder = new URIBuilder()                    .setScheme("http")                    .setHost(urlAddr);
            params.forEach((x, y) -> {                builder.setParameter(x, y);            });            return builder.build();        } catch (Exception e) {            e.printStackTrace();        }        return null;    }

}

package com.zhong.cli.shell;

import com.zhong.cli.service.PressureTestService;import org.springframework.beans.factory.annotation.Autowired;import org.springframework.shell.standard.ShellComponent;import org.springframework.shell.standard.ShellMethod;
import java.net.URISyntaxException;import java.util.HashMap;import java.util.concurrent.ExecutionException;
//@ShellComponentpublic class CommandShell {
    @Autowired    PressureTestService pressureTestService;
    @ShellMethod(value = "please input url,request num,Concurrent number,param.", key = "sum")    public String add(String url,Integer requestNum,Integer concurrentNum) throws URISyntaxException, InterruptedException, ExecutionException {        String result = "";        for(int i =0 ;i<requestNum;i++){            result = result.concat(pressureTestService.request(url,concurrentNum,new HashMap<>()));        }        return  result;    }
}