120 lines
3.5 KiB
C++
120 lines
3.5 KiB
C++
#include "wsserver.hpp"
|
|
|
|
#include <crow.h>
|
|
|
|
#include "utils/allocnum.hpp"
|
|
|
|
sv::WSServer::WSServer(uint16_t port)
|
|
{
|
|
ws_thread_ = std::make_unique<std::thread>([this, port]() {
|
|
crow::SimpleApp app;
|
|
{
|
|
std::lock_guard<std::mutex> lock(mtx_);
|
|
|
|
app_ptr_ = (void*)&app;
|
|
|
|
CROW_WEBSOCKET_ROUTE(app, "/ws")
|
|
.onopen([&](crow::websocket::connection& conn) {
|
|
CROW_LOG_INFO << "new websocket connection from " << conn.get_remote_ip();
|
|
|
|
std::unique_lock<std::mutex> lock(mtx_);
|
|
conn.set_nodelay();
|
|
|
|
WSConnId conn_id = utils::AllocNum(id2conn_, last_id_);
|
|
|
|
// register connection
|
|
id2conn_[conn_id] = &conn;
|
|
conn.userdata(reinterpret_cast<void*>(static_cast<uintptr_t>(conn_id)));
|
|
// push connection event
|
|
PushEvent(lock, {WSE_CONNECTED, conn_id});
|
|
})
|
|
.onclose([&](crow::websocket::connection& conn, const std::string& reason, uint16_t) {
|
|
CROW_LOG_INFO << "websocket connection closed: " << reason;
|
|
|
|
std::unique_lock<std::mutex> lock(mtx_);
|
|
|
|
WSConnId conn_id = static_cast<uint32_t>(reinterpret_cast<uintptr_t>(conn.userdata()));
|
|
|
|
// push disonnected event
|
|
PushEvent(lock, {WSE_DISCONNECTED, conn_id});
|
|
|
|
id2conn_.erase(conn_id);
|
|
})
|
|
.onmessage([&](crow::websocket::connection& conn, const std::string& data, bool is_binary) {
|
|
if (!is_binary)
|
|
return; // only accept binary messages here
|
|
|
|
std::unique_lock<std::mutex> lock(mtx_);
|
|
|
|
WSConnId conn_id = static_cast<uint32_t>(reinterpret_cast<uintptr_t>(conn.userdata()));
|
|
PushEvent(lock, {WSE_MESSAGE, conn_id, data});
|
|
});
|
|
|
|
// CROW_ROUTE(app, "/")
|
|
// ([] {
|
|
// crow::mustache::context x;
|
|
// x["servername"] = "127.0.0.1";
|
|
// auto page = crow::mustache::load("ws.html");
|
|
// return page.render(x);
|
|
// });
|
|
}
|
|
|
|
app.port(port).run();
|
|
|
|
std::lock_guard<std::mutex> lock(mtx_);
|
|
|
|
// push exit event
|
|
events_.emplace_back(WSE_EXIT);
|
|
|
|
app_ptr_ = nullptr;
|
|
});
|
|
}
|
|
|
|
bool sv::WSServer::PollEvent(WSEvent& out_event)
|
|
{
|
|
std::lock_guard<std::mutex> lock(mtx_);
|
|
|
|
if (events_.empty())
|
|
return false;
|
|
|
|
out_event = std::move(events_.front());
|
|
events_.pop_front();
|
|
not_full_.notify_one();
|
|
return true;
|
|
}
|
|
|
|
void sv::WSServer::Send(WSConnId conn_id, std::string data)
|
|
{
|
|
std::lock_guard<std::mutex> lock(mtx_);
|
|
|
|
auto it = id2conn_.find(conn_id);
|
|
if (it == id2conn_.end())
|
|
{
|
|
std::cerr << "attempted to send message to unknown conn ID " << conn_id << std::endl;
|
|
return;
|
|
}
|
|
|
|
(*it->second).send_binary(std::move(data));
|
|
}
|
|
|
|
void sv::WSServer::Exit()
|
|
{
|
|
std::lock_guard<std::mutex> lock(mtx_);
|
|
if (app_ptr_)
|
|
((crow::SimpleApp*)app_ptr_)->stop();
|
|
}
|
|
|
|
sv::WSServer::~WSServer()
|
|
{
|
|
Exit();
|
|
if (ws_thread_ && ws_thread_->joinable())
|
|
ws_thread_->join();
|
|
}
|
|
|
|
void sv::WSServer::PushEvent(std::unique_lock<std::mutex>& lock, const WSEvent& event)
|
|
{
|
|
not_full_.wait(lock, [this] { return events_.size() < MAX_QUEUE_SIZE; });
|
|
events_.emplace_back(event);
|
|
|
|
}
|