ServletOutputStream.h

Go to the documentation of this file.

/***************************************************************************
 *   Copyright (C) 2004 by Ilya A. Volynets-Evenbakh                       *
 *   ilya@total-knowledge.com                                              *
 *                                                                         *
 *   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 ied 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.,                                       *
 *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
 ***************************************************************************/
#ifndef CSERVSERVLETOUTPUTSTREAM_H
#define CSERVSERVLETOUTPUTSTREAM_H

#include <connection.h>

namespace container {

class HttpServletResponseImpl;
class ServletOutputStreamBuf: public std::streambuf
{
    Connection& m_con;
    HttpServletResponseImpl& m_resp;
    size_t m_bufSz;
public:
    ServletOutputStreamBuf(Connection& con,HttpServletResponseImpl& resp)
        : m_con(con)
        , m_resp(resp)
    {
        m_bufSz=m_con.epptr()-m_con.pbase();
    }

    size_t getBufSize(){return m_bufSz;}
protected:
    virtual int sync();
    virtual int showmanyc() const {return 0;}
    virtual std::streamsize xsgetn(char* s, std::streamsize n)
        {return 0;}
    virtual int underflow(){return traits_type::eof();}
    virtual int pbackfail();
    virtual std::streamsize xsputn(const char* s,std::streamsize n);
    virtual int overflow(int c);
    virtual std::streambuf* setbuf(char* s,std::streamsize n);
};

class ServletOutputStream : public std::ostream
{
private:
    ServletOutputStreamBuf m_buf;
public:
    ServletOutputStream(Connection&,HttpServletResponseImpl&);
    virtual ~ServletOutputStream();
    friend class HttpServletResponseImpl;
protected:
    virtual void setBufferSize(size_t sz);
    virtual size_t getBufferSize();
};

}

#endif