com.emc.mongoose.base.item.io.ItemInputFactory Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of mongoose-base Show documentation
Mongoose is a high-load storage performance testing tool
There is a newer version: 4.3.10
package com.emc.mongoose.base.item.io;

import com.emc.mongoose.base.item.Item;
import com.emc.mongoose.base.item.ItemFactory;
import com.emc.mongoose.base.item.ItemType;
import com.emc.mongoose.base.item.op.Operation;
import com.emc.mongoose.base.logging.LogUtil;
import com.emc.mongoose.base.logging.Loggers;
import com.emc.mongoose.base.storage.driver.StorageDriver;
import com.github.akurilov.commons.io.Input;
import com.github.akurilov.commons.io.file.BinFileInput;
import com.github.akurilov.confuse.Config;
import java.io.IOException;
import java.nio.file.Path;
import java.nio.file.Paths;
import org.apache.logging.log4j.Level;

public interface ItemInputFactory {

	static > Input createItemInput(
					final Config itemConfig, final int batchSize, final StorageDriver storageDriver) {
		Input itemInput = null;

		final ItemType itemType = ItemType.valueOf(itemConfig.stringVal("type").toUpperCase());
		final ItemFactory itemFactory = ItemType.getItemFactory(itemType);
		final Config itemInputConfig = itemConfig.configVal("input");
		final String itemInputFile = itemInputConfig.stringVal("file");

		if (itemInputFile != null && !itemInputFile.isEmpty()) {
			itemInput = createFileItemInput(itemFactory, itemInputFile);
			Loggers.MSG.debug("Using the file \"{}\" as items input", itemInputFile);
		} else {
			final String itemInputPath = itemInputConfig.stringVal("path");
			if (itemInputPath != null && !itemInputPath.isEmpty()) {
				itemInput = createPathItemInput(itemConfig, batchSize, itemFactory, itemInputPath, storageDriver);
				Loggers.MSG.debug("Using the storage path \"{}\" as items input", itemInputPath);
			}
		}

		return itemInput;
	}

	static  Input createFileItemInput(
					final ItemFactory itemFactory, final String itemInputFile) {

		Input fileItemInput = null;

		final Path itemInputFilePath = Paths.get(itemInputFile);
		try {
			if (itemInputFile.endsWith(".csv")) {
				try {
					fileItemInput = new CsvFileItemInput<>(itemInputFilePath, itemFactory);
				} catch (final NoSuchMethodException e) {
					throw new AssertionError(e);
				}
			} else {
				fileItemInput = new BinFileInput<>(itemInputFilePath);
			}
		} catch (final IOException e) {
			LogUtil.exception(Level.WARN, e, "Failed to open the item input file \"{}\"", itemInputFile);
		}

		return fileItemInput;
	}

	static > Input createPathItemInput(
					final Config itemConfig,
					final int batchSize,
					final ItemFactory itemFactory,
					final String itemInputPath,
					final StorageDriver storageDriver) {
		Input itemInput = null;
		try {
			final var namingConfig = itemConfig.configVal("naming");
			final var prefix = namingConfig.stringVal("prefix");
			final var radix = namingConfig.intVal("radix");
			itemInput = new StorageItemInput<>(
							storageDriver, batchSize, itemFactory, itemInputPath, prefix, radix);
		} catch (final IllegalStateException | IllegalArgumentException e) {
			LogUtil.exception(Level.WARN, e, "Failed to initialize the data input");
		}
		return itemInput;
	}
}