Replaced Client based implementation to composition instead of inheritance.

1 files changed, 11 insertions, 1 deletions

diff --git a/network-handler/nodeclient.cpp b/network-handler/nodeclient.cpp
index 8486f4b..9b026ba 100644
--- a/network-handler/nodeclient.cpp
+++ b/network-handler/nodeclient.cpp
@@ -3,10 +3,20 @@
 #include <iostream>
 
 NodeClient::NodeClient(boost::asio::ip::tcp::socket &&socket)
-: Client(std::move(socket))
+: client(std::move(socket))
 {}
 
 void NodeClient::handle_message(std::vector<uint8_t> message)
 {
 	std::cout << std::string(message.begin(), message.end()) << std::endl;
 }
+
+void NodeClient::receive()
+{
+	client.receive(std::bind(&NodeClient::handle_message, this, std::placeholders::_1));
+}
+
+void NodeClient::on_done(Client::OnDoneFT done) {
+	client.on_done(done);
+}
+