59 lines
1.9 KiB
C++
59 lines
1.9 KiB
C++
#include "App.h"
|
|
#include <thread>
|
|
#include <algorithm>
|
|
|
|
int main() {
|
|
/* ws->getUserData returns one of these */
|
|
struct PerSocketData {
|
|
|
|
};
|
|
|
|
/* Simple echo websocket server, using multiple threads */
|
|
std::vector<std::thread *> threads(std::thread::hardware_concurrency());
|
|
|
|
std::transform(threads.begin(), threads.end(), threads.begin(), [](std::thread */*t*/) {
|
|
return new std::thread([]() {
|
|
|
|
/* Very simple WebSocket echo server */
|
|
uWS::App().ws<PerSocketData>("/*", {
|
|
/* Settings */
|
|
.compression = uWS::SHARED_COMPRESSOR,
|
|
.maxPayloadLength = 16 * 1024,
|
|
.idleTimeout = 10,
|
|
.maxBackpressure = 1 * 1024 * 1024,
|
|
/* Handlers */
|
|
.upgrade = nullptr,
|
|
.open = [](auto */*ws*/) {
|
|
|
|
},
|
|
.message = [](auto *ws, std::string_view message, uWS::OpCode opCode) {
|
|
ws->send(message, opCode);
|
|
},
|
|
.drain = [](auto */*ws*/) {
|
|
/* Check getBufferedAmount here */
|
|
},
|
|
.ping = [](auto */*ws*/, std::string_view) {
|
|
|
|
},
|
|
.pong = [](auto */*ws*/, std::string_view) {
|
|
|
|
},
|
|
.close = [](auto */*ws*/, int /*code*/, std::string_view /*message*/) {
|
|
|
|
}
|
|
}).listen(9001, [](auto *listen_socket) {
|
|
if (listen_socket) {
|
|
std::cout << "Thread " << std::this_thread::get_id() << " listening on port " << 9001 << std::endl;
|
|
} else {
|
|
std::cout << "Thread " << std::this_thread::get_id() << " failed to listen on port 9001" << std::endl;
|
|
}
|
|
}).run();
|
|
|
|
});
|
|
});
|
|
|
|
std::for_each(threads.begin(), threads.end(), [](std::thread *t) {
|
|
t->join();
|
|
});
|
|
}
|