adding in mutex for local store of messages

This commit is contained in:
talksik
2024-01-27 13:16:55 -08:00
parent b5039f7c43
commit df816494f4
2 changed files with 22 additions and 2 deletions
+5
View File
@@ -3,5 +3,10 @@
"directory": "/home/talksik/code/homechat/server/build",
"command": "/usr/bin/c++ -I/home/talksik/code/homechat/server/src/include -g -o CMakeFiles/HomechatServer.dir/src/Main.cpp.o -c /home/talksik/code/homechat/server/src/Main.cpp",
"file": "/home/talksik/code/homechat/server/src/Main.cpp"
},
{
"directory": "/home/talksik/code/homechat/server/build",
"command": "/usr/bin/c++ -I/home/talksik/code/homechat/server/src/include -g -o CMakeFiles/HomechatServer.dir/src/MessageCache.cpp.o -c /home/talksik/code/homechat/server/src/MessageCache.cpp",
"file": "/home/talksik/code/homechat/server/src/MessageCache.cpp"
}
]
+17 -2
View File
@@ -12,6 +12,7 @@ int main()
{
std::cout << "Welcome to the homechat server!\n";
std::mutex message_cache_mutex;
std::unique_ptr<MessageCache> message_cache = std::make_unique<MessageCache>();
/* ws->getUserData returns one of these */
@@ -39,7 +40,7 @@ int main()
/* Handlers */
.upgrade = nullptr,
.open =
[](uWS::WebSocket<false, true, PerSocketData> *ws) {
[&](uWS::WebSocket<false, true, PerSocketData> *ws) {
auto user_data = ws->getUserData();
user_data->created_at = DateTime::now();
@@ -49,15 +50,29 @@ int main()
std::cout << "Failed to subscribe to * channel\n";
ws->close();
}
std::lock_guard<std::mutex> lock(message_cache_mutex);
auto messages = message_cache->last_ten_messages();
ws->send(std::string("Welcome to the homechat server! Here are the last 10 messages: \n"), uWS::OpCode::TEXT, false);
for (auto &message : messages)
{
ws->send(message, uWS::OpCode::TEXT, false);
}
},
.message =
[](uWS::WebSocket<false, true, PerSocketData> *ws, std::string_view message, uWS::OpCode opCode) {
[&](uWS::WebSocket<false, true, PerSocketData> *ws, std::string_view message, uWS::OpCode opCode) {
/* This is the opposite of what you probably want; compress if message is LARGER than 16 kb
* the reason we do the opposite here; compress if SMALLER than 16 kb is to allow for
* benchmarking of large message sending without compression */
std::cout << "Message received: " << message << std::endl;
ws->send(message, opCode, false);
ws->publish("*", message, opCode, false);
std::lock_guard<std::mutex> lock(message_cache_mutex);
if (!message_cache->add_message(std::string(message)))
{
std::cerr << "Failed to add message to cache" << std::endl;
}
},
.dropped =
[](auto * /*ws*/, std::string_view /*message*/, uWS::OpCode /*opCode*/) {