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
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
|
#include "server_ws.hpp"
#include "client_ws.hpp"
using namespace std;
typedef SimpleWeb::SocketServer<SimpleWeb::WS> WsServer;
typedef SimpleWeb::SocketClient<SimpleWeb::WS> WsClient;
int main() {
//WebSocket (WS)-server at port 8080 using 1 thread
WsServer server;
server.config.port=8080;
//Example 1: echo WebSocket endpoint
// Added debug messages for example use of the callbacks
// Test with the following JavaScript:
// var ws=new WebSocket("ws://localhost:8080/echo");
// ws.onmessage=function(evt){console.log(evt.data);};
// ws.send("test");
auto& echo=server.endpoint["^/echo/?$"];
echo.on_message=[&server](shared_ptr<WsServer::Connection> connection, shared_ptr<WsServer::Message> message) {
//WsServer::Message::string() is a convenience function for:
//stringstream data_ss;
//data_ss << message->rdbuf();
//auto message_str = data_ss.str();
auto message_str=message->string();
cout << "Server: Message received: \"" << message_str << "\" from " << (size_t)connection.get() << endl;
cout << "Server: Sending message \"" << message_str << "\" to " << (size_t)connection.get() << endl;
auto send_stream=make_shared<WsServer::SendStream>();
*send_stream << message_str;
//server.send is an asynchronous function
server.send(connection, send_stream, [](const boost::system::error_code& ec){
if(ec) {
cout << "Server: Error sending message. " <<
//See http://www.boost.org/doc/libs/1_55_0/doc/html/boost_asio/reference.html, Error Codes for error code meanings
"Error: " << ec << ", error message: " << ec.message() << endl;
}
});
};
echo.on_open=[](shared_ptr<WsServer::Connection> connection) {
cout << "Server: Opened connection " << (size_t)connection.get() << endl;
};
//See RFC 6455 7.4.1. for status codes
echo.on_close=[](shared_ptr<WsServer::Connection> connection, int status, const string& /*reason*/) {
cout << "Server: Closed connection " << (size_t)connection.get() << " with status code " << status << endl;
};
//See http://www.boost.org/doc/libs/1_55_0/doc/html/boost_asio/reference.html, Error Codes for error code meanings
echo.on_error=[](shared_ptr<WsServer::Connection> connection, const boost::system::error_code& ec) {
cout << "Server: Error in connection " << (size_t)connection.get() << ". " <<
"Error: " << ec << ", error message: " << ec.message() << endl;
};
//Example 2: Echo thrice
// Demonstrating queuing of messages by sending a received message three times back to the client.
// send_stream2 is automatically queued by the library after the first send call,
// and send_stream3 is queued after the first send call through its callback
// Test with the following JavaScript:
// var ws=new WebSocket("ws://localhost:8080/echo_thrice");
// ws.onmessage=function(evt){console.log(evt.data);};
// ws.send("test");
auto& echo_thrice=server.endpoint["^/echo_thrice/?$"];
echo_thrice.on_message=[&server](shared_ptr<WsServer::Connection> connection, shared_ptr<WsServer::Message> message) {
auto message_str=message->string();
auto send_stream1=make_shared<WsServer::SendStream>();
*send_stream1 << message_str;
//server.send is an asynchronous function
server.send(connection, send_stream1, [&server, connection, message_str](const boost::system::error_code& ec) {
if(!ec) {
auto send_stream3=make_shared<WsServer::SendStream>();
*send_stream3 << message_str;
server.send(connection, send_stream3); //Sent after send_stream1 is sent, and most likely after send_stream2
}
});
//Do not reuse send_stream1 here as it most likely is not sent yet
auto send_stream2=make_shared<WsServer::SendStream>();
*send_stream2 << message_str;
server.send(connection, send_stream2); //Most likely queued, and sent after send_stream1
};
//Example 3: Echo to all WebSocket endpoints
// Sending received messages to all connected clients
// Test with the following JavaScript on more than one browser windows:
// var ws=new WebSocket("ws://localhost:8080/echo_all");
// ws.onmessage=function(evt){console.log(evt.data);};
// ws.send("test");
auto& echo_all=server.endpoint["^/echo_all/?$"];
echo_all.on_message=[&server](shared_ptr<WsServer::Connection> /*connection*/, shared_ptr<WsServer::Message> message) {
auto message_str=message->string();
//echo_all.get_connections() can also be used to solely receive connections on this endpoint
for(auto &a_connection: server.get_connections()) {
auto send_stream=make_shared<WsServer::SendStream>();
*send_stream << message_str;
//server.send is an asynchronous function
server.send(a_connection, send_stream);
}
};
thread server_thread([&server](){
//Start WS-server
server.start();
});
//Wait for server to start so that the client can connect
this_thread::sleep_for(chrono::seconds(1));
//Example 4: Client communication with server
//Possible output:
//Server: Opened connection 140184920260656
//Client: Opened connection
//Client: Sending message: "Hello"
//Server: Message received: "Hello" from 140184920260656
//Server: Sending message "Hello" to 140184920260656
//Client: Message received: "Hello"
//Client: Sending close connection
//Server: Closed connection 140184920260656 with status code 1000
//Client: Closed connection with status code 1000
WsClient client("localhost:8080/echo");
client.on_message=[&client](shared_ptr<WsClient::Message> message) {
auto message_str=message->string();
cout << "Client: Message received: \"" << message_str << "\"" << endl;
cout << "Client: Sending close connection" << endl;
client.send_close(1000);
};
client.on_open=[&client]() {
cout << "Client: Opened connection" << endl;
string message="Hello";
cout << "Client: Sending message: \"" << message << "\"" << endl;
auto send_stream=make_shared<WsClient::SendStream>();
*send_stream << message;
client.send(send_stream);
};
client.on_close=[](int status, const string& /*reason*/) {
cout << "Client: Closed connection with status code " << status << endl;
};
//See http://www.boost.org/doc/libs/1_55_0/doc/html/boost_asio/reference.html, Error Codes for error code meanings
client.on_error=[](const boost::system::error_code& ec) {
cout << "Client: Error: " << ec << ", error message: " << ec.message() << endl;
};
client.start();
server_thread.join();
return 0;
}
|
