qgsbufferserverresponse.h

Go to the documentation of this file.

/***************************************************************************
                          qgsbufferserverresponse.h

  Define response wrapper for storing responsea in buffer
  -------------------
  begin                : 2017-01-03
  copyright            : (C) 2017 by David Marteau
  email                : david dot marteau at 3liz dot 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.                                   *
 *                                                                         *
 ***************************************************************************/
#ifndef QGSBUFFERSERVERRESPONSE_H
#define QGSBUFFERSERVERRESPONSE_H

#include "qgis_server.h"
#include "qgis.h"
#include "qgsserverresponse.h"

#include <QBuffer>
#include <QByteArray>
#include <QMap>

class SERVER_EXPORT QgsBufferServerResponse: public QgsServerResponse
{
  public:

    QgsBufferServerResponse();

    QgsBufferServerResponse( const QgsBufferServerResponse & ) = delete;

    void setHeader( const QString &key, const QString &value ) override;

    void removeHeader( const QString &key ) override;

    QString header( const QString &key ) const override;

    QMap<QString, QString> headers() const override { return mHeaders; }

    bool headersSent() const override;

    void setStatusCode( int code ) override;

    int statusCode() const override { return mStatusCode; }

    void sendError( int code,  const QString &message ) override;

    QIODevice *io() override;

    void finish() override;

    void flush() override;

    void clear() override;

    QByteArray data() const override;

    void truncate() override;

    QByteArray body() const { return mBody; }


  private:

#ifdef SIP_RUN
    QgsBufferServerResponse( const QgsBufferServerResponse & ) SIP_FORCE;
#endif

    QMap<QString, QString> mHeaders;
    QBuffer                mBuffer;
    QByteArray             mBody;
    bool                   mFinished = false;
    bool                   mHeadersSent = false;
    int                    mStatusCode = 200;
};

#endif