Updated mutithreaded example

Grzegorz Jabłoński
1 parent 3a0a35ab
Showing 1 changed file with 13 additions and 6 deletions
13-daytime8.cpp
@@ -50,9 +50,12 @@ public:
 private:
   tcp_connection(asio::io_context &io_context) : socket_(io_context) {}
-  void handle_write(const asio::error_code & /*error*/,
-                    size_t /*bytes_transferred*/) {
-    std::cout << "Written!" << std::endl;
+  void handle_write(const asio::error_code &error, size_t bytes_transferred) {
+    std::cout << error.category().name() << " : " << error.value() << " : " << error.message() << std::endl;
+    if (error)
+      std::cout << "Error" << std::endl;
+    else
+      std::cout << "Written " << bytes_transferred << " bytes" << std::endl;
   }
   tcp::socket socket_;
@@ -90,9 +93,7 @@ private:
   tcp::acceptor acceptor_;
 };
-void WorkerThread(std::shared_ptr<asio::io_context> io_svc) {
-  io_svc->run();
-}
+void WorkerThread(std::shared_ptr<asio::io_context> io_svc) { io_svc->run(); }
 int main() {
   try {
@@ -107,6 +108,12 @@ int main() {
     tcp_server server(*io_context);
+    std::cout << "Press ENTER key to exit!" << std::endl;
+
+    std::cin.get();
+
+    io_context->stop();
+
     for (auto &i : threads)
       i.join();