ch.algotrader.report.ListReporter.java Source code

Introduction

Here is the source code for ch.algotrader.report.ListReporter.java

Source

/***********************************************************************************
 * AlgoTrader Enterprise Trading Framework
 *
 * Copyright (C) 2015 AlgoTrader GmbH - All rights reserved
 *
 * All information contained herein is, and remains the property of AlgoTrader GmbH.
 * The intellectual and technical concepts contained herein are proprietary to
 * AlgoTrader GmbH. Modification, translation, reverse engineering, decompilation,
 * disassembly or reproduction of this material is strictly forbidden unless prior
 * written permission is obtained from AlgoTrader GmbH
 *
 * Fur detailed terms and conditions consult the file LICENSE.txt or contact
 *
 * AlgoTrader GmbH
 * Aeschstrasse 6
 * 8834 Schindellegi
 ***********************************************************************************/

package ch.algotrader.report;

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

import org.apache.commons.io.FileUtils;
import org.supercsv.cellprocessor.ift.CellProcessor;
import org.supercsv.io.CsvListWriter;
import org.supercsv.prefs.CsvPreference;

/**
 * SuperCSV Writer that writes Maps to the specified CSV-File.
 *
 * @author <a href="mailto:aflury@algotrader.ch">Andy Flury</a>
 */
public class ListReporter extends Report {

    private final CellProcessor[] processor;
    private final CsvListWriter writer;

    public ListReporter(File file, String[] header) throws IOException {

        this(file, header, null);
    }

    public ListReporter(File file, String[] header, CellProcessor[] processor) throws IOException {

        File parent = file.getParentFile();
        if (!parent.exists()) {
            FileUtils.forceMkdir(parent);
        }

        this.processor = processor;

        this.writer = new CsvListWriter(new FileWriter(file, false), CsvPreference.EXCEL_PREFERENCE);

        this.writer.writeHeader(header);

        ReportManager.registerReport(this);
    }

    public void write(List<?> row) throws IOException {

        if (this.processor != null) {
            this.writer.write(row, this.processor);
        } else {
            this.writer.write(row);
        }
    }

    public void writeAndFlush(List<?> row) throws IOException {

        if (this.processor != null) {
            this.writer.write(row, this.processor);
        } else {
            this.writer.write(row);
        }
        this.writer.flush();
    }

    public void write(Object... row) throws IOException {

        this.writer.write(row);
    }

    public void writeAndFlush(Object... row) throws IOException {

        this.writer.write(row);
        this.writer.flush();
    }

    @Override
    public void close() throws IOException {

        this.writer.close();
    }
}