A Complete Example - mgm-tp/perfload GitHub Wiki

package com.mgmtp.perfload.example.driver;

import static com.google.common.collect.Maps.newHashMapWithExpectedSize;
import static org.apache.commons.io.IOUtils.lineIterator;

import java.io.IOException;
import java.io.InputStream;
import java.util.Map;

import org.apache.commons.io.LineIterator;

import com.google.inject.Provides;
import com.mgmtp.perfload.core.client.web.config.AbstractWebLtModule;
import com.mgmtp.perfload.core.client.web.config.WebLtModule;
import com.mgmtp.perfload.core.common.util.PropertiesMap;

public class ExampleModule extends AbstractWebLtModule {

	public ExampleModule(final PropertiesMap testplanProperties) {
		super(testplanProperties);
	}

	@Override
	protected void doConfigureWebModule() {
		bindRequestFlowEventListener().to(ExampleListener.class);
		install(new WebLtModule(testplanProperties));
	}

	@TestData
	@Provides
	Map<String, String> provideTestData() {
		ClassLoader cl = Thread.currentThread().getContextClassLoader();
		try (InputStream is = cl.getResourceAsStream("testdata/fibonacci.txt")) {
			Map<String, String> result = newHashMapWithExpectedSize(20);
			for (LineIterator it = lineIterator(is, "UTF-8"); it.hasNext();) {
				String line = it.nextLine();
				if (line.startsWith("#")) {
					continue;
				}
				String[] columns = line.split(";");
				result.put(columns[0], columns[1]);
			}
			return result;
		} catch (IOException ex) {
			throw new IllegalStateException("Error reading test data.", ex);
		}
	}
}

package com.mgmtp.perfload.example.driver;

import static java.lang.annotation.ElementType.FIELD;
import static java.lang.annotation.ElementType.METHOD;
import static java.lang.annotation.ElementType.PARAMETER;
import static java.lang.annotation.RetentionPolicy.RUNTIME;

import java.lang.annotation.Retention;
import java.lang.annotation.Target;

import javax.inject.Qualifier;

@Retention(RUNTIME)
@Target({ FIELD, PARAMETER, METHOD })
@Qualifier
public @interface TestData {
	//
}

package com.mgmtp.perfload.example.driver;

import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Random;

import javax.inject.Inject;
import javax.inject.Provider;
import javax.inject.Singleton;

import net.jcip.annotations.Immutable;
import net.jcip.annotations.ThreadSafe;

import com.google.common.collect.ImmutableList;
import com.mgmtp.perfload.core.client.util.PlaceholderContainer;
import com.mgmtp.perfload.core.client.web.event.LtListenerAdapter;
import com.mgmtp.perfload.core.client.web.event.RequestFlowEvent;

@Singleton
public class ExampleListener extends LtListenerAdapter {
	private final Provider<PlaceholderContainer> placeholderContainerProvider;
	private final List<Entry<String, String>> testDataEntries;
	private final Random random = new Random();

	@Inject
	public ExampleListener(final Provider<PlaceholderContainer> placeholderContainerProvider,
			@TestData final Map<String, String> testData) {
		this.placeholderContainerProvider = placeholderContainerProvider;
		this.testDataEntries = ImmutableList.copyOf(testData.entrySet());
	}

	@Override
	public void beforeRequest(final RequestFlowEvent event) {
		int index = random.nextInt(testDataEntries.size());
		Entry<String, String> entry = testDataEntries.get(index);

		PlaceholderContainer placeholderContainer = placeholderContainerProvider.get();
		placeholderContainer.put("n", entry.getKey());
		placeholderContainer.put("fibn", entry.getValue());
	}
}