Skip to content

Commit b5445a0

Browse files
committed
Don't have block_service as an Option
1 parent 5cc2c13 commit b5445a0

4 files changed

Lines changed: 11 additions & 13 deletions

File tree

validator_client/src/http_api/mod.rs

Lines changed: 2 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,7 @@ impl From<String> for Error {
7373
pub struct Context<T: SlotClock, E: EthSpec> {
7474
pub task_executor: TaskExecutor,
7575
pub api_secret: ApiSecret,
76-
pub block_service: Option<BlockService<T, E>>,
76+
pub block_service: BlockService<T, E>,
7777
pub validator_store: Option<Arc<ValidatorStore<T, E>>>,
7878
pub validator_dir: Option<PathBuf>,
7979
pub secrets_dir: Option<PathBuf>,
@@ -172,15 +172,7 @@ pub fn serve<T: 'static + SlotClock + Clone, E: EthSpec>(
172172
};
173173

174174
let inner_block_service = ctx.block_service.clone();
175-
let block_service_filter = warp::any()
176-
.map(move || inner_block_service.clone())
177-
.and_then(|block_service: Option<_>| async move {
178-
block_service.ok_or_else(|| {
179-
warp_utils::reject::custom_not_found(
180-
"block service is not initialized.".to_string(),
181-
)
182-
})
183-
});
175+
let block_service_filter = warp::any().map(move || inner_block_service.clone());
184176

185177
let inner_validator_store = ctx.validator_store.clone();
186178
let validator_store_filter = warp::any()

validator_client/src/http_api/test_utils.rs

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
use crate::doppelganger_service::DoppelgangerService;
22
use crate::key_cache::{KeyCache, CACHE_FILENAME};
3+
use crate::BlockServiceBuilder;
34
use crate::{
45
http_api::{ApiSecret, Config as HttpConfig, Context},
56
initialized_validators::{InitializedValidators, OnDecryptFailure},
@@ -127,7 +128,9 @@ impl ApiTester {
127128
let context = Arc::new(Context {
128129
task_executor: test_runtime.task_executor.clone(),
129130
api_secret,
130-
block_service: None,
131+
block_service: BlockServiceBuilder::new()
132+
.build()
133+
.expect("Should build block service"),
131134
validator_dir: Some(validator_dir.path().into()),
132135
secrets_dir: Some(secrets_dir.path().into()),
133136
validator_store: Some(validator_store.clone()),

validator_client/src/http_api/tests.rs

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
mod keystores;
55

66
use crate::doppelganger_service::DoppelgangerService;
7+
use crate::BlockServiceBuilder;
78
use crate::{
89
http_api::{ApiSecret, Config as HttpConfig, Context},
910
initialized_validators::InitializedValidators,
@@ -115,7 +116,9 @@ impl ApiTester {
115116
let context = Arc::new(Context {
116117
task_executor: test_runtime.task_executor.clone(),
117118
api_secret,
118-
block_service: None,
119+
block_service: BlockServiceBuilder::new()
120+
.build()
121+
.expect("Should build block service"),
119122
validator_dir: Some(validator_dir.path().into()),
120123
secrets_dir: Some(secrets_dir.path().into()),
121124
validator_store: Some(validator_store.clone()),

validator_client/src/lib.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -572,7 +572,7 @@ impl<E: EthSpec> ProductionValidatorClient<E> {
572572
let ctx = Arc::new(http_api::Context {
573573
task_executor: self.context.executor.clone(),
574574
api_secret,
575-
block_service: Some(self.block_service.clone()),
575+
block_service: self.block_service.clone(),
576576
validator_store: Some(self.validator_store.clone()),
577577
validator_dir: Some(self.config.validator_dir.clone()),
578578
secrets_dir: Some(self.config.secrets_dir.clone()),

0 commit comments

Comments
 (0)