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_sip.h"
#include "qgsserverresponse.h"
 
#include <QBuffer>
#include <QByteArray>
#include <QMap>
#include <QString>

class SERVER_EXPORT QgsBufferServerResponse : public QgsServerResponse
{
  public:
    QgsBufferServerResponse();
    QgsBufferServerResponse( const QgsBufferServerResponse & ) = delete;

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

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

    void removeHeader( const QString &key ) override;

    Q_DECL_DEPRECATED QString header( const QString &key ) const override SIP_DEPRECATED;

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

    virtual QList<QString> fullHeader( const QString &key ) const override;

    Q_DECL_DEPRECATED QMap<QString, QString> headers() const override SIP_DEPRECATED;

    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, QList<QString>> mHeaders;
    QBuffer mBuffer;
    QByteArray mBody;
    bool mFinished = false;
    bool mHeadersSent = false;
    int mStatusCode = 200;
};
 
#endif