diff --git a/prover/src/main.rs b/prover/src/main.rs index 8b7cb1279..8122ef3cc 100644 --- a/prover/src/main.rs +++ b/prover/src/main.rs @@ -39,6 +39,62 @@ struct Args { log_file: Option, } +fn start_loop<'a>( + config: &'a Config, + prover: &'a Prover<'a>, + task_cache: Rc, + log_file: Option, + current_dsn: Option, +) -> Option { + let mut new_dsn: Option = None; + let should_stop = || { + new_dsn = prover.coordinator_client.borrow().get_sentry_dsn(); + + current_dsn != new_dsn + }; + + let _guard = current_dsn.clone().map(|dsn| { + log::info!("successfully get dsn from coordinator"); + let gurad = sentry::init(( + dsn, + sentry::ClientOptions { + release: Some(version::get_version_cow()), + environment: Some(utils::get_environment()), + ..Default::default() + }, + )); + utils::set_logger_with_sentry(log_file); + gurad + }); + + _guard.iter().for_each(|_| { + sentry::configure_scope(|scope| { + scope.set_tag("prover_type", config.prover_type); + scope.set_tag("partner_name", config.partner_name()); + scope.set_tag("prover_name", config.prover_name.clone()); + + let public_key = sentry::protocol::Value::from(prover.get_public_key()); + scope.set_extra("public_key", public_key); + }); + + sentry::capture_message("test message on start", sentry::Level::Info); + }); + + log::info!( + "prover start successfully. name: {}, type: {:?}, publickey: {}, version: {}", + config.prover_name, + config.prover_type, + prover.get_public_key(), + version::get_version(), + ); + + let task_processor = TaskProcessor::new(prover, task_cache); + + task_processor.start(should_stop); + + new_dsn +} + fn start() -> Result<()> { let args = Args::parse(); @@ -64,54 +120,15 @@ fn start() -> Result<()> { let prover = Prover::new(&config, coordinator_listener)?; - let _guard = prover - .coordinator_client - .borrow() - .get_sentry_dsn() - .map(|dsn| { - log::info!("successfully get dsn from coordinator"); - let gurad = Some(sentry::init(( - dsn, - sentry::ClientOptions { - release: Some(version::get_version_cow()), - environment: Some(utils::get_environment()), - ..Default::default() - }, - ))); - utils::set_logger_with_sentry(args.log_file); - gurad - }); - - _guard.iter().for_each(|_| { - sentry::configure_scope(|scope| { - scope.set_tag("prover_type", config.prover_type); - scope.set_tag("partner_name", config.partner_name()); - scope.set_tag("prover_name", config.prover_name.clone()); - }); - - sentry::capture_message("test message on start", sentry::Level::Info); - }); - - _guard.iter().for_each(|_| { - sentry::configure_scope(|scope| { - let public_key = sentry::protocol::Value::from(prover.get_public_key()); - scope.set_extra("public_key", public_key); - }); - }); - - log::info!( - "prover start successfully. name: {}, type: {:?}, publickey: {}, version: {}", - config.prover_name, - config.prover_type, - prover.get_public_key(), - version::get_version(), - ); - - let task_processor = TaskProcessor::new(&prover, task_cache); - - task_processor.start(); - - Ok(()) + loop { + start_loop( + &config, + &prover, + task_cache.clone(), + args.log_file.clone(), + None, + ); + } } fn main() { diff --git a/prover/src/task_processor.rs b/prover/src/task_processor.rs index 366a8d978..f92990fb1 100644 --- a/prover/src/task_processor.rs +++ b/prover/src/task_processor.rs @@ -16,8 +16,15 @@ impl<'a> TaskProcessor<'a> { TaskProcessor { prover, task_cache } } - pub fn start(&self) { + pub fn start(&self, mut should_stop: F) + where + F: FnMut() -> bool, + { loop { + if should_stop() { + log::info!("task processor should stop."); + break; + } log::info!("start a new round."); if let Err(err) = self.prove_and_submit() { if err.is::() {