org.seedstack.io.supercsv.SuperCsvParser Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of io-supercsv Show documentation
There is a newer version: 3.0.1
/**
 * Copyright (c) 2013-2016, The SeedStack authors 
 *
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 */
package org.seedstack.io.supercsv;

import org.seedstack.io.RendererErrorCode;
import org.seedstack.io.spi.AbstractTemplateParser;
import org.seedstack.seed.SeedException;
import org.supercsv.cellprocessor.ift.CellProcessor;
import org.supercsv.io.CsvBeanReader;
import org.supercsv.io.ICsvBeanReader;

import javax.inject.Named;
import java.io.Closeable;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.List;

/**
 * Parser for SuperCSV.
 *
 * @param  parsed object
 * @author [email protected]
 *         Date: 25/03/14
 */
@Named("SuperCSV")
public class SuperCsvParser extends AbstractTemplateParser {

    private static final String PARAM = "param";
    private static final String TEMPLATE_NAME = "templateName";

    @Override
    public List parse(InputStream inputStream, Class clazz) {
        ICsvBeanReader beanReader = null;
        List beans = new ArrayList();

        // Checks params nullity
        SeedException.createNew(RendererErrorCode.INCORRECT_PARAM).put(PARAM, "inputStream").throwsIfNull(inputStream);
        SeedException.createNew(RendererErrorCode.INCORRECT_PARAM).put(PARAM, "clazz").throwsIfNull(clazz);

        try {
            beanReader = new CsvBeanReader(new InputStreamReader(inputStream, template.getCharsetName()), template.getPreferences());

            if (template.showHeader()) {
                beanReader.getHeader(true);
            }

            // the header elements are used to map the values to the bean (names must match)
            final List confList = template.getReadingCellProcessors();
            final CellProcessor[] processors = confList.toArray(new CellProcessor[confList.size()]);
            final String[] fields = template.getFields().toArray(new String[template.getFields().size()]);

            T bean;
            while ((bean = beanReader.read(clazz, fields, processors)) != null) {
                beans.add(bean);
            }

        } catch (Exception e) {
            throw SeedException.wrap(e, RendererErrorCode.SUPER_CSV_EXCEPTION).put(TEMPLATE_NAME, template.getName());
        } finally {
            closeQuietly(beanReader);
        }
        return beans;
    }

    private void closeQuietly(Closeable closeable) {
        try {
            if (closeable != null) {
                closeable.close();
            }
        } catch (IOException e) {
            throw SeedException.wrap(e, RendererErrorCode.SUPER_CSV_EXCEPTION).put(TEMPLATE_NAME, template.getName());
        }
    }
}