com.emc.mongoose.api.model.item.ItemInfoFileOutput 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 com.github.akurilov.commons.io.Output;
import com.github.akurilov.commons.io.file.TextFileOutput;

import com.emc.mongoose.api.model.io.task.IoTask;

import java.io.IOException;
import java.nio.file.Path;
import java.util.ArrayList;
import java.util.List;

/**
 Created by kurila on 09.01.17.
 */
public final class ItemInfoFileOutput
implements Output {
	
	private final Output itemInfoOutput;

	public ItemInfoFileOutput(final Path filePath)
	throws IOException {
		itemInfoOutput = new TextFileOutput(filePath);
	}
	
	@Override
	public final boolean put(final O ioResult)
	throws IOException {
		if(ioResult == null) { // poison
			close();
			return true;
		}
		return itemInfoOutput.put(ioResult.getItem().toString());
	}
	
	@Override
	public final int put(final List ioResults, final int from, final int to)
	throws IOException {
		final int n = to - from;
		final List itemsInfo = new ArrayList<>(n);
		O ioResult;
		for(int i = from; i < to; i ++) {
			ioResult = ioResults.get(i);
			if(ioResult == null) { // poison
				try {
					return itemInfoOutput.put(itemsInfo, 0, i);
				} finally {
					close();
				}
			}
			itemsInfo.add(ioResult.getItem().toString());
		}
		return itemInfoOutput.put(itemsInfo, 0, n);
	}
	
	@Override
	public final int put(final List ioResults)
	throws IOException {
		final List itemsInfo = new ArrayList<>(ioResults.size());
		for(final O nextIoResult : ioResults) {
			if(nextIoResult == null) { // poison
				try {
					return itemInfoOutput.put(itemsInfo);
				} finally {
					close();
				}
			}
			itemsInfo.add(nextIoResult.getItem().toString());
		}
		return itemInfoOutput.put(itemsInfo);
	}
	
	@Override
	public final Input getInput()
	throws IOException {
		throw new AssertionError();
	}
	
	@Override
	public final void close()
	throws IOException {
		itemInfoOutput.close();
	}
}