http_response_body_stream.hpp

#pragma once
 
#include <string>
 
#include <userver/server/http/http_response.hpp>
#include <userver/server/request/response_base.hpp>
 
USERVER_NAMESPACE_BEGIN
 
namespace server::handlers {
class HttpHandlerBase;
}
 
namespace server::http {
 
class ResponseBodyStream final {
public:
    ResponseBodyStream(ResponseBodyStream&&) = default;
    ~ResponseBodyStream();
 
    // Send a chunk of response data. It may NOT generate
    // exactly one HTTP chunk per call to PushBodyChunk().
    // May not be called after SetBody().
    void PushBodyChunk(std::string&& chunk, engine::Deadline deadline);
 
    // Set full response body instead of sending chunks.
    // May not be called after PushBodyChunk().
    void SetBody(std::string&& body);
 
    void SetHeader(const std::string&, const std::string&);
 
    void SetHeader(std::string_view, const std::string&);
 
    void SetEndOfHeaders();
 
    void SetStatusCode(int status_code);
 
    void SetStatusCode(HttpStatus status);
 
private:
    friend class server::handlers::HttpHandlerBase;
 
    ResponseBodyStream(HttpResponse::Producer&& queue_producer, HttpResponse& http_response);
 
    bool headers_ended_{false};
    bool headers_end_sent_{false};
    HttpResponse::Producer queue_producer_;
    HttpResponse& http_response_;
};
 
}  // namespace server::http
 
USERVER_NAMESPACE_END