com.automationrockstars.gir.ui.part.MetricsService Maven / Gradle / Ivy

Go to download
package com.automationrockstars.gir.ui.part;

import java.lang.reflect.Method;
import java.text.SimpleDateFormat;
import java.util.Comparator;
import java.util.Date;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;

import org.openqa.selenium.logging.LogEntry;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.automationrockstars.base.ConfigLoader;
import com.automationrockstars.design.gir.webdriver.DriverFactory;
import com.google.common.base.CharMatcher;
import com.google.common.collect.FluentIterable;
import com.google.common.collect.Maps;
import com.google.gson.Gson;

public class MetricsService {

	private static final Logger LOG = LoggerFactory.getLogger(MetricsService.class);
	public static void finished(Object host, Method method, Object[] args, long elapsed) {
		LOG.info("Execution of {} took {} nanos",method.getName(),elapsed);

		if (ConfigLoader.config().getBoolean("performance.run",false) && DriverFactory.isPhantom()){
			logNow(host.toString(), method.getName(),"");
		}

	}

	public static void logNow(String host, String method, String... pageFilter){
		SimpleDateFormat sdf = new SimpleDateFormat("MM:dd-HH:mm:ss.SSS");
		if (DriverFactory.isPhantom()){
			for (LogEntry log : DriverFactory.getDriver().manage().logs().get("har").getAll()){
				Map timings = timings(log.getMessage(), pageFilter);
				MetricFileWriter.put(sdf.format(new Date(log.getTimestamp())), host, method, pageAndTime(timings));
				MetricsRabbitWriter.log(host, method, data(timings));
			}
		}
	}
	
	
	private static Gson gson = new Gson();
	public static synchronized String pageAndTime(Map timings){
		String result = "";
		for (String page : timings.keySet()){
			result = String.format("%s, %s, %s",result,page, timings.get(page));
		}
		return result;
	}

	@SuppressWarnings({ "unchecked", "rawtypes" })
	private synchronized static Map timings(String har, String... pageFilter){
		Map content = gson.fromJson(har, HashMap.class);

		List entries = (List) ((Map)content.get("log")).get("entries");
		Map result = Maps.newHashMap();
		for (Map entry : entries){
			if (entry.containsKey("pageref") ){
				boolean match = false;
				for (String filter : pageFilter){
					match = match || entry.get("pageref").toString().contains(filter);
				}
				if (match){
					String pageRef = entry.get("pageref").toString();
					int i =0;
					while (result.containsKey(pageRef)){
						pageRef += i++;
					}
					result.put(pageRef, entry.get("timings").toString());
				}
			}
		}
		return result;
	}

	public static synchronized Map data(Map timings){
		Long timeStamp = System.currentTimeMillis();
		Map result = Maps.newHashMap();
		try {
			
			String timingsString = FluentIterable.from(timings.entrySet()).toSortedList(new Comparator>(){

				@Override
				public int compare(Entry arg0, Entry arg1) {
					return arg0.getKey().length() - arg1.getKey().length();
				}}).iterator().next().getValue();
			if (timingsString != null){
				String[] times = CharMatcher.JAVA_DIGIT.or(CharMatcher.anyOf(",.")).retainFrom(timingsString).split(",");
				Float val = 0F;
				for (String time : times){
					val += Float.valueOf(time);
				}
				result.put("value", Math.round(val));
				result.put("timeStamp",timeStamp*1000000);
			}

		} catch (Exception e){
			LOG.debug("Error with data {}",e.getMessage());
		}
		return result;
	}


}