Updated asio_compatibility.hpp

1 files changed, 6 insertions, 0 deletions

diff --git a/asio_compatibility.hpp b/asio_compatibility.hpp
index ca332c2..a9abd7c 100644
--- a/asio_compatibility.hpp
+++ b/asio_compatibility.hpp
@@ -50,6 +50,9 @@ namespace SimpleWeb {
   void async_resolve(asio::ip::tcp::resolver &resolver, const std::pair<std::string, std::string> &host_port, handler_type &&handler) {
     resolver.async_resolve(host_port.first, host_port.second, std::forward<handler_type>(handler));
   }
+  inline asio::executor_work_guard<io_context::executor_type> make_work_guard(io_context &context) {
+    return asio::make_work_guard(context);
+  }
 #else
   using io_context = asio::io_service;
   using resolver_results = asio::ip::tcp::resolver::iterator;
@@ -73,6 +76,9 @@ namespace SimpleWeb {
   void async_resolve(asio::ip::tcp::resolver &resolver, const std::pair<std::string, std::string> &host_port, handler_type &&handler) {
     resolver.async_resolve(asio::ip::tcp::resolver::query(host_port.first, host_port.second), std::forward<handler_type>(handler));
   }
+  inline io_context::work make_work_guard(io_context &context) {
+    return io_context::work(context);
+  }
 #endif
 } // namespace SimpleWeb