com.emc.mongoose.base.item.io.CsvItemOutput 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.io;

import static com.github.akurilov.commons.lang.Exceptions.throwUnchecked;

import com.emc.mongoose.base.Constants;
import com.emc.mongoose.base.item.Item;
import com.emc.mongoose.base.item.ItemFactory;
import com.github.akurilov.commons.io.Output;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.nio.charset.StandardCharsets;
import java.util.List;

/**
* The data item output writing into the specified file human-readable data item records using the
* CSV format
*/
public abstract class CsvItemOutput implements Output {
	//
	protected final ItemFactory itemFactory;
	protected final BufferedWriter itemsDst;

	//
	protected CsvItemOutput(final OutputStream out, final ItemFactory itemFactory)
					throws IOException {
		itemsDst = new BufferedWriter(new OutputStreamWriter(out, StandardCharsets.UTF_8), Constants.MIB);
		this.itemFactory = itemFactory;
	}

	//
	@Override
	public boolean put(final I item) {
		try {
			itemsDst.write(item.toString());
			itemsDst.newLine();
			return true;
		} catch (final IOException e) {
			throwUnchecked(e);
		}
		return false;
	}

	//
	@Override
	public int put(final List buffer, final int from, final int to) {
		int i = from;
		while (i < to && put(buffer.get(i))) {
			i++;
		}
		return i - from;
	}

	//
	@Override
	public final int put(final List items) {
		return put(items, 0, items.size());
	}

	//
	@Override
	public void close() {
		try {
			itemsDst.close();
		} catch (final IOException e) {
			throwUnchecked(e);
		}
	}

	//
	@Override
	public String toString() {
		return "csvItemOutput<" + itemsDst + ">";
	}
}