com.github.bjuvensjo.rsimulator.proxy.ResponseHandler Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of rsimulator-proxy Show documentation

${project.artifactId} module

There is a newer version: 3.1.1

package com.github.bjuvensjo.rsimulator.proxy;

import java.io.IOException;
import java.io.InputStream;
import java.net.HttpURLConnection;
import java.util.List;
import java.util.Map;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;

import org.apache.commons.io.IOUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.google.inject.Singleton;

/**
 * @author Anders Bälter
 * @author Magnus Bjuvensjö
 */
@Singleton
public class ResponseHandler {
    private Logger log = LoggerFactory.getLogger(ResponseHandler.class);

    public void handle(HttpServletResponse response, HttpURLConnection connection) throws IOException {
        copyResponseHeaders(response, connection);
        copyResponse(response, connection);
    }
    
    private void copyResponseHeaders(HttpServletResponse response, HttpURLConnection con) throws IOException {
        //response.setContentType(con.getContentType());
        response.setStatus(con.getResponseCode());
        Map> headerFields = con.getHeaderFields();
        for (Map.Entry> entry : headerFields.entrySet()) {
            String key = entry.getKey();
            if (key != null) {
                for (String value : entry.getValue()) {
                    if (value != null) {
                        response.addHeader(key, value);
                        log.debug("copied response header {} : {}", key, value);
                    }
                }
            }
        }
    }

    private void copyResponse(HttpServletResponse response, HttpURLConnection connection) throws IOException {
        InputStream in = connection.getInputStream();
        ServletOutputStream out = response.getOutputStream();
        IOUtils.copy(in, out);
    }
}