package com.github.http;

import com.github.underscore.lodash.U;
import java.io.File;
import java.net.URL;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.time.LocalDateTime;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicLong;
import org.HdrHistogram.AtomicHistogram;
import org.HdrHistogram.Histogram;
import org.apache.commons.cli.BasicParser;
import org.apache.commons.cli.CommandLine;
import org.apache.commons.cli.CommandLineParser;
import org.apache.commons.cli.HelpFormatter;
import org.apache.commons.cli.Options;
import org.apache.commons.cli.ParseException;
import org.eclipse.jetty.http.HttpFields;
import org.eclipse.jetty.toolchain.perf.HistogramSnapshot;
import org.eclipse.jetty.util.ssl.SslContextFactory;
import org.eclipse.jetty.util.thread.ScheduledExecutorScheduler;
import org.eclipse.jetty.util.thread.Scheduler;
import org.mortbay.jetty.load.generator.HTTP1ClientTransportBuilder;
import org.mortbay.jetty.load.generator.HTTPClientTransportBuilder;
import org.mortbay.jetty.load.generator.LoadGenerator;
import org.mortbay.jetty.load.generator.Resource;
import org.mortbay.jetty.load.generator.util.MonitoringThreadPoolExecutor;

public class HttpLoadGenerator {

    private final Scheduler scheduler = new ScheduledExecutorScheduler();

    private LoadGenerator.Builder prepareLoadGenerator(HTTPClientTransportBuilder clientTransportBuilder, String path) {
        HttpFields headers = new HttpFields();
        headers.put("User-Agent", "Mozilla/5.0 (X11; Ubuntu; Linux x86_64; rv:52.0) Gecko/20100101 Firefox/52.0");
        headers.put("Accept", "text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8");
        headers.put("Accept-Language", "en-US,en;q=0.5");
        headers.put("Cookie", "__utma=124097164.2025215041.1465995519.1483973120.1485461487.58; __utmz=124097164.1480932641.29.9.utmcsr=localhost:8080|utmccn=(referral)|utmcmd=referral|utmcct=/; wp-settings-3=editor%3Dhtml%26wplink%3D1%26post_dfw%3Doff%26posts_list_mode%3Dlist; wp-settings-time-3=1483536385; wp-settings-time-4=1485794804; wp-settings-4=editor%3Dhtml; _ga=GA1.2.2025215041.1465995519; wordpress_google_apps_login=30a7b62f9ae5db1653367cafa3accacd; PHPSESSID=r8rr7hnl7kttpq40q7bkbcn5c2; ckon1703=sject1703_bfc34a0618c85; JCS_INENREF=; JCS_INENTIM=1489507850637; _gat=1");
        return new LoadGenerator.Builder()
                .threads(1)
                .port(80)
                .httpClientTransportBuilder(clientTransportBuilder)
                .resource(new Resource(path).requestHeaders(headers))
                .scheduler(scheduler);
    }

    public void startLoadTest(URL url, int resourceRate) throws Exception {
        MonitoringThreadPoolExecutor executor = new MonitoringThreadPoolExecutor(1024, 60, TimeUnit.SECONDS);

        Histogram treeHistogram = new AtomicHistogram(TimeUnit.MICROSECONDS.toNanos(1),
                TimeUnit.SECONDS.toNanos(10), 3);
        Histogram rootHistogram = new AtomicHistogram(TimeUnit.MICROSECONDS.toNanos(1),
                TimeUnit.SECONDS.toNanos(10), 3);

        LoadGenerator loadGenerator = prepareLoadGenerator(new HTTP1ClientTransportBuilder(), url.getPath())
                .warmupIterationsPerThread(0)
                .iterationsPerThread(10)
                .runFor(30, TimeUnit.SECONDS)
                .usersPerThread(100)
                .channelsPerUser(1)
                .resourceRate(resourceRate)
                .sslContextFactory(new SslContextFactory(true))
                .host(url.getHost())
                .port(url.getPort() == -1 ? 80 : url.getPort())
                .executor(executor)
                .resourceListener((Resource.TreeListener) info -> {
                    rootHistogram.recordValue(info.getResponseTime() - info.getRequestTime());
                    treeHistogram.recordValue(info.getTreeTime() - info.getRequestTime());
                })
                .build();
        AtomicLong responses = new AtomicLong();
        List<Long> requestsStatistic = new ArrayList<>();
        Runnable helloRunnable = () -> {
            long responsesForLastSecond = rootHistogram.getTotalCount() - responses.get();
            requestsStatistic.add(responsesForLastSecond);
            responses.set(rootHistogram.getTotalCount());
            double averageResponses = requestsStatistic.stream().mapToDouble(Double::valueOf)
                    .average().orElse(Double.NaN);
            System.err.printf("%s amount of responses for the last second: %d, avarege request rate: %.3f%n",
                    LocalDateTime.now().toString(),
                    responsesForLastSecond,
                    averageResponses);
        };

        ScheduledExecutorService scheduledExecutor = Executors.newScheduledThreadPool(1);
        scheduledExecutor.scheduleAtFixedRate(helloRunnable, 0, 1, TimeUnit.SECONDS);

        loadGenerator.begin().join();
        scheduledExecutor.shutdown();

        HistogramSnapshot rootSnapshot = new HistogramSnapshot(rootHistogram, 20, "root response time", "us", TimeUnit.NANOSECONDS::toMicros);
        System.err.println(rootSnapshot);
        System.err.printf("client thread pool - max_threads: %d, max_queue_size: %d, max_queue_latency: %dms%n%n",
                executor.getMaxActiveThreads(),
                executor.getMaxQueueSize(),
                TimeUnit.NANOSECONDS.toMillis(executor.getMaxQueueLatency())
        );

        executor.shutdown();
    }

    public static void main(String[] args) throws Exception {
        Options options = new Options();
        options.addOption("c", "config", true, "The config file.");
        options.addOption("?", "help", false, "This help text.");
        
        if (args.length > 0) {
            CommandLineParser parser = new BasicParser();
            CommandLine cmd;
            try {
                cmd = parser.parse(options, args);
            } catch (ParseException e) {
                help(options);
                return;
            }
            if (cmd.hasOption("?") || !cmd.hasOption("c")) {
                help(options);
                return;
            }
            String config = cmd.getOptionValue("c");
            if (new File(config).isFile()) {
                try {
                    final byte[] bytes = Files.readAllBytes(Paths.get(config, new String[0]));
                    String text = new String(bytes, StandardCharsets.UTF_8);
                    Map<String, Object> result = (Map<String, Object>) U.fromJson(text);
                    String url = (String) result.get("url");
                    Long requestRate = (Long) result.get("requestRate");
                    HttpLoadGenerator httpLoadGenerator = new HttpLoadGenerator();
                    httpLoadGenerator.startLoadTest(new URL(url), requestRate.intValue());
                } catch (Exception ex) {
                    System.out.println(config + " - " + ex.getMessage());
                }
            }
        } else {
            help(options);
        }
    }
        
    private static void help(Options options) {
        HelpFormatter formater = new HelpFormatter();
        formater.printHelp(100, "java -jar http-load-generator.jar", "", options, "");
    }
}

<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
    <modelVersion>4.0.0</modelVersion>
    <groupId>com.github.javadev</groupId>
    <artifactId>http-load-generator</artifactId>
    <version>1.0-SNAPSHOT</version>
    <packaging>jar</packaging>
    <name>Http load generator</name>
    <description>The http load generator</description>

    <properties>
        <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
        <maven.compiler.source>1.8</maven.compiler.source>
        <maven.compiler.target>1.8</maven.compiler.target>
    </properties>

    <build>
        <finalName>${project.artifactId}</finalName>
        <plugins>
            <plugin>
                <groupId>org.apache.maven.plugins</groupId>
                <artifactId>maven-jar-plugin</artifactId>
                <version>3.0.2</version>
                <configuration>
                    <archive>
                        <manifest>
                            <mainClass>com.github.http.HttpLoadGenerator</mainClass>
                        </manifest>
                    </archive>
                </configuration>
            </plugin>
            <plugin>
                <groupId>org.apache.maven.plugins</groupId>
                <artifactId>maven-shade-plugin</artifactId>
                <version>2.4.3</version>
                <executions>
                    <execution>
                        <phase>package</phase>
                        <goals>
                            <goal>shade</goal>
                        </goals>
                        <configuration>
                            <filters>
                                <filter>
                                    <artifact>*:*</artifact>
                                    <excludes>
                                        <exclude>META-INF/maven/**</exclude>
                                        <exclude>META-INF/COPYRIGHT.html</exclude>
                                        <exclude>META-INF/LICENSE*</exclude>
                                        <exclude>META-INF/NOTICE*</exclude>
                                        <exclude>META-INF/README.txt</exclude>
                                        <exclude>META-INF/DEPENDENCIES*</exclude>
                                        <exclude>LICENSE.txt</exclude>
                                        <exclude>rhinoDiff.txt</exclude>
                                        <exclude>license/**</exclude>
                                    </excludes>
                                </filter>
                            </filters>
                            <dependencyReducedPomLocation>${project.build.directory}/dependency-reduced-pom.xml</dependencyReducedPomLocation>
                        </configuration>
                    </execution>
                </executions>
            </plugin>
        </plugins>
    </build>

    <dependencies>
        <dependency>
            <groupId>org.mortbay.jetty.loadgenerator</groupId>
            <artifactId>jetty-load-generator-client</artifactId>
            <version>1.0.0-BETA0</version>
        </dependency>
        <dependency>
            <groupId>commons-cli</groupId>
            <artifactId>commons-cli</artifactId>
            <version>1.2</version>
        </dependency>
        <dependency>
            <groupId>com.github.javadev</groupId>
            <artifactId>underscore</artifactId>
            <version>1.42</version>
        </dependency>
        <dependency>
            <groupId>junit</groupId>
            <artifactId>junit</artifactId>
            <version>4.12</version>
            <scope>test</scope>
        </dependency>
    </dependencies>
</project>