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

Added callback to start and accept_and_run

Browse source
This commit is contained in:
eidheim 2019-12-29 10:35:14 +01:00
commit d156326598
1 changed files with 13 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

19
server_http.hpp
View file

@ -129,7 +129,7 @@ namespace SimpleWeb {
/// Send the content of the response stream to client. The callback is called when the send has completed.
///
/// Use this function if you need to recursively send parts of a longer message, or when using server-sent events.
void send(const std::function<void(const error_code &)> &callback = nullptr) noexcept {
void send(std::function<void(const error_code &)> callback = nullptr) noexcept {
session->connection->set_timeout(timeout_content);
std::shared_ptr<asio::streambuf> streambuf = std::move(this->streambuf);
@ -137,7 +137,7 @@ namespace SimpleWeb {
rdbuf(this->streambuf.get());
LockGuard lock(send_queue_mutex);
send_queue.emplace_back(streambuf, callback);
send_queue.emplace_back(std::move(streambuf), std::move(callback));
if(send_queue.size() == 1)
send_from_queue();
}
@ -432,7 +432,8 @@ namespace SimpleWeb {
/// If you know the server port in advance, use start() instead.
/// Accept requests, and if io_service was not set before calling bind(), run the internal io_service instead.
/// Call after bind().
void accept_and_run() {
/// The callback parameter is called after the server is accepting connections.
void accept_and_run(std::function<void()> callback = nullptr) {
acceptor->listen();
accept();
@ -440,6 +441,9 @@ namespace SimpleWeb {
if(io_service->stopped())
restart(*io_service);
if(callback)
io_service->post(std::move(callback));
// If thread_pool_size>1, start m_io_service.run() in (thread_pool_size-1) threads for thread-pooling
threads.clear();
for(std::size_t c = 1; c < config.thread_pool_size; c++) {
@ -456,12 +460,15 @@ namespace SimpleWeb {
for(auto &t : threads)
t.join();
}
else if(callback)
io_service->post(std::move(callback));
}
/// Start the server by calling bind() and accept_and_run()
void start() {
/// Start the server by calling bind() and accept_and_run().
/// The callback parameter is called after the server is accepting connections.
void start(std::function<void()> callback = nullptr) {
bind();
accept_and_run();
accept_and_run(std::move(callback));
}
/// Stop accepting new requests, and close current connections.