-
Notifications
You must be signed in to change notification settings - Fork 7
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Multi job runner + SDK main macro (#346)
- Loading branch information
Showing
24 changed files
with
720 additions
and
403 deletions.
There are no files selected for viewing
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
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
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 |
|---|---|---|
| @@ -1,146 +1,30 @@ | ||
| use color_eyre::{eyre::eyre, Result}; | ||
| use gadget_sdk::config::{ContextConfig, GadgetCLICoreSettings, GadgetConfiguration, StdGadgetConfiguration}; | ||
| use gadget_sdk::{ | ||
| tangle_subxt::tangle_testnet_runtime::api::{ | ||
| self, | ||
| runtime_types::{sp_core::ecdsa, tangle_primitives::services}, | ||
| }, | ||
| tx, | ||
| info | ||
| }; | ||
| use std::io::Write; | ||
| use incredible_squaring_blueprint as blueprint; | ||
| use structopt::StructOpt; | ||
| use gadget_sdk::events_watcher::InitializableEventHandler; | ||
| use gadget_sdk::keystore::KeystoreUriSanitizer; | ||
| use gadget_sdk::keystore::sp_core_subxt::Pair; | ||
| use gadget_sdk::run::GadgetRunner; | ||
| use gadget_sdk::info; | ||
| use gadget_sdk::job_runner::MultiJobRunner; | ||
| use gadget_sdk::tangle_subxt::subxt::tx::Signer; | ||
| use gadget_sdk::tangle_subxt::tangle_testnet_runtime::api::runtime_types::tangle_primitives::services::PriceTargets; | ||
| struct TangleGadgetRunner { | ||
| env: GadgetConfiguration<parking_lot::RawRwLock>, | ||
| } | ||
|
|
||
| #[async_trait::async_trait] | ||
| impl GadgetRunner for TangleGadgetRunner { | ||
| type Error = color_eyre::eyre::Report; | ||
|
|
||
| fn config(&self) -> &StdGadgetConfiguration { | ||
| todo!() | ||
| } | ||
|
|
||
| async fn register(&mut self) -> Result<()> { | ||
| // TODO: Use the function in blueprint-test-utils | ||
| if self.env.test_mode { | ||
| info!("Skipping registration in test mode"); | ||
| return Ok(()); | ||
| } | ||
|
|
||
| let client = self.env.client().await.map_err(|e| eyre!(e))?; | ||
| let signer = self | ||
| .env | ||
| .first_sr25519_signer() | ||
| .map_err(|e| eyre!(e)) | ||
| .map_err(|e| eyre!(e))?; | ||
| let ecdsa_pair = self.env.first_ecdsa_signer().map_err(|e| eyre!(e))?; | ||
|
|
||
| let xt = api::tx().services().register( | ||
| self.env.blueprint_id, | ||
| services::OperatorPreferences { | ||
| key: ecdsa::Public(ecdsa_pair.signer().public().0), | ||
| approval: services::ApprovalPrefrence::None, | ||
| // TODO: Set the price targets | ||
| price_targets: PriceTargets { | ||
| cpu: 0, | ||
| mem: 0, | ||
| storage_hdd: 0, | ||
| storage_ssd: 0, | ||
| storage_nvme: 0, | ||
| }, | ||
| }, | ||
| Default::default(), | ||
| ); | ||
|
|
||
| // send the tx to the tangle and exit. | ||
| let result = tx::tangle::send(&client, &signer, &xt).await?; | ||
| info!("Registered operator with hash: {:?}", result); | ||
| Ok(()) | ||
| } | ||
|
|
||
| async fn benchmark(&self) -> std::result::Result<(), Self::Error> { | ||
| todo!() | ||
| } | ||
|
|
||
| async fn run(&mut self) -> Result<()> { | ||
| let client = self.env.client().await.map_err(|e| eyre!(e))?; | ||
| let signer = self.env.first_sr25519_signer().map_err(|e| eyre!(e))?; | ||
|
|
||
| info!("Starting the event watcher for {} ...", signer.account_id()); | ||
|
|
||
| let x_square = blueprint::XsquareEventHandler { | ||
| service_id: self.env.service_id.unwrap(), | ||
| client: client.clone(), | ||
| signer, | ||
| }; | ||
| use incredible_squaring_blueprint as blueprint; | ||
|
|
||
| let finished_rx = x_square | ||
| .init_event_handler() | ||
| .await | ||
| .expect("Event Listener init already called"); | ||
| let res = finished_rx.await; | ||
| gadget_sdk::error!("Event handler finished with {res:?}"); | ||
| #[gadget_sdk::main(env)] | ||
| async fn main() { | ||
| let client = env.client().await.map_err(|e| eyre!(e))?; | ||
| let signer = env.first_sr25519_signer().map_err(|e| eyre!(e))?; | ||
|
|
||
| Ok(()) | ||
| } | ||
| } | ||
| info!("Starting the event watcher for {} ...", signer.account_id()); | ||
|
|
||
| #[tokio::main] | ||
| #[allow(clippy::needless_return)] | ||
| async fn main() -> Result<()> { | ||
| gadget_sdk::logging::setup_log(); | ||
| // Load the environment and create the gadget runner | ||
| let config = ContextConfig::from_args(); | ||
| let env = gadget_sdk::config::load(config.clone()).expect("Failed to load environment"); | ||
| let mut runner = Box::new(TangleGadgetRunner { env: env.clone() }); | ||
| let x_square = blueprint::XsquareEventHandler { | ||
| service_id: env.service_id.unwrap(), | ||
| client: client.clone(), | ||
| signer, | ||
| }; | ||
|
|
||
| info!("~~~ Executing the incredible squaring blueprint ~~~"); | ||
|
|
||
| check_for_test(&env, &config)?; | ||
|
|
||
| info!("Registering..."); | ||
| // Register the operator if needed | ||
| if env.should_run_registration() { | ||
| runner.register().await?; | ||
| } | ||
|
|
||
| info!("Running..."); | ||
| // Run the gadget / AVS | ||
| runner.run().await?; | ||
| MultiJobRunner::new(&env) | ||
| .with_job() | ||
| .with_default_price_targets() | ||
| .finish(x_square) | ||
| .run() | ||
| .await?; | ||
|
|
||
| info!("Exiting..."); | ||
| Ok(()) | ||
| } | ||
|
|
||
| #[allow(irrefutable_let_patterns)] | ||
| fn check_for_test( | ||
| _env: &GadgetConfiguration<parking_lot::RawRwLock>, | ||
| config: &ContextConfig, | ||
| ) -> Result<()> { | ||
| // create a file to denote we have started | ||
| if let GadgetCLICoreSettings::Run { | ||
| keystore_uri: base_path, | ||
| test_mode, | ||
| .. | ||
| } = &config.gadget_core_settings | ||
| { | ||
| if !*test_mode { | ||
| return Ok(()); | ||
| } | ||
| let path = base_path.sanitize_file_path().join("test_started.tmp"); | ||
| let mut file = std::fs::File::create(&path)?; | ||
| file.write_all(b"test_started")?; | ||
| info!("Successfully wrote test file to {}", path.display()) | ||
| } | ||
|
|
||
| Ok(()) | ||
| } |
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.