send_cluster_candidate_statements

Signed-off-by: iceseer <[email protected]>
qdrvm · Mar 31, 2024 · 7621937 · 7621937
1 parent a962179
commit 7621937
Show file tree

Hide file tree

Showing 3 changed files with 33 additions and 4 deletions.
diff --git a/core/network/types/collator_messages_vstaging.hpp b/core/network/types/collator_messages_vstaging.hpp
@@ -137,9 +137,9 @@ namespace kagome::network::vstaging {
     scale::BitVec validated_in_group;
 
     StatementFilter() = default;
-    StatementFilter(size_t len) {
-      seconded_in_group.bits.assign(len, false);
-      validated_in_group.bits.assign(len, false);
+    StatementFilter(size_t len, bool val = false) {
+      seconded_in_group.bits.assign(len, val);
+      validated_in_group.bits.assign(len, val);
     }
 
     bool has_len(size_t len) const {

diff --git a/core/parachain/validator/impl/parachain_processor.cpp b/core/parachain/validator/impl/parachain_processor.cpp
@@ -912,11 +912,16 @@ namespace kagome::parachain {
                validator->validatorIndex(),
                relay_parent);
 
+    std::optional<GroupIndex> our_group;
+    if (groups_p) {
+      our_group = groups_p->byValidatorIndex(validator->validatorIndex());
+    }
     return RelayParentState{
         .prospective_parachains_mode = mode,
         .assignment = assignment,
         .seconded = {},
         .our_index = validator->validatorIndex(),
+        .our_group = our_group,
         .required_collator = required_collator,
         .collations = {},
         .table_context =
@@ -2146,7 +2151,30 @@ namespace kagome::parachain {
   }
 
   void ParachainProcessorImpl::send_cluster_candidate_statements(
-      const CandidateHash &candidate_hash, const RelayHash &relay_parent) {}
+      const CandidateHash &candidate_hash, const RelayHash &relay_parent) {
+    auto relay_parent_state = tryGetStateByRelayParent(relay_parent);
+    if (!relay_parent_state) {
+      return;
+    }
+
+    auto local_group = relay_parent_state->get().our_group;
+    if (!local_group) {
+      return;
+    }
+
+    auto group = relay_parent_state->get().groups->get(*local_group);
+    if (!group) {
+      return;
+    }
+    auto group_size = group->size();
+
+    relay_parent_state->get().statement_store->groupStatements(
+        *group,
+        candidate_hash,
+        network::vstaging::StatementFilter(group_size, true),
+        [&](const IndexedAndSigned<network::vstaging::CompactStatement>
+                &statement) { circulate_statement(relay_parent, statement); });
+  }
 
   void ParachainProcessorImpl::apply_post_confirmation(
       const PostConfirmation &post_confirmation) {

diff --git a/core/parachain/validator/parachain_processor.hpp b/core/parachain/validator/parachain_processor.hpp
@@ -252,6 +252,7 @@ namespace kagome::parachain {
       std::optional<network::ParachainId> assignment;
       std::optional<primitives::BlockHash> seconded;
       std::optional<network::ValidatorIndex> our_index;
+      std::optional<network::GroupIndex> our_group;
       std::optional<network::CollatorPublicKey> required_collator;
 
       Collations collations;