com.emc.mongoose.api.common.supply.RangeDefinedLongSupplier 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;

import com.github.akurilov.commons.math.Random;

import java.io.IOException;

/**
 Created by kurila on 07.03.17.
 */
public class RangeDefinedLongSupplier
implements BatchLongSupplier {
	
	private final long min;
	private final long range;
	private final Random rnd;
	
	/**
	 @param min "unbounded" range will be used if Long.MAX_VALUE > max - min
	 @param max "unbounded" range will be if less than min
	 */
	public RangeDefinedLongSupplier(final long seed, final long min, final long max) {
		rnd = new Random(seed);
		this.min = min;
		this.range = max - min + 1;
	}

	@Override
	public final long getAsLong() {
		if(range < 1) {
			return rnd.nextLong();
		} else {
			return min + rnd.nextLong(range);
		}
	}
	
	@Override
	public final int get(final long[] buffer, final int limit) {
		final int _limit = Math.min(buffer.length, limit);
		if(range < 0) {
			for(int i = 0; i < _limit; i ++) {
				buffer[i] = rnd.nextLong();
			}
		} else {
			for(int i = 0; i < _limit; i ++) {
				buffer[i] = min + rnd.nextLong(range);
			}
		}
		return _limit;
	}
	
	@Override
	public final long skip(final long count) {
		for(long i = 0; i < count; i ++) {
			rnd.nextLong();
		}
		return count;
	}
	
	@Override
	public final void reset() {
		rnd.reset();
	}
	
	@Override
	public void close()
	throws IOException {
	}
}