Multithreaded server

Grzegorz Jabłoński
1 parent f68ef015
Showing 3 changed files with 157 additions and 1 deletions
12-threads.cpp
13-daytime8.cpp
makefile
+/* multithreads.cpp */
+#include <asio/ts/io_context.hpp>
+#include <asio/ts/net.hpp>
+#include <asio/ts/socket.hpp>
+
+#include <iostream>
+
+int a = 0;
+
+void WorkerThread(std::shared_ptr<asio::io_context> io_svc, int counter) {
+  std::cout << counter << ".\n";
+  io_svc->run();
+  std::cout << "End.\n";
+}
+
+int main(void) {
+
+  auto io_svc = std::make_shared<asio::io_context>();
+
+  auto worker = std::make_shared<asio::io_context::work>(*io_svc);
+
+  std::cout << "Press ENTER key to exit!" << std::endl;
+
+  std::vector<std::thread> threads;
+  for (int i = 0; i < 5; i++)
+    threads.emplace_back(WorkerThread, io_svc, i);
+
+  io_svc->post([] { std::cout << "Hello\n"; });
+
+  std::cin.get();
+
+  io_svc->stop();
+
+  for (auto &i : threads)
+    i.join();
+
+  return 0;
+}
 \ No newline at end of file
+//
+// server.cpp
+// ~~~~~~~~~~
+//
+// Copyright (c) 2003-2020 Christopher M. Kohlhoff (chris at kohlhoff dot com)
+//
+// Distributed under the Boost Software License, Version 1.0. (See accompanying
+// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
+//
+
+#include <asio.hpp>
+#include <ctime>
+#include <iostream>
+#include <string>
+
+using asio::ip::tcp;
+
+std::shared_ptr<asio::io_context> io_context;
+
+std::string make_daytime_string() {
+  using namespace std; // For time_t, time and ctime;
+  time_t now = time(0);
+
+  sleep(10);
+
+  char buf[26];
+  return ctime_r(&now, buf);
+}
+
+class tcp_connection : public std::enable_shared_from_this<tcp_connection> {
+public:
+  typedef std::shared_ptr<tcp_connection> pointer;
+
+  static pointer create(asio::io_context &io_context) {
+    return pointer(new tcp_connection(io_context));
+  }
+
+  tcp::socket &socket() { return socket_; }
+
+  void start() {
+    io_context->post([self = shared_from_this()] {
+      self->message_ = make_daytime_string();
+      asio::async_write(self->socket_, asio::buffer(self->message_),
+                        std::bind(&tcp_connection::handle_write, self,
+                                  std::placeholders::_1,
+                                  std::placeholders::_2));
+    });
+  }
+
+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;
+  }
+
+  tcp::socket socket_;
+  std::string message_;
+};
+
+class tcp_server {
+public:
+  tcp_server(asio::io_context &io_context)
+      : io_context_(io_context),
+        acceptor_(io_context, tcp::endpoint(tcp::v4(), 13)) {
+    start_accept();
+  }
+
+private:
+  void start_accept() {
+    tcp_connection::pointer new_connection =
+        tcp_connection::create(io_context_);
+
+    acceptor_.async_accept(new_connection->socket(),
+                           std::bind(&tcp_server::handle_accept, this,
+                                     new_connection, std::placeholders::_1));
+  }
+
+  void handle_accept(tcp_connection::pointer new_connection,
+                     const asio::error_code &error) {
+    if (!error) {
+      new_connection->start();
+    }
+
+    start_accept();
+  }
+
+  asio::io_context &io_context_;
+  tcp::acceptor acceptor_;
+};
+
+void WorkerThread(std::shared_ptr<asio::io_context> io_svc) {
+  io_svc->run();
+}
+
+int main() {
+  try {
+    io_context = std::make_shared<asio::io_context>();
+
+    auto worker = std::make_shared<asio::io_context::work>(*io_context);
+
+    std::vector<std::thread> threads;
+
+    for (int i = 0; i < 5; i++)
+      threads.emplace_back(WorkerThread, io_context);
+
+    tcp_server server(*io_context);
+
+    for (auto &i : threads)
+      i.join();
+
+  } catch (std::exception &e) {
+    std::cerr << e.what() << std::endl;
+  }
+
+  return 0;
+}
@@ -4,7 +4,7 @@ CFLAGS=-ggdb -Wall -pedantic -D_REENTRANT -DASIO_STANDALONE -DASIO_HAS_CO_AWAIT 
 %: %.cpp
 	g++-10 -std=c++20 -fcoroutines $(CFLAGS) $< -o $@ -lpthread
  
-EXECS = 00-timer1 01-timer2 02-timer3 03-timer4 04-timer5 05-daytime1 06-daytime2 07-daytime3 08-daytime4 09-daytime5 10-daytime6 11-daytime7
+EXECS = 00-timer1 01-timer2 02-timer3 03-timer4 04-timer5 05-daytime1 06-daytime2 07-daytime3 08-daytime4 09-daytime5 10-daytime6 11-daytime7 12-threads 13-daytime8
  
 all: $(EXECS)