org.nuiton.util.ZipStreamEncoder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of nuiton-utils Show documentation
Library of usefull class to be used in any project.
There is a newer version: 3.1
/*
 * #%L
 * Nuiton Utils
 * %%
 * Copyright (C) 2004 - 2010 CodeLutin
 * %%
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as 
 * published by the Free Software Foundation, either version 3 of the 
 * License, or (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Lesser Public License for more details.
 * 
 * You should have received a copy of the GNU General Lesser Public 
 * License along with this program.  If not, see
 * .
 * #L%
 */

package org.nuiton.util;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

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

/** The Class ZipStreamEncoder. */
public class ZipStreamEncoder extends Thread {

    /** Logger. */
    private static final Log log = LogFactory.getLog(ZipStreamEncoder.class);

    /** The Constant BUFFER. */
    static final int BUFFER = 2048;

    /** The files. */
    private Map files;

    /** The zos. */
    private ZipOutputStream zos;

    /**
     * Instantiates a new zip stream encoder.
     *
     * @param files the files
     * @param os    the os
     */
    public ZipStreamEncoder(Map files, OutputStream os) {
        this.files = files;

        zos = new ZipOutputStream(os);
        zos.setMethod(ZipOutputStream.DEFLATED);
        zos.setLevel(Deflater.BEST_COMPRESSION);
    }

    /* (non-Javadoc)
     * @see java.lang.Thread#run()
     */
    @Override
    public void run() {
        byte data[] = new byte[BUFFER];
        try {
            for (Map.Entry kv : files.entrySet()) {
                ZipEntry entry = new ZipEntry(kv.getKey());
                InputStream origin = kv.getValue();
                zos.putNextEntry(entry);
                int count;
                while ((count = origin.read(data, 0, BUFFER)) != -1) {
                    zos.write(data, 0, count);
                }
                origin.close();
            }

            zos.close();
        } catch (IOException e) {
            for (Map.Entry kv : files.entrySet()) {
                InputStream origin = kv.getValue();
                try {
                    origin.close();
                } catch (IOException ioe) {
                    log.error("Impossible to close " + kv.getKey());
                }
            }
            log.error("Impossible to compress in stream");
            throw new RuntimeException("Impossible to compress in stream");
        }
    }

}