14-daytime8.cpp
3.03 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
//
// 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::string make_daytime_string() {
using namespace std; // For time_t, time and ctime;
time_t now = time(0);
sleep(5);
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() {
message_ = make_daytime_string();
asio::async_write(socket_, asio::buffer(message_),
std::bind(&tcp_connection::handle_write, shared_from_this(),
std::placeholders::_1, std::placeholders::_2));
}
~tcp_connection()
{
std::cout << "~tcp_connection()" << std::endl;
}
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 << 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_;
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) {
io_context_.post(std::bind(&tcp_connection::start, new_connection));
}
start_accept();
}
asio::io_context &io_context_;
tcp::acceptor acceptor_;
};
int main() {
try {
asio::io_context io_context;
asio::executor_work_guard work_guard(io_context.get_executor()); //prevent run() from exiting immediately after all work is done
std::vector<std::thread> threads;
for (int i = 0; i < 5; i++)
threads.emplace_back([&io_context]{io_context.run();});
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();
} catch (std::exception &e) {
std::cerr << e.what() << std::endl;
}
return 0;
}