com.emc.mongoose.base.item.naming.ItemNameInputImpl 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.naming;

import com.github.akurilov.commons.io.Input;
import it.unimi.dsi.fastutil.longs.Long2LongFunction;

import java.util.List;

public final class ItemNameInputImpl
				implements ItemNameInput {

	private final long initialId;
	private final Long2LongFunction idFunction;
	private volatile long lastId;
	private final Input prefixInput;
	private final int radix;

	public ItemNameInputImpl(
					final Long2LongFunction idFunction, final long offset, final Input prefixInput, final int radix) {
		this.initialId = offset;
		this.lastId = initialId;
		this.idFunction = idFunction;
		this.prefixInput = prefixInput;
		this.radix = radix;
	}

	@Override
	public final long lastId() {
		return lastId;
	}

	private void eval() {
		lastId = idFunction.applyAsLong(lastId);
	}

	private String convert() {
		return prefixInput.get() + Long.toString(lastId, radix);
	}

	@Override
	public final String get() {
		eval();
		return convert();
	}

	@Override
	public final int get(final List buffer, final int limit) {
		for (var i = 0; i < limit; i++) {
			eval();
			buffer.add(convert());
		}
		return limit;
	}

	@Override
	public final long skip(final long count) {
		for (var i = 0L; i < count; i++) {
			idFunction.applyAsLong(lastId);
		}
		return count;
	}

	@Override
	public final void reset() {
		lastId = initialId;
	}

	@Override
	public final void close()
					throws Exception {
		prefixInput.close();
	}
}