From f839d011c641d89693102982f6739b99fcd45fca Mon Sep 17 00:00:00 2001 From: pmendes Date: Thu, 2 Jan 2025 13:23:11 +0000 Subject: [PATCH] Rename get_{,hpke_}receiver_configs --- crates/daphne-server/src/roles/aggregator.rs | 2 +- crates/daphne/src/hpke.rs | 2 +- crates/daphne/src/roles/helper/handle_agg_job.rs | 4 +++- crates/daphne/src/roles/leader/mod.rs | 4 +++- crates/daphne/src/testing/mod.rs | 2 +- 5 files changed, 9 insertions(+), 5 deletions(-) diff --git a/crates/daphne-server/src/roles/aggregator.rs b/crates/daphne-server/src/roles/aggregator.rs index 060453b68..26b1a2d55 100644 --- a/crates/daphne-server/src/roles/aggregator.rs +++ b/crates/daphne-server/src/roles/aggregator.rs @@ -422,7 +422,7 @@ impl HpkeProvider for crate::App { .unwrap_or(false)) } - async fn get_receiver_configs<'s>( + async fn get_hpke_receiver_configs<'s>( &'s self, version: DapVersion, ) -> Result, DapError> { diff --git a/crates/daphne/src/hpke.rs b/crates/daphne/src/hpke.rs index 0c550ab7a..4d770dde7 100644 --- a/crates/daphne/src/hpke.rs +++ b/crates/daphne/src/hpke.rs @@ -235,7 +235,7 @@ pub trait HpkeProvider { task_id: Option<&TaskId>, ) -> Result, DapError>; - async fn get_receiver_configs<'s>( + async fn get_hpke_receiver_configs<'s>( &'s self, version: DapVersion, ) -> Result, DapError>; diff --git a/crates/daphne/src/roles/helper/handle_agg_job.rs b/crates/daphne/src/roles/helper/handle_agg_job.rs index bf6fad839..58d852b25 100644 --- a/crates/daphne/src/roles/helper/handle_agg_job.rs +++ b/crates/daphne/src/roles/helper/handle_agg_job.rs @@ -119,7 +119,9 @@ impl HandleAggJob { let task_id = request.task_id; let part_batch_sel = request.payload.part_batch_sel.clone(); let initialized_reports = task_config.consume_agg_job_req( - &aggregator.get_receiver_configs(task_config.version).await?, + &aggregator + .get_hpke_receiver_configs(task_config.version) + .await?, aggregator.valid_report_time_range(), &task_id, request.payload, diff --git a/crates/daphne/src/roles/leader/mod.rs b/crates/daphne/src/roles/leader/mod.rs index 503461805..aed51c207 100644 --- a/crates/daphne/src/roles/leader/mod.rs +++ b/crates/daphne/src/roles/leader/mod.rs @@ -302,7 +302,9 @@ async fn run_agg_job( // Prepare AggregationJobInitReq. let agg_job_id = AggregationJobId(thread_rng().gen()); let (agg_job_state, agg_job_init_req) = task_config.produce_agg_job_req( - aggregator.get_receiver_configs(task_config.version).await?, + aggregator + .get_hpke_receiver_configs(task_config.version) + .await?, aggregator.valid_report_time_range(), task_id, part_batch_sel, diff --git a/crates/daphne/src/testing/mod.rs b/crates/daphne/src/testing/mod.rs index ce074d95f..23b3d2025 100644 --- a/crates/daphne/src/testing/mod.rs +++ b/crates/daphne/src/testing/mod.rs @@ -640,7 +640,7 @@ impl HpkeProvider for InMemoryAggregator { Ok(&self.hpke_receiver_config_list[0].config) } - async fn get_receiver_configs<'s>( + async fn get_hpke_receiver_configs<'s>( &'s self, _version: DapVersion, ) -> Result, DapError> {