Fix zeromq#230

williamleong · Jan 23, 2025 · de7947b · de7947b
1 parent ba4230d
commit de7947b
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/src/zmqpp/socket.cpp b/src/zmqpp/socket.cpp
@@ -118,13 +118,21 @@ void socket::close()
 
 bool socket::send(std::string const& str, bool dont_block/* = false */)
 {
-	return send(str, (dont_block) ? socket::dont_wait : socket::normal);
+	message_t msg(str);
+	return send(msg, (dont_block) ? socket::dont_wait : socket::normal);
 }
 
 bool socket::receive(std::string &str, bool dont_block /* = false */)
 {
-	// Unable to use message wrapper as this could be multipart legacy fallback
-	return receive(str, (dont_block) ? socket::dont_wait : socket::normal);
+	message msg;
+
+    bool ret = receive(msg, (dont_block) ? socket::dont_wait : socket::normal);
+    if (ret)
+    {
+        msg.get(str, 0);
+    }
+
+    return ret;
 }
 
 bool socket::send(zmqpp::signal sig, bool dont_block/* = false */)