com.neotys.neoload.model.readers.loadrunner.method.LoadRunnerSupportedMethods Maven / Gradle / Ivy

Go to download
package com.neotys.neoload.model.readers.loadrunner.method;

import com.neotys.neoload.model.readers.loadrunner.customaction.CustomActionMappingLoader;
import com.neotys.neoload.model.readers.loadrunner.customaction.ImmutableMappingMethod;

import java.util.HashMap;
import java.util.Map;

public class LoadRunnerSupportedMethods {
	
	private final CustomActionMappingLoader customActionMappingLoader;
	private final Map supportedMethods;

	private ContainerInFileMethod containerInFileMethod;
	
	public LoadRunnerSupportedMethods(final String additionalCustomActionMappingContent) {
		super();
		this.customActionMappingLoader = new CustomActionMappingLoader(additionalCustomActionMappingContent);
		this.supportedMethods = new HashMap<>();
	}
	
	public LoadRunnerMethod get(final String methodName) {		
		if(supportedMethods.containsKey(methodName)){
			return supportedMethods.get(methodName);
		}
		final LoadRunnerMethod method = internalLoadMethod(methodName);
		if (containerInFileMethod != null && containerInFileMethod.containsContainer(methodName)) {
			return containerInFileMethod;
		}
		supportedMethods.put(methodName, method);
		return method;
	}
	
	public ImmutableMappingMethod getCustomActionMappingMethod(String methodName) {
		return customActionMappingLoader.getMethod(methodName);
	}
	
	private final LoadRunnerMethod internalLoadMethod(final String methodName) {
		if (methodName == null) {
			return null;
		}										
		if (customActionMappingLoader.getMethod(methodName)!=null) {
			return new CustomActionMethod();
		}
		final String lowerCaseMethodName = methodName.replaceAll("_", "").toLowerCase();
		if (lowerCaseMethodName.length() < 2) {
			return null;
		}
		final String fullQualifiedMethodName = "com.neotys.neoload.model.readers.loadrunner.method."
				+ lowerCaseMethodName.substring(0, 1).toUpperCase() + lowerCaseMethodName.substring(1) + "Method";
		try {
			return (LoadRunnerMethod) Class.forName(fullQualifiedMethodName).getConstructor().newInstance();
		} catch (Exception e) {
		}
		if("lr_start_sub_transaction".equals(methodName)) return new LrstarttransactionMethod();
		if("lr_end_sub_transaction".equals(methodName)) return new LrendtransactionMethod();
		if("lr_param_sprintf".equals(methodName)) return new SprintfMethod();
		return null;
	}

	public void setContainerInFileMethod(final ContainerInFileMethod containerInFileMethod) {
		this.containerInFileMethod = containerInFileMethod;
	}
}