New subscription manager (#1071)

Fix #886
2025-11-26 22:55:53 +00:00 · 2024-01-08 14:45:57 +00:00
parent 07bd4b0760
commit eb1831c489
37 changed files with 4427 additions and 2098 deletions
--- a/src/feed/impl/SingleFeedBase.cpp
+++ b/src/feed/impl/SingleFeedBase.cpp
@@ -0,0 +1,90 @@
+//------------------------------------------------------------------------------
+/*
+    This file is part of clio: https://github.com/XRPLF/clio
+    Copyright (c) 2024, the clio developers.
+
+    Permission to use, copy, modify, and distribute this software for any
+    purpose with or without fee is hereby granted, provided that the above
+    copyright notice and this permission notice appear in all copies.
+
+    THE  SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
+    WITH  REGARD  TO  THIS  SOFTWARE  INCLUDING  ALL  IMPLIED  WARRANTIES  OF
+    MERCHANTABILITY  AND  FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
+    ANY  SPECIAL,  DIRECT,  INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
+    WHATSOEVER  RESULTING  FROM  LOSS  OF USE, DATA OR PROFITS, WHETHER IN AN
+    ACTION  OF  CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
+    OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
+*/
+//==============================================================================
+
+#include "feed/impl/SingleFeedBase.h"
+
+#include "feed/Types.h"
+#include "feed/impl/TrackableSignal.h"
+#include "feed/impl/Util.h"
+#include "util/log/Logger.h"
+
+#include <boost/asio/io_context.hpp>
+#include <boost/asio/post.hpp>
+#include <boost/asio/strand.hpp>
+
+#include <cstdint>
+#include <memory>
+#include <string>
+#include <utility>
+
+namespace feed::impl {
+
+SingleFeedBase::SingleFeedBase(boost::asio::io_context& ioContext, std::string const& name)
+    : strand_(boost::asio::make_strand(ioContext)), subCount_(getSubscriptionsGaugeInt(name)), name_(name)
+{
+}
+
+void
+SingleFeedBase::sub(SubscriberSharedPtr const& subscriber)
+{
+    auto const weakPtr = std::weak_ptr(subscriber);
+    auto const added = signal_.connectTrackableSlot(subscriber, [weakPtr](std::shared_ptr<std::string> const& msg) {
+        if (auto connectionPtr = weakPtr.lock())
+            connectionPtr->send(msg);
+    });
+
+    if (added) {
+        LOG(logger_.debug()) << subscriber->tag() << "Subscribed " << name_;
+        ++subCount_.get();
+        subscriber->onDisconnect.connect([this](SubscriberPtr connectionDisconnecting) {
+            unsubInternal(connectionDisconnecting);
+        });
+    };
+}
+
+void
+SingleFeedBase::unsub(SubscriberSharedPtr const& subscriber)
+{
+    unsubInternal(subscriber.get());
+}
+
+void
+SingleFeedBase::pub(std::string msg) const
+{
+    boost::asio::post(strand_, [this, msg = std::move(msg)]() mutable {
+        auto const msgPtr = std::make_shared<std::string>(std::move(msg));
+        signal_.emit(msgPtr);
+    });
+}
+
+std::uint64_t
+SingleFeedBase::count() const
+{
+    return subCount_.get().value();
+}
+
+void
+SingleFeedBase::unsubInternal(SubscriberPtr subscriber)
+{
+    if (signal_.disconnect(subscriber)) {
+        LOG(logger_.debug()) << subscriber->tag() << "Unsubscribed " << name_;
+        --subCount_.get();
+    }
+}
+}  // namespace feed::impl