    001/*
    002 * Copyright (c) 2009 The openGion Project.
    003 *
    004 * Licensed under the Apache License, Version 2.0 (the "License");
    005 * you may not use this file except in compliance with the License.
    006 * You may obtain a copy of the License at
    007 *
    008 *     http://www.apache.org/licenses/LICENSE-2.0
    009 *
    010 * Unless required by applicable law or agreed to in writing, software
    011 * distributed under the License is distributed on an "AS IS" BASIS,
    012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
    013 * either express or implied. See the License for the specific language
    014 * governing permissions and limitations under the License.
    015 */
    016package org.opengion.hayabusa.filter;
    017
    018import org.opengion.fukurou.util.Closer;
    019
    020import java.io.PrintWriter;
    021import java.io.IOException;
    022import java.io.OutputStreamWriter;
    023import javax.servlet.ServletOutputStream;
    024import javax.servlet.http.HttpServletResponseWrapper;
    025import javax.servlet.http.HttpServletResponse;
    026
    027/**
    028 * GZIPFilter で使用する、GZIP圧縮されたレスポンスのラッパクラスです。
    029 *
    030 * @og.group フィルター処理
    031 *
    032 * @version  4.0
    033 * @author   Kazuhiko Hasegawa
    034 * @since    JDK5.0,
    035 */
    036public class GZIPResponseWrapper extends HttpServletResponseWrapper {
    037        protected HttpServletResponse origResponse = null;
    038        protected ServletOutputStream stream = null;
    039        protected PrintWriter writer = null;
    040
    041        /**
    042         * コンストラクター
    043         *
    044         * @param       response        HttpServletResponseオブジェクト
    045         */
    046        public GZIPResponseWrapper(final HttpServletResponse response) {
    047                super(response);
    048                origResponse = response;
    049        }
    050
    051        /**
    052         * ServletOutputStream の実体である GZIPResponseStream を作成して返します。
    053         *
    054         * @return      ServletOutputStreamオブジェクト
    055         * @throws IOException オブジェクトの作成に失敗したとき、throw されます。
    056         */
    057        public ServletOutputStream createOutputStream() throws IOException {
    058                return (new GZIPResponseStream(origResponse));
    059        }
    060
    061        /**
    062         * 内部ストリーム を クローズします。
    063         *
    064         */
    065        public void finishResponse() {
    066                Closer.ioClose( writer );
    067                Closer.ioClose( stream );
    068        }
    069
    070        /**
    071         * 内部ストリームの flush() メソッドを呼び出します。
    072         *
    073         */
    074        @Override
    075        public void flushBuffer() throws IOException {
    076                stream.flush();
    077        }
    078
    079        /**
    080         * 内部ServletOutputStreamを返します。
    081         *
    082         * 内部オブジェクトが存在しない場合は、新規に作成します。
    083         *
    084         * @return      ServletOutputStreamオブジェクト
    085         */
    086        @Override
    087        public ServletOutputStream getOutputStream() throws IOException {
    088                if(writer != null) {
    089                        throw new IllegalStateException("getWriter() has already been called!");
    090                }
    091
    092                if(stream == null) {
    093                        stream = createOutputStream();
    094                }
    095                return (stream);
    096        }
    097
    098        /**
    099         * 内部PrintWriterを返します。
    100         *
    101         * 内部オブジェクトが存在しない場合は、新規に作成します。
    102         *
    103         * @return PrintWriterオブジェクト
    104         */
    105        @Override
    106        public PrintWriter getWriter() throws IOException {
    107                if(writer != null) {
    108                        return (writer);
    109                }
    110
    111                if(stream != null) {
    112                        throw new IllegalStateException("getOutputStream() has already been called!");
    113                }
    114
    115                stream = createOutputStream();
    116                writer = new PrintWriter(new OutputStreamWriter(stream, "UTF-8"));
    117                return (writer);
    118        }
    119
    120        /**
    121         * 内部ストリームのデータ長を設定します(何もしません)。
    122         *
    123         * @param       length  データ長
    124         */
    125        @Override
    126        public void setContentLength(final int length) {
    127                // ここでは処理を行いません。
    128        }
    129}