001 package org.apache.myfaces.tobago.webapp;
002
003 /*
004 * Licensed to the Apache Software Foundation (ASF) under one or more
005 * contributor license agreements. See the NOTICE file distributed with
006 * this work for additional information regarding copyright ownership.
007 * The ASF licenses this file to You under the Apache License, Version 2.0
008 * (the "License"); you may not use this file except in compliance with
009 * the License. You may obtain a copy of the License at
010 *
011 * http://www.apache.org/licenses/LICENSE-2.0
012 *
013 * Unless required by applicable law or agreed to in writing, software
014 * distributed under the License is distributed on an "AS IS" BASIS,
015 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
016 * See the License for the specific language governing permissions and
017 * limitations under the License.
018 */
019
020 import org.apache.commons.logging.Log;
021 import org.apache.commons.logging.LogFactory;
022
023 import javax.servlet.ServletOutputStream;
024 import javax.servlet.http.HttpServletResponse;
025 import javax.servlet.http.HttpServletResponseWrapper;
026 import java.io.IOException;
027 import java.io.PrintWriter;
028 import java.io.StringWriter;
029
030 /*
031 * Created: Nov 29, 2004 4:48:29 PM
032 * User: bommel
033 * $Id: TobagoResponse.java 601107 2007-12-04 22:12:14Z bommel $
034 */
035 public class TobagoResponse extends HttpServletResponseWrapper {
036
037 private static final Log LOG = LogFactory.getLog(TobagoResponse.class);
038
039 private PrintWriter printWriter = null;
040 private StringWriter bufferedWriter = null;
041
042
043 public TobagoResponse(HttpServletResponse base) {
044 super(base);
045 }
046
047 public void setBuffering() {
048 if (bufferedWriter == null) {
049 bufferedWriter = new StringWriter();
050 printWriter = new PrintWriter(bufferedWriter);
051 }
052 }
053
054 public String getBufferedString() {
055 if (bufferedWriter != null) {
056 printWriter.flush();
057 return bufferedWriter.toString();
058 } else {
059 return "";
060 }
061 }
062
063 public ServletOutputStream getOutputStream() throws IOException {
064 LOG.debug("***** getOutputStream() from " + new Exception().getStackTrace()[1]);
065 return getResponse().getOutputStream();
066 }
067
068 public PrintWriter getWriter() throws IOException {
069 LOG.debug("***** getWriter() from " + new Exception().getStackTrace()[1]);
070 if (printWriter != null) {
071 return printWriter;
072 }
073 return getResponse().getWriter();
074 }
075
076
077 public void setContentType(String s) {
078 LOG.debug("***** setContentType(" + s + ") from " + new Exception().getStackTrace()[1]);
079 getResponse().setContentType(s);
080 }
081
082
083 }