add: ETCM-12351 dolos governed map data source

Cargo.toml

@@ -188,6 +188,7 @@ fork-tree = { version = "13.0.1" }
 ureq = { version = "3.1.2", default-features = false }
 url = { version = "2.5.7", default-features = false }
 blockfrost-openapi = { version = "0.1.75", default-features = false }
+chrono = { version = "0.4.31", default-features = false }
 
 # substrate dependencies
 frame-benchmarking = { default-features = false, git = "https://github.com/paritytech/polkadot-sdk.git", tag = "polkadot-stable2509" }

demo/node/src/data_sources.rs

@@ -103,12 +103,11 @@ pub fn create_mock_data_sources()
 
 // TODO Currently uses db-sync for unimplemented Dolos data sources
 pub async fn create_dolos_data_sources(
-	metrics_opt: Option<McFollowerMetrics>,
+	_metrics_opt: Option<McFollowerMetrics>,
 ) -> std::result::Result<DataSources, Box<dyn Error + Send + Sync + 'static>> {
 	let dolos_client = partner_chains_dolos_data_sources::get_connection_from_env()?;
-	let pool = partner_chains_db_sync_data_sources::get_connection_from_env().await?;
-	let block = Arc::new(
-		partner_chains_db_sync_data_sources::BlockDataSourceImpl::new_from_env(pool.clone())
+	let block_dolos = Arc::new(
+		partner_chains_dolos_data_sources::BlockDataSourceImpl::new_from_env(dolos_client.clone())
 			.await?,
 	);
 	Ok(DataSources {
@@ -118,37 +117,26 @@ pub async fn create_dolos_data_sources(
 			),
 		),
 		mc_hash: Arc::new(partner_chains_dolos_data_sources::McHashDataSourceImpl::new(
-			dolos_client.clone(),
+			block_dolos.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,
+			partner_chains_dolos_data_sources::StakeDistributionDataSourceImpl::new(
+				dolos_client.clone(),
 			),
 		),
 		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,
+			partner_chains_dolos_data_sources::GovernedMapDataSourceImpl::new(
+				dolos_client.clone(),
 			),
 		),
+		bridge: Arc::new(partner_chains_dolos_data_sources::TokenBridgeDataSourceImpl::new(
+			dolos_client,
+		)),
 	})
 }
 

dev/local-environment/setup.sh

@@ -3,7 +3,7 @@
 PARTNER_CHAINS_NODE_IMAGE="ghcr.io/input-output-hk/partner-chains/partner-chains-node-unstable:latest"
 CARDANO_IMAGE="ghcr.io/intersectmbo/cardano-node:10.5.1"
 DBSYNC_IMAGE="ghcr.io/intersectmbo/cardano-db-sync:13.6.0.5"
-DOLOS_IMAGE="ghcr.io/txpipe/dolos:1.0.0-beta.8"
+DOLOS_IMAGE="ghcr.io/txpipe/dolos:1.0.0-rc.2"
 OGMIOS_IMAGE="cardanosolutions/ogmios:v6.13.0"
 POSTGRES_IMAGE="postgres:17.2"
 TESTS_IMAGE="python:3.12-slim"

toolkit/data-sources/db-sync/Cargo.toml

@@ -16,7 +16,7 @@ sqlx = { workspace = true }
 db-sync-sqlx = { workspace = true }
 tokio = { workspace = true, features = ["full"] }
 futures = { workspace = true }
-chrono = "0.4.31"
+chrono = { workspace = true }
 hex = { workspace = true }
 hex-literal = { workspace = true }
 itertools = { workspace = true }

toolkit/data-sources/db-sync/src/candidates/cached.rs

@@ -20,6 +20,7 @@ pub type ArcMut<T> = Arc<Mutex<T>>;
 
 type AriadneParametersCacheKey = (McEpochNumber, PolicyId, PolicyId);
 type CandidatesCacheKey = (McEpochNumber, String);
+/// Cached candidate data source
 pub struct CandidateDataSourceCached {
 	inner: CandidatesDataSourceImpl,
 	get_ariadne_parameters_for_epoch_cache:
@@ -96,6 +97,7 @@ impl CandidateDataSourceCacheConfig {
 }
 
 impl CandidateDataSourceCached {
+	/// Creates new instance of the data source
 	pub fn new(
 		inner: CandidatesDataSourceImpl,
 		candidates_for_epoch_cache_size: usize,
@@ -114,6 +116,7 @@ impl CandidateDataSourceCached {
 		}
 	}
 
+	/// Creates a new instance of the data source, reading configuration from the environment.
 	pub fn new_from_env(
 		inner: CandidatesDataSourceImpl,
 		candidates_for_epoch_cache_size: usize,

toolkit/data-sources/db-sync/src/candidates/mod.rs

@@ -17,7 +17,7 @@ use sqlx::PgPool;
 use std::collections::HashMap;
 use std::error::Error;
 
-mod cached;
+pub mod cached;
 
 #[cfg(test)]
 mod tests;

toolkit/data-sources/db-sync/src/lib.rs

@@ -103,7 +103,7 @@ pub use crate::block::{BlockDataSourceImpl, DbSyncBlockDataSourceConfig};
 #[cfg(feature = "bridge")]
 pub use crate::bridge::{TokenBridgeDataSourceImpl, cache::CachedTokenBridgeDataSourceImpl};
 #[cfg(feature = "candidate-source")]
-pub use crate::candidates::CandidatesDataSourceImpl;
+pub use crate::candidates::{CandidatesDataSourceImpl, cached::CandidateDataSourceCached};
 #[cfg(feature = "governed-map")]
 pub use crate::governed_map::{GovernedMapDataSourceCachedImpl, GovernedMapDataSourceImpl};
 #[cfg(feature = "mc-hash")]

toolkit/data-sources/dolos/Cargo.toml

@@ -40,6 +40,9 @@ cardano-serialization-lib = { workspace = true }
 itertools = { workspace = true }
 figment = { workspace = true }
 tokio = { workspace = true }
+derive-new = { workspace = true }
+chrono = { workspace = true }
+bech32 = { workspace = true }
 
 [features]
 default = []