jmeter.java

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

				remote_hosts_string = JMeterUtils.getPropDefault(
	                    "remote_hosts", //$NON-NLS-1$ 
	                    "127.0.0.1");//$NON-NLS-1$				
			}
		}
		if (testFile == null) {
			throw new IllegalUserActionException();
		}
		String argument = testFile.getArgument();
        if (argument == null) {
            throw new IllegalUserActionException();
        }
        if (logFile == null) {
			driver.run(argument, null, remoteStart != null,remote_hosts_string);
		} else {
			driver.run(argument, logFile.getArgument(), remoteStart != null,remote_hosts_string);
		}
	}

    // run test in batch mode
	private void run(String testFile, String logFile, boolean remoteStart, String remote_hosts_string) {
		FileInputStream reader = null;
		try {
			File f = new File(testFile);
			if (!f.exists() || !f.isFile()) {
				println("Could not open " + testFile);
				return;
			}
			FileServer.getFileServer().setBasedir(f.getAbsolutePath());

			reader = new FileInputStream(f);
			log.info("Loading file: " + f);

			HashTree tree = SaveService.loadTree(reader);

            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 replaceableControllers = new SearchByClass(ReplaceableController.class);
            tree.traverse(replaceableControllers);
            Collection replaceableControllersRes = replaceableControllers.getSearchResults();
            for (Iterator iter = replaceableControllersRes.iterator(); iter.hasNext();) {
                ReplaceableController replaceableController = (ReplaceableController) iter.next();
                replaceableController.resolveReplacementSubTree(root);
            }

			// Remove the disabled items
			// For GUI runs this is done in Start.java
			convertSubTree(tree);

			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 + ">");
				println("Creating summariser <" + summariserName + ">");
				Summariser summer = new Summariser(summariserName);
				tree.add(tree.getArray()[0], summer);
			}
			List engines = new LinkedList();
			tree.add(tree.getArray()[0], new ListenToTest(parent, (remoteStart && remoteStop) ? engines : null));
			println("Created the tree successfully");
			JMeterEngine engine = null;
			if (!remoteStart) {
				engine = new StandardJMeterEngine();
				engine.configure(tree);
				long now=System.currentTimeMillis();
				println("Starting the test @ "+new Date(now)+" ("+now+")");
				engine.runTest();
			} else {
				java.util.StringTokenizer st = new java.util.StringTokenizer(remote_hosts_string, ",");//$NON-NLS-1$
				while (st.hasMoreElements()) {
					String el = (String) st.nextElement();
					println("Configuring remote engine for " + el);
					engines.add(doRemoteInit(el.trim(), tree));
				}
				println("Starting remote engines");
				long now=System.currentTimeMillis();
				println("Starting the test @ "+new Date(now)+" ("+now+")");
				Iterator iter = engines.iterator();
				while (iter.hasNext()) {
					engine = (JMeterEngine) iter.next();
					engine.runTest();
				}
				println("Remote engines have been started");
			}
		} catch (Exception e) {
			System.out.println("Error in NonGUIDriver " + e.toString());
			log.error("", e);
        } finally {
            JOrphanUtils.closeQuietly(reader);
        }
	}

    /**
     * Refactored from AbstractAction.java
     * 
     * @param tree
     */
    public static void convertSubTree(HashTree tree) {
        Iterator iter = new LinkedList(tree.list()).iterator();
        while (iter.hasNext()) {
            Object o = iter.next();
            if (o instanceof TestElement) {
                TestElement item = (TestElement) o;
                if (item.isEnabled()) {
                    if (item instanceof ReplaceableController) {
                        // HACK: force the controller to load its tree
                        ReplaceableController rc = (ReplaceableController) item
                            .clone();
                        HashTree subTree = tree.getTree(item);
                        if (subTree != null) {
                            HashTree replacementTree = rc
                                .getReplacementSubTree();
                            if (replacementTree != null) {
                                convertSubTree(replacementTree);
                                tree.replace(item, rc);
                                tree.set(rc, replacementTree);
                            }
                        } else {
                            convertSubTree(tree.getTree(item));
                        }
                    } else {
                        convertSubTree(tree.getTree(item));
                    }
                } else
                    tree.remove(item);
            } else {
                JMeterTreeNode item = (JMeterTreeNode) o;
                if (item.isEnabled()) {
                    // Replacement only needs to occur when starting the engine
                    // @see StandardJMeterEngine.run()
                    if (item.getUserObject() instanceof ReplaceableController) {
                        ReplaceableController rc = (ReplaceableController) item
                            .getTestElement();
                        HashTree subTree = tree.getTree(item);

                        if (subTree != null) {
                            HashTree replacementTree = rc
                                .getReplacementSubTree();
                            if (replacementTree != null) {
                                convertSubTree(replacementTree);
                                tree.replace(item, rc);
                                tree.set(rc, replacementTree);
                            }
                        }
                    } else {
                        convertSubTree(tree.getTree(item));
                        TestElement testElement = item.getTestElement();
                        tree.replace(item, testElement);
                    }
                 } else {
                    tree.remove(item);
                }
            }
        }
    }

	private JMeterEngine doRemoteInit(String hostName, HashTree testTree) {
		JMeterEngine engine = null;
		try {
			engine = new ClientJMeterEngine(hostName);
		} catch (Exception e) {
			log.fatalError("Failure connecting to remote host", e);
			System.err.println("Failure connecting to remote host"+e);
			System.exit(1);
		}
		engine.configure(testTree);
		engine.setProperties(remoteProps);
		return engine;
	}

	/*
	 * 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 System.exit to deal with the Naming Timer Thread.
	 */
	private static class ListenToTest implements TestListener, Runnable, Remoteable {
		private int started = 0; // keep track of remote tests

		//NOT YET USED private JMeter _parent;

		private List engines;
		
		public ListenToTest(JMeter parent, List engines) {
			//_parent = parent;
			this.engines=engines;
		}

		public synchronized void testEnded(String host) {
			started--;
			long now=System.currentTimeMillis();
			log.info("Finished remote host: " + host + " ("+now+")");
			if (started <= 0) {
				Thread stopSoon = new Thread(this);
				stopSoon.start();
			}
		}

		public void testEnded() {
			long now = System.currentTimeMillis();
			println("Tidying up ...    @ "+new Date(now)+" ("+now+")");
			println("... end of run");
		}

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

		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.
		 */
		public void run() {
			long now = System.currentTimeMillis();
			println("Tidying up ...    @ "+new Date(now)+" ("+now+")");
            /*
             * Note: although it should not be necessary to call System.exit here, in the case
             * of a remote test, a Timer thread seems to be generated by the Naming.lookup()
             * method, and it does not die.
             */
			if (engines!=null){ // it will be null unless remoteStop = true
				System.out.println("Exitting remote servers");
				Iterator it = engines.iterator();
				while(it.hasNext()){
					JMeterEngine e = (JMeterEngine) it.next();
					e.exit();
				}
			}
			try {
				Thread.sleep(5000); // Allow listeners to close files
			} catch (InterruptedException ignored) {
			}
			println("... end of run");
			System.exit(0);
		}

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

	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.AbstractVisualizer",    "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$
	};

	public String[][] getIconMappings() {
		final String defaultIconProp = "org/apache/jmeter/images/icon.properties"; //$NON-NLS-1$
		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];
			if (icons.length > 1)
				iconlist[i][2] = icons[1];
			i++;
		}
		return iconlist;
	}

	public String[][] getResourceBundles() {
		return new String[0][];
	}
	
	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$
	}
}