Rodeos with Streaming Plugin

.gitmodules

@@ -22,3 +22,6 @@
 [submodule "libraries/fc/include/fc/crypto/webauthn_json"]
 	path = libraries/fc/include/fc/crypto/webauthn_json
 	url = https://github.com/Tencent/rapidjson/
+[submodule "libraries/amqp-cpp"]
+	path = libraries/amqp-cpp
+	url = https://github.com/CopernicaMarketingSoftware/AMQP-CPP

CMakeLists.txt

@@ -222,6 +222,8 @@ configure_file(${CMAKE_SOURCE_DIR}/libraries/rocksdb/LICENSE.Apache
                ${CMAKE_BINARY_DIR}/licenses/eosio/LICENSE.rocksdb COPYONLY)
 configure_file(${CMAKE_SOURCE_DIR}/libraries/rocksdb/LICENSE.leveldb
                ${CMAKE_BINARY_DIR}/licenses/eosio/LICENSE.leveldb COPYONLY)
+configure_file(${CMAKE_SOURCE_DIR}/libraries/amqp-cpp/LICENSE
+               ${CMAKE_BINARY_DIR}/licenses/eosio/LICENSE.amqpcpp COPYONLY)
 
 install(FILES LICENSE DESTINATION ${CMAKE_INSTALL_FULL_DATAROOTDIR}/licenses/eosio/ COMPONENT base)
 install(FILES libraries/softfloat/COPYING.txt DESTINATION ${CMAKE_INSTALL_FULL_DATAROOTDIR}/licenses/eosio/ RENAME LICENSE.softfloat COMPONENT base)
@@ -233,6 +235,7 @@ install(FILES libraries/yubihsm/LICENSE DESTINATION ${CMAKE_INSTALL_FULL_DATAROO
 install(FILES libraries/eos-vm/LICENSE DESTINATION ${CMAKE_INSTALL_FULL_DATAROOTDIR}/licenses/eosio/ RENAME LICENSE.eos-vm COMPONENT base)
 install(FILES libraries/rocksdb/LICENSE.Apache DESTINATION ${CMAKE_INSTALL_FULL_DATAROOTDIR}/licenses/eosio/ RENAME LICENSE.rocksdb COMPONENT base)
 install(FILES libraries/rocksdb/LICENSE.leveldb DESTINATION ${CMAKE_INSTALL_FULL_DATAROOTDIR}/licenses/eosio/ RENAME LICENSE.leveldb COMPONENT base)
+install(FILES libraries/amqp-cpp/LICENSE DESTINATION ${CMAKE_INSTALL_FULL_DATAROOTDIR}/licenses/eosio/ RENAME LICENSE.amqpcpp COMPONENT base)
 
 add_custom_target(base-install
   COMMAND "${CMAKE_COMMAND}" --build "${CMAKE_BINARY_DIR}"

libraries/CMakeLists.txt

@@ -45,3 +45,8 @@ set_property(GLOBAL PROPERTY CTEST_CUSTOM_TESTS_IGNORE
   attest pbkdf2 parsing ${_CTEST_CUSTOM_TESTS_IGNORE}")
 
 add_subdirectory( rodeos )
+
+find_package(OpenSSL REQUIRED)
+option(AMQP-CPP_LINUX_TCP CACHE ON)
+add_subdirectory( amqp-cpp EXCLUDE_FROM_ALL )
+target_include_directories(amqpcpp PRIVATE "${OPENSSL_INCLUDE_DIR}")

programs/rodeos/CMakeLists.txt

@@ -1,9 +1,12 @@
-add_executable( ${RODEOS_EXECUTABLE_NAME} 
+file(GLOB HEADERS *.hpp streams/*.hpp)
+add_executable( ${RODEOS_EXECUTABLE_NAME}
   cloner_plugin.cpp
+  streamer_plugin.cpp
   main.cpp
   rocksdb_plugin.cpp
   wasm_ql_http.cpp
   wasm_ql_plugin.cpp
+  ${HEADERS}
 )
 
 if( UNIX AND NOT APPLE )
@@ -29,11 +32,13 @@ endif()
 
 configure_file(config.hpp.in config.hpp ESCAPE_QUOTES)
 
-target_include_directories(${RODEOS_EXECUTABLE_NAME} PUBLIC ${CMAKE_CURRENT_BINARY_DIR} ${CMAKE_CURRENT_SOURCE_DIR}/../../libraries/abieos/src )
+target_include_directories(${RODEOS_EXECUTABLE_NAME} PUBLIC ${CMAKE_CURRENT_BINARY_DIR}
+  ${CMAKE_CURRENT_SOURCE_DIR}/../../libraries/abieos/src
+  ${CMAKE_CURRENT_SOURCE_DIR}/../../libraries/amqp-cpp/include)
 
 target_link_libraries( ${RODEOS_EXECUTABLE_NAME}
         PRIVATE appbase version
-        PRIVATE rodeos_lib fc ${CMAKE_DL_LIBS} ${PLATFORM_SPECIFIC_LIBS} )
+        PRIVATE rodeos_lib fc amqpcpp ${CMAKE_DL_LIBS} ${PLATFORM_SPECIFIC_LIBS})
 
 copy_bin( ${RODEOS_EXECUTABLE_NAME} )
 install( TARGETS

programs/rodeos/cloner_plugin.cpp

@@ -36,9 +36,10 @@ struct cloner_config : ship_client::connection_config {
 };
 
 struct cloner_plugin_impl : std::enable_shared_from_this<cloner_plugin_impl> {
-   std::shared_ptr<cloner_config>  config = std::make_shared<cloner_config>();
-   std::shared_ptr<cloner_session> session;
-   boost::asio::deadline_timer     timer;
+   std::shared_ptr<cloner_config>                                           config = std::make_shared<cloner_config>();
+   std::shared_ptr<cloner_session>                                          session;
+   boost::asio::deadline_timer                                              timer;
+   std::function<void(const char* data, uint64_t data_size)>                streamer = {};
 
    cloner_plugin_impl() : timer(app().get_io_service()) {}
 
@@ -157,9 +158,13 @@ struct cloner_session : ship_client::connection_callbacks, std::enable_shared_fr
       rodeos_snapshot->write_block_info(result);
       rodeos_snapshot->write_deltas(result, [] { return app().is_quiting(); });
 
-      // todo: remove
-      if (filter)
-         filter->process(*rodeos_snapshot, result, bin, [](const char*, uint64_t) {});
+      if (filter) {
+         filter->process(*rodeos_snapshot, result, bin, [&](const char* data, uint64_t data_size) {
+            if (my->streamer) {
+               my->streamer(data, data_size);
+            }
+         });
+      }
 
       rodeos_snapshot->end_block(result, false);
       return true;
@@ -235,4 +240,8 @@ void cloner_plugin::plugin_shutdown() {
    ilog("cloner_plugin stopped");
 }
 
+void cloner_plugin::set_streamer(std::function<void(const char* data, uint64_t data_size)> streamer_func) {
+   my->streamer = streamer_func;
+}
+
 } // namespace b1

programs/rodeos/cloner_plugin.hpp

@@ -15,6 +15,8 @@ class cloner_plugin : public appbase::plugin<cloner_plugin> {
    void         plugin_startup();
    void         plugin_shutdown();
 
+   void set_streamer(std::function<void(const char* data, uint64_t data_size)> streamer_function);
+
  private:
    std::shared_ptr<struct cloner_plugin_impl> my;
 };

programs/rodeos/streamer_plugin.cpp

@@ -0,0 +1,77 @@
+// copyright defined in LICENSE.txt
+
+#include "streamer_plugin.hpp"
+#include "streams/logger.hpp"
+#include "streams/rabbitmq.hpp"
+#include "streams/stream.hpp"
+
+#include <abieos.hpp>
+#include <eosio/abi.hpp>
+#include <eosio/chain/exceptions.hpp>
+#include <fc/exception/exception.hpp>
+#include <memory>
+
+namespace b1 {
+
+using namespace appbase;
+using namespace std::literals;
+
+struct streamer_plugin_impl {
+   std::vector<std::unique_ptr<stream_handler>> streams;
+};
+
+static abstract_plugin& _streamer_plugin = app().register_plugin<streamer_plugin>();
+
+streamer_plugin::streamer_plugin() : my(std::make_shared<streamer_plugin_impl>()) {}
+
+streamer_plugin::~streamer_plugin() {}
+
+void streamer_plugin::set_program_options(options_description& cli, options_description& cfg) {
+   auto op = cfg.add_options();
+   op("stream-rabbits", bpo::value<std::vector<string>>()->composing(),
+      "RabbitMQ Streams if any; Format: amqp://USER:PASSWORD@ADDRESS:PORT/QUEUE[/ROUTING_KEYS, ...]");
+   op("stream-loggers", bpo::value<std::vector<string>>()->composing(),
+      "Logger Streams if any; Format: [routing_keys, ...]");
+}
+
+void streamer_plugin::plugin_initialize(const variables_map& options) {
+   try {
+      if (options.count("stream-loggers")) {
+         auto loggers = options.at("stream-loggers").as<std::vector<std::string>>();
+         initialize_loggers(my->streams, loggers);
+      }
+
+      if (options.count("stream-rabbits")) {
+         auto rabbits = options.at("stream-rabbits").as<std::vector<std::string>>();
+         initialize_rabbits(app().get_io_service(), my->streams, rabbits);
+      }
+
+      ilog("initialized streams: ${streams}", ("streams", my->streams.size()));
+   }
+   FC_LOG_AND_RETHROW()
+}
+
+void streamer_plugin::plugin_startup() {
+   cloner_plugin* cloner = app().find_plugin<cloner_plugin>();
+   EOS_ASSERT( cloner, eosio::chain::plugin_config_exception, "cloner_plugin not found" );
+   cloner->set_streamer([this](const char* data, uint64_t data_size) { stream_data(data, data_size); });
+}
+
+void streamer_plugin::plugin_shutdown() {}
+
+void streamer_plugin::stream_data(const char* data, uint64_t data_size) {
+   eosio::input_stream bin(data, data_size);
+   stream_wrapper      res = eosio::from_bin<stream_wrapper>(bin);
+   const auto&         sw  = std::get<stream_wrapper_v0>(res);
+   publish_to_streams(sw);
+}
+
+void streamer_plugin::publish_to_streams(const stream_wrapper_v0& sw) {
+   for (const auto& stream : my->streams) {
+      if (stream->check_route(sw.route)) {
+         stream->publish(sw.data.data(), sw.data.size());
+      }
+   }
+}
+
+} // namespace b1

programs/rodeos/streamer_plugin.hpp

@@ -0,0 +1,38 @@
+// copyright defined in LICENSE.txt
+
+#pragma once
+#include <appbase/application.hpp>
+#include <eosio/abi.hpp>
+
+#include "cloner_plugin.hpp"
+
+namespace b1 {
+
+struct stream_wrapper_v0 {
+   eosio::name       route;
+   std::vector<char> data;
+};
+EOSIO_REFLECT(stream_wrapper_v0, route, data);
+using stream_wrapper = std::variant<stream_wrapper_v0>;
+
+class streamer_plugin : public appbase::plugin<streamer_plugin> {
+
+ public:
+   APPBASE_PLUGIN_REQUIRES((cloner_plugin))
+
+   streamer_plugin();
+   virtual ~streamer_plugin();
+
+   virtual void set_program_options(appbase::options_description& cli, appbase::options_description& cfg) override;
+   void         plugin_initialize(const appbase::variables_map& options);
+   void         plugin_startup();
+   void         plugin_shutdown();
+   void         stream_data(const char* data, uint64_t data_size);
+
+ private:
+   std::shared_ptr<struct streamer_plugin_impl> my;
+
+   void publish_to_streams(const stream_wrapper_v0& sw);
+};
+
+} // namespace b1

programs/rodeos/streams/logger.hpp

@@ -0,0 +1,32 @@
+#pragma once
+
+#include "stream.hpp"
+#include <fc/log/logger.hpp>
+
+namespace b1 {
+
+class logger : public stream_handler {
+   std::vector<eosio::name> routes_;
+
+ public:
+   logger(std::vector<eosio::name> routes) : routes_(std::move(routes)) {
+      ilog("logger initialized");
+   }
+
+   const std::vector<eosio::name>& get_routes() const override { return routes_; }
+
+   void publish(const char* data, uint64_t data_size) override {
+      ilog("logger stream [${data_size}] >> ${data}", ("data", data)("data_size", data_size));
+   }
+};
+
+inline void initialize_loggers(std::vector<std::unique_ptr<stream_handler>>& streams,
+                               const std::vector<std::string>&               loggers) {
+   for (const auto& routings : loggers) {
+      std::vector<eosio::name> routing_keys = extract_routings(routings);
+      logger                   logger_streamer{ std::move(routing_keys) };
+      streams.emplace_back(std::make_unique<logger>(std::move(logger_streamer)));
+   }
+}
+
+} // namespace b1

programs/rodeos/streams/rabbitmq.hpp

@@ -0,0 +1,86 @@
+#pragma once
+
+#include "stream.hpp"
+#include "amqpcpp.h"
+#include "amqpcpp/libboostasio.h"
+#include "amqpcpp/linux_tcp.h"
+#include <fc/log/logger.hpp>
+#include <memory>
+
+namespace b1 {
+
+class rabbitmq_handler;
+
+class rabbitmq : public stream_handler {
+   std::shared_ptr<rabbitmq_handler>    handler_;
+   std::shared_ptr<AMQP::TcpConnection> connection_;
+   std::shared_ptr<AMQP::TcpChannel>    channel_;
+   std::string                          name_;
+   std::vector<eosio::name>             routes_;
+
+ public:
+   rabbitmq(boost::asio::io_service& io_service, std::vector<eosio::name> routes, std::string address, std::string name)
+       : name_(std::move(name)), routes_(std::move(routes)) {
+      AMQP::Address amqp_address(address);
+      ilog("Connecting to RabbitMQ address ${a} - Queue: ${q}...", ("a", std::string(amqp_address))("q", name_));
+
+      handler_    = std::make_shared<rabbitmq_handler>(io_service);
+      connection_ = std::make_shared<AMQP::TcpConnection>(handler_.get(), amqp_address);
+      channel_    = std::make_shared<AMQP::TcpChannel>(connection_.get());
+      declare_queue();
+   }
+
+   const std::vector<eosio::name>& get_routes() const override { return routes_; }
+
+   void publish(const char* data, uint64_t data_size) override { channel_->publish("", name_, data, data_size, 0); }
+
+ private:
+   void declare_queue() {
+      auto& queue = channel_->declareQueue(name_, AMQP::durable);
+      queue.onSuccess([](const std::string& name, uint32_t messagecount, uint32_t consumercount) {
+         ilog("RabbitMQ Connected Successfully!\n Queue ${q} - Messages: ${mc} - Consumers: ${cc}",
+              ("q", name)("mc", messagecount)("cc", consumercount));
+      });
+      queue.onError([](const char* error_message) {
+         throw std::runtime_error("RabbitMQ Queue error: " + std::string(error_message));
+      });
+   }
+};
+
+class rabbitmq_handler : public AMQP::LibBoostAsioHandler {
+ public:
+   explicit rabbitmq_handler(boost::asio::io_service& io_service) : AMQP::LibBoostAsioHandler(io_service) {}
+
+   void onError(AMQP::TcpConnection* connection, const char* message) {
+      throw std::runtime_error("rabbitmq connection failed: " + std::string(message));
+   }
+};
+
+inline void initialize_rabbits(boost::asio::io_service&                      io_service,
+                               std::vector<std::unique_ptr<stream_handler>>& streams,
+                               const std::vector<std::string>&               rabbits) {
+   for (std::string rabbit : rabbits) {
+      rabbit            = rabbit.substr(7, rabbit.length());
+      size_t pos        = rabbit.find("/");
+      size_t pos_router = rabbit.find_last_of("/");
+      bool   has_router = pos_router != pos;
+
+      std::vector<eosio::name> routings = {};
+      if (has_router) {
+         routings = extract_routings(rabbit.substr(pos_router + 1, rabbit.length()));
+         rabbit.erase(pos_router, rabbit.length());
+      }
+
+      std::string queue_name = "stream.default";
+      if (pos != std::string::npos) {
+         queue_name = rabbit.substr(pos + 1, rabbit.length());
+         rabbit.erase(pos, rabbit.length());
+      }
+
+      std::string address = "amqp://" + rabbit;
+      rabbitmq rmq{ io_service, std::move(routings), std::move(address), std::move(queue_name) };
+      streams.emplace_back(std::make_unique<rabbitmq>(std::move(rmq)));
+   }
+}
+
+} // namespace b1