uno.anahata.mapacho.servlet.DownloadResponse Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of mapacho-server Show documentation
The newest version!
/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package uno.anahata.mapacho.servlet;

import java.io.*;
import java.util.Date;
import java.util.zip.Deflater;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServletResponse;
import lombok.Getter;
import lombok.Setter;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.compress.compressors.gzip.GzipCompressorOutputStream;
import org.apache.commons.compress.compressors.gzip.GzipParameters;
import org.apache.commons.compress.compressors.pack200.Pack200CompressorOutputStream;
import org.apache.commons.compress.compressors.pack200.Pack200Strategy;
import org.apache.commons.io.IOUtils;
import uno.anahata.mapacho.common.http.HttpHeaders;
import uno.anahata.mapacho.common.io.CachingOutputStream;
import static uno.anahata.mapacho.common.http.HttpHeaders.*;

/**
 *
 * @author pablo
 */
@Getter
@Slf4j
public abstract class DownloadResponse {

    protected Date lastModified;

    @Setter
    protected String mimeType;

    protected long contentLength;

    /**
     * Uncompressed content length
     */
    protected long realContentLength;

    @Setter
    protected String contentEncoding;

    @Setter
    protected String versionId;

    @Setter
    @Getter
    protected boolean pack = false;

    @Setter
    @Getter
    protected boolean gz = false;

    @Setter
    @Getter
    protected String targetCacheName;

    protected DownloadResponse() {

    }

    public void setLastModified(Date lastModified) {
        this.lastModified = lastModified;
        if (lastModified != null) {
            lastModified = new Date((lastModified.getTime() / 1000) * 1000);
        }
    }

    private OutputStream getOutputStream(HttpServletResponse resp) throws IOException {
        OutputStream effectiveOs = resp.getOutputStream();

        if (targetCacheName != null) {
            effectiveOs = new CachingOutputStream(effectiveOs, new File(MapachoServlet.getCacheDirectory(),
                    targetCacheName));
        }

        if (gz) {
            log.debug("Streaming: " + this + " will gz with commons gz");
            GzipParameters params = new GzipParameters();
            params.setCompressionLevel(Deflater.BEST_COMPRESSION);
            effectiveOs = new GzipCompressorOutputStream(effectiveOs, params);
            //effectiveOs = new DeflaterOutputStream(effectiveOs, new Deflater(Deflater.BEST_COMPRESSION), true);
            //effectiveOs = new BufferedOutputStream(effectiveOs);            
        }

        if (pack) {
            log.debug("Streaming: " + this + " will pack");
            effectiveOs = new Pack200CompressorOutputStream(effectiveOs, Pack200Strategy.IN_MEMORY);
        }

        if (pack || gz) {
            realContentLength = contentLength;
            contentLength = 0;
        }

        return effectiveOs;

    }

    public void sendResponse(DownloadRequest dr, HttpServletResponse resp) throws IOException, ServletException {
        try {

            try (InputStream is = getInputStream(); OutputStream os = getOutputStream(resp)) {
                if (is == null) {
                    log.debug("Not Found " + dr);
                    resp.sendError(HttpServletResponse.SC_NOT_FOUND);
                    return;
                }

                //set last modified header        
                //send the bloody thing
                if (dr.getIfModifiedSince() != null && !lastModified.after(dr.getIfModifiedSince())) {
                    log.debug("Not modified " + this);
                    resp.sendError(HttpServletResponse.SC_NOT_MODIFIED);
                    return;
                }

                if (lastModified != null) {
                    log.debug(HEADER_LASTMOD + ": " + contentLength);
                    resp.setDateHeader(HEADER_LASTMOD, lastModified.getTime());
                }

                if (versionId != null) {
                    log.debug(HEADER_JNLP_VERSION + ": " + versionId);
                    resp.setHeader(HEADER_JNLP_VERSION, versionId);
                }

                if (contentLength != 0) {
                    log.debug("Content-Length: " + contentLength);
                    resp.setContentLengthLong(contentLength);
                }

                if (realContentLength != 0) {
                    log.debug(HttpHeaders.HEADER_REAL_CONTENT_LENGTH + ":" + realContentLength);
                    resp.setHeader(HttpHeaders.HEADER_REAL_CONTENT_LENGTH, realContentLength + "");
                }

                if (contentEncoding != null) {
                    log.debug(HEADER_CONTENT_ENCODING + ": " + contentEncoding);
                    resp.setHeader(HEADER_CONTENT_ENCODING, contentEncoding);
                }

                if (mimeType != null) {
                    log.debug("Content-Type (MimeType): " + mimeType);
                    resp.setContentType(mimeType);
                }

                log.debug("Streaming: " + is);

                IOUtils.copy(is, os);

            }
        } catch (Exception e) {
            log.error("Exception in sendResponse", e);
            throw new ServletException(e);
        }

    }

    protected abstract InputStream getInputStream() throws Exception;

    @Override
    public String toString() {
        return " " + ", lastModified=" + lastModified + ", mimeType=" + mimeType + ", contentLength=" + contentLength + '}';
    }

    //        if (dreq.isSupportsDeflate()) {            
//            try {
//                //file = AnahataCompressionUtils.compress(CompressionType.DEFLATE, file);
//                os = new DeflaterOutputStream(os, new Deflater(Deflater.BEST_COMPRESSION));
//                //resp.setHeader("encoding", "deflate");
//                resp.set
//            } catch (Exception e) {
//                log.warn("Could not compress file " + file, e);
//            }
//        }
}