com.emc.mongoose.api.model.item.NewItemInput Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mongoose-api-model 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.model.item;

import com.github.akurilov.commons.io.Input;

import java.io.IOException;
import java.util.List;
/**
 Created by andrey on 01.12.16.
 */
public class NewItemInput
implements Input {

	protected final ItemFactory itemFactory;
	protected final IdStringInput idInput;

	public NewItemInput(final ItemFactory itemFactory, final IdStringInput idInput) {
		this.itemFactory = itemFactory;
		this.idInput = idInput;
	}

	@Override
	public I get()
	throws IOException {
		return itemFactory.getItem(idInput.get());
	}

	@Override
	public int get(final List buffer, final int maxCount)
	throws IOException {
		for(int i = 0; i < maxCount; i ++) {
			buffer.add(itemFactory.getItem(idInput.get()));
		}
		return maxCount;
	}

	/**
	 * Skips the specified count of the new item ids
	 * @param itemsCount count of items which should be skipped from the beginning
	 * @throws IOException doesn't throw
	 */
	@Override
	public final long skip(final long itemsCount)
	throws IOException {
		return idInput.skip(itemsCount);
	}

	@Override
	public final void reset()
	throws IOException {
		idInput.reset();
	}

	@Override
	public final void close()
	throws IOException {
		itemFactory.close();
		idInput.close();
	}

	@Override
	public String toString() {
		return "New" + (itemFactory instanceof DataItemFactory ? "Data" : "") + "Items";
	}
}