Merge pull request #907 from redboltz/add_dispatch_wrapper

Added dispatch wrapper.
redboltz · Nov 30, 2021 · b4caab7 · b4caab7
2 parents 96743d4 + e0e03ad
commit b4caab7
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 0 deletions.
diff --git a/include/mqtt/tcp_endpoint.hpp b/include/mqtt/tcp_endpoint.hpp
@@ -76,6 +76,13 @@ class tcp_endpoint : public socket {
         );
     }
 
+    MQTT_ALWAYS_INLINE void dispatch(std::function<void()> handler) override final {
+        as::dispatch(
+            strand_,
+            force_move(handler)
+        );
+    }
+
     MQTT_ALWAYS_INLINE as::ip::tcp::socket::lowest_layer_type& lowest_layer() override final {
         return tcp_.lowest_layer();
     }

diff --git a/include/mqtt/type_erased_socket.hpp b/include/mqtt/type_erased_socket.hpp
@@ -26,6 +26,7 @@ class socket {
     virtual void async_write(std::vector<as::const_buffer>, std::function<void(error_code, std::size_t)>) = 0;
     virtual std::size_t write(std::vector<as::const_buffer>, boost::system::error_code&) = 0;
     virtual void post(std::function<void()>) = 0;
+    virtual void dispatch(std::function<void()>) = 0;
     virtual as::ip::tcp::socket::lowest_layer_type& lowest_layer() = 0;
     virtual any native_handle() = 0;
     virtual void clean_shutdown_and_close(boost::system::error_code&) = 0;

diff --git a/include/mqtt/ws_endpoint.hpp b/include/mqtt/ws_endpoint.hpp
@@ -137,6 +137,13 @@ class ws_endpoint : public socket {
         );
     }
 
+    MQTT_ALWAYS_INLINE void dispatch(std::function<void()> handler) override final {
+        as::dispatch(
+            strand_,
+            force_move(handler)
+        );
+    }
+
 #if BOOST_VERSION >= 107000
 
     MQTT_ALWAYS_INLINE as::ip::tcp::socket::lowest_layer_type& lowest_layer() override final {