com.emc.mongoose.load.step.weighted.WeightedLoadStepExtension Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mongoose-load-step-weighted Show documentation

Mongoose is a high-load storage performance testing tool

The newest version!

package com.emc.mongoose.load.step.weighted;

import com.emc.mongoose.base.env.Extension;
import com.emc.mongoose.base.env.ExtensionBase;
import com.emc.mongoose.base.load.step.LoadStepFactory;
import static com.emc.mongoose.base.Constants.APP_NAME;

import com.emc.mongoose.base.metrics.MetricsManager;
import com.github.akurilov.confuse.Config;
import com.github.akurilov.confuse.SchemaProvider;

import com.github.akurilov.confuse.io.yaml.YamlSchemaProviderBase;

import java.io.InputStream;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;

public final class WeightedLoadStepExtension
				extends ExtensionBase
				implements LoadStepFactory {

	public static final String TYPE = "WeightedLoad";

	private static final SchemaProvider SCHEMA_PROVIDER = new YamlSchemaProviderBase() {

		@Override
		protected final InputStream schemaInputStream() {
			return getClass().getResourceAsStream("/config-schema-load-generator-weight.yaml");
		}

		@Override
		public final String id() {
			return APP_NAME;
		}
	};

	private static final String DEFAULTS_FILE_NAME = "defaults-load-generator-weight.yaml";

	private static final List RES_INSTALL_FILES = Collections.unmodifiableList(
					Arrays.asList("config/" + DEFAULTS_FILE_NAME));

	@Override
	public final String id() {
		return TYPE;
	}

	@Override
	@SuppressWarnings("unchecked")
	public final T createLocal(
					final Config baseConfig, final List extensions, final List contextConfigs,
					final MetricsManager metricsManager) {
		return (T) new WeightedLoadStepLocal(baseConfig, extensions, contextConfigs, metricsManager);
	}

	@Override
	@SuppressWarnings("unchecked")
	public final U createClient(
					final Config baseConfig, final List extensions, final MetricsManager metricsManager) {
		return (U) new WeightedLoadStepClient(baseConfig, extensions, null, metricsManager);
	}

	@Override
	public final SchemaProvider schemaProvider() {
		return SCHEMA_PROVIDER;
	}

	@Override
	protected final String defaultsFileName() {
		return DEFAULTS_FILE_NAME;
	}

	@Override
	protected final List resourceFilesToInstall() {
		return RES_INSTALL_FILES;
	}
}