Response.java :  » Google-tech » gap-data » gap » Java Open Source

/*
 * Gap Data
 * Copyright (C) 2009 John Pritchard
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License as
 * published by the Free Software Foundation; either version 2 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 Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
 * 02110-1301 USA.
 */
package gap;

import gap.util.DevNullOutputStream;

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

import java.io.IOException;
import java.io.OutputStreamWriter;
import java.io.PrintWriter;
import java.nio.charset.Charset;

/**
 * Created by {@link gap.service.Servlet}.
 * 
 * @author jdp
 */
public class Response 
    extends javax.servlet.http.HttpServletResponseWrapper
    implements gap.data.DataInheritance.Notation
{
    /**
     * Universal charset
     */
    public final static Charset UTF8 = Charset.forName("UTF-8");

    private final static java.lang.ThreadLocal<Response> RTL = new java.lang.ThreadLocal<Response>();

    public final static Response Get(){
        return RTL.get();
    }
    public final static void Exit(){
        RTL.remove();
    }


    private ServletOutputStream out;
    private PrintWriter writer;


    public Response(HttpServletResponse req){
        super(req);
        this.setCharacterEncoding("UTF-8");

        RTL.set(this);
    }


    public PrintWriter openDevNull(){
        this.out = new DevNullOutputStream();
        this.writer = new PrintWriter(new OutputStreamWriter(this.out,UTF8));
        return this.writer;
    }
    public void closeDevNull(){
    }
    public void setContentLength(){

        DevNullOutputStream devNull = (DevNullOutputStream)this.out;
        int contentLength = devNull.getCount();
        if (0 < contentLength)
            this.setContentLength(contentLength);
    }
    public ServletOutputStream getOutputStream() throws IOException {
        ServletOutputStream out = this.out;
        if (null != out)
            return out;
        else
            return super.getOutputStream();
    }
    public PrintWriter getWriter() throws IOException {
        PrintWriter writer = this.writer;
        if (null != writer)
            return writer;
        else
            return super.getWriter();
    }
    public void println(String line) throws IOException {
        this.getWriter().println(line);
    }
    public void printf(String format, Object... args) throws IOException {
        this.getWriter().printf(format,args);
    }
    public void write(String string) throws IOException {
        this.getWriter().write(string);
    }
    public void setContentTypeText(){
        this.setContentType("text/plain");
    }
    public void setContentTypeHtml(){
        this.setContentType("text/html");
    }
    public void setContentTypeXml(){
        this.setContentType("text/xml");
    }
    public void setContentTypeXml(Request req){
        if (req.accept("application/xml"))
            this.setContentType("application/xml");
        else
            this.setContentType("text/xml");
    }
    public void setContentTypeJson(){
        this.setContentType("application/json");
    }

}