jyapayne/Simple-Web-Server
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Allow calling Response::send multiple times: copy the buffer before async_write

Browse source
This commit is contained in:
Alex Fraser 2018-05-18 15:01:01 +10:00 • committed by eidheim
commit 6b91839cb6
2 changed files with 32 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

10
server_http.hpp
View file

@ -89,9 +89,17 @@ namespace SimpleWeb {
/// Use this function if you need to recursively send parts of a longer message
void send(const std::function<void(const error_code &)> &callback = nullptr) noexcept {
// Take a snapshot of the stream buffer. Otherwise async_write may not
// have consumed the buffer before the next call, in which case the
// data from the first call may be sent twice. Use a captured shared
// pointer to keep the snapshot alive during the async call.
auto stream_snapshot = std::make_shared<asio::streambuf>();
std::ostream snapshot_writer(stream_snapshot.get());
snapshot_writer << &streambuf;
session->connection->set_timeout(timeout_content);
auto self = this->shared_from_this(); // Keep Response instance alive through the following async_write
asio::async_write(*session->connection->socket, streambuf, [self, callback](const error_code &ec, std::size_t /*bytes_transferred*/) {
asio::async_write(*session->connection->socket, *stream_snapshot, [self, callback, stream_snapshot](const error_code &ec, std::size_t /*bytes_transferred*/) {
self->session->connection->cancel_timeout();
auto lock = self->session->connection->handler_runner->continue_lock();
if(!lock)