com.emc.mongoose.api.model.item.CsvFileItemOutput 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.file.FileOutput;

import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.StandardOpenOption;

/**
 Created by kurila on 30.06.15.
 */
public class CsvFileItemOutput
extends CsvItemOutput
implements FileOutput {
	
	protected Path itemsFilePath;
	
	public CsvFileItemOutput(final Path itemsFilePath, final ItemFactory itemFactory)
	throws IOException {
		super(
			Files.newOutputStream(
				itemsFilePath, StandardOpenOption.WRITE, StandardOpenOption.CREATE
			),
			itemFactory
		);
		this.itemsFilePath = itemsFilePath;
	}
	
	public CsvFileItemOutput(final ItemFactory itemFactory)
	throws IOException {
		this(Files.createTempFile(null, ".csv"), itemFactory);
		this.itemsFilePath.toFile().deleteOnExit();
	}
	
	@Override
	public CsvFileItemInput getInput()
	throws IOException {
		try {
			return new CsvFileItemInput<>(itemsFilePath, itemFactory);
		} catch(final NoSuchMethodException e) {
			throw new IOException(e);
		}
	}
	
	@Override
	public String toString() {
		return "csvFileItemOutput<" + itemsFilePath.getFileName() + ">";
	}
	
	@Override
	public final Path getFilePath() {
		return itemsFilePath;
	}
}