com.webtide.jetty.load.generator.jenkins.LoadGeneratorProcessRunner.java Source code

Introduction

Here is the source code for com.webtide.jetty.load.generator.jenkins.LoadGeneratorProcessRunner.java

Source

//
//  ========================================================================
//  Copyright (c) 1995-2016 Webtide LLC, Olivier Lamy
//  ------------------------------------------------------------------------
//  All rights reserved. This program and the accompanying materials
//  are made available under the terms of the Eclipse Public License v1.0
//  and Apache License v2.0 which accompanies this distribution.
//
//      The Eclipse Public License is available at
//      http://www.eclipse.org/legal/epl-v10.html
//
//      The Apache License v2.0 is available at
//      http://www.opensource.org/licenses/apache2.0.php
//
//  You may elect to redistribute this code under either of these licenses.
//  ========================================================================

package com.webtide.jetty.load.generator.jenkins;

import hudson.FilePath;
import hudson.Launcher;
import hudson.model.JDK;
import hudson.model.Node;
import hudson.model.TaskListener;
import hudson.remoting.Channel;
import hudson.util.ArgumentListBuilder;
import jenkins.model.Jenkins;
import jenkins.security.MasterToSlaveCallable;
import org.apache.commons.lang.StringUtils;
import org.eclipse.jetty.client.HttpClient;
import org.eclipse.jetty.client.api.ContentResponse;
import org.eclipse.jetty.util.SocketAddressResolver;
import org.mortbay.jetty.load.generator.LoadGenerator;
import org.mortbay.jetty.load.generator.Resource;

import java.io.Serializable;
import java.lang.reflect.Method;
import java.util.List;
import java.util.Map;
import java.util.concurrent.TimeUnit;

/**
 *
 */
public class LoadGeneratorProcessRunner implements Serializable {

    public void runProcess(TaskListener taskListener, FilePath workspace, Launcher launcher, String jdkName, //
            Node currentNode, List<Resource.NodeListener> nodeListeners, //
            List<LoadGenerator.Listener> loadGeneratorListeners, //
            List<String> args, String jvmExtraArgs, String alpnVersion, //
            Map<String, String> jdkVersionAlpnBootVersions) throws Exception {

        Channel channel = null;

        try {
            long start = System.nanoTime();

            JDK jdk = StringUtils.isEmpty(jdkName) ? //
                    null : Jenkins.getInstance().getJDK(jdkName).forNode(currentNode, taskListener);

            channel = new LoadGeneratorProcessFactory().buildChannel(taskListener, jdk, workspace, launcher,
                    jvmExtraArgs);

            // alpn version from jdk
            if (StringUtils.isEmpty(alpnVersion) && !StringUtils.equals(alpnVersion, "N/A")) {

                String javaVersion = findJavaVersion(channel, taskListener);
                alpnVersion = jdkVersionAlpnBootVersions.get(javaVersion);
                // download alpn jar
            }

            channel.call(new LoadCaller(args, nodeListeners, loadGeneratorListeners));

            long end = System.nanoTime();

            taskListener.getLogger().println("remote LoadGenerator execution done: " //
                    + TimeUnit.NANOSECONDS.toMillis(end - start) //
                    + " ms ");
        } finally {
            if (channel != null) {
                channel.close();
            }
        }

    }

    private static class LoadCaller extends MasterToSlaveCallable<Void, Exception> implements Serializable {
        private final List<String> args;

        private final List<?> nodeListeners;

        private final List<?> loadGeneratorListeners;

        public LoadCaller(List<String> args, List<?> nodeListeners, List<?> loadGeneratorListeners) {
            this.args = args;
            this.nodeListeners = nodeListeners;
            this.loadGeneratorListeners = loadGeneratorListeners;
        }

        @Override
        public Void call() throws Exception {
            ClassLoader classLoader = Thread.currentThread().getContextClassLoader();

            Class jenkinsRemoteStarterClazz = classLoader
                    .loadClass("org.mortbay.jetty.load.generator.starter.JenkinsRemoteStarter");

            Method setResponseTimeListeners = jenkinsRemoteStarterClazz.getMethod("setNodeListeners", List.class);
            setResponseTimeListeners.invoke(null, nodeListeners);

            Method setLoadGeneratorListeners = jenkinsRemoteStarterClazz.getMethod("setLoadGeneratorListeners",
                    List.class);
            setLoadGeneratorListeners.invoke(null, loadGeneratorListeners);

            Method launch = jenkinsRemoteStarterClazz.getMethod("launch", List.class);

            launch.invoke(null, args);

            return null;
        }
    }

    private String findJavaVersion(Channel channel, TaskListener taskListener) throws Exception {
        String javaVersion = channel.call(new JavaVersionCallable());
        taskListener.getLogger().println("found javaVersion " + javaVersion);
        return javaVersion;
    }

    private static class JavaVersionCallable extends MasterToSlaveCallable<String, Exception>
            implements Serializable {
        @Override
        public String call() throws Exception {
            return System.getProperty("java.version");
        }
    }

}