com.emc.mongoose.base.item.op.OperationsBuilderImpl 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.op;

import static com.github.akurilov.commons.lang.Exceptions.throwUnchecked;

import com.emc.mongoose.base.config.ConstantValueInput;
import com.emc.mongoose.base.item.Item;
import com.emc.mongoose.base.storage.Credential;
import com.github.akurilov.commons.io.Input;
import java.io.IOException;
import java.util.List;
import java.util.Map;

/** Created by kurila on 14.07.16. */
public class OperationsBuilderImpl>
				implements OperationsBuilder {

	protected final int originIndex;

	protected OpType opType = OpType.CREATE; // by default
	protected String inputPath = null;

	protected Input outputPathInput;
	protected boolean constOutputPathFlag;
	protected String constOutputPath;

	protected Input credentialInput;
	protected boolean constCredFlag;
	protected Credential constCred;

	protected Map credentialsByPath = null;

	public OperationsBuilderImpl(final int originIndex) {
		this.originIndex = originIndex;
	}

	@Override
	public final int originIndex() {
		return originIndex;
	}

	@Override
	public final OpType opType() {
		return opType;
	}

	@Override
	public final OperationsBuilderImpl opType(final OpType opType) {
		this.opType = opType;
		return this;
	}

	public final String inputPath() {
		return inputPath;
	}

	@Override
	public final OperationsBuilderImpl inputPath(final String inputPath) {
		this.inputPath = inputPath;
		return this;
	}

	@Override
	public final OperationsBuilderImpl outputPathInput(final Input ops) {
		this.outputPathInput = ops;
		if (outputPathInput == null) {
			constOutputPathFlag = true;
			constOutputPath = null;
		} else if (outputPathInput instanceof ConstantValueInput) {
			constOutputPathFlag = true;
			constOutputPath = outputPathInput.get();
		} else {
			constOutputPathFlag = false;
		}
		return this;
	}

	@Override
	public final OperationsBuilderImpl credentialInput(
					final Input credentialInput) {
		this.credentialInput = credentialInput;
		if (credentialInput == null) {
			constCredFlag = true;
			constCred = Credential.NONE;
		} else if (credentialInput instanceof ConstantValueInput) {
			constCredFlag = true;
			constCred = credentialInput.get();
		} else {
			constCredFlag = false;
		}
		return this;
	}

	@Override
	public OperationsBuilderImpl credentialsByPath(final Map credByPath) {
		this.credentialsByPath = credByPath;
		return this;
	}

	@Override
	@SuppressWarnings("unchecked")
	public O buildOp(final I item) throws IOException {
		final String outputPath = getNextOutputPath();
		return (O) new OperationImpl(
						originIndex, opType, item, inputPath, outputPath, getNextCredential(outputPath));
	}

	@Override
	@SuppressWarnings("unchecked")
	public void buildOps(final List items, final List buff) throws IOException {
		String outputPath;
		for (final I item : items) {
			outputPath = getNextOutputPath();
			buff.add(
							(O) new OperationImpl<>(
											originIndex, opType, item, inputPath, outputPath, getNextCredential(outputPath)));
		}
	}

	protected final String getNextOutputPath() {
		return constOutputPathFlag ? constOutputPath : outputPathInput.get();
	}

	protected final Credential getNextCredential(final String path) {
		return constCredFlag ? constCred : credentialsByPath.get(path);
	}

	@Override
	public void close() {
		inputPath = null;
		try {
			if (outputPathInput != null) {
				outputPathInput.close();
				outputPathInput = null;
			}
			if (credentialInput != null) {
				credentialInput.close();
				credentialInput = null;
			}
			if (credentialsByPath != null) {
				credentialsByPath.clear();
				credentialsByPath = null;
			}
		} catch (final Exception e) {
			throwUnchecked(e);
		}
	}
}