Switch to message buffer / dispatcher for websocket messages

plugins/base/utils/connection-manager.cpp

@@ -199,6 +199,11 @@ void Connection::Save(obs_data_t *obj) const
 	obs_data_set_int(obj, "version", 1);
 }
 
+WebsocketMessageBuffer Connection::RegisterForEvents()
+{
+	return _client.RegisterForEvents();
+}
+
 void Connection::UseOBSWebsocketProtocol(bool useOBSWSProtocol)
 {
 	_useOBSWSProtocol = useOBSWSProtocol;

plugins/base/utils/connection-manager.hpp

@@ -40,9 +40,9 @@ public:
 	void SendMsg(const std::string &msg);
 	void Load(obs_data_t *obj);
 	void Save(obs_data_t *obj) const;
-	std::string GetName() { return _name; }
-	std::vector<WSMessage> &Events() { return _client.Events(); }
-	bool IsUsingOBSProtocol() { return _useOBSWSProtocol; }
+	std::string GetName() const { return _name; }
+	WebsocketMessageBuffer RegisterForEvents();
+	bool IsUsingOBSProtocol() const { return _useOBSWSProtocol; }
 
 private:
 	void UseOBSWebsocketProtocol(bool);

plugins/base/utils/websocket-helpers.cpp

@@ -14,14 +14,14 @@ using websocketpp::lib::placeholders::_2;
 using websocketpp::lib::bind;
 
 #define RPC_VERSION 1
+#undef DispatchMessage
 
 constexpr char VendorName[] = "AdvancedSceneSwitcher";
 constexpr char VendorRequest[] = "AdvancedSceneSwitcherMessage";
 constexpr char VendorEvent[] = "AdvancedSceneSwitcherEvent";
 obs_websocket_vendor vendor;
 
-static void clearWebsocketMessages();
-static std::vector<WSMessage> websocketMessages;
+static WebsocketMessageDispatcher websocketMessageDispatcher;
 static void registerWebsocketVendor();
 
 static bool setup();
@@ -29,36 +29,13 @@ static bool setupDone = setup();
 
 bool setup()
 {
-	AddIntervalResetStep(clearWebsocketMessages);
 	AddPluginPostLoadStep(registerWebsocketVendor);
 	return true;
 }
 
-std::vector<WSMessage> &GetWebsocketMessages()
+WebsocketMessageBuffer RegisterForWebsocketMessages()
 {
-	return websocketMessages;
-}
-
-static void clearWebsocketMessages()
-{
-	websocketMessages.erase(std::remove_if(websocketMessages.begin(),
-					       websocketMessages.end(),
-					       [](const WSMessage &message) {
-						       return message.processed;
-					       }),
-				websocketMessages.end());
-	for (auto &connection : GetConnections()) {
-		auto c = dynamic_cast<Connection *>(connection.get());
-		if (!c) {
-			continue;
-		}
-		auto &messages = c->Events();
-		messages.erase(std::remove_if(messages.begin(), messages.end(),
-					      [](const WSMessage &message) {
-						      return message.processed;
-					      }),
-			       messages.end());
-	}
+	return websocketMessageDispatcher.RegisterClient();
 }
 
 void SendWebsocketEvent(const std::string &eventMsg)
@@ -79,8 +56,7 @@ static void receiveWebsocketMessage(obs_data_t *request_data, obs_data_t *,
 	}
 
 	auto msg = obs_data_get_string(request_data, "message");
-	auto lock = LockContext();
-	websocketMessages.emplace_back(msg);
+	websocketMessageDispatcher.DispatchMessage(msg);
 	vblog(LOG_INFO, "received message: %s", msg);
 }
 
@@ -247,6 +223,11 @@ void WSConnection::SendRequest(const std::string &msg)
 	Send(msg);
 }
 
+WebsocketMessageBuffer WSConnection::RegisterForEvents()
+{
+	return _dispatcher.RegisterClient();
+}
+
 WSConnection::Status WSConnection::GetStatus() const
 {
 	return _status;
@@ -329,8 +310,8 @@ void WSConnection::HandleEvent(obs_data_t *msg)
 		return;
 	}
 	auto eventDataNested = obs_data_get_obj(eventData, "eventData");
-	auto lock = LockContext();
-	_messages.emplace_back(obs_data_get_string(eventDataNested, "message"));
+	_dispatcher.DispatchMessage(
+		obs_data_get_string(eventDataNested, "message"));
 	vblog(LOG_INFO, "received event msg \"%s\"",
 	      obs_data_get_string(eventDataNested, "message"));
 	obs_data_release(eventDataNested);
@@ -361,9 +342,8 @@ void WSConnection::OnGenericMessage(connection_hdl, client::message_ptr message)
 		return;
 	}
 
-	auto lock = LockContext();
 	const auto payload = message->get_payload();
-	_messages.emplace_back(payload);
+	_dispatcher.DispatchMessage(payload);
 	vblog(LOG_INFO, "received event msg \"%s\"", payload.c_str());
 }
 

plugins/base/utils/websocket-helpers.hpp

@@ -1,4 +1,6 @@
 #pragma once
+#include "message-buffer.hpp"
+#include "message-dispatcher.hpp"
 
 #include <set>
 #include <QtCore/QObject>
@@ -21,16 +23,12 @@
 namespace advss {
 
 using websocketpp::connection_hdl;
-
-struct WSMessage {
-	WSMessage(const std::string &m) : message(m) {}
-	std::string message = "";
-	bool processed = false;
-};
+using WebsocketMessageBuffer = std::shared_ptr<MessageBuffer<std::string>>;
+using WebsocketMessageDispatcher = MessageDispatcher<std::string>;
 
 void SendWebsocketEvent(const std::string &);
 std::string ConstructVendorRequestMessage(const std::string &message);
-std::vector<WSMessage> &GetWebsocketMessages();
+[[nodiscard]] WebsocketMessageBuffer RegisterForWebsocketMessages();
 
 class WSConnection : public QObject {
 	using server = websocketpp::server<websocketpp::config::asio>;
@@ -44,7 +42,7 @@ public:
 		     bool _reconnect, int reconnectDelay = 10);
 	void Disconnect();
 	void SendRequest(const std::string &msg);
-	std::vector<WSMessage> &Events() { return _messages; }
+	[[nodiscard]] WebsocketMessageBuffer RegisterForEvents();
 	std::string GetFail() { return _failMsg; }
 
 	enum class Status {
@@ -82,7 +80,7 @@ private:
 	std::atomic<Status> _status = {Status::DISCONNECTED};
 	std::atomic_bool _disconnect{false};
 
-	std::vector<WSMessage> _messages;
+	WebsocketMessageDispatcher _dispatcher;
 };
 
 } // namespace advss