1 files changed, 5 insertions, 15 deletions
diff --git a/node/nextnode.cpp b/node/nextnode.cpp
index d38200d..f6c4454 100644
--- a/node/nextnode.cpp
+++ b/node/nextnode.cpp
@@ -1,34 +1,24 @@
 #include "nextnode.hpp"
 
-#include "connect.hpp"
-
 #include "logging.hpp"
 
 using namespace boost::asio::ip;
 
 NextNode::NextNode(tcp::socket&& socket)
-: Client(std::move(socket))
+: client(std::move(socket))
 {}
 
 void NextNode::send(std::string message)
 {
-	auto handler = [](boost::system::error_code const& ec, std::size_t bytes_transferred) {
-		BOOST_LOG_TRIVIAL(trace) << "sent message";
-		if(ec) {
-			BOOST_LOG_TRIVIAL(fatal) << ec;
-			throw std::runtime_error("unable to send message");
-		}
-	};
-
-	socket.async_send(boost::asio::buffer(message), 0, handler);
+	client.send(message);
 }
 
-void NextNode::connect(std::string next_host, std::string next_port, std::function<void ()> on_connect)
+void NextNode::async_connect(std::string next_host, std::string next_port, std::function<void ()> on_connect)
 {
-	async_connect(socket, next_host, next_port, on_connect);
+	client.async_connect(next_host, next_port, on_connect);
 }
 
 void NextNode::close()
 {
-	socket.close();
+	client.close();
 }