Skip to content

Commit ab76e8c

Browse files
committed
Simplify build_combined_affinity_filter to take StatementProtocolConfig
1 parent b30bc35 commit ab76e8c

File tree

1 file changed

+15
-24
lines changed

1 file changed

+15
-24
lines changed

light-base/src/json_rpc_service/background.rs

Lines changed: 15 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -773,14 +773,10 @@ pub(super) async fn run<TPlat: PlatformRef>(
773773
if matches!(version, network_service::StatementProtocolVersion::V2) {
774774
me.v2_statement_peers.insert(peer_id.clone());
775775
if !me.statement_subscriptions.is_empty() {
776-
let spc = me
777-
.statement_protocol_config
778-
.as_ref()
779-
.expect("V2 peers require statement protocol; qed");
780776
let combined_filter = build_combined_affinity_filter(
781777
&me.statement_subscriptions,
782-
spc.bloom_seed(),
783-
spc.false_positive_rate(),
778+
me.statement_protocol_config.as_ref()
779+
.expect("V2 peers require statement protocol; qed"),
784780
);
785781
let _ = me
786782
.network_service
@@ -2923,14 +2919,10 @@ pub(super) async fn run<TPlat: PlatformRef>(
29232919
.insert(subscription_id.clone(), filter);
29242920

29252921
if !me.v2_statement_peers.is_empty() {
2926-
let spc = me
2927-
.statement_protocol_config
2928-
.as_ref()
2929-
.expect("V2 peers require statement protocol; qed");
2930-
let combined_filter = build_combined_affinity_filter(
2922+
let combined_filter = build_combined_affinity_filter(
29312923
&me.statement_subscriptions,
2932-
spc.bloom_seed(),
2933-
spc.false_positive_rate(),
2924+
me.statement_protocol_config.as_ref()
2925+
.expect("V2 peers require statement protocol; qed"),
29342926
);
29352927
for peer_id in me.v2_statement_peers.clone() {
29362928
let _ = me
@@ -2955,14 +2947,10 @@ pub(super) async fn run<TPlat: PlatformRef>(
29552947
let existed = me.statement_subscriptions.remove(&subscription).is_some();
29562948

29572949
if existed && !me.v2_statement_peers.is_empty() {
2958-
let spc = me
2959-
.statement_protocol_config
2960-
.as_ref()
2961-
.expect("V2 peers require statement protocol; qed");
2962-
let combined_filter = build_combined_affinity_filter(
2950+
let combined_filter = build_combined_affinity_filter(
29632951
&me.statement_subscriptions,
2964-
spc.bloom_seed(),
2965-
spc.false_positive_rate(),
2952+
me.statement_protocol_config.as_ref()
2953+
.expect("V2 peers require statement protocol; qed"),
29662954
);
29672955
for peer_id in me.v2_statement_peers.clone() {
29682956
let _ = me
@@ -6111,8 +6099,7 @@ fn build_combined_affinity_filter(
61116099
smoldot::json_rpc::methods::TopicFilter,
61126100
fnv::FnvBuildHasher,
61136101
>,
6114-
seed: u128,
6115-
false_positive_rate: f64,
6102+
config: &network_service::StatementProtocolConfig,
61166103
) -> network_service::AffinityFilter {
61176104
use smoldot::json_rpc::methods::TopicFilter;
61186105

@@ -6121,13 +6108,17 @@ fn build_combined_affinity_filter(
61216108
for filter in subscriptions.values() {
61226109
match filter {
61236110
TopicFilter::Any => {
6124-
return network_service::AffinityFilter::match_all(seed);
6111+
return network_service::AffinityFilter::match_all(config.bloom_seed());
61256112
}
61266113
TopicFilter::MatchAll(topics) | TopicFilter::MatchAny(topics) => {
61276114
all_topics.extend(topics.iter());
61286115
}
61296116
}
61306117
}
61316118

6132-
network_service::AffinityFilter::from_topics(all_topics.into_iter(), seed, false_positive_rate)
6119+
network_service::AffinityFilter::from_topics(
6120+
all_topics.into_iter(),
6121+
config.bloom_seed(),
6122+
config.false_positive_rate(),
6123+
)
61336124
}

0 commit comments

Comments
 (0)