com.loadcoder.load.scenario.FinishedExecution Maven / Gradle / Ivy

Go to download
/*******************************************************************************
 * Copyright (C) 2018 Stefan Vahlgren at Loadcoder
 * 
 * This file is part of Loadcoder.
 * 
 * Loadcoder is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 * 
 * Loadcoder is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see .
 ******************************************************************************/
package com.loadcoder.load.scenario;

import java.io.File;
import java.io.IOException;
import java.util.List;
import java.util.Map;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.loadcoder.load.exceptions.RuntimeResultStorageNotActivatedException;
import com.loadcoder.load.exceptions.NoResultOrFormatterException;
import com.loadcoder.result.Logs;
import com.loadcoder.result.Result;
import com.loadcoder.result.TransactionExecutionResult;

public class FinishedExecution {

	private Execution s;
	private Logger log = LoggerFactory.getLogger(this.getClass());
	private final static String RESULTFILE_DEFAULT = "result.log";

	protected FinishedExecution(Execution s) {
		this.s = s;
	}

	/**
	 * Returns a Result for the default log file (result.log)
	 * 
	 * @return a new Result instance from the finished load, assuming that the name
	 *         of the file containing the result is result.log So if the directory
	 *         for the result is set to /foo/bar, the the Result will be generated
	 *         from the file /foo/bar/result.log
	 * @throws NoResultOrFormatterException will be thrown if either the result is
	 *                                      missing or the formatter can format the
	 *                                      file
	 */
	public Result getReportedResultFromResultFile() throws NoResultOrFormatterException {
		return getReportedResultFromResultFile(RESULTFILE_DEFAULT);
	}

	/**
	 * Returns a Result
	 * 
	 * @param fileName is the name of the file. So if fileName is result.log and the
	 *                 directory for the result is set to /foo/bar, the the Result
	 *                 will be generated from the file /foo/bar/result.log
	 * @return a new Result instance from the result file
	 * @throws NoResultOrFormatterException will be thrown if either the result is
	 *                                      missing or the formatter can format the
	 *                                      file
	 */
	public Result getReportedResultFromResultFile(String fileName) throws NoResultOrFormatterException {
		File resultDir = Logs.getLogDir();
		File resultFile = new File(resultDir, fileName);
		return getReportedResultFromResultFile(resultFile);
	}

	/**
	 * Returns a Result
	 * 
	 * @param resultFile is the file where the result is stored, that will be used
	 *                   to generate the Result
	 * @return a new Result instance from the result file
	 * @throws NoResultOrFormatterException will be thrown if either the result is
	 *                                      missing or the formatter can format the
	 *                                      file
	 */
	public Result getReportedResultFromResultFile(File resultFile) throws NoResultOrFormatterException {
		if (s.getResultFormatter() != null) {
			try {
				long startTime = System.currentTimeMillis();
				Result result = s.getResultFormatter().toResultList(resultFile);
				long executionTime = System.currentTimeMillis() - startTime;
				log.debug(String.format("Time taken to generate result: %s ms", executionTime));

				return result;
			} catch (IOException ioe) {
				throw new NoResultOrFormatterException("Could not read the resultFile " + resultFile, ioe);
			}
		}
		throw new NoResultOrFormatterException("The report can not be produced"
				+ " since the ResultFormatter or/and ResultDestination " + "seems to be missning in the scenario");
	}

	protected class ResultFromMemory extends Result {

		protected ResultFromMemory(Map> resultLists) {
			super(resultLists);
		}

	}

	public Result getResultFromMemory() {
		if (s.getTransactionExecutionResultBuffer() == null) {
			throw new RuntimeResultStorageNotActivatedException(
					"In memory storage of results was not activated for this execution. "
							+ "The results are by default written to the configured logger appender");

		}
		List list = s.getTransactionExecutionResultBuffer().swap();
		Map> map = TransactionExecutionResult.getResultListAsMap(list);
		Result r = new ResultFromMemory(map);
		return r;
	}

}