[Jakarta-jmeter Wiki] Update of "MeterMojo" by borja

[Apache Wiki <wi...@apache.org>]

Dear Wiki user,

You have subscribed to a wiki page or wiki category on "Jakarta-jmeter Wiki" for change notification.

The following page has been changed by borja:
http://wiki.apache.org/jakarta-jmeter/MeterMojo

New page:
package org.apache.jmeter;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileReader;
import java.io.FileWriter;
import java.io.IOException;
import java.lang.Thread.UncaughtExceptionHandler;
import java.security.Permission;
import java.text.DateFormat;
import java.text.SimpleDateFormat;
import java.util.Arrays;
import java.util.Date;
import java.util.List;
import java.util.regex.Pattern;

import org.apache.commons.io.IOUtils;
import org.apache.maven.plugin.AbstractMojo;
import org.apache.maven.plugin.MojoExecutionException;
import org.apache.maven.plugin.MojoFailureException;
import org.apache.tools.ant.DirectoryScanner;

/**
 * JMeter Maven plugin.
 * @author Tim McCune
 * @goal jmeter
 */
public class JMeterMojo extends AbstractMojo {

	private static final Pattern PAT_ERROR = Pattern.compile(".*\\s+ERROR\\s+.*");
	
	/**
	 * @parameter
	 */
	private List<String> includes;

	/**
	 * @parameter
	 */
	private List<String> excludes;

	/**
	 * @parameter expression="${basedir}/src/test/jmeter"
	 */
	private File srcDir;

	/**
	 * @parameter expression="jmeter-reports"
	 */
	private File reportDir;

	/**
	 * @parameter expression="${basedir}/src/test/jmeter/jmeter.properties"
	 */
	private File jmeterProps;

	/**
	 * @parameter
	 */
	private boolean remote;
	
	private File workDir;
	private File saveServiceProps;
	private File jmeterLog;
	private DateFormat fmt = new SimpleDateFormat("yyMMdd");

	/**
	 * Run all JMeter tests.
	 */
	public void execute() throws MojoExecutionException, MojoFailureException {
		initSystemProps();
		try {
			DirectoryScanner scanner = new DirectoryScanner();
			scanner.setBasedir(srcDir);
			scanner.setIncludes(includes == null ? new String[] { "**/*.jmx" }
					: includes.toArray(new String[] {}));
			if (excludes != null) {
				scanner.setExcludes(excludes.toArray(new String[] {}));
			}
			scanner.scan();
			for (String file : scanner.getIncludedFiles()) {
				executeTest(new File(srcDir, file));
			}
			checkForErrors();
		} finally {
			saveServiceProps.delete();
		}
	}
	
	private void checkForErrors() throws MojoExecutionException, MojoFailureException {
		try {
			BufferedReader in = new BufferedReader(new FileReader(jmeterLog));
			String line;
			while ( (line = in.readLine()) != null) {
				if (PAT_ERROR.matcher(line).find()) {
					throw new MojoFailureException("There were test errors");
				}
			}
			in.close();
		} catch (IOException e) {
			throw new MojoExecutionException("Can't read log file", e);
		}
	}
	
	private void initSystemProps() throws MojoExecutionException {
		workDir = new File("target" + File.separator + "jmeter");
		workDir.mkdirs();
		createSaveServiceProps();
		jmeterLog = new File(workDir, "jmeter.log");
		try {
			System.setProperty("log_file", jmeterLog.getCanonicalPath());
		} catch (IOException e) {
			throw new MojoExecutionException("Can't get canonical path for log file", e);
		}
	}
	
	/**
	 * This mess is necessary because JMeter must load this info from a file.
	 * Resources won't work.
	 */
	private void createSaveServiceProps() throws MojoExecutionException {
		saveServiceProps = new File(workDir, "saveservice.properties");
		try {
			FileWriter out = new FileWriter(saveServiceProps);
			IOUtils.copy(Thread.currentThread().getContextClassLoader()
					.getResourceAsStream("saveservice.properties"), out);
			out.flush();
			out.close();
			System.setProperty("saveservice_properties",
					File.separator + "target" + File.separator + "jmeter" + File.separator +
					"saveservice.properties");
		} catch (IOException e) {
			throw new MojoExecutionException("Could not create temporary saveservice.properties", e);
		}
	}

	/**
	 * Executes a single JMeter test by building up a list of command line
	 * parameters to pass to JMeter.start().
	 */
	private void executeTest(File test) throws MojoExecutionException {
		try {
			getLog().info("Executing test: " + test.getCanonicalPath());
			String reportFileName = test.getName().substring(0,
					test.getName().lastIndexOf(".")) + "-" +
					fmt.format(new Date()) + ".xml";
			List<String> args = Arrays.asList("-n",
					"-t", test.getCanonicalPath(),
					"-l", reportDir.toString() + File.separator + reportFileName,
					"-p", jmeterProps.toString(),
					"-d", System.getProperty("user.dir"));
			if (remote) {
				args.add("-r");
			}
			// This mess is necessary because JMeter likes to use System.exit.
			// We need to trap the exit call.
			SecurityManager oldManager=System.getSecurityManager();
	    System.setSecurityManager(new SecurityManager() {
	      @Override public void checkExit(int status) { throw new ExitException(status);}
	      @Override public void checkPermission(Permission perm, Object context) {}
	      @Override public void checkPermission(Permission perm) {}
	    });
	    UncaughtExceptionHandler oldHandler = Thread.getDefaultUncaughtExceptionHandler();
	    Thread.setDefaultUncaughtExceptionHandler(new UncaughtExceptionHandler() {
				public void uncaughtException(Thread t, Throwable e) {
					if (e instanceof ExitException && ((ExitException) e).getCode() == 0) {
						return;	//Ignore
					}
					getLog().error("Error in thread " + t.getName());
				}
	    });
			try {
				// This mess is necessary because the only way to know when JMeter
				// is done is to wait for all of the threads that it spawned to exit.
				int startThreadCount = Thread.activeCount();
				new JMeter().start(args.toArray(new String[]{}));
				int activeThreadCount;
				while ( (activeThreadCount = Thread.activeCount()) > startThreadCount) {
					try {
						Thread.sleep(1000);
					} catch (InterruptedException e) {
						break;
					}
				}
			} catch (ExitException e) {
				if (e.getCode() != 0) {
					throw new MojoExecutionException("Test failed", e);
				}
			} finally {
				System.setSecurityManager(oldManager);
				Thread.setDefaultUncaughtExceptionHandler(oldHandler);
			}
		} catch (IOException e) {
			throw new MojoExecutionException("Can't execute test", e);
		}
	}

	private static class ExitException extends SecurityException {
		private static final long serialVersionUID = 5544099211927987521L;

		public int _rc;

		public ExitException(int rc) {
			super(Integer.toString(rc));
			_rc = rc;
		}
		
		public int getCode() {
			return _rc;
		}
	}

}

---------------------------------------------------------------------
To unsubscribe, e-mail: jmeter-dev-unsubscribe@jakarta.apache.org
For additional commands, e-mail: jmeter-dev-help@jakarta.apache.org