123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384 |
- //
- // Copyright (c) 2016-2019 Vinnie Falco (vinnie dot falco at gmail 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)
- //
- // Official repository: https://github.com/vinniefalco/CppCon2018
- //
- //------------------------------------------------------------------------------
- /*
- WebSocket chat server, multi-threaded
- This implements a multi-user chat room using WebSocket. The
- `io_context` runs on any number of threads, specified at
- the command line.
- */
- //------------------------------------------------------------------------------
- #include "listener.hpp"
- #include "shared_state.hpp"
- #include <boost/asio/signal_set.hpp>
- #include <boost/smart_ptr.hpp>
- #include <iostream>
- #include <vector>
- int
- main(int argc, char* argv[])
- {
- // Check command line arguments.
- if (argc != 5)
- {
- std::cerr <<
- "Usage: websocket-chat-multi <address> <port> <doc_root> <threads>\n" <<
- "Example:\n" <<
- " websocket-chat-server 0.0.0.0 8080 . 5\n";
- return EXIT_FAILURE;
- }
- auto address = net::ip::make_address(argv[1]);
- auto port = static_cast<unsigned short>(std::atoi(argv[2]));
- auto doc_root = argv[3];
- auto const threads = std::max<int>(1, std::atoi(argv[4]));
- // The io_context is required for all I/O
- net::io_context ioc;
- // Create and launch a listening port
- boost::make_shared<listener>(
- ioc,
- tcp::endpoint{address, port},
- boost::make_shared<shared_state>(doc_root))->run();
- // Capture SIGINT and SIGTERM to perform a clean shutdown
- net::signal_set signals(ioc, SIGINT, SIGTERM);
- signals.async_wait(
- [&ioc](boost::system::error_code const&, int)
- {
- // Stop the io_context. This will cause run()
- // to return immediately, eventually destroying the
- // io_context and any remaining handlers in it.
- ioc.stop();
- });
- // Run the I/O service on the requested number of threads
- std::vector<std::thread> v;
- v.reserve(threads - 1);
- for(auto i = threads - 1; i > 0; --i)
- v.emplace_back(
- [&ioc]
- {
- ioc.run();
- });
- ioc.run();
- // (If we get here, it means we got a SIGINT or SIGTERM)
- // Block until all the threads exit
- for(auto& t : v)
- t.join();
- return EXIT_SUCCESS;
- }
|