    001    // Copyright 2006, 2007 The Apache Software Foundation
    002    //
    003    // Licensed under the Apache License, Version 2.0 (the "License");
    004    // you may not use this file except in compliance with the License.
    005    // You may obtain a copy of the License at
    006    //
    007    //     http://www.apache.org/licenses/LICENSE-2.0
    008    //
    009    // Unless required by applicable law or agreed to in writing, software
    010    // distributed under the License is distributed on an "AS IS" BASIS,
    011    // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    012    // See the License for the specific language governing permissions and
    013    // limitations under the License.
    014
    015    package org.apache.tapestry5.internal.util;
    016
    017    import java.io.PrintWriter;
    018    import java.io.StringWriter;
    019
    020    /**
    021     * Utility for collecting the output of a {@link java.io.PrintWriter}.
    022     */
    023    public class PrintOutCollector
    024    {
    025        private final StringWriter stringWriter;
    026
    027        private PrintWriter printWriter;
    028
    029        public PrintOutCollector()
    030        {
    031            stringWriter = new StringWriter();
    032            printWriter = new PrintWriter(stringWriter);
    033        }
    034
    035        public PrintWriter getPrintWriter()
    036        {
    037            return printWriter;
    038        }
    039
    040        /**
    041         * Closes the {@link PrintWriter} and returns the accumulated text output.
    042         */
    043        public String getPrintOut()
    044        {
    045            printWriter.close();
    046            return stringWriter.toString();
    047
    048        }
    049
    050    }