Jmeter源码系列(1) - NewDriver 类详解-Jmeter 的启动器

写 bug 的大耳朵图图
·

写在前面的话

Jmeter 全称(Apache JMeter)是一个开源的、功能强大的性能测试工具,用于对各种应用程序和协议进行功能、负载、压力和性能测试。它被广泛应用于软件开发和计划阶段,以确保应用程序在各种负载情况下的稳定性和可靠性。 本系列将从 Jmeter 代码层面陆续剖析其实现原理,包括但不限于 Jmeter 设计思路,Jmeter 核心对象/接口/方法。如有错误,敬请指正!

NewDriver

NewDriver 是 org.apache.jmeter 包下的一个类,如下是 NewDriver 源码中的类说明

/**
 * Main class for JMeter - sets up initial classpath and the loader.
 */

从这个说明中,我们可以知道,这个类提供了 2 个主要功能:

  • 初始化 classpath
  • 初始化一个 loader, 这个 loader 其实就是一个动态类加载器

以下内容摘抄自 NewDriver 源码,在源码中会使用注释来说明关键代码的作用,最后也会做总结,让我们开始吧

public final class NewDriver {

    /**
     * 定义一堆常量,会在 static 代码块中使用
     */
    private static final String CLASSPATH_SEPARATOR = File.pathSeparator;
    private static final String OS_NAME = System.getProperty("os.name");
    private static final String OS_NAME_LC = OS_NAME.toLowerCase(java.util.Locale.ENGLISH);
    private static final String JAVA_CLASS_PATH = "java.class.path";
    private static final String JMETER_LOGFILE_SYSTEM_PROPERTY = "jmeter.logfile";
    private static final String HEADLESS_MODE_PROPERTY = "java.awt.headless";
    /**
     * 动态类加载器,继承自 URLClassLoader,提供了一个静态方法 updateLoader(URL [] urls) 实现了动态加载 jar
     * 的功能。
     */
    private static final DynamicClassLoader loader;
    private static final String JMETER_INSTALLATION_DIRECTORY;
    private static final List<Exception> EXCEPTIONS_IN_INIT = new ArrayList<>();

    static {
        final List<URL> jars = new ArrayList<>();
        /**
         * 启动时从 jvm 获取 classpath
         */
        final String initiaClasspath = System.getProperty(JAVA_CLASS_PATH);
        String tmpDir;
        /**
         * 按照指定标记符来分割给定的字符串,但是 StringTokenizer 是一个遗留类,出于兼容性原因而保留,建议使用 String 的拆分方法或 java.util.regex 包。
         * 顺便说一下,Jmeter 源码中会包含非常多的过时的方法或者写法,有些是因为 Jmeter 本身开发较早,当时的 jdk 版本没有我们常用的新方法,
         * 有些则是因为当时 jdk 早期版本存在 bug,jmeter 会使用另一种写法来规避这些 bug,当然,现在这些 bug 可能已经修复了,不过 jmeter 的源码中
         * 任然会保留这部分注释
         */
        StringTokenizer tok = new StringTokenizer(initiaClasspath, File.pathSeparator);
        /**
         * 对 mac 系统做了单独的判断,我也没有深究为啥要单独处理,不晓得现在还需不需要这么写
         */
        if (tok.countTokens() == 1|| (tok.countTokens()  == 2 && OS_NAME_LC.startsWith("mac os x"))) {
            File jar = new File(tok.nextToken());
            try {
                tmpDir = jar.getCanonicalFile().getParentFile().getParent();
            } catch (IOException e) {
                tmpDir = null;
            }
        } else {
            /**
             * 从 jvm 获取 jmeter.home 属性,没有的话就默认从环境变量 JMETER_HOME 取值,当然这个值也不一定有,因为不是所有人都会配置 JMETER_HOME 这个环境变量
             * 其实从这边开始,大家就会发现,Jmeter 会经常使用 System.getProperty 来获取一些属性,在后面的代码中我们也会经常见到这样的代码
             */
            tmpDir = System.getProperty("jmeter.home", System.getenv("JMETER_HOME"));
            if (tmpDir == null || tmpDir.length() == 0) {
                File userDir = new File(System.getProperty("user.dir"));
                tmpDir = userDir.getAbsoluteFile().getParent();
            }
        }
        if (tmpDir == null) {
            tmpDir = System.getenv("JMETER_HOME");
        }
        JMETER_INSTALLATION_DIRECTORY = tmpDir;
        boolean usesUNC = OS_NAME_LC.startsWith("windows");
        StringBuilder classpath = new StringBuilder();
        /**
         * 下面的几个目录大家就很眼熟了,就是 Jmeter 解压后,主目录下的文件夹,里面都是 Jmeter 可能用到的一些 jar 包
         */
        File[] libDirs = new File[] { new File(JMETER_INSTALLATION_DIRECTORY + File.separator + "lib"),
                new File(JMETER_INSTALLATION_DIRECTORY + File.separator + "lib" + File.separator + "ext"),
                new File(JMETER_INSTALLATION_DIRECTORY + File.separator + "lib" + File.separator + "junit")};
        for (File libDir : libDirs) {
            File[] libJars = libDir.listFiles((dir, name) -> name.endsWith(".jar"));
            if (libJars == null) {
                new Throwable("Could not access " + libDir).printStackTrace();
                continue;
            }
            /**
             * 不晓得为啥要排个序
             */
            Arrays.sort(libJars);
            for (File libJar : libJars) {
                try {
                    String s = libJar.getPath();
                    if (usesUNC) {
                        if (s.startsWith("\\\\") && !s.startsWith("\\\\\\")) {
                            s = "\\\\" + s;
                        } else if (s.startsWith("//") && !s.startsWith("///")) {
                            s = "//" + s;
                        }
                    }
                    jars.add(new File(s).toURI().toURL());
                    classpath.append(CLASSPATH_SEPARATOR);
                    classpath.append(s);
                } catch (MalformedURLException e) {
                    EXCEPTIONS_IN_INIT.add(new Exception("Error adding jar:"+libJar.getAbsolutePath(), e));
                }
            }
        }
        System.setProperty(JAVA_CLASS_PATH, initiaClasspath + classpath.toString());
        /**
         * 类加载器会加载扫描到的这些 jar 包,为 Jmeter 真正启动做好准备
         */
        loader = AccessController.doPrivileged(
                (PrivilegedAction<DynamicClassLoader>) () ->
                        new DynamicClassLoader(jars.toArray(new URL[jars.size()]))
        );
    }
}

从上面的代码中,我们可以看到,NewDriver 在实例化时,会执行一个静态代码块,主要作用就是加载 Jmeter 安装目录下的 jar 包。

Main方法介绍

下面介绍 NewDriver 的 main 方法,这个方法就是整个 Jmeter 启动的入口方法。

public static void main(String[] args) {
        /**
         * 检查初始化是不是报错了
         */
        if(!EXCEPTIONS_IN_INIT.isEmpty()) {
            System.err.println("Configuration error during init, see exceptions:"+exceptionsToString(EXCEPTIONS_IN_INIT));
        } else {
            /**
             * 设置当前线程的类加载器,也就是 Jmeter 自己写的那个动态类加载器
             */
            Thread.currentThread().setContextClassLoader(loader);
            /**
             * 配置一些日志属性,不重要
             */
            setLoggingProperties(args);
            try {
                /**
                 * 判断要不要用 GUI 模式启动,默认 true,也可以通过 Jmeter 命令行参数 -n 来指定使用非 GUI 模式启动
                 */
                if(System.getProperty(HEADLESS_MODE_PROPERTY) == null && shouldBeHeadless(args)) {
                    System.setProperty(HEADLESS_MODE_PROPERTY, "true");
                }
                /**
                 * 获取 Jmeter 类,作用类似于 Class.forName(String clazzName)
                 */
                Class<?> initialClass = loader.loadClass("org.apache.jmeter.JMeter");
                /**
                 * 获取 Jmeter 实例
                 */
                Object instance = initialClass.getDeclaredConstructor().newInstance();
                /**
                 * 获取 Jmeter.start方法,并调用
                 */
                Method startup = initialClass.getMethod("start", new Class[] { new String[0].getClass() });
                startup.invoke(instance, new Object[] { args });
            } catch(Throwable e){ 
                e.printStackTrace();
                System.err.println("JMeter home directory was detected as: "+JMETER_INSTALLATION_DIRECTORY);
            }
        }
    }

main 方法其实很简单直接,就是看下是不是要启动 GUI,然后就是通过反射调用 Jmeter 的 start 方法,来开始测试。 综上,NewDriver 其实就是一个启动器,正如其所在源码模块 launcher 一样,他的作用就是为 Jmeter 真正启动做好准备。 好了,NewDriver 就介绍完了,下一章将介绍 Jmeter 这个核心类,以及调用其 start(String[] args) 之后会发生什么...

2
社区准则 博客 联系 社区 状态
主题