org.jruby.rack.servlet.ServletRackResponseEnvironment Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jruby-rack Show documentation

A servlet bridge for (Ruby-based) Rack applications that allow them to run in Java Application servers using JRuby.

There is a newer version: 1.2.2

/*
 * Copyright (c) 2010-2012 Engine Yard, Inc.
 * Copyright (c) 2007-2009 Sun Microsystems, Inc.
 * This source code is available under the MIT license.
 * See the file LICENSE.txt for details.
 */

package org.jruby.rack.servlet;

import org.jruby.rack.RackResponse;
import org.jruby.rack.RackResponseEnvironment;

import java.io.IOException;
import java.util.Iterator;
import java.util.Map;
import java.util.Set;

import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpServletResponseWrapper;

/**
 * Servlet response wrapper Rack (response) implementation.
 * 
 * @author nicksieger
 */
public class ServletRackResponseEnvironment extends HttpServletResponseWrapper 
    implements RackResponseEnvironment {
    
    public ServletRackResponseEnvironment(HttpServletResponse response) {
        super(response);
    }
    
    public void defaultRespond(final RackResponse response) throws IOException {
        setStatus(response.getStatus());
        @SuppressWarnings("unchecked")
        final Set headers = response.getHeaders().entrySet();
        for ( Iterator it = headers.iterator(); it.hasNext(); ) {
            final Map.Entry entry = it.next();
            final String key = entry.getKey().toString();
            final Object value = entry.getValue();
            addHeader(key, value != null ? value.toString() : null);
        }
        getWriter().write(response.getBody());
    }
    
}