eden-emu/cpp-httplib
14
0
Fork 0
mirror of https://github.com/yhirose/cpp-httplib.git synced 2025-05-15 01:08:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge pull request #85 from davidgfnet/streaming

Browse source 
Implementing streaming Responses
This commit is contained in:
yhirose 2018-08-01 17:56:09 -04:00 • committed by GitHub
commit f2daaf1b85
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 71 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

37
httplib.h
View file

@ -146,6 +146,7 @@ struct Response {
int status;
Headers headers;
std::string body;
std::function<std::string (uint64_t offset)> streamcb;
bool has_header(const char* key) const;
std::string get_header_value(const char* key) const;
@ -964,6 +965,17 @@ inline bool from_hex_to_i(const std::string& s, size_t i, size_t cnt, int& val)
return true;
}
inline std::string from_i_to_hex(uint64_t n)
{
const char *charset = "0123456789abcdef";
std::string ret;
do {
ret = charset[n & 15] + ret;
n >>= 4;
} while (n > 0);
return ret;
}
inline size_t to_utf8(int code, char* buff)
{
if (code < 0x0080) {
@ -1587,13 +1599,34 @@ inline void Server::write_response(Stream& strm, bool last_connection, const Req
auto length = std::to_string(res.body.size());
res.set_header("Content-Length", length.c_str());
} else if (res.streamcb) {
// Streamed response
bool chunked_response = !res.has_header("Content-Length");
if (chunked_response)
res.set_header("Transfer-Encoding", "chunked");
}
detail::write_headers(strm, res);
// Body
if (!res.body.empty() && req.method != "HEAD") {
strm.write(res.body.c_str(), res.body.size());
if (req.method != "HEAD") {
if (!res.body.empty()) {
strm.write(res.body.c_str(), res.body.size());
} else if (res.streamcb) {
bool chunked_response = !res.has_header("Content-Length");
uint64_t offset = 0;
bool data_available = true;
while (data_available) {
std::string chunk = res.streamcb(offset);
offset += chunk.size();
data_available = !chunk.empty();
// Emit chunked response header and footer for each chunk
if (chunked_response)
chunk = detail::from_i_to_hex(chunk.size()) + "\r\n" + chunk + "\r\n";
if (strm.write(chunk.c_str(), chunk.size()) < 0)
break; // Stop on error
}
}
}
// Log