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
|
#pragma once
#include <boost/asio.hpp>
#include <string>
#include <boost/beast/core/tcp_stream.hpp>
#include <boost/beast/websocket/stream.hpp>
#include "RemoteEndPoint.h"
#include "stream.h"
#include "threaded_queue.h"
namespace lsp
{
class Log;
}
namespace lsp
{
class websocket_stream_wrapper : public istream, public ostream
{
public:
websocket_stream_wrapper(boost::beast::websocket::stream<boost::beast::tcp_stream>& _w);
boost::beast::websocket::stream<boost::beast::tcp_stream>& ws_;
std::atomic<bool> quit {};
std::shared_ptr<MultiQueueWaiter> request_waiter;
ThreadedQueue<char> on_request;
std::string error_message;
bool fail() override;
bool eof() override;
bool good() override;
websocket_stream_wrapper& read(char* str, std::streamsize count) override;
int get() override;
bool bad() override;
websocket_stream_wrapper& write(std::string const& c) override;
websocket_stream_wrapper& write(std::streamsize _s) override;
websocket_stream_wrapper& flush() override;
void clear() override;
std::string what() override;
};
/// The top-level class of the HTTP server.
class WebSocketServer
{
public:
WebSocketServer(WebSocketServer const&) = delete;
WebSocketServer& operator=(WebSocketServer const&) = delete;
~WebSocketServer();
/// Construct the server to listen on the specified TCP address and port, and
/// serve up files from the given directory.
explicit WebSocketServer(
std::string const& user_agent, std::string const& address, std::string const& port,
std::shared_ptr<MessageJsonHandler> json_handler, std::shared_ptr<Endpoint> localEndPoint, lsp::Log&,
uint32_t _max_workers = 2
);
/// Run the server's io_context loop.
void run();
void stop();
RemoteEndPoint point;
private:
struct Data;
/// Perform an asynchronous accept operation.
void do_accept();
/// Wait for a request to stop the server.
void do_stop();
Data* d_ptr = nullptr;
};
} // namespace lsp
|
