chunked_reader.hpp

浏览该文件的文档.

#ifndef NEFORCE_NETWORK_HTTP_CHUNKED_READER_HPP__
#define NEFORCE_NETWORK_HTTP_CHUNKED_READER_HPP__

 
#include "NeForce/core/container/unordered_map.hpp"
#include "NeForce/core/utility/byte_size.hpp"
#include "NeForce/network/tcp/tcp_socket.hpp"
NEFORCE_BEGIN_NAMESPACE__
NEFORCE_BEGIN_HTTP__


class chunked_body_reader {
public:
    byte_size max_chunk_size{1_MB};   
    byte_size max_total_size{100_MB}; 
    milliseconds read_timeout{30000}; 
 
    explicit chunked_body_reader(tcp_socket& socket) :
    socket_(&socket) {}

    bool next_chunk(byte_vector& out);

    NEFORCE_NODISCARD const unordered_map<string, string>& trailers() const noexcept { return trailers_; }

    NEFORCE_NODISCARD byte_size total_read() const noexcept { return total_read_; }

    NEFORCE_NODISCARD bool is_complete() const noexcept { return complete_; }
 
private:
    tcp_socket* socket_;
    byte_size total_read_{0};
    bool complete_{false};
    bool trailers_parsed_{false};
    unordered_map<string, string> trailers_;
    byte_vector buffer_;
 
    bool read_line(string& line);
    void parse_trailers();
};
 // HTTP
 
NEFORCE_END_HTTP__
NEFORCE_END_NAMESPACE__
#endif // NEFORCE_NETWORK_HTTP_CHUNKED_READER_HPP__