com.wadpam.open.io.ZipCsvsConverter Maven / Gradle / Ivy

Go to download
package com.wadpam.open.io;

import java.io.IOException;
import java.io.OutputStream;
import java.util.Map;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

import org.apache.poi.ss.util.WorkbookUtil;

/**
 *
 * @author os
 */
public class ZipCsvsConverter extends CsvConverter {

    @Override
    public Object preDao(OutputStream out, Object arg, Object preExport, Object preDao, 
            String tableName, Iterable columns, Map headers, 
            int daoIndex, D dao) {

        final String safeSheetName = WorkbookUtil.createSafeSheetName(tableName, '_');
        try {
            ((ZipOutputStream) out).putNextEntry(new ZipEntry(safeSheetName + ".csv"));

            return super.preDao(out, arg, preExport, preDao, tableName, columns, headers, daoIndex, dao);
        }
        catch (IOException e) {
            throw new RuntimeException(e);
        }
    }

    @Override
    public Object postDao(OutputStream out, Object arg, Object preExport, Object preDao, Object postDao, D dao) {
        Object res = super.postDao(out, arg, preExport, preDao, postDao, dao);

        try {
            ((ZipOutputStream) out).closeEntry();
        }
        catch (IOException e) {
            throw new RuntimeException(e);
        }

        return res;
    }

}