com.emc.mongoose.api.common.supply.async.AsyncStringSupplierFactory Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of mongoose-api-common Show documentation
Mongoose is a high-load storage performance testing tool
There is a newer version: 4.0.0-alpha5
package com.emc.mongoose.api.common.supply.async;

import com.emc.mongoose.api.common.supply.BatchSupplier;
import com.emc.mongoose.api.common.supply.SupplierFactory;
import com.emc.mongoose.api.common.exception.OmgShootMyFootException;
import com.github.akurilov.coroutines.CoroutinesProcessor;

import java.text.ParseException;
import java.util.Date;
import java.util.regex.Matcher;

import static org.apache.commons.lang.time.DateUtils.parseDate;

public final class AsyncStringSupplierFactory>
implements SupplierFactory {

	private static final AsyncStringSupplierFactory>
		INSTANCE = new AsyncStringSupplierFactory<>();

	private CoroutinesProcessor coroutinesProcessor;

	private AsyncStringSupplierFactory() {
	}

	public static AsyncStringSupplierFactory> getInstance(
		final CoroutinesProcessor coroutinesProcessor
	) {
		return INSTANCE.setCoroutinesProcessor(coroutinesProcessor);
	}

	private AsyncStringSupplierFactory> setCoroutinesProcessor(
		final CoroutinesProcessor coroutinesProcessor
	) {
		this.coroutinesProcessor = coroutinesProcessor;
		return this;
	}

	/**
	 *
	 * @param type - a type of the generator
	 * @return a suitable generator
	 */
	@Override @SuppressWarnings("unchecked")
	public final G createSupplier(
		final char type, final String seedStr, final String formatStr, final String rangeStr
	) throws OmgShootMyFootException {
		long seed = System.nanoTime() ^ System.currentTimeMillis();
		if(seedStr != null && !seedStr.isEmpty()) {
			try {
				seed = Long.parseLong(seedStr);
			} catch(final NumberFormatException e) {
				throw new OmgShootMyFootException(
					"Seed value is not a 64 bit integer: \"" + seedStr + "\""
				);
			}
		}

		switch(type) {

			case 'd' : {
				long min = Long.MIN_VALUE;
				long max = Long.MAX_VALUE;
				if(rangeStr != null && !rangeStr.isEmpty()) {
					final Matcher matcher = LONG_PATTERN.matcher(rangeStr);
					if(matcher.find()) {
						min = Long.parseLong(matcher.group(1));
						max = Long.parseLong(matcher.group(2));
					} else {
						throw new OmgShootMyFootException();
					}
				}
				return (G) new AsyncRangeDefinedLongFormattingSupplier(
					coroutinesProcessor, seed, min, max, formatStr
				);
			}

			case 'f' : {
				double min = 0;
				double max = 1;
				if(rangeStr != null && !rangeStr.isEmpty()) {
					final Matcher matcher = DOUBLE_PATTERN.matcher(rangeStr);
					if(matcher.find()) {
						min = Double.parseDouble(matcher.group(1));
						max = Double.parseDouble(matcher.group(2));
					} else {
						throw new OmgShootMyFootException();
					}
				}
				return (G) new AsyncRangeDefinedDoubleFormattingSupplier(
					coroutinesProcessor, seed, min, max, formatStr
				);
			}

			case 'D': {
				Date min = new Date(0);
				Date max = new Date();
				if(rangeStr != null && !rangeStr.isEmpty()) {
					final Matcher matcher = DATE_PATTERN.matcher(rangeStr);
					if(matcher.find()) {
						try {
							min = parseDate(matcher.group(1), INPUT_DATE_FMT_STRINGS);
							max = parseDate(matcher.group(6), INPUT_DATE_FMT_STRINGS);
						} catch(final ParseException e) {
							throw new OmgShootMyFootException("Failed to parse the pattern");
						}
					} else {
						throw new OmgShootMyFootException();
					}
				}
				return (G) new AsyncRangeDefinedDateFormattingSupplier(
					coroutinesProcessor, seed, min, max, formatStr
				);
			}

			default:
				throw new OmgShootMyFootException("Unknown format type: '" + type + "'");
		}
	}

}