diff options
-rw-r--r-- | server_main.cpp | 13 |
1 files changed, 9 insertions, 4 deletions
diff --git a/server_main.cpp b/server_main.cpp index 8da6477..145f8a2 100644 --- a/server_main.cpp +++ b/server_main.cpp | |||
@@ -22,8 +22,9 @@ using socket_type = websocketpp::server<websocketpp::config::asio>; | |||
22 | 22 | ||
23 | class server { | 23 | class server { |
24 | public: | 24 | public: |
25 | server(const cardset& cards, const imagestore& images, std::mt19937& rng) | 25 | server(const cardset& cards, const imagestore& images, int port, |
26 | : cards_(cards), images_(images), rng_(rng) { | 26 | std::mt19937& rng) |
27 | : cards_(cards), images_(images), port_(port), rng_(rng) { | ||
27 | socket_.init_asio(); | 28 | socket_.init_asio(); |
28 | 29 | ||
29 | socket_.set_message_handler([this](websocketpp::connection_hdl connection, | 30 | socket_.set_message_handler([this](websocketpp::connection_hdl connection, |
@@ -33,11 +34,13 @@ class server { | |||
33 | } | 34 | } |
34 | 35 | ||
35 | void run() { | 36 | void run() { |
36 | socket_.listen(9002); | 37 | socket_.listen(port_); |
37 | socket_.start_accept(); | 38 | socket_.start_accept(); |
38 | 39 | ||
39 | asio::post(std::bind(&server::cleanup_thread, this)); | 40 | asio::post(std::bind(&server::cleanup_thread, this)); |
40 | 41 | ||
42 | std::cout << "Listening on port " << port_ << "..." << std::endl; | ||
43 | |||
41 | socket_.run(); | 44 | socket_.run(); |
42 | } | 45 | } |
43 | 46 | ||
@@ -198,6 +201,7 @@ class server { | |||
198 | 201 | ||
199 | const cardset& cards_; | 202 | const cardset& cards_; |
200 | const imagestore& images_; | 203 | const imagestore& images_; |
204 | const int port_; | ||
201 | 205 | ||
202 | std::mutex rng_mutex_; | 206 | std::mutex rng_mutex_; |
203 | std::mt19937& rng_; | 207 | std::mt19937& rng_; |
@@ -224,7 +228,8 @@ int main(int argc, char** argv) { | |||
224 | 228 | ||
225 | cardset cards(config_data["cards_path"]); | 229 | cardset cards(config_data["cards_path"]); |
226 | imagestore images(config_data["cache_path"]); | 230 | imagestore images(config_data["cache_path"]); |
231 | int port = config_data["port"]; | ||
227 | 232 | ||
228 | server app(cards, images, rng); | 233 | server app(cards, images, port, rng); |
229 | app.run(); | 234 | app.run(); |
230 | } | 235 | } |