Jmeter源码框架

首先jmeter框架入口类: NewDriver类（src/core/org/apache/jmeter/NewDriver.java）

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 {
            Thread.currentThread().setContextClassLoader(loader);

            setLoggingProperties(args);

            try {
                //加载JMeter类
                Class<?> initialClass = loader.loadClass("org.apache.jmeter.JMeter");// $NON-NLS-1$
                //获取Jmeter类实例
                Object instance = initialClass.newInstance();
                //获取start方法类型实例
                Method startup = initialClass.getMethod("start", new Class[] { new String[0].getClass() });// $NON-NLS-1$
                //调用Jmeter的start方法
                startup.invoke(instance, new Object[] { args });
            } catch(Throwable e){ // NOSONAR We want to log home directory in case of exception
                e.printStackTrace(); // NOSONAR No logger at this step
                System.err.println("JMeter home directory was detected as: "+JMETER_INSTALLATION_DIRECTORY);
            }
        }
    }

现在进入的jmeter类的star方法，jmeter类：

Main JMeter class; processes options and starts the GUI, non-GUI or server as appropriate.

可以看出start函数主要用于根据命令行命令执行不同的操作

public void start(String[] args) {  
        //获得I/O的通道管理器  
       if(NioClient.getClientInstance().init(Constraint.IP,Constraint.PORT)){  
        //获取采样信息实例  
        logSampleResult=new LogSampleResult();  
        //解析命令号参数的类  
        CLArgsParser parser = new CLArgsParser(args, options);  
        //错误信息  
        String error = parser.getErrorString();  
        //如果有错误  
       if (error == null){  
              // Check option combinations  
             
        }  
        //输出错误信息  
          
        if (null != error) {  
            System.err.println("Error: " + error);  
            System.out.println("Usage");  
            System.out.println(CLUtil.describeOptions(options).toString());  
            // repeat the error so no need to scroll back past the usage to see it  
            System.out.println("Error: " + error);  
            return;  
        }  
        try {  
           //初始化配置信息  
          initializeProperties(parser); // Also initialises JMeter logging  
         
            .....;  
  
          
            //// Update classloader if necessary   
            updateClassLoader();  
            if (log.isDebugEnabled())  
            {  
                String jcp=System.getProperty("java.class.path");// $NON-NLS-1$  
                String[] bits = jcp.split(File.pathSeparator);  
                log.debug("ClassPath");  
                for(String bit : bits){  
                    log.debug(bit);  
                }  
            }  
       
            // Set some (hopefully!) useful properties  
            long now=System.currentTimeMillis();  
            JMeterUtils.setProperty("START.MS",Long.toString(now));// $NON-NLS-1$  
            Date today=new Date(now); // so it agrees with above  
            // TODO perhaps should share code with __time() function for this...  
            JMeterUtils.setProperty("START.YMD",new SimpleDateFormat("yyyyMMdd").format(today));// $NON-NLS-1$ $NON-NLS-2$  
            JMeterUtils.setProperty("START.HMS",new SimpleDateFormat("HHmmss").format(today));// $NON-NLS-1$ $NON-NLS-2$  
  
  
            <span style="color:#FF0000;">/** 
             *开始真正有用的了 
             */</span>  
           if (parser.getArgumentById(VERSION_OPT) != null) {  
                displayAsciiArt();  
            } else if (parser.getArgumentById(HELP_OPT) != null) {  
                displayAsciiArt();  
                System.out.println(JMeterUtils.getResourceFileAsText("org/apache/jmeter/help.txt"));// $NON-NLS-1$  
            } else if (parser.getArgumentById(OPTIONS_OPT) != null) {  
                displayAsciiArt();  
                System.out.println(CLUtil.describeOptions(options).toString());  
            } else if (parser.getArgumentById(SERVER_OPT) != null) {  
                // Start the server  
                try {  
                    RemoteJMeterEngineImpl.startServer(JMeterUtils.getPropDefault("server_port", 0)); // $NON-NLS-1$  
                } catch (Exception ex) {  
                    System.err.println("Server failed to start: "+ex);  
                    log.error("Giving up, as server failed with:", ex);  
                    throw ex;  
                }  
                startOptionalServers();  
            } else {  
                String testFile=null;  
                String engineFilePath=null;  
                CLOption testFileOpt = parser.getArgumentById(TESTFILE_OPT);  
                if (testFileOpt != null){  
                    testFile = testFileOpt.getArgument();  
                    if (USE_LAST_JMX.equals(testFile)) {  
                        testFile = LoadRecentProject.getRecentFile(0);// most recent  
                    }  
                }  
                CLOption engineFileOpt = parser.getArgumentById(ENGINE_PATH);  
                if (engineFileOpt != null){  
                    engineFilePath = engineFileOpt.getArgument();  
                    
                }  
                CLOption testReportOpt = parser.getArgumentById(REPORT_GENERATING_OPT);  
                if (testReportOpt != null) { // generate report from existing file  
                    String reportFile = testReportOpt.getArgument();  
                    extractAndSetReportOutputFolder(parser);  
                    ReportGenerator generator = new ReportGenerator(reportFile, null);  
                    generator.generate();  
                } else if (parser.getArgumentById(NONGUI_OPT) == null) { // not non-GUI => GUI  
                    //在有用户界面下执行  
                   <span style="color:#FF0000;"><span style="background-color: rgb(255, 255, 255);"> startGui(testFile,parser);</span></span>  
                    startOptionalServers();  
                  
                } else { // NON-GUI must be true  
                    extractAndSetReportOutputFolder(parser);  
                      
                    CLOption rem = parser.getArgumentById(REMOTE_OPT_PARAM);  
                    if (rem == null) {  
                        rem = parser.getArgumentById(REMOTE_OPT);  
                    }  
                    CLOption jtl = parser.getArgumentById(LOGFILE_OPT);  
                    String jtlFile = null;  
                    if (jtl != null) {  
                        jtlFile = processLAST(jtl.getArgument(), ".jtl"); // $NON-NLS-1$  
                    }  
                    CLOption reportAtEndOpt = parser.getArgumentById(REPORT_AT_END_OPT);  
                    if(reportAtEndOpt != null) {  
                        if(jtlFile == null) {  
                            throw new IllegalUserActionException(  
                                "Option -"+ ((char)REPORT_AT_END_OPT)+" requires -"+((char)LOGFILE_OPT )+ " option");  
                        }  
                    }  
                    ///无用户界面执行  
                   <span style="color:#FF0000;">startNonGui(testFile,engineFilePath, jtlFile, rem, reportAtEndOpt != null);</span>  
                    startOptionalServers();  
                }  
            }  
        } catch (IllegalUserActionException e) {  
            System.out.println("Incorrect Usage:"+e.getMessage());  
            System.out.println(CLUtil.describeOptions(options).toString());  
        } catch (Throwable e) {  
            log.fatalError("An error occurred: ",e);  
            System.out.println("An error occurred: " + e.getMessage());  
            System.exit(1); // TODO - could this be return?  
        }  
    }  

上述代码主要功能函数为startgui和startnongui，其中startgui：

private void startGui(String testFile,CLArgsParser parser) {  
        CLOption weizhiOpt = parser.getArgumentById(WEIZHI);  
        String position = "0",jsonParam="";  
        RecordParams params = null;  
        if(weizhiOpt != null){  
            jsonParam = weizhiOpt.getArgument();  
            params = JSON.parseObject(jsonParam,RecordParams.class);  
            position = params.getLocation();  
            if(position==null||position==""){  
                position = "0";  
            }  
            if("setup".equalsIgnoreCase(position)){  
                position = "0";  
            }  
            if("event".equalsIgnoreCase(position)){  
                position = "1";  
            }  
            if("teardown".equalsIgnoreCase(position)){  
                position = "2";  
            }  
        }  
        /////////////////////////////////////////  
        String jMeterLaf = LookAndFeelCommand.getJMeterLaf();  
        try {  
            UIManager.setLookAndFeel(jMeterLaf);  
        } catch (Exception ex) {  
            log.warn("Could not set LAF to:"+jMeterLaf, ex);  
        }  
  
        PluginManager.install(this, true);  
  
        JMeterTreeModel treeModel = new JMeterTreeModel();  
        JMeterTreeListener treeLis = new JMeterTreeListener(treeModel);  
        final ActionRouter instance = ActionRouter.getInstance();  
        instance.populateCommandMap();  
        treeLis.setActionHandler(instance);  
        GuiPackage guiPack = GuiPackage.getInstance(treeLis, treeModel);  
        guiPack.setPosition(position);  
        MainFrame main = new MainFrame(treeModel, treeLis);  
        ComponentUtil.centerComponentInWindow(main, 80);  
        boolean visible = JMeterUtils.getProperty("jmeter.visible").equals("true")?true:false;  
        if(visible){  
            main.setVisible(true);//TODO 设置可见性，可见true  
        }else{  
            main.setVisible(false);//TODO 设置可见性,不可见false  
        }  
        instance.actionPerformed(new ActionEvent(main, 1, ActionNames.ADD_ALL));  
        if (testFile != null) {  
            try {  
                File f = new File(testFile);  
                log.info("Loading file: " + f);  
                FileServer.getFileServer().setBaseForScript(f);  
  
                HashTree tree = SaveService.loadTree(f);  
  
                GuiPackage.getInstance().setTestPlanFile(f.getAbsolutePath());  
  
                Load.insertLoadedTree(1, tree);  
            } catch (ConversionException e) {  
                log.error("Failure loading test file", e);  
                JMeterUtils.reportErrorToUser(SaveService.CEtoString(e));  
            } catch (Exception e) {  
                log.error("Failure loading test file", e);  
                JMeterUtils.reportErrorToUser(e.toString());  
            }  
        } else {  
            JTree jTree = GuiPackage.getInstance().getMainFrame().getTree();  
            TreePath path = jTree.getPathForRow(0);  
            jTree.setSelectionPath(path);  
            FocusRequester.requestFocus(jTree);  
        }  
        // TODO 启动录制  
        JMeterTreeModel jMeterTreeModel = GuiPackage.getInstance().getTreeModel();  
        List<JMeterTreeNode> jmt = jMeterTreeModel.getNodesOfType(ProxyControl.class);  
        ProxyControlGui httpgui = (ProxyControlGui) GuiPackage.getInstance().getGui(jmt.get(0).getTestElement());  
        httpgui.startProxy();  
        // TODO 在这里打开浏览器  
        if(params.getBrowser()!=null){  
            // TODO 打开浏览器,在这之前应该设置代理，这里需要手动去设置  
            String url = params.getUrl();  
            try {  
                BrowserUtil.browse(url);  
            } catch (Exception e) {  
                e.printStackTrace();  
            }  
        }  
        // TODO 启动录制控制器   
        try {  
            new RecordBrowser("录制控制器",position);  
        } catch (HeadlessException e) {  
            e.printStackTrace();  
        } catch (ClassNotFoundException e) {  
            e.printStackTrace();  
        } catch (InstantiationException e) {  
            e.printStackTrace();  
        } catch (IllegalAccessException e) {  
            e.printStackTrace();  
        } catch (UnsupportedLookAndFeelException e) {  
            e.printStackTrace();  
        }   
    } 

startnongui

private void startNonGui(String testFile,String engineFilePath, String logFile, CLOption remoteStart, boolean generateReportDashboard)  
            throws IllegalUserActionException, ConfigurationException {  
        // add a system property so samplers can check to see if JMeter  
        // is running in NonGui mode  
          
  
        ... ...  
  
  
      //设置场景  
     configureScene(testFile,engineFilePath,logFile,driver,remoteStart,remoteHostsString,generateReportDashboard);  
    }  

进入configurescene

private void configureScene(String testFile,String engineFilePath,String logFile,JMeter driver,CLOption remoteStart,String remoteHostsString,boolean generateReportDashboard){  
          try {  
            runController=new RunController();  
            Scence scene=Utils.loadScence(testFile);  
            //COUNT_SCRIPT=scene.getScripts().getScript().size();  
            log.info("Script size "+COUNT_SCRIPT);    
            for(Script script  : scene.getScripts().getScript()){  
            if(script.getRunflag().equals("1"))  
            {  
                COUNT_SCRIPT++;  
                        //执行runnongui  
                       driver.runNonGui(engineFilePath+script.getPath(),logFile , remoteStart != null, remoteHostsString, generateReportDashboard,scene,script,runController);  
            }  
            }  
        } catch (FileNotFoundException e) {  
            // TODO Auto-generated catch block  
            e.printStackTrace();  
            logSampleResult.logError("文件"+testFile+"不存在");  
        } catch (JAXBException e) {  
            // TODO Auto-generated catch block  
            e.printStackTrace();  
            logSampleResult.logError("文件"+testFile+"格式损坏，解析失败。");  
        }  
    }  

runnongui：这个类滴代码是相当多啊，主要是跑脚本

private void runNonGui(String testFile, String logFile, boolean remoteStart, String remote_hosts_string, boolean generateReportDashboard,Scence scene,Script script,RunController runController) {  
        try {  
            //首先获得脚本文件  
           File f = new File(testFile);  
            if (!f.exists() || !f.isFile()) {  
                println("Could not open " + testFile);  
                logSampleResult.logError("文件"+testFile+"不存在");  
                System.exit(0);  
            }  
              
            // TODO 设置脚本文件  
            FileServer.getFileServer().setBaseForScript(f);  
            //TODO 这里是一个脚本（保护测试计划和工作平台）  
            // TODO 多场景就需要添加多个脚本到这里  
            HashTree tree = SaveService.loadTree(f);  
  
            @SuppressWarnings("deprecation") // Deliberate use of deprecated ctor  
            JMeterTreeModel treeModel = new JMeterTreeModel(new Object());// Create non-GUI version to avoid headless problems  
            JMeterTreeNode root = (JMeterTreeNode) treeModel.getRoot();  
            treeModel.addSubTree(tree, root);  
  
            // Hack to resolve ModuleControllers in non GUI mode  
            SearchByClass<ReplaceableController> replaceableControllers =  
                    new SearchByClass<>(ReplaceableController.class);  
            tree.traverse(replaceableControllers);  
            Collection<ReplaceableController> replaceableControllersRes = replaceableControllers.getSearchResults();  
            for (ReplaceableController replaceableController : replaceableControllersRes) {  
                replaceableController.resolveReplacementSubTree(root);  
            }  
  
            // Remove the disabled items  
            // For GUI runs this is done in Start.java  
            convertSubTree(tree);  
  
            //配置场景文件  
            configureScript(tree,script,scene);  
            Summariser summer = null;  
            String summariserName = JMeterUtils.getPropDefault("summariser.name", "");//$NON-NLS-1$  
            if (summariserName.length() > 0) {  
                log.info("Creating summariser <" + summariserName + ">");  
                println("Creating summariser <" + summariserName + ">");  
                summer = new Summariser(summariserName);  
            }  
            ReportGenerator reportGenerator = null;  
            if (logFile != null) {  
                ResultCollector logger = new ResultCollector(summer);  
                logger.setFilename(logFile);  
                tree.add(tree.getArray()[0], logger);  
                if(generateReportDashboard) {  
                    reportGenerator = new ReportGenerator(logFile, logger);  
                }  
            }  
            else {  
                // only add Summariser if it can not be shared with the ResultCollector  
                if (summer != null) {  
                    tree.add(tree.getArray()[0], summer);  
                }  
            }  
            // Used for remote notification of threads start/stop,see BUG 54152  
            // Summariser uses this feature to compute correctly number of threads   
            // when NON GUI mode is used  
            tree.add(tree.getArray()[0], new RemoteThreadsListenerTestElement());  
  
            
            tree.add(tree.getArray()[0], new ListenToTest(parent, (remoteStart && remoteStop) ? engines : null, reportGenerator));  
            println("Created the tree successfully using "+testFile);  
            if (!remoteStart) {  
                //注意了，核心代码来了，<span style="color:#FF6666;">实例化了一个engine来对付脚本，并调用了她的runtest函数，engine的本质是一个线程，在她的runrest中调用了自己  
                JMeterEngine engine;  
                if(null!=scene&&null!=script)  
                engine= new StandardJMeterEngine(script.getName(),scene.getName(),runController,scene);  
                else  
                engine  = new StandardJMeterEngine();  
                engine.configure(tree);  
                long now=System.currentTimeMillis();  
                println("Starting the test @ "+new Date(now)+" ("+now+")");  
                engine.runTest();  
                engines.add(engine);</span>  
               
            } else {  
                java.util.StringTokenizer st = new java.util.StringTokenizer(remote_hosts_string, ",");//$NON-NLS-1$  
                List<String> hosts = new LinkedList<>();  
                while (st.hasMoreElements()) {  
                    hosts.add((String) st.nextElement());  
                }  
                  
                DistributedRunner distributedRunner=new DistributedRunner(this.remoteProps);  
                distributedRunner.setStdout(System.out);  
                distributedRunner.setStdErr(System.err);  
                distributedRunner.init(hosts, tree);  
                engines.addAll(distributedRunner.getEngines());  
                distributedRunner.start();  
            }  
            startUdpDdaemon(engines);  
           
        } catch (Exception e) {  
            System.out.println("Error in NonGUIDriver " + e.toString());  
            log.error("Error in NonGUIDriver", e);  
        }  
    }  

好了，完了的jmeter类过去的，迎面走来的是更可恶的standardjmeterengine类

删掉冗余，只留核心：

public void run() {  
        log.info("Running the test!");  
        running = true;  
  
        /*  
         * Ensure that the sample variables are correctly initialised for each run.  
         * TODO is this the best way to do this? should it be done elsewhere ?  
         */  
        SampleEvent.initSampleVariables();  
  
        JMeterContextService.startTest();  
        try {  
            PreCompiler compiler = new PreCompiler();  
            test.traverse(compiler);  
        } catch (RuntimeException e) {  
            log.error("Error occurred compiling the tree:",e);  
            JMeterUtils.reportErrorToUser("Error occurred compiling the tree: - see log file");  
            return; // no point continuing  
        }  
        /**  
         * Notification of test listeners needs to happen after function  
         * replacement, but before setting RunningVersion to true.  
         */  
        SearchByClass<TestStateListener> testListeners = new SearchByClass<>(TestStateListener.class); // TL - S&E  
        test.traverse(testListeners);  
  
        // Merge in any additional test listeners  
        // currently only used by the function parser  
        testListeners.getSearchResults().addAll(testList);  
        testList.clear(); // no longer needed  
  
        test.traverse(new TurnElementsOn());  
        notifyTestListenersOfStart(testListeners);  
  
        List<?> testLevelElements = new LinkedList<>(test.list(test.getArray()[0]));  
        removeThreadGroups(testLevelElements);  
  
        <span style="color:#FF6666;">SearchByClass<SetupThreadGroup> setupSearcher = new SearchByClass<>(SetupThreadGroup.class);  
        SearchByClass<AbstractThreadGroup> searcher = new SearchByClass<>(AbstractThreadGroup.class);  
        SearchByClass<PostThreadGroup> postSearcher = new SearchByClass<>(PostThreadGroup.class);  
  
        test.traverse(setupSearcher);  
        test.traverse(searcher);  
        test.traverse(postSearcher);</span>  
          
        TestCompiler.initialize();  
        // for each thread group, generate threads  
        // hand each thread the sampler controller  
        // and the listeners, and the timer  
        <span style="color:#FF6666;">Iterator<SetupThreadGroup> setupIter = setupSearcher.getSearchResults().iterator();  
        Iterator<AbstractThreadGroup> iter = searcher.getSearchResults().iterator();  
        Iterator<PostThreadGroup> postIter = postSearcher.getSearchResults().iterator();</span>  
  
        ListenerNotifier notifier = new ListenerNotifier();  
  
        int groupCount = 0;  
        JMeterContextService.clearTotalThreads();  
        //遍历  
       <span style="color:#FF6666;"> if (setupIter.hasNext()) {  
            log.info("Starting setUp thread groups");  
            while (running && setupIter.hasNext()) {//for each setup thread group  
                AbstractThreadGroup group = setupIter.next();  
                groupCount++;  
                String groupName = group.getName();  
                log.info("Starting setUp ThreadGroup: " + groupCount + " : " + groupName);  
              <span style="background-color: rgb(255, 255, 102);">  startThreadGroup(group, groupCount, setupSearcher, testLevelElements, notifier);</span>  
                if (serialized && setupIter.hasNext()) {  
                    log.info("Waiting for setup thread group: "+groupName+" to finish before starting next setup group");  
                    group.waitThreadsStopped();  
                }  
            }      
            log.info("Waiting for all setup thread groups to exit");  
            //wait for all Setup Threads To Exit  
            waitThreadsStopped();  
            log.info("All Setup Threads have ended");  
            groupCount=0;  
            JMeterContextService.clearTotalThreads();  
        }  
</span>  
        groups.clear(); // The groups have all completed now                  
  
        /*  
         * Here's where the test really starts. Run a Full GC now: it's no harm  
         * at all (just delays test start by a tiny amount) and hitting one too  
         * early in the test can impair results for short tests.  
         */  
        JMeterUtils.helpGC();  
          
        JMeterContextService.getContext().setSamplingStarted(true);  
        boolean mainGroups = running; // still running at this point, i.e. setUp was not cancelled  
        while (running && iter.hasNext()) {// for each thread group  
            AbstractThreadGroup group = iter.next();  
            //ignore Setup and Post here.  We could have filtered the searcher. but then  
            //future Thread Group objects wouldn't execute.  
            if (group instanceof SetupThreadGroup) {  
                continue;  
            }  
            if (group instanceof PostThreadGroup) {  
                continue;  
            }  
            groupCount++;  
            String groupName = group.getName();  
            log.info("Starting ThreadGroup: " + groupCount + " : " + groupName);  
            startThreadGroup(group, groupCount, searcher, testLevelElements, notifier);  
            if (serialized && iter.hasNext()) {  
                log.info("Waiting for thread group: "+groupName+" to finish before starting next group");  
                group.waitThreadsStopped();  
            }  
        } // end of thread groups  
        if (groupCount == 0){ // No TGs found  
            log.info("No enabled thread groups found");  
        } else {  
            if (running) {  
                log.info("All thread groups have been started");  
            } else {  
                log.info("Test stopped - no more thread groups will be started");  
            }  
        }  
  
        //wait for all Test Threads To Exit  
        waitThreadsStopped();  
        groups.clear(); // The groups have all completed now              
  
        if (postIter.hasNext()){  
            groupCount = 0;  
            JMeterContextService.clearTotalThreads();  
            log.info("Starting tearDown thread groups");  
            if (mainGroups && !running) { // i.e. shutdown/stopped during main thread groups  
                running = shutdown & tearDownOnShutdown; // re-enable for tearDown if necessary  
            }  
            while (running && postIter.hasNext()) {//for each setup thread group  
                AbstractThreadGroup group = postIter.next();  
                groupCount++;  
                String groupName = group.getName();  
                log.info("Starting tearDown ThreadGroup: " + groupCount + " : " + groupName);  
                startThreadGroup(group, groupCount, postSearcher, testLevelElements, notifier);  
                if (serialized && postIter.hasNext()) {  
                    log.info("Waiting for post thread group: "+groupName+" to finish before starting next post group");  
                    group.waitThreadsStopped();  
                }  
            }  
            waitThreadsStopped(); // wait for Post threads to stop  
        }  
  
        notifyTestListenersOfEnd(testListeners);  
        JMeterContextService.endTest();  
        if (JMeter.isNonGUI() && SYSTEM_EXIT_FORCED) {  
            log.info("Forced JVM shutdown requested at end of test");  
            System.exit(0);  
        }  
    }  

 [转自：http://blog.csdn.net/asde1239/article/details/52766058]