dolos AuthoritySelectionDataSource draft

Author: kpinter-iohk

demo/node/src/data_sources.rs

@@ -7,7 +7,7 @@ use sidechain_mc_hash::McHashDataSource;
 use sp_block_participation::inherent_data::BlockParticipationDataSource;
 use sp_governed_map::GovernedMapDataSource;
 use sp_partner_chains_bridge::TokenBridgeDataSource;
-use std::{error::Error, sync::Arc};
+use std::{error::Error, sync::Arc, time::Duration};
 
 pub const DATA_SOURCE_VAR: &str = "CARDANO_DATA_SOURCE";
 
@@ -81,7 +81,7 @@ pub(crate) async fn create_cached_data_sources(
 			ServiceError::Application(format!("Failed to create mock data sources: {err}").into())
 		}),
 
-		DataSourceType::Dolos => create_dolos_data_sources().map_err(|err| {
+		DataSourceType::Dolos => create_dolos_data_sources(metrics_opt).await.map_err(|err| {
 			ServiceError::Application(format!("Failed to create dolos data sources: {err}").into())
 		}),
 	}
@@ -101,16 +101,58 @@ pub fn create_mock_data_sources()
 	})
 }
 
-pub fn create_dolos_data_sources()
--> std::result::Result<DataSources, Box<dyn Error + Send + Sync + 'static>> {
-	use partner_chains_dolos_data_sources::*;
+// Currently uses db-sync for unimplemented dolos data sources
+pub async fn create_dolos_data_sources(
+	metrics_opt: Option<McFollowerMetrics>,
+) -> std::result::Result<DataSources, Box<dyn Error + Send + Sync + 'static>> {
+	let dolos_client = partner_chains_dolos_data_sources::client::MiniBFClient::new(
+		"http://localhost:3000",
+		Duration::from_secs(5),
+	);
+	let pool = partner_chains_db_sync_data_sources::get_connection_from_env().await?;
+	// block data source is reused between mc_hash and sidechain_rpc to share cache
+	let block = Arc::new(
+		partner_chains_db_sync_data_sources::BlockDataSourceImpl::new_from_env(pool.clone())
+			.await?,
+	);
 	Ok(DataSources {
-		sidechain_rpc: Arc::new(SidechainRpcDataSourceImpl::new()),
-		mc_hash: Arc::new(McHashDataSourceImpl::new()),
-		authority_selection: Arc::new(AuthoritySelectionDataSourceImpl::new()),
-		block_participation: Arc::new(StakeDistributionDataSourceImpl::new()),
-		governed_map: Arc::new(GovernedMapDataSourceImpl::default()),
-		bridge: Arc::new(TokenBridgeDataSourceImpl::new()),
+		sidechain_rpc: Arc::new(
+			partner_chains_dolos_data_sources::SidechainRpcDataSourceImpl::new(
+				dolos_client.clone(),
+			),
+		),
+		mc_hash: Arc::new(partner_chains_dolos_data_sources::McHashDataSourceImpl::new(
+			dolos_client.clone(),
+		)),
+		authority_selection: Arc::new(
+			partner_chains_dolos_data_sources::AuthoritySelectionDataSourceImpl::new(
+				dolos_client.clone(),
+			),
+		),
+		block_participation: Arc::new(
+			partner_chains_db_sync_data_sources::StakeDistributionDataSourceImpl::new(
+				pool.clone(),
+				metrics_opt.clone(),
+				STAKE_CACHE_SIZE,
+			),
+		),
+		governed_map: Arc::new(
+			partner_chains_db_sync_data_sources::GovernedMapDataSourceCachedImpl::new(
+				pool.clone(),
+				metrics_opt.clone(),
+				GOVERNED_MAP_CACHE_SIZE,
+				block.clone(),
+			)
+			.await?,
+		),
+		bridge: Arc::new(
+			partner_chains_db_sync_data_sources::CachedTokenBridgeDataSourceImpl::new(
+				pool,
+				metrics_opt,
+				block,
+				BRIDGE_TRANSFER_CACHE_LOOKAHEAD,
+			),
+		),
 	})
 }