blob: 5206525cebea1f8d51f2ee6ed7b27a7e1d9346cb (
plain)
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
|
#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(const std::string& 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(const WebSocketServer&) = delete;
WebSocketServer& operator=(const WebSocketServer&) = delete;
~WebSocketServer();
/// Construct the server to listen on the specified TCP address and port, and
/// serve up files from the given directory.
explicit WebSocketServer(const std::string& user_agent, const std::string& address, const std::string& 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
|