jmeterreport.java

来自「测试工具」· Java 代码 · 共 523 行 · 第 1/2 页

			if (logFile != null) {
				ResultCollector logger = new ResultCollector();
				logger.setFilename(logFile);
				tree.add(tree.getArray()[0], logger);
			}
			String summariserName = JMeterUtils.getPropDefault(
					"summariser.name", "");//$NON-NLS-1$
			if (summariserName.length() > 0) {
				log.info("Creating summariser <" + summariserName + ">");
				System.out.println("Creating summariser <" + summariserName + ">");
				Summariser summer = new Summariser(summariserName);
				tree.add(tree.getArray()[0], summer);
			}
			tree.add(tree.getArray()[0], new ListenToTest(parent));
			System.out.println("Created the tree successfully");
            /**
			JMeterEngine engine = null;
			if (!remoteStart) {
				engine = new StandardJMeterEngine();
				engine.configure(tree);
				System.out.println("Starting the test");
				engine.runTest();
			} else {
				String remote_hosts_string = JMeterUtils.getPropDefault(
						"remote_hosts", "127.0.0.1");
				java.util.StringTokenizer st = new java.util.StringTokenizer(
						remote_hosts_string, ",");
				List engines = new LinkedList();
				while (st.hasMoreElements()) {
					String el = (String) st.nextElement();
					System.out.println("Configuring remote engine for " + el);
					// engines.add(doRemoteInit(el.trim(), tree));
				}
				System.out.println("Starting remote engines");
				Iterator iter = engines.iterator();
				while (iter.hasNext()) {
					engine = (JMeterEngine) iter.next();
					engine.runTest();
				}
				System.out.println("Remote engines have been started");
			}
            **/
		} catch (Exception e) {
			System.out.println("Error in NonGUIDriver " + e.toString());
			log.error("", e);
		}
        finally{
            JOrphanUtils.closeQuietly(reader);
        }
	}

    
    /**
     * 
     * @param args
     */
    public void start(String[] args) {
        CLArgsParser parser = new CLArgsParser(args, options);
        if (null != parser.getErrorString()) {
            System.err.println("Error: " + parser.getErrorString());
            System.out.println("Usage");
            System.out.println(CLUtil.describeOptions(options).toString());
            return;
        }
        try {
            initializeProperties(parser);
            log.info("Version " + JMeterUtils.getJMeterVersion());
            log.info("java.version=" + System.getProperty("java.version"));
            log.info(JMeterUtils.getJMeterCopyright());
            if (parser.getArgumentById(VERSION_OPT) != null) {
                System.out.println(JMeterUtils.getJMeterCopyright());
                System.out.println("Version " + JMeterUtils.getJMeterVersion());
            } else if (parser.getArgumentById(HELP_OPT) != null) {
                System.out.println(JMeterUtils.getResourceFileAsText("org/apache/jmeter/help.txt"));
            } else if (parser.getArgumentById(NONGUI_OPT) == null) {
                startGui(parser.getArgumentById(TESTFILE_OPT));
            } else {
                startNonGui(parser.getArgumentById(TESTFILE_OPT), parser.getArgumentById(LOGFILE_OPT));
            }
        } catch (Exception e) {
            e.printStackTrace();
            System.out.println("An error occurred: " + e.getMessage());
            System.exit(-1);
        }
    }
    
    private void initializeProperties(CLArgsParser parser) {
        if (parser.getArgumentById(PROPFILE_OPT) != null) {
            JMeterUtils.getProperties(parser.getArgumentById(PROPFILE_OPT).getArgument());
        } else {
            JMeterUtils.getProperties(NewDriver.getJMeterDir() + File.separator + "bin" + File.separator
                    + "jmeter.properties");
        }

        // 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());
        }

        // Process command line property definitions (can occur multiple times)

        Properties jmeterProps = JMeterUtils.getJMeterProperties();
        List clOptions = parser.getArguments();
        int size = clOptions.size();

        for (int i = 0; i < size; i++) {
            CLOption option = (CLOption) clOptions.get(i);
            String name = option.getArgument(0);
            String value = option.getArgument(1);

            switch (option.getDescriptor().getId()) {
            case PROPFILE2_OPT: // Bug 33920 - allow multiple props
                File f = new File(name);
                try {
                    jmeterProps.load(new FileInputStream(f));
                } 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);
                }
                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 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;
            }
        }

    }
    
    /**
     * Code copied from AbstractAction.java and modified to suit TestElements
     * 
     * @param tree
     */
    private void convertSubTree(HashTree tree) {// TODO check build dependencies
        Iterator iter = new LinkedList(tree.list()).iterator();
        while (iter.hasNext()) {
            TestElement item = (TestElement) iter.next();
            if (item.isEnabled()) {
                // This is done for GUI runs in JMeterTreeModel.addSubTree()
                if (item instanceof ReportPlan) {
                    ReportPlan tp = (ReportPlan) item;
                }
                // TODO handle ReplaceableControllers
                // if (item instanceof ReplaceableController)
                // {
                // System.out.println("Replaceable "+item.getClass().getName());
                // HashTree subTree = tree.getTree(item);
                //
                // if (subTree != null)
                // {
                // ReplaceableController rc =
                // (ReplaceableController) item;//.createTestElement();
                // rc.replace(subTree);
                // convertSubTree(subTree);
                // tree.replace(item, rc.getReplacement());
                // }
                // }
                // else
                {
                    // System.out.println("NonReplaceable
                    // "+item.getClass().getName());
                    convertSubTree(tree.getTree(item));
                    // TestElement testElement = item.createTestElement();
                    // tree.replace(item, testElement);
                }
            } else {
                // System.out.println("Disabled "+item.getClass().getName());
                tree.remove(item);
            }
        }
    }
    
    /**
     * Listen to test and exit program after test completes, after a 5 second
     * delay to give listeners a chance to close out their files.
     */
    private static class ListenToTest implements TestListener, Runnable, Remoteable {
        int started = 0;

        private JMeterReport _parent;

        private ListenToTest(JMeterReport parent) {
            _parent = parent;
        }

        public synchronized void testEnded(String host) {
            started--;
            log.info("Remote host " + host + " finished");
            if (started == 0) {
                testEnded();
            }
        }

        public void testEnded() {
            Thread stopSoon = new Thread(this);
            stopSoon.start();
        }

        public synchronized void testStarted(String host) {
            started++;
            log.info("Started remote host: " + host);
        }

        public void testStarted() {
            log.info(JMeterUtils.getResString("running_test"));
        }

        /**
         * 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.
         */
        public void run() {
            System.out.println("Tidying up ...");
            try {
                Thread.sleep(5000);
            } catch (InterruptedException e) {
                // ignored
            }
            System.out.println("... end of run");
            _parent.testEnded = true;
        }

        /**
         * @see TestListener#testIterationStart(LoopIterationEvent)
         */
        public void testIterationStart(LoopIterationEvent event) {
            // ignored
        }
    }

}