com.emc.mongoose.item.io.NewDataItemInput Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mongoose Show documentation

Mongoose is a high-load storage performance testing tool

The newest version!

package com.emc.mongoose.item.io;

import com.emc.mongoose.item.DataItem;
import com.emc.mongoose.item.IdStringInput;
import com.emc.mongoose.item.ItemFactory;
import com.github.akurilov.commons.io.Input;
import com.github.akurilov.commons.system.SizeInBytes;

import java.io.IOException;
import java.util.List;

public final class NewDataItemInput
extends NewItemInput
implements Input {
	
	private final SizeInBytes dataSize;
	
	public NewDataItemInput(
		final ItemFactory itemFactory, final IdStringInput idInput, final SizeInBytes dataSize
	) {
		super(itemFactory, idInput);
		this.dataSize = dataSize;
	}
	
	@Override
	public final D get()
	throws IOException {
		return itemFactory.getItem(idInput.get(), idInput.getAsLong(), dataSize.get());
	}
	
	@Override
	public final int get(final List buffer, final int maxCount)
	throws IOException {
		for(int i = 0; i < maxCount; i ++) {
			buffer.add(itemFactory.getItem(idInput.get(), idInput.getAsLong(), dataSize.get()));
		}
		return maxCount;
	}
	
	@Override
	public final String toString() {
		return super.toString() + "(" + dataSize.toString() + ")";
	}
}