Add handshake to tx protocol (#1460)

* Kagome now writes roles to tx propagation protocol handshake
qdrvm · Jan 10, 2023 · bfc6a68 · bfc6a68
1 parent 6ea24a6
commit bfc6a68
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/core/network/impl/protocols/propagate_transactions_protocol.cpp b/core/network/impl/protocols/propagate_transactions_protocol.cpp
@@ -39,6 +39,7 @@ namespace kagome::network {
                   kPropagateTransactionsProtocol, genesis_hash, chain_spec),
               log::createLogger(kPropagateTransactionsProtocolName,
                                 "propagate_transactions_protocol")),
+        app_config_{app_config},
         babe_(std::move(babe)),
         extrinsic_observer_(std::move(extrinsic_observer)),
         stream_engine_(std::move(stream_engine)),
@@ -232,7 +233,7 @@ namespace kagome::network {
       std::function<void(outcome::result<void>)> &&cb) {
     auto read_writer = std::make_shared<ScaleMessageReadWriter>(stream);
 
-    read_writer->write(NoData{},
+    read_writer->write(app_config_.roles(),
                        [stream = std::move(stream),
                         direction,
                         wp = weak_from_this(),

diff --git a/core/network/impl/protocols/propagate_transactions_protocol.hpp b/core/network/impl/protocols/propagate_transactions_protocol.hpp
@@ -84,6 +84,7 @@ namespace kagome::network {
     const static inline auto kPropagateTransactionsProtocolName =
         "PropagateTransactionsProtocol"s;
     ProtocolBaseImpl base_;
+    const application::AppConfiguration &app_config_;
     std::shared_ptr<consensus::babe::Babe> babe_;
     std::shared_ptr<ExtrinsicObserver> extrinsic_observer_;
     std::shared_ptr<StreamEngine> stream_engine_;