美文网首页
CountDownLatch实现并发请求

CountDownLatch实现并发请求

作者: 惜时流光沿途留殇 | 来源:发表于2020-09-24 12:55 被阅读0次

    1.URL工具类

    import java.io.BufferedReader;
    import java.io.IOException;
    import java.io.InputStreamReader;
    import java.io.PrintWriter;
    import java.io.UnsupportedEncodingException;
    import java.net.URL;
    import java.net.URLConnection;
    import java.util.Iterator;
    import java.util.List;
    import java.util.Map;
    import java.util.Map.Entry;
    
    public class HttpSender {
    
        public static void main(String[] args) {
        }
    
        /**
         * 向指定URL发送GET方法的请求
         */
        public static String sendGet(String url, String param) throws UnsupportedEncodingException, IOException {
            return sendGet(url, param, null);
        }
        public static String sendGet(String url, String param, Map<String, String> header) throws UnsupportedEncodingException, IOException {
            String result = "";
            BufferedReader in = null;
            String urlNameString = url + "?" + param;
            URL realUrl = new URL(urlNameString);
            // 打开和URL之间的连接
            URLConnection connection = realUrl.openConnection();
            //设置超时时间
            connection.setConnectTimeout(5000);
            connection.setReadTimeout(15000);
            // 设置通用的请求属性
            if (header!=null) {
                Iterator<Entry<String, String>> it =header.entrySet().iterator();
                while(it.hasNext()){
                    Entry<String, String> entry = it.next();
                    System.out.println(entry.getKey()+":::"+entry.getValue());
                    connection.setRequestProperty(entry.getKey(), entry.getValue());
                }
            }
            connection.setRequestProperty("accept", "*/*");
            connection.setRequestProperty("connection", "Keep-Alive");
            connection.setRequestProperty("user-agent","Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1;SV1)");
            // 建立实际的连接
            connection.connect();
            // 获取所有响应头字段
            Map<String, List<String>> map = connection.getHeaderFields();
            // 遍历所有的响应头字段
            for (String key : map.keySet()) {
                System.out.println(key + "--->" + map.get(key));
            }
            // 定义 BufferedReader输入流来读取URL的响应,设置utf8防止中文乱码
            in = new BufferedReader(new InputStreamReader(connection.getInputStream(), "utf-8"));
            String line;
            while ((line = in.readLine()) != null) {
                result += line;
            }
            if (in != null) {
                in.close();
            }
            return result;
        }
    
        /**
         * 向指定 URL 发送POST方法的请求
         */
        public static String sendPost(String url, String param) throws UnsupportedEncodingException, IOException {
            return sendPost(url, param, null);
        }
    
        public static String sendPost(String url, String param, Map<String, String> header) throws UnsupportedEncodingException, IOException {
            PrintWriter out = null;
            BufferedReader in = null;
            String result = "";
            URL realUrl = new URL(url);
            // 打开和URL之间的连接
            URLConnection conn = realUrl.openConnection();
            //设置超时时间
            conn.setConnectTimeout(5000);
            conn.setReadTimeout(15000);
            // 设置通用的请求属性
            if (header!=null) {
                for (Entry<String, String> entry : header.entrySet()) {
                    conn.setRequestProperty(entry.getKey(), entry.getValue());
                }
            }
            conn.setRequestProperty("accept", "*/*");
            conn.setRequestProperty("connection", "Keep-Alive");
            conn.setRequestProperty("user-agent",
                    "Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1;SV1)");
            // 发送POST请求必须设置如下两行
            conn.setDoOutput(true);
            conn.setDoInput(true);
            // 获取URLConnection对象对应的输出流
            out = new PrintWriter(conn.getOutputStream());
            // 发送请求参数
            out.print(param);
            // flush输出流的缓冲
            out.flush();
            // 定义BufferedReader输入流来读取URL的响应
            in = new BufferedReader(
                    new InputStreamReader(conn.getInputStream(), "utf8"));
            String line;
            while ((line = in.readLine()) != null) {
                result += line;
            }
            if(out!=null){
                out.close();
            }
            if(in!=null){
                in.close();
            }
            return result;
        }
    }
    

    2.测试类

    import java.io.IOException;
    import java.util.concurrent.CountDownLatch;
    import java.util.concurrent.ExecutorService;
    import java.util.concurrent.Executors;
    import java.util.concurrent.Semaphore;
    
    public class CountExample {
    
        // 请求总数
        public static int clientTotal = 5000;
    
        // 同时并发执行的线程数
        public static int threadTotal = 200;
    
        public static int count = 0;
    
        public static void main(String[] args) throws Exception {
            ExecutorService executorService = Executors.newCachedThreadPool();
            //信号量,此处用于控制并发的线程数
            final Semaphore semaphore = new Semaphore(threadTotal);
            //闭锁,可实现计数器递减
            final CountDownLatch countDownLatch = new CountDownLatch(clientTotal);
            for (int i = 0; i < clientTotal ; i++) {
                executorService.execute(() -> {
                    try {
                        //执行此方法用于获取执行许可,当总计未释放的许可数不超过200时,
                        //允许通行,否则线程阻塞等待,直到获取到许可。
                         semaphore.acquire();
                        System.out.println(test());
                        
    
                    } catch (Exception e) {
                        //log.error("exception", e);
                        e.printStackTrace();
                    }finally {
                        //释放许可
                        semaphore.release();
                        //闭锁减一
                        countDownLatch.countDown();
                    }
    
                });
            }
            countDownLatch.await();//线程阻塞,直到闭锁值为0时,阻塞才释放,继续往下执行
            executorService.shutdown();
    
        }
    
        private static String test() throws IOException {
            return HttpSender.sendGet("http://localhost:8888/captcha","");
        }
    }
    

    相关文章

      网友评论

          本文标题:CountDownLatch实现并发请求

          本文链接:https://www.haomeiwen.com/subject/qopzyktx.html