week07 作业

import com.google.common.collect.MinMaxPriorityQueue;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.atomic.AtomicInteger;
/**
 * @author yin.jianfeng
 * @date 2020/7/22
 */
public class Test {
    String url;
    int cn;
    int frequency;
    public Test(String url, int cn, int frequency) {
        this.url = url;
        this.cn = cn;
        this.frequency = frequency;
    }
    public static Test parseArgs(String[] args) {
        String url = args[0];
        int cn = Integer.parseInt(args[1]);
        int time = Integer.parseInt(args[2]);
        return new Test(url, cn, time);
    }
    static class Statistics {
        private MinMaxPriorityQueue<Long> queue;
        private long totalTime;
        private int frequency;
        public Statistics(Test test) {
            this.frequency = test.frequency;
            queue = MinMaxPriorityQueue.maximumSize(test.frequency * 95 / 100).create();
        }
        public void print() {
            System.out.println("avg:" + totalTime / frequency);
            System.out.println("95%:" + queue.peekLast());
        }
        public void record(long time) {
            totalTime += time;
            queue.add(time);
        }
    }
    public Statistics doJob(CountDownLatch latch) {
        Statistics statistics = new Statistics();
        AtomicInteger frequency = new AtomicInteger(this.frequency);
        for (int i = 0; i < cn; i++) {
            new Thread(
                    () -> {
                        if (frequency.decrementAndGet() <= 0) {
                            latch.countDown();
                            return;
                        }
                        long start = System.currentTimeMillis();
                        HttpUtils.get(url);
                        long time = System.currentTimeMillis() - start;
                        statistics.record(time);
                    }
            ).start();
        }
        return statistics;
    }
    public void run() {
        CountDownLatch latch = new CountDownLatch(cn);
        Statistics statistics = doJob(latch);
        try {
            latch.await();
        } catch (Exception e) {
            e.printStackTrace();
        }
        statistics.print();
    }
    public static void main(String[] args) {
        Test.parseArgs(args).run();
    }
}