| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You under the Apache License, Version 2.0 |
| * (the "License"); you may not use this file except in compliance with |
| * the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| * |
| */ |
| |
| package org.apache.jmeter; |
| |
| import java.awt.event.ActionEvent; |
| import java.io.File; |
| import java.io.FileInputStream; |
| import java.io.FileNotFoundException; |
| import java.io.IOException; |
| import java.io.InputStream; |
| import java.lang.Thread.UncaughtExceptionHandler; |
| import java.net.Authenticator; |
| import java.net.DatagramPacket; |
| import java.net.DatagramSocket; |
| import java.net.InetAddress; |
| import java.net.MalformedURLException; |
| import java.net.SocketException; |
| import java.text.SimpleDateFormat; |
| import java.util.Collection; |
| import java.util.Date; |
| import java.util.Enumeration; |
| import java.util.LinkedList; |
| import java.util.List; |
| import java.util.Locale; |
| import java.util.Properties; |
| import java.util.StringTokenizer; |
| import java.util.concurrent.TimeUnit; |
| import java.util.concurrent.atomic.AtomicInteger; |
| |
| import javax.swing.JTree; |
| import javax.swing.UIManager; |
| import javax.swing.tree.TreePath; |
| |
| import org.apache.commons.cli.avalon.CLArgsParser; |
| import org.apache.commons.cli.avalon.CLOption; |
| import org.apache.commons.cli.avalon.CLOptionDescriptor; |
| import org.apache.commons.cli.avalon.CLUtil; |
| import org.apache.commons.io.IOUtils; |
| import org.apache.jmeter.control.ReplaceableController; |
| import org.apache.jmeter.engine.ClientJMeterEngine; |
| import org.apache.jmeter.engine.DistributedRunner; |
| import org.apache.jmeter.engine.JMeterEngine; |
| import org.apache.jmeter.engine.RemoteJMeterEngineImpl; |
| import org.apache.jmeter.engine.StandardJMeterEngine; |
| import org.apache.jmeter.exceptions.IllegalUserActionException; |
| import org.apache.jmeter.gui.GuiPackage; |
| import org.apache.jmeter.gui.MainFrame; |
| import org.apache.jmeter.gui.action.ActionNames; |
| import org.apache.jmeter.gui.action.ActionRouter; |
| import org.apache.jmeter.gui.action.Load; |
| import org.apache.jmeter.gui.action.LoadRecentProject; |
| import org.apache.jmeter.gui.action.LookAndFeelCommand; |
| import org.apache.jmeter.gui.tree.JMeterTreeListener; |
| import org.apache.jmeter.gui.tree.JMeterTreeModel; |
| import org.apache.jmeter.gui.tree.JMeterTreeNode; |
| import org.apache.jmeter.gui.util.FocusRequester; |
| import org.apache.jmeter.plugin.JMeterPlugin; |
| import org.apache.jmeter.plugin.PluginManager; |
| import org.apache.jmeter.report.config.ConfigurationException; |
| import org.apache.jmeter.report.dashboard.GenerationException; |
| import org.apache.jmeter.report.dashboard.ReportGenerator; |
| import org.apache.jmeter.reporters.ResultCollector; |
| import org.apache.jmeter.reporters.Summariser; |
| import org.apache.jmeter.samplers.Remoteable; |
| import org.apache.jmeter.samplers.SampleEvent; |
| import org.apache.jmeter.save.SaveService; |
| import org.apache.jmeter.services.FileServer; |
| import org.apache.jmeter.testelement.TestElement; |
| import org.apache.jmeter.testelement.TestStateListener; |
| import org.apache.jmeter.threads.RemoteThreadsListenerTestElement; |
| import org.apache.jmeter.util.BeanShellInterpreter; |
| import org.apache.jmeter.util.BeanShellServer; |
| import org.apache.jmeter.util.JMeterUtils; |
| import org.apache.jorphan.collections.HashTree; |
| import org.apache.jorphan.collections.SearchByClass; |
| import org.apache.jorphan.gui.ComponentUtil; |
| import org.apache.jorphan.logging.LoggingManager; |
| import org.apache.jorphan.reflect.ClassTools; |
| import org.apache.jorphan.util.HeapDumper; |
| import org.apache.jorphan.util.JMeterException; |
| import org.apache.jorphan.util.JOrphanUtils; |
| import org.apache.log.Logger; |
| |
| import com.thoughtworks.xstream.converters.ConversionException; |
| |
| /** |
| * Main JMeter class; processes options and starts the GUI, non-GUI or server as appropriate. |
| */ |
| public class JMeter implements JMeterPlugin { |
| private static final Logger log = LoggingManager.getLoggerForClass(); |
| |
| public static final int UDP_PORT_DEFAULT = 4445; // needed for ShutdownClient |
| |
| public static final String HTTP_PROXY_PASS = "http.proxyPass"; // $NON-NLS-1$ |
| |
| public static final String HTTP_PROXY_USER = "http.proxyUser"; // $NON-NLS-1$ |
| |
| public static final String JMETER_NON_GUI = "JMeter.NonGui"; // $NON-NLS-1$ |
| |
| public static final String JMETER_REPORT_OUTPUT_DIR_PROPERTY = |
| "jmeter.reportgenerator.outputdir"; //$NON-NLS-1$ |
| |
| // Icons size in the JMeter tree |
| public static final String TREE_ICON_SIZE = "jmeter.tree.icons.size"; //$NON-NLS-1$ |
| |
| public static final String DEFAULT_TREE_ICON_SIZE = "19x19"; //$NON-NLS-1$ |
| |
| protected static final String KEY_SIZE = "<SIZE>"; //$NON-NLS-1$ |
| |
| // If the -t flag is to "LAST", then the last loaded file (if any) is used |
| private static final String USE_LAST_JMX = "LAST"; |
| // If the -j or -l flag is set to LAST or LAST.log|LAST.jtl, then the last loaded file name is used to |
| // generate the log file name by removing .JMX and replacing it with .log|.jtl |
| |
| private static final int PROXY_PASSWORD = 'a';// $NON-NLS-1$ |
| private static final int JMETER_HOME_OPT = 'd';// $NON-NLS-1$ |
| private static final int HELP_OPT = 'h';// $NON-NLS-1$ |
| private static final int OPTIONS_OPT = '?';// $NON-NLS-1$ |
| // jmeter.log |
| private static final int JMLOGFILE_OPT = 'j';// $NON-NLS-1$ |
| // sample result log file |
| private static final int LOGFILE_OPT = 'l';// $NON-NLS-1$ |
| private static final int NONGUI_OPT = 'n';// $NON-NLS-1$ |
| private static final int PROPFILE_OPT = 'p';// $NON-NLS-1$ |
| private static final int PROPFILE2_OPT = 'q';// $NON-NLS-1$ |
| private static final int REMOTE_OPT = 'r';// $NON-NLS-1$ |
| private static final int SERVER_OPT = 's';// $NON-NLS-1$ |
| private static final int TESTFILE_OPT = 't';// $NON-NLS-1$ |
| private static final int PROXY_USERNAME = 'u';// $NON-NLS-1$ |
| private static final int VERSION_OPT = 'v';// $NON-NLS-1$ |
| private static final int REPORT_GENERATING_OPT = 'g';// $NON-NLS-1$ |
| private static final int REPORT_AT_END_OPT = 'e';// $NON-NLS-1$ |
| private static final int REPORT_OUTPUT_FOLDER_OPT = 'o';// $NON-NLS-1$ |
| |
| private static final int SYSTEM_PROPERTY = 'D';// $NON-NLS-1$ |
| private static final int JMETER_GLOBAL_PROP = 'G';// $NON-NLS-1$ |
| private static final int PROXY_HOST = 'H';// $NON-NLS-1$ |
| private static final int JMETER_PROPERTY = 'J';// $NON-NLS-1$ |
| private static final int LOGLEVEL = 'L';// $NON-NLS-1$ |
| private static final int NONPROXY_HOSTS = 'N';// $NON-NLS-1$ |
| private static final int PROXY_PORT = 'P';// $NON-NLS-1$ |
| private static final int REMOTE_OPT_PARAM = 'R';// $NON-NLS-1$ |
| private static final int SYSTEM_PROPFILE = 'S';// $NON-NLS-1$ |
| private static final int REMOTE_STOP = 'X';// $NON-NLS-1$ |
| |
| /** |
| * Define the understood options. Each CLOptionDescriptor contains: |
| * <ul> |
| * <li>The "long" version of the option. Eg, "help" means that "--help" |
| * will be recognised.</li> |
| * <li>The option flags, governing the option's argument(s).</li> |
| * <li>The "short" version of the option. Eg, 'h' means that "-h" will be |
| * recognised.</li> |
| * <li>A description of the option.</li> |
| * </ul> |
| */ |
| private static final CLOptionDescriptor D_OPTIONS_OPT = |
| new CLOptionDescriptor("?", CLOptionDescriptor.ARGUMENT_DISALLOWED, OPTIONS_OPT, |
| "print command line options and exit"); |
| private static final CLOptionDescriptor D_HELP_OPT = |
| new CLOptionDescriptor("help", CLOptionDescriptor.ARGUMENT_DISALLOWED, HELP_OPT, |
| "print usage information and exit"); |
| private static final CLOptionDescriptor D_VERSION_OPT = |
| new CLOptionDescriptor("version", CLOptionDescriptor.ARGUMENT_DISALLOWED, VERSION_OPT, |
| "print the version information and exit"); |
| private static final CLOptionDescriptor D_PROPFILE_OPT = |
| new CLOptionDescriptor("propfile", CLOptionDescriptor.ARGUMENT_REQUIRED, PROPFILE_OPT, |
| "the jmeter property file to use"); |
| private static final CLOptionDescriptor D_PROPFILE2_OPT = |
| new CLOptionDescriptor("addprop", CLOptionDescriptor.ARGUMENT_REQUIRED |
| | CLOptionDescriptor.DUPLICATES_ALLOWED, PROPFILE2_OPT, |
| "additional JMeter property file(s)"); |
| private static final CLOptionDescriptor D_TESTFILE_OPT = |
| new CLOptionDescriptor("testfile", CLOptionDescriptor.ARGUMENT_REQUIRED, TESTFILE_OPT, |
| "the jmeter test(.jmx) file to run"); |
| private static final CLOptionDescriptor D_LOGFILE_OPT = |
| new CLOptionDescriptor("logfile", CLOptionDescriptor.ARGUMENT_REQUIRED, LOGFILE_OPT, |
| "the file to log samples to"); |
| private static final CLOptionDescriptor D_JMLOGFILE_OPT = |
| new CLOptionDescriptor("jmeterlogfile", CLOptionDescriptor.ARGUMENT_REQUIRED, JMLOGFILE_OPT, |
| "jmeter run log file (jmeter.log)"); |
| private static final CLOptionDescriptor D_NONGUI_OPT = |
| new CLOptionDescriptor("nongui", CLOptionDescriptor.ARGUMENT_DISALLOWED, NONGUI_OPT, |
| "run JMeter in nongui mode"); |
| private static final CLOptionDescriptor D_SERVER_OPT = |
| new CLOptionDescriptor("server", CLOptionDescriptor.ARGUMENT_DISALLOWED, SERVER_OPT, |
| "run the JMeter server"); |
| private static final CLOptionDescriptor D_PROXY_HOST = |
| new CLOptionDescriptor("proxyHost", CLOptionDescriptor.ARGUMENT_REQUIRED, PROXY_HOST, |
| "Set a proxy server for JMeter to use"); |
| private static final CLOptionDescriptor D_PROXY_PORT = |
| new CLOptionDescriptor("proxyPort", CLOptionDescriptor.ARGUMENT_REQUIRED, PROXY_PORT, |
| "Set proxy server port for JMeter to use"); |
| private static final CLOptionDescriptor D_NONPROXY_HOSTS = |
| new CLOptionDescriptor("nonProxyHosts", CLOptionDescriptor.ARGUMENT_REQUIRED, NONPROXY_HOSTS, |
| "Set nonproxy host list (e.g. *.apache.org|localhost)"); |
| private static final CLOptionDescriptor D_PROXY_USERNAME = |
| new CLOptionDescriptor("username", CLOptionDescriptor.ARGUMENT_REQUIRED, PROXY_USERNAME, |
| "Set username for proxy server that JMeter is to use"); |
| private static final CLOptionDescriptor D_PROXY_PASSWORD = |
| new CLOptionDescriptor("password", CLOptionDescriptor.ARGUMENT_REQUIRED, PROXY_PASSWORD, |
| "Set password for proxy server that JMeter is to use"); |
| private static final CLOptionDescriptor D_JMETER_PROPERTY = |
| new CLOptionDescriptor("jmeterproperty", CLOptionDescriptor.DUPLICATES_ALLOWED |
| | CLOptionDescriptor.ARGUMENTS_REQUIRED_2, JMETER_PROPERTY, |
| "Define additional JMeter properties"); |
| private static final CLOptionDescriptor D_JMETER_GLOBAL_PROP = |
| new CLOptionDescriptor("globalproperty", CLOptionDescriptor.DUPLICATES_ALLOWED |
| | CLOptionDescriptor.ARGUMENTS_REQUIRED_2, JMETER_GLOBAL_PROP, |
| "Define Global properties (sent to servers)\n\t\te.g. -Gport=123 or -Gglobal.properties"); |
| private static final CLOptionDescriptor D_SYSTEM_PROPERTY = |
| new CLOptionDescriptor("systemproperty", CLOptionDescriptor.DUPLICATES_ALLOWED |
| | CLOptionDescriptor.ARGUMENTS_REQUIRED_2, SYSTEM_PROPERTY, |
| "Define additional system properties"); |
| private static final CLOptionDescriptor D_SYSTEM_PROPFILE = |
| new CLOptionDescriptor("systemPropertyFile", CLOptionDescriptor.DUPLICATES_ALLOWED |
| | CLOptionDescriptor.ARGUMENT_REQUIRED, SYSTEM_PROPFILE, |
| "additional system property file(s)"); |
| private static final CLOptionDescriptor D_LOGLEVEL = |
| new CLOptionDescriptor("loglevel", CLOptionDescriptor.DUPLICATES_ALLOWED |
| | CLOptionDescriptor.ARGUMENTS_REQUIRED_2, LOGLEVEL, |
| "[category=]level e.g. jorphan=INFO or jmeter.util=DEBUG"); |
| private static final CLOptionDescriptor D_REMOTE_OPT = |
| new CLOptionDescriptor("runremote", CLOptionDescriptor.ARGUMENT_DISALLOWED, REMOTE_OPT, |
| "Start remote servers (as defined in remote_hosts)"); |
| private static final CLOptionDescriptor D_REMOTE_OPT_PARAM = |
| new CLOptionDescriptor("remotestart", CLOptionDescriptor.ARGUMENT_REQUIRED, REMOTE_OPT_PARAM, |
| "Start these remote servers (overrides remote_hosts)"); |
| private static final CLOptionDescriptor D_JMETER_HOME_OPT = |
| new CLOptionDescriptor("homedir", CLOptionDescriptor.ARGUMENT_REQUIRED, JMETER_HOME_OPT, |
| "the jmeter home directory to use"); |
| private static final CLOptionDescriptor D_REMOTE_STOP = |
| new CLOptionDescriptor("remoteexit", CLOptionDescriptor.ARGUMENT_DISALLOWED, REMOTE_STOP, |
| "Exit the remote servers at end of test (non-GUI)"); |
| private static final CLOptionDescriptor D_REPORT_GENERATING_OPT = |
| new CLOptionDescriptor("reportonly", |
| CLOptionDescriptor.ARGUMENT_REQUIRED, REPORT_GENERATING_OPT, |
| "generate report dashboard only, from a test results file", |
| new CLOptionDescriptor[]{ D_NONGUI_OPT, D_REMOTE_OPT, D_REMOTE_OPT_PARAM, D_LOGFILE_OPT }); // disallowed |
| private static final CLOptionDescriptor D_REPORT_AT_END_OPT = |
| new CLOptionDescriptor("reportatendofloadtests", |
| CLOptionDescriptor.ARGUMENT_DISALLOWED, REPORT_AT_END_OPT, |
| "generate report dashboard after load test"); |
| private static final CLOptionDescriptor D_REPORT_OUTPUT_FOLDER_OPT = |
| new CLOptionDescriptor("reportoutputfolder", |
| CLOptionDescriptor.ARGUMENT_REQUIRED, REPORT_OUTPUT_FOLDER_OPT, |
| "output folder for report dashboard"); |
| |
| private static final CLOptionDescriptor[] options = new CLOptionDescriptor[] { |
| D_OPTIONS_OPT, |
| D_HELP_OPT, |
| D_VERSION_OPT, |
| D_PROPFILE_OPT, |
| D_PROPFILE2_OPT, |
| D_TESTFILE_OPT, |
| D_LOGFILE_OPT, |
| D_JMLOGFILE_OPT, |
| D_NONGUI_OPT, |
| D_SERVER_OPT, |
| D_PROXY_HOST, |
| D_PROXY_PORT, |
| D_NONPROXY_HOSTS, |
| D_PROXY_USERNAME, |
| D_PROXY_PASSWORD, |
| D_JMETER_PROPERTY, |
| D_JMETER_GLOBAL_PROP, |
| D_SYSTEM_PROPERTY, |
| D_SYSTEM_PROPFILE, |
| D_LOGLEVEL, |
| D_REMOTE_OPT, |
| D_REMOTE_OPT_PARAM, |
| D_JMETER_HOME_OPT, |
| D_REMOTE_STOP, |
| D_REPORT_GENERATING_OPT, |
| D_REPORT_AT_END_OPT, |
| D_REPORT_OUTPUT_FOLDER_OPT, |
| }; |
| |
| public JMeter() { |
| } |
| |
| |
| private JMeter parent; |
| |
| /** Properties to be sent to remote servers */ |
| private Properties remoteProps; |
| |
| /** should remote engines be stopped at end of non-GUI test? */ |
| private boolean remoteStop; |
| |
| /** |
| * Starts up JMeter in GUI mode |
| */ |
| private void startGui(String testFile) { |
| 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.initInstance(treeLis, treeModel); |
| MainFrame main = new MainFrame(treeModel, treeLis); |
| ComponentUtil.centerComponentInWindow(main, 80); |
| main.setVisible(true); |
| 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); |
| } |
| } |
| |
| /** |
| * Takes the command line arguments and uses them to determine how to |
| * startup JMeter. |
| * |
| * Called reflectively by {@link NewDriver#main(String[])} |
| * @param args The arguments for JMeter |
| */ |
| public void start(String[] args) { |
| CLArgsParser parser = new CLArgsParser(args, options); |
| String error = parser.getErrorString(); |
| if (error == null){// Check option combinations |
| boolean gui = parser.getArgumentById(NONGUI_OPT)==null; |
| boolean nonGuiOnly = parser.getArgumentById(REMOTE_OPT)!=null |
| || parser.getArgumentById(REMOTE_OPT_PARAM)!=null |
| || parser.getArgumentById(REMOTE_STOP)!=null; |
| if (gui && nonGuiOnly) { |
| error = "-r and -R and -X are only valid in non-GUI mode"; |
| } |
| } |
| 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 |
| /* |
| * The following is needed for HTTPClient. |
| * (originally tried doing this in HTTPSampler2, |
| * but it appears that it was done too late when running in GUI mode) |
| * Set the commons logging default to Avalon Logkit, if not already defined |
| */ |
| if (System.getProperty("org.apache.commons.logging.Log") == null) { // $NON-NLS-1$ |
| System.setProperty("org.apache.commons.logging.Log" // $NON-NLS-1$ |
| , "org.apache.commons.logging.impl.LogKitLogger"); // $NON-NLS-1$ |
| } |
| |
| Thread.setDefaultUncaughtExceptionHandler(new UncaughtExceptionHandler() { |
| @Override |
| public void uncaughtException(Thread t, Throwable e) { |
| if (!(e instanceof ThreadDeath)) { |
| log.error("Uncaught exception: ", e); |
| System.err.println("Uncaught Exception " + e + ". See log file for details."); |
| } |
| } |
| }); |
| |
| log.info(JMeterUtils.getJMeterCopyright()); |
| log.info("Version " + JMeterUtils.getJMeterVersion()); |
| logProperty("java.version"); //$NON-NLS-1$ |
| logProperty("java.vm.name"); //$NON-NLS-1$ |
| logProperty("os.name"); //$NON-NLS-1$ |
| logProperty("os.arch"); //$NON-NLS-1$ |
| logProperty("os.version"); //$NON-NLS-1$ |
| logProperty("file.encoding"); // $NON-NLS-1$ |
| log.info("Max memory ="+ Runtime.getRuntime().maxMemory()); |
| log.info("Available Processors ="+ Runtime.getRuntime().availableProcessors()); |
| log.info("Default Locale=" + Locale.getDefault().getDisplayName()); |
| log.info("JMeter Locale=" + JMeterUtils.getLocale().getDisplayName()); |
| log.info("JMeterHome=" + JMeterUtils.getJMeterHome()); |
| logProperty("user.dir"," ="); //$NON-NLS-1$ |
| log.info("PWD ="+new File(".").getCanonicalPath());//$NON-NLS-1$ |
| log.info("IP: "+JMeterUtils.getLocalHostIP() |
| +" Name: "+JMeterUtils.getLocalHostName() |
| +" FullName: "+JMeterUtils.getLocalHostFullName()); |
| setProxy(parser); |
| |
| 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$ |
| |
| 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; |
| 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 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 |
| startGui(testFile); |
| 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"); |
| } |
| } |
| startNonGui(testFile, jtlFile, rem, reportAtEndOpt != null); |
| 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? |
| } |
| } |
| |
| /** |
| * Extract option JMeter#REPORT_OUTPUT_FOLDER_OPT and if defined sets property |
| * {@link JMeter#JMETER_REPORT_OUTPUT_DIR_PROPERTY} after checking folder can |
| * be safely written to |
| * @param parser {@link CLArgsParser} |
| * @throws IllegalArgumentException |
| */ |
| private void extractAndSetReportOutputFolder(CLArgsParser parser) |
| throws IllegalArgumentException { |
| CLOption reportOutputFolderOpt = parser |
| .getArgumentById(REPORT_OUTPUT_FOLDER_OPT); |
| if(reportOutputFolderOpt != null) { |
| String reportOutputFolder = parser.getArgumentById(REPORT_OUTPUT_FOLDER_OPT).getArgument(); |
| File reportOutputFolderAsFile = new File(reportOutputFolder); |
| |
| JOrphanUtils.canSafelyWriteToFolder(reportOutputFolderAsFile); |
| log.info("Setting property '"+JMETER_REPORT_OUTPUT_DIR_PROPERTY+"' to:'"+reportOutputFolderAsFile.getAbsolutePath()+"'"); |
| JMeterUtils.setProperty(JMETER_REPORT_OUTPUT_DIR_PROPERTY, |
| reportOutputFolderAsFile.getAbsolutePath()); |
| } |
| } |
| |
| /** |
| * Displays as ASCII Art Apache JMeter version + Copyright notice |
| */ |
| private void displayAsciiArt() { |
| try (InputStream inputStream = JMeter.class.getResourceAsStream("jmeter_as_ascii_art.txt")) { |
| if(inputStream != null) { |
| String text = IOUtils.toString(inputStream); |
| System.out.println(text); |
| } |
| } catch (Exception e1) { |
| System.out.println(JMeterUtils.getJMeterCopyright()); |
| System.out.println("Version " + JMeterUtils.getJMeterVersion()); |
| } |
| } |
| |
| // Update classloader if necessary |
| private void updateClassLoader() { |
| updatePath("search_paths",";", true); //$NON-NLS-1$//$NON-NLS-2$ |
| updatePath("user.classpath",File.pathSeparator, true);//$NON-NLS-1$ |
| updatePath("plugin_dependency_paths",";", false);//$NON-NLS-1$ |
| } |
| |
| private void updatePath(String property, String sep, boolean cp) { |
| String userpath= JMeterUtils.getPropDefault(property,"");// $NON-NLS-1$ |
| if (userpath.length() <= 0) { return; } |
| log.info(property+"="+userpath); //$NON-NLS-1$ |
| StringTokenizer tok = new StringTokenizer(userpath, sep); |
| while(tok.hasMoreTokens()) { |
| String path=tok.nextToken(); |
| File f=new File(path); |
| if (!f.canRead() && !f.isDirectory()) { |
| log.warn("Can't read "+path); |
| } else { |
| if (cp) { |
| log.info("Adding to classpath and loader: "+path); |
| try { |
| NewDriver.addPath(path); |
| } catch (MalformedURLException e) { |
| log.warn("Error adding: "+path+" "+e.getLocalizedMessage()); |
| } |
| } else { |
| log.info("Adding to loader: "+path); |
| NewDriver.addURL(path); |
| } |
| } |
| } |
| } |
| |
| /** |
| * |
| */ |
| private void startOptionalServers() { |
| int bshport = JMeterUtils.getPropDefault("beanshell.server.port", 0);// $NON-NLS-1$ |
| String bshfile = JMeterUtils.getPropDefault("beanshell.server.file", "");// $NON-NLS-1$ $NON-NLS-2$ |
| if (bshport > 0) { |
| log.info("Starting Beanshell server (" + bshport + "," + bshfile + ")"); |
| Runnable t = new BeanShellServer(bshport, bshfile); |
| t.run(); |
| } |
| |
| // Should we run a beanshell script on startup? |
| String bshinit = JMeterUtils.getProperty("beanshell.init.file");// $NON-NLS-1$ |
| if (bshinit != null){ |
| log.info("Run Beanshell on file: "+bshinit); |
| try { |
| BeanShellInterpreter bsi = new BeanShellInterpreter(); |
| bsi.source(bshinit); |
| } catch (ClassNotFoundException e) { |
| log.warn("Could not start Beanshell: "+e.getLocalizedMessage()); |
| } catch (JMeterException e) { |
| log.warn("Could not process Beanshell file: "+e.getLocalizedMessage()); |
| } |
| } |
| |
| int mirrorPort=JMeterUtils.getPropDefault("mirror.server.port", 0);// $NON-NLS-1$ |
| if (mirrorPort > 0){ |
| log.info("Starting Mirror server (" + mirrorPort + ")"); |
| try { |
| Object instance = ClassTools.construct( |
| "org.apache.jmeter.protocol.http.control.HttpMirrorControl",// $NON-NLS-1$ |
| mirrorPort); |
| ClassTools.invoke(instance,"startHttpMirror"); |
| } catch (JMeterException e) { |
| log.warn("Could not start Mirror server",e); |
| } |
| } |
| } |
| |
| /** |
| * Sets a proxy server for the JVM if the command line arguments are |
| * specified. |
| */ |
| private void setProxy(CLArgsParser parser) throws IllegalUserActionException { |
| if (parser.getArgumentById(PROXY_USERNAME) != null) { |
| Properties jmeterProps = JMeterUtils.getJMeterProperties(); |
| if (parser.getArgumentById(PROXY_PASSWORD) != null) { |
| String u, p; |
| Authenticator.setDefault(new ProxyAuthenticator(u = parser.getArgumentById(PROXY_USERNAME) |
| .getArgument(), p = parser.getArgumentById(PROXY_PASSWORD).getArgument())); |
| log.info("Set Proxy login: " + u + "/" + p); |
| jmeterProps.setProperty(HTTP_PROXY_USER, u);//for Httpclient |
| jmeterProps.setProperty(HTTP_PROXY_PASS, p);//for Httpclient |
| } else { |
| String u; |
| Authenticator.setDefault(new ProxyAuthenticator(u = parser.getArgumentById(PROXY_USERNAME) |
| .getArgument(), "")); |
| log.info("Set Proxy login: " + u); |
| jmeterProps.setProperty(HTTP_PROXY_USER, u); |
| } |
| } |
| if (parser.getArgumentById(PROXY_HOST) != null && parser.getArgumentById(PROXY_PORT) != null) { |
| String h = parser.getArgumentById(PROXY_HOST).getArgument(); |
| String p = parser.getArgumentById(PROXY_PORT).getArgument(); |
| System.setProperty("http.proxyHost", h );// $NON-NLS-1$ |
| System.setProperty("https.proxyHost", h);// $NON-NLS-1$ |
| System.setProperty("http.proxyPort", p);// $NON-NLS-1$ |
| System.setProperty("https.proxyPort", p);// $NON-NLS-1$ |
| log.info("Set http[s].proxyHost: " + h + " Port: " + p); |
| } else if (parser.getArgumentById(PROXY_HOST) != null || parser.getArgumentById(PROXY_PORT) != null) { |
| throw new IllegalUserActionException(JMeterUtils.getResString("proxy_cl_error"));// $NON-NLS-1$ |
| } |
| |
| if (parser.getArgumentById(NONPROXY_HOSTS) != null) { |
| String n = parser.getArgumentById(NONPROXY_HOSTS).getArgument(); |
| System.setProperty("http.nonProxyHosts", n );// $NON-NLS-1$ |
| System.setProperty("https.nonProxyHosts", n );// $NON-NLS-1$ |
| log.info("Set http[s].nonProxyHosts: "+n); |
| } |
| } |
| |
| private void initializeProperties(CLArgsParser parser) { |
| if (parser.getArgumentById(PROPFILE_OPT) != null) { |
| JMeterUtils.loadJMeterProperties(parser.getArgumentById(PROPFILE_OPT).getArgument()); |
| } else { |
| JMeterUtils.loadJMeterProperties(NewDriver.getJMeterDir() + File.separator |
| + "bin" + File.separator // $NON-NLS-1$ |
| + "jmeter.properties");// $NON-NLS-1$ |
| } |
| |
| if (parser.getArgumentById(JMLOGFILE_OPT) != null){ |
| String jmlogfile=parser.getArgumentById(JMLOGFILE_OPT).getArgument(); |
| jmlogfile = processLAST(jmlogfile, ".log");// $NON-NLS-1$ |
| JMeterUtils.setProperty(LoggingManager.LOG_FILE,jmlogfile); |
| } |
| |
| JMeterUtils.initLogging(); |
| JMeterUtils.initLocale(); |
| // Bug 33845 - allow direct override of Home dir |
| if (parser.getArgumentById(JMETER_HOME_OPT) == null) { |
| JMeterUtils.setJMeterHome(NewDriver.getJMeterDir()); |
| } else { |
| JMeterUtils.setJMeterHome(parser.getArgumentById(JMETER_HOME_OPT).getArgument()); |
| } |
| |
| Properties jmeterProps = JMeterUtils.getJMeterProperties(); |
| remoteProps = new Properties(); |
| |
| // Add local JMeter properties, if the file is found |
| String userProp = JMeterUtils.getPropDefault("user.properties",""); //$NON-NLS-1$ |
| if (userProp.length() > 0){ //$NON-NLS-1$ |
| FileInputStream fis=null; |
| try { |
| File file = JMeterUtils.findFile(userProp); |
| if (file.canRead()){ |
| log.info("Loading user properties from: "+file.getCanonicalPath()); |
| fis = new FileInputStream(file); |
| Properties tmp = new Properties(); |
| tmp.load(fis); |
| jmeterProps.putAll(tmp); |
| LoggingManager.setLoggingLevels(tmp);//Do what would be done earlier |
| } |
| } catch (IOException e) { |
| log.warn("Error loading user property file: " + userProp, e); |
| } finally { |
| JOrphanUtils.closeQuietly(fis); |
| } |
| } |
| |
| // Add local system properties, if the file is found |
| String sysProp = JMeterUtils.getPropDefault("system.properties",""); //$NON-NLS-1$ |
| if (sysProp.length() > 0){ |
| FileInputStream fis=null; |
| try { |
| File file = JMeterUtils.findFile(sysProp); |
| if (file.canRead()){ |
| log.info("Loading system properties from: "+file.getCanonicalPath()); |
| fis = new FileInputStream(file); |
| System.getProperties().load(fis); |
| } |
| } catch (IOException e) { |
| log.warn("Error loading system property file: " + sysProp, e); |
| } finally { |
| JOrphanUtils.closeQuietly(fis); |
| } |
| } |
| |
| // Process command line property definitions |
| // These can potentially occur multiple times |
| |
| List<CLOption> clOptions = parser.getArguments(); |
| for (CLOption option : clOptions) { |
| String name = option.getArgument(0); |
| String value = option.getArgument(1); |
| FileInputStream fis = null; |
| |
| switch (option.getDescriptor().getId()) { |
| |
| // Should not have any text arguments |
| case CLOption.TEXT_ARGUMENT: |
| throw new IllegalArgumentException("Unknown arg: " + option.getArgument()); |
| |
| case PROPFILE2_OPT: // Bug 33920 - allow multiple props |
| try { |
| fis = new FileInputStream(new File(name)); |
| Properties tmp = new Properties(); |
| tmp.load(fis); |
| jmeterProps.putAll(tmp); |
| LoggingManager.setLoggingLevels(tmp);//Do what would be done earlier |
| } catch (FileNotFoundException e) { |
| log.warn("Can't find additional property file: " + name, e); |
| } catch (IOException e) { |
| log.warn("Error loading additional property file: " + name, e); |
| } finally { |
| JOrphanUtils.closeQuietly(fis); |
| } |
| break; |
| case SYSTEM_PROPFILE: |
| log.info("Setting System properties from file: " + name); |
| try { |
| fis = new FileInputStream(new File(name)); |
| System.getProperties().load(fis); |
| } catch (IOException e) { |
| log.warn("Cannot find system property file " + e.getLocalizedMessage()); |
| } finally { |
| JOrphanUtils.closeQuietly(fis); |
| } |
| break; |
| case SYSTEM_PROPERTY: |
| if (value.length() > 0) { // Set it |
| log.info("Setting System property: " + name + "=" + value); |
| System.getProperties().setProperty(name, value); |
| } else { // Reset it |
| log.warn("Removing System property: " + name); |
| System.getProperties().remove(name); |
| } |
| break; |
| case JMETER_PROPERTY: |
| if (value.length() > 0) { // Set it |
| log.info("Setting JMeter property: " + name + "=" + value); |
| jmeterProps.setProperty(name, value); |
| } else { // Reset it |
| log.warn("Removing JMeter property: " + name); |
| jmeterProps.remove(name); |
| } |
| break; |
| case JMETER_GLOBAL_PROP: |
| if (value.length() > 0) { // Set it |
| log.info("Setting Global property: " + name + "=" + value); |
| remoteProps.setProperty(name, value); |
| } else { |
| File propFile = new File(name); |
| if (propFile.canRead()) { |
| log.info("Setting Global properties from the file " + name); |
| try { |
| fis = new FileInputStream(propFile); |
| remoteProps.load(fis); |
| } catch (FileNotFoundException e) { |
| log.warn("Could not find properties file: " + e.getLocalizedMessage()); |
| } catch (IOException e) { |
| log.warn("Could not load properties file: " + e.getLocalizedMessage()); |
| } finally { |
| JOrphanUtils.closeQuietly(fis); |
| } |
| } |
| } |
| break; |
| case LOGLEVEL: |
| if (value.length() > 0) { // Set category |
| log.info("LogLevel: " + name + "=" + value); |
| LoggingManager.setPriority(value, name); |
| } else { // Set root level |
| log.warn("LogLevel: " + name); |
| LoggingManager.setPriority(name); |
| } |
| break; |
| case REMOTE_STOP: |
| remoteStop = true; |
| break; |
| default: |
| // ignored |
| break; |
| } |
| } |
| |
| String sampleVariables = (String) jmeterProps.get(SampleEvent.SAMPLE_VARIABLES); |
| if (sampleVariables != null){ |
| remoteProps.put(SampleEvent.SAMPLE_VARIABLES, sampleVariables); |
| } |
| jmeterProps.put("jmeter.version", JMeterUtils.getJMeterVersion()); |
| } |
| |
| /* |
| * Checks for LAST or LASTsuffix. |
| * Returns the LAST name with .JMX replaced by suffix. |
| */ |
| private String processLAST(String jmlogfile, String suffix) { |
| if (USE_LAST_JMX.equals(jmlogfile) || USE_LAST_JMX.concat(suffix).equals(jmlogfile)){ |
| String last = LoadRecentProject.getRecentFile(0);// most recent |
| final String JMX_SUFFIX = ".JMX"; // $NON-NLS-1$ |
| if (last.toUpperCase(Locale.ENGLISH).endsWith(JMX_SUFFIX)){ |
| jmlogfile=last.substring(0, last.length() - JMX_SUFFIX.length()).concat(suffix); |
| } |
| } |
| return jmlogfile; |
| } |
| |
| private void startNonGui(String testFile, 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 |
| System.setProperty(JMETER_NON_GUI, "true");// $NON-NLS-1$ |
| JMeter driver = new JMeter();// TODO - why does it create a new instance? |
| driver.remoteProps = this.remoteProps; |
| driver.remoteStop = this.remoteStop; |
| driver.parent = this; |
| PluginManager.install(this, false); |
| |
| String remoteHostsString = null; |
| if (remoteStart != null) { |
| remoteHostsString = remoteStart.getArgument(); |
| if (remoteHostsString == null) { |
| remoteHostsString = JMeterUtils.getPropDefault( |
| "remote_hosts", //$NON-NLS-1$ |
| "127.0.0.1");//$NON-NLS-1$ |
| } |
| } |
| if (testFile == null) { |
| throw new IllegalUserActionException("Non-GUI runs require a test plan"); |
| } |
| driver.runNonGui(testFile, logFile, remoteStart != null, remoteHostsString, generateReportDashboard); |
| } |
| |
| // run test in batch mode |
| private void runNonGui(String testFile, String logFile, boolean remoteStart, String remote_hosts_string, boolean generateReportDashboard) { |
| try { |
| File f = new File(testFile); |
| if (!f.exists() || !f.isFile()) { |
| println("Could not open " + testFile); |
| return; |
| } |
| FileServer.getFileServer().setBaseForScript(f); |
| |
| 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); |
| |
| 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()); |
| |
| List<JMeterEngine> engines = new LinkedList<>(); |
| tree.add(tree.getArray()[0], new ListenToTest(parent, (remoteStart && remoteStop) ? engines : null, reportGenerator)); |
| println("Created the tree successfully using "+testFile); |
| if (!remoteStart) { |
| JMeterEngine engine = new StandardJMeterEngine(); |
| engine.configure(tree); |
| long now=System.currentTimeMillis(); |
| println("Starting the test @ "+new Date(now)+" ("+now+")"); |
| engine.runTest(); |
| engines.add(engine); |
| } 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); |
| } |
| } |
| |
| /** |
| * Remove disabled elements |
| * Replace the ReplaceableController with the target subtree |
| * |
| * @param tree The {@link HashTree} to convert |
| */ |
| public static void convertSubTree(HashTree tree) { |
| LinkedList<Object> copyList = new LinkedList<>(tree.list()); |
| for (Object o : copyList) { |
| if (o instanceof TestElement) { |
| TestElement item = (TestElement) o; |
| if (item.isEnabled()) { |
| if (item instanceof ReplaceableController) { |
| ReplaceableController rc = ensureReplaceableControllerIsLoaded(item); |
| |
| HashTree subTree = tree.getTree(item); |
| if (subTree != null) { |
| HashTree replacementTree = rc.getReplacementSubTree(); |
| if (replacementTree != null) { |
| convertSubTree(replacementTree); |
| tree.replaceKey(item, rc); |
| tree.set(rc, replacementTree); |
| } |
| } |
| } else { // not Replaceable Controller |
| convertSubTree(tree.getTree(item)); |
| } |
| } else { // Not enabled |
| tree.remove(item); |
| } |
| } else { // Not a TestElement |
| JMeterTreeNode item = (JMeterTreeNode) o; |
| if (item.isEnabled()) { |
| // Replacement only needs to occur when starting the engine |
| // @see StandardJMeterEngine.run() |
| if (item.getUserObject() instanceof ReplaceableController) { |
| TestElement controllerAsItem = item.getTestElement(); |
| ReplaceableController rc = ensureReplaceableControllerIsLoaded(controllerAsItem); |
| |
| HashTree subTree = tree.getTree(item); |
| |
| if (subTree != null) { |
| HashTree replacementTree = rc.getReplacementSubTree(); |
| if (replacementTree != null) { |
| convertSubTree(replacementTree); |
| tree.replaceKey(item, rc); |
| tree.set(rc, replacementTree); |
| } |
| } |
| } else { // Not a ReplaceableController |
| convertSubTree(tree.getTree(item)); |
| TestElement testElement = item.getTestElement(); |
| tree.replaceKey(item, testElement); |
| } |
| } else { // Not enabled |
| tree.remove(item); |
| } |
| } |
| } |
| } |
| |
| /** |
| * Ensures the {@link ReplaceableController} is loaded |
| * @param item {@link TestElement} |
| * @return {@link ReplaceableController} loaded |
| */ |
| private static ReplaceableController ensureReplaceableControllerIsLoaded( |
| TestElement item) { |
| ReplaceableController rc; |
| // TODO this bit of code needs to be tidied up |
| // Unfortunately ModuleController is in components, not core |
| if (item.getClass().getName().equals("org.apache.jmeter.control.ModuleController")){ // Bug 47165 |
| rc = (ReplaceableController) item; |
| } else { |
| // HACK: force the controller to load its tree |
| rc = (ReplaceableController) item.clone(); |
| } |
| return rc; |
| } |
| |
| /* |
| * Listen to test and handle tidyup after non-GUI test completes. |
| * If running a remote test, then after waiting a few seconds for listeners to finish files, |
| * it calls ClientJMeterEngine.tidyRMI() to deal with the Naming Timer Thread. |
| */ |
| private static class ListenToTest implements TestStateListener, Runnable, Remoteable { |
| private final AtomicInteger started = new AtomicInteger(0); // keep track of remote tests |
| |
| private final List<JMeterEngine> engines; |
| |
| private final ReportGenerator reportGenerator; |
| |
| /** |
| * @param unused JMeter unused for now |
| * @param engines List<JMeterEngine> |
| * @param reportGenerator {@link ReportGenerator} |
| */ |
| public ListenToTest(JMeter unused, List<JMeterEngine> engines, ReportGenerator reportGenerator) { |
| this.engines=engines; |
| this.reportGenerator = reportGenerator; |
| } |
| |
| @Override |
| // N.B. this is called by a daemon RMI thread from the remote host |
| public void testEnded(String host) { |
| long now=System.currentTimeMillis(); |
| log.info("Finished remote host: " + host + " ("+now+")"); |
| if (started.decrementAndGet() <= 0) { |
| Thread stopSoon = new Thread(this); |
| // the calling thread is a daemon; this thread must not be |
| // see Bug 59391 |
| stopSoon.setDaemon(false); |
| stopSoon.start(); |
| } |
| } |
| |
| @Override |
| public void testEnded() { |
| long now = System.currentTimeMillis(); |
| println("Tidying up ... @ "+new Date(now)+" ("+now+")"); |
| checkForRemainingThreads(); |
| try { |
| generateReport(); |
| } catch (Exception e) { |
| System.err.println("Error generating the report: "+e); |
| log.error("Error generating the report",e); |
| } |
| println("... end of run"); |
| } |
| |
| @Override |
| public void testStarted(String host) { |
| started.incrementAndGet(); |
| long now=System.currentTimeMillis(); |
| log.info("Started remote host: " + host + " ("+now+")"); |
| } |
| |
| @Override |
| public void testStarted() { |
| long now=System.currentTimeMillis(); |
| log.info(JMeterUtils.getResString("running_test")+" ("+now+")");//$NON-NLS-1$ |
| } |
| |
| /** |
| * This is a hack to allow listeners a chance to close their files. Must |
| * implement a queue for sample responses tied to the engine, and the |
| * engine won't deliver testEnded signal till all sample responses have |
| * been delivered. Should also improve performance of remote JMeter |
| * testing. |
| */ |
| @Override |
| public void run() { |
| long now = System.currentTimeMillis(); |
| println("Tidying up remote @ "+new Date(now)+" ("+now+")"); |
| if (engines!=null){ // it will be null unless remoteStop = true |
| println("Exiting remote servers"); |
| for (JMeterEngine e : engines){ |
| e.exit(); |
| } |
| } |
| try { |
| TimeUnit.SECONDS.sleep(5); // Allow listeners to close files |
| } catch (InterruptedException ignored) { |
| } |
| ClientJMeterEngine.tidyRMI(log); |
| println("... end of run"); |
| generateReport(); |
| checkForRemainingThreads(); |
| } |
| |
| /** |
| * Generate report |
| */ |
| private void generateReport() { |
| if(reportGenerator != null) { |
| try { |
| log.info("Generating Dashboard"); |
| reportGenerator.generate(); |
| log.info("Dashboard generated"); |
| } catch (GenerationException ex) { |
| log.error("Error generating dashboard:"+ex.getMessage(), ex); |
| } |
| } |
| } |
| |
| /** |
| * Runs daemon thread which waits a short while; |
| * if JVM does not exit, lists remaining non-daemon threads on stdout. |
| */ |
| private void checkForRemainingThreads() { |
| // This cannot be a JMeter class variable, because properties |
| // are not initialised until later. |
| final int REMAIN_THREAD_PAUSE = |
| JMeterUtils.getPropDefault("jmeter.exit.check.pause", 2000); // $NON-NLS-1$ |
| |
| if (REMAIN_THREAD_PAUSE > 0) { |
| Thread daemon = new Thread(){ |
| @Override |
| public void run(){ |
| try { |
| TimeUnit.MILLISECONDS.sleep(REMAIN_THREAD_PAUSE); // Allow enough time for JVM to exit |
| } catch (InterruptedException ignored) { |
| } |
| // This is a daemon thread, which should only reach here if there are other |
| // non-daemon threads still active |
| System.out.println("The JVM should have exitted but did not."); |
| System.out.println("The following non-daemon threads are still running (DestroyJavaVM is OK):"); |
| JOrphanUtils.displayThreads(false); |
| } |
| |
| }; |
| daemon.setDaemon(true); |
| daemon.start(); |
| } else if(REMAIN_THREAD_PAUSE<=0) { |
| if(log.isDebugEnabled()) { |
| log.debug("jmeter.exit.check.pause is <= 0, JMeter won't check for unterminated non-daemon threads"); |
| } |
| } |
| } |
| |
| } |
| |
| private static void println(String str) { |
| System.out.println(str); |
| } |
| |
| private static final String[][] DEFAULT_ICONS = { |
| { "org.apache.jmeter.control.gui.TestPlanGui", "org/apache/jmeter/images/beaker.gif" }, //$NON-NLS-1$ $NON-NLS-2$ |
| { "org.apache.jmeter.timers.gui.AbstractTimerGui", "org/apache/jmeter/images/timer.gif" }, //$NON-NLS-1$ $NON-NLS-2$ |
| { "org.apache.jmeter.threads.gui.ThreadGroupGui", "org/apache/jmeter/images/thread.gif" }, //$NON-NLS-1$ $NON-NLS-2$ |
| { "org.apache.jmeter.visualizers.gui.AbstractListenerGui", "org/apache/jmeter/images/meter.png" }, //$NON-NLS-1$ $NON-NLS-2$ |
| { "org.apache.jmeter.config.gui.AbstractConfigGui", "org/apache/jmeter/images/testtubes.png" }, //$NON-NLS-1$ $NON-NLS-2$ |
| { "org.apache.jmeter.processor.gui.AbstractPreProcessorGui", "org/apache/jmeter/images/leafnode.gif"}, //$NON-NLS-1$ $NON-NLS-2$ |
| { "org.apache.jmeter.processor.gui.AbstractPostProcessorGui","org/apache/jmeter/images/leafnodeflip.gif"},//$NON-NLS-1$ $NON-NLS-2$ |
| { "org.apache.jmeter.control.gui.AbstractControllerGui", "org/apache/jmeter/images/knob.gif" }, //$NON-NLS-1$ $NON-NLS-2$ |
| { "org.apache.jmeter.control.gui.WorkBenchGui", "org/apache/jmeter/images/clipboard.gif" }, //$NON-NLS-1$ $NON-NLS-2$ |
| { "org.apache.jmeter.samplers.gui.AbstractSamplerGui", "org/apache/jmeter/images/pipet.png" }, //$NON-NLS-1$ $NON-NLS-2$ |
| { "org.apache.jmeter.assertions.gui.AbstractAssertionGui", "org/apache/jmeter/images/question.gif"} //$NON-NLS-1$ $NON-NLS-2$ |
| }; |
| |
| @Override |
| public String[][] getIconMappings() { |
| final String defaultIconProp = "org/apache/jmeter/images/icon.properties"; //$NON-NLS-1$ |
| final String iconSize = JMeterUtils.getPropDefault(TREE_ICON_SIZE, DEFAULT_TREE_ICON_SIZE); |
| String iconProp = JMeterUtils.getPropDefault("jmeter.icons", defaultIconProp);//$NON-NLS-1$ |
| Properties p = JMeterUtils.loadProperties(iconProp); |
| if (p == null && !iconProp.equals(defaultIconProp)) { |
| log.info(iconProp + " not found - using " + defaultIconProp); |
| iconProp = defaultIconProp; |
| p = JMeterUtils.loadProperties(iconProp); |
| } |
| if (p == null) { |
| log.info(iconProp + " not found - using inbuilt icon set"); |
| return DEFAULT_ICONS; |
| } |
| log.info("Loaded icon properties from " + iconProp); |
| String[][] iconlist = new String[p.size()][3]; |
| Enumeration<?> pe = p.keys(); |
| int i = 0; |
| while (pe.hasMoreElements()) { |
| String key = (String) pe.nextElement(); |
| String[] icons = JOrphanUtils.split(p.getProperty(key), " ");//$NON-NLS-1$ |
| iconlist[i][0] = key; |
| iconlist[i][1] = icons[0].replace(KEY_SIZE, iconSize); |
| if (icons.length > 1) { |
| iconlist[i][2] = icons[1].replace(KEY_SIZE, iconSize); |
| } |
| i++; |
| } |
| return iconlist; |
| } |
| |
| @Override |
| public String[][] getResourceBundles() { |
| return new String[0][]; |
| } |
| |
| /** |
| * Check if JMeter is running in non-GUI mode. |
| * |
| * @return true if JMeter is running in non-GUI mode. |
| */ |
| public static boolean isNonGUI(){ |
| return "true".equals(System.getProperty(JMeter.JMETER_NON_GUI)); //$NON-NLS-1$ |
| } |
| |
| private void logProperty(String prop){ |
| log.info(prop+"="+System.getProperty(prop));//$NON-NLS-1$ |
| } |
| private void logProperty(String prop,String separator){ |
| log.info(prop+separator+System.getProperty(prop));//$NON-NLS-1$ |
| } |
| |
| private static void startUdpDdaemon(final List<JMeterEngine> engines) { |
| int port = JMeterUtils.getPropDefault("jmeterengine.nongui.port", UDP_PORT_DEFAULT); // $NON-NLS-1$ |
| int maxPort = JMeterUtils.getPropDefault("jmeterengine.nongui.maxport", 4455); // $NON-NLS-1$ |
| if (port > 1000){ |
| final DatagramSocket socket = getSocket(port, maxPort); |
| if (socket != null) { |
| Thread waiter = new Thread("UDP Listener"){ |
| @Override |
| public void run() { |
| waitForSignals(engines, socket); |
| } |
| }; |
| waiter.setDaemon(true); |
| waiter.start(); |
| } else { |
| System.out.println("Failed to create UDP port"); |
| } |
| } |
| } |
| |
| private static void waitForSignals(final List<JMeterEngine> engines, DatagramSocket socket) { |
| byte[] buf = new byte[80]; |
| System.out.println("Waiting for possible Shutdown/StopTestNow/Heapdump message on port "+socket.getLocalPort()); |
| DatagramPacket request = new DatagramPacket(buf, buf.length); |
| try { |
| while(true) { |
| socket.receive(request); |
| InetAddress address = request.getAddress(); |
| // Only accept commands from the local host |
| if (address.isLoopbackAddress()){ |
| String command = new String(request.getData(), request.getOffset(), request.getLength(),"ASCII"); |
| System.out.println("Command: "+command+" received from "+address); |
| log.info("Command: "+command+" received from "+address); |
| if (command.equals("StopTestNow")){ |
| for(JMeterEngine engine : engines) { |
| engine.stopTest(true); |
| } |
| } else if (command.equals("Shutdown")) { |
| for(JMeterEngine engine : engines) { |
| engine.stopTest(false); |
| } |
| } else if (command.equals("HeapDump")) { |
| HeapDumper.dumpHeap(); |
| } else { |
| System.out.println("Command: "+command+" not recognised "); |
| } |
| } |
| } |
| } catch (Exception e) { |
| System.out.println(e); |
| } finally { |
| socket.close(); |
| } |
| } |
| |
| private static DatagramSocket getSocket(int udpPort, int udpPortMax) { |
| DatagramSocket socket = null; |
| int i = udpPort; |
| while (i<= udpPortMax) { |
| try { |
| socket = new DatagramSocket(i); |
| break; |
| } catch (SocketException e) { |
| i++; |
| } |
| } |
| |
| return socket; |
| } |
| } |
