-
Notifications
You must be signed in to change notification settings - Fork 616
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
12 changed files
with
984 additions
and
621 deletions.
There are no files selected for viewing
Large diffs are not rendered by default.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,194 @@ | ||
| use anyhow::{bail, Context, Error, Ok, Result}; | ||
| use ethers_core::types::U64; | ||
|
|
||
| use std::{cell::RefCell, rc::Rc}; | ||
|
|
||
| use crate::{ | ||
| config::Config, | ||
| coordinator_client::{listener::Listener, types::*, CoordinatorClient}, | ||
| geth_client::GethClient, | ||
| key_signer::KeySigner, | ||
| types::{ProofFailureType, ProofStatus, ProverType, TaskType}, | ||
| utils::{get_prover_type, get_circuit_types}, | ||
| zk_circuits_handler::{CircuitsHandler, CircuitsHandlerProvider}, | ||
| }; | ||
|
|
||
| use super::types::{ProofDetail, Task}; | ||
|
|
||
| pub struct Prover<'a> { | ||
| config: &'a Config, | ||
| key_signer: Rc<KeySigner>, | ||
| circuits_handler_provider: RefCell<CircuitsHandlerProvider<'a>>, | ||
| coordinator_client: RefCell<CoordinatorClient<'a>>, | ||
| geth_client: Option<Rc<RefCell<GethClient>>>, | ||
| } | ||
|
|
||
| impl<'a> Prover<'a> { | ||
| pub fn new(config: &'a Config, coordinator_listener: Box<dyn Listener>) -> Result<Self> { | ||
| let keystore_path = &config.keystore_path; | ||
| let keystore_password = &config.keystore_password; | ||
|
|
||
| let geth_client = if config | ||
| .prover_types | ||
| .iter() | ||
| .any(|element| *element == ProverType::Chunk) | ||
| { | ||
| Some(Rc::new(RefCell::new( | ||
| GethClient::new( | ||
| &config.prover_name, | ||
| &config.l2geth.as_ref().unwrap().endpoint, | ||
| ) | ||
| .context("failed to create l2 geth_client")?, | ||
| ))) | ||
| } else { | ||
| None | ||
| }; | ||
|
|
||
| let provider = CircuitsHandlerProvider::new(config, geth_client.clone()) | ||
| .context("failed to create circuits handler provider")?; | ||
|
|
||
| let vks = provider.init_vks(config.prover_types.clone(), config, geth_client.clone()); | ||
|
|
||
| let key_signer = Rc::new(KeySigner::new(keystore_path, keystore_password)?); | ||
| let coordinator_client = | ||
| CoordinatorClient::new(config, Rc::clone(&key_signer), coordinator_listener, vks) | ||
| .context("failed to create coordinator_client")?; | ||
|
|
||
| let prover = Prover { | ||
| config, | ||
| key_signer: Rc::clone(&key_signer), | ||
| circuits_handler_provider: RefCell::new(provider), | ||
| coordinator_client: RefCell::new(coordinator_client), | ||
| geth_client, | ||
| }; | ||
|
|
||
| Ok(prover) | ||
| } | ||
|
|
||
| pub fn get_public_key(&self) -> String { | ||
| self.key_signer.get_public_key() | ||
| } | ||
|
|
||
| pub fn fetch_task(&self) -> Result<Task> { | ||
| log::info!("[prover] start to fetch_task"); | ||
|
|
||
| let task_types: Vec<TaskType> = | ||
| self.config | ||
| .prover_types | ||
| .iter() | ||
| .fold(Vec::new(), |mut acc, prover_type| { | ||
| acc.extend(get_circuit_types(*prover_type)); | ||
| acc | ||
| }); | ||
|
|
||
| let mut req = GetTaskRequest { | ||
| task_types, | ||
| prover_height: None, | ||
| }; | ||
|
|
||
| if self | ||
| .config | ||
| .prover_types | ||
| .iter() | ||
| .any(|element| *element == ProverType::Chunk) | ||
| { | ||
| let latest_block_number = self.get_latest_block_number_value()?; | ||
| if let Some(v) = latest_block_number { | ||
| if v.as_u64() == 0 { | ||
| bail!("omit to prove task of the genesis block") | ||
| } | ||
| req.prover_height = Some(v.as_u64()); | ||
| } else { | ||
| log::error!("[prover] failed to fetch latest confirmed block number, got None"); | ||
| bail!("failed to fetch latest confirmed block number, got None") | ||
| } | ||
| } | ||
| let resp = self.coordinator_client.borrow_mut().get_task(&req)?; | ||
|
|
||
| match resp.data { | ||
| Some(d) => Ok(Task::from(d)), | ||
| None => { | ||
| bail!("data of get_task empty, while error_code is success. there may be something wrong in response data or inner logic.") | ||
| } | ||
| } | ||
| } | ||
|
|
||
| pub fn prove_task(&self, task: &Task) -> Result<ProofDetail> { | ||
| let prover_type = match get_prover_type(task.task_type) { | ||
| Some(pt) => Ok(pt), | ||
| None => { | ||
| bail!("unsupported prover_type.") | ||
| } | ||
| }?; | ||
| log::info!("[prover] start to prove_task, task id: {}", task.id); | ||
| let handler: Rc<Box<dyn CircuitsHandler>> = self | ||
| .circuits_handler_provider | ||
| .borrow_mut() | ||
| .get_circuits_handler(&task.hard_fork_name, prover_type) | ||
| .context("failed to get circuit handler")?; | ||
| self.do_prove(task, handler) | ||
| } | ||
|
|
||
| fn do_prove(&self, task: &Task, handler: Rc<Box<dyn CircuitsHandler>>) -> Result<ProofDetail> { | ||
| let mut proof_detail = ProofDetail { | ||
| id: task.id.clone(), | ||
| proof_type: task.task_type, | ||
| ..Default::default() | ||
| }; | ||
|
|
||
| proof_detail.proof_data = handler.get_proof_data(task.task_type, task)?; | ||
| Ok(proof_detail) | ||
| } | ||
|
|
||
| pub fn submit_proof(&self, proof_detail: ProofDetail, task: &Task) -> Result<()> { | ||
| log::info!( | ||
| "[prover] start to submit_proof, task id: {}", | ||
| proof_detail.id | ||
| ); | ||
|
|
||
| let request = SubmitProofRequest { | ||
| uuid: task.uuid.clone(), | ||
| task_id: proof_detail.id, | ||
| task_type: proof_detail.proof_type, | ||
| status: ProofStatus::Ok, | ||
| proof: proof_detail.proof_data, | ||
| ..Default::default() | ||
| }; | ||
|
|
||
| self.do_submit(&request) | ||
| } | ||
|
|
||
| pub fn submit_error( | ||
| &self, | ||
| task: &Task, | ||
| failure_type: ProofFailureType, | ||
| error: Error, | ||
| ) -> Result<()> { | ||
| log::info!("[prover] start to submit_error, task id: {}", task.id); | ||
| let request = SubmitProofRequest { | ||
| uuid: task.uuid.clone(), | ||
| task_id: task.id.clone(), | ||
| task_type: task.task_type, | ||
| status: ProofStatus::Error, | ||
| failure_type: Some(failure_type), | ||
| failure_msg: Some(format!("{:#}", error)), | ||
| ..Default::default() | ||
| }; | ||
| self.do_submit(&request) | ||
| } | ||
|
|
||
| fn do_submit(&self, request: &SubmitProofRequest) -> Result<()> { | ||
| self.coordinator_client.borrow_mut().submit_proof(request)?; | ||
| Ok(()) | ||
| } | ||
|
|
||
| fn get_latest_block_number_value(&self) -> Result<Option<U64>> { | ||
| let number = self | ||
| .geth_client | ||
| .as_ref() | ||
| .unwrap() | ||
| .borrow_mut() | ||
| .block_number()?; | ||
| Ok(number.as_number()) | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.