chore: read connection urls vector from config

MarcosNicolau · MarcosNicolau · commit 51085bda817c · 2025-12-19T12:13:53.000-03:00
diff --git a/aggregation_mode/Cargo.lock b/aggregation_mode/Cargo.lock
diff --git a/aggregation_mode/db/src/orchestrator.rs b/aggregation_mode/db/src/orchestrator.rs
@@ -66,7 +66,7 @@ impl std::fmt::Display for DbOrchestartorError {
 
 impl DbOrchestartor {
     pub fn try_new(
-        connection_urls: &[&str],
+        connection_urls: &[String],
         retry_config: RetryConfig,
     ) -> Result<Self, DbOrchestartorError> {
         if connection_urls.is_empty() {
diff --git a/aggregation_mode/gateway/src/config.rs b/aggregation_mode/gateway/src/config.rs
@@ -5,7 +5,7 @@ use serde::{Deserialize, Serialize};
 #[derive(Clone, Debug, Deserialize, Serialize)]
 pub struct Config {
     pub port: u16,
-    pub db_connection_url: String,
+    pub db_connection_urls: Vec<String>,
     pub network: String,
     pub max_daily_proofs_per_user: i64,
 }
diff --git a/aggregation_mode/gateway/src/db.rs b/aggregation_mode/gateway/src/db.rs
@@ -14,7 +14,7 @@ pub enum DbError {
 }
 
 impl Db {
-    pub async fn try_new(connection_urls: &[&str]) -> Result<Self, DbError> {
+    pub async fn try_new(connection_urls: &[String]) -> Result<Self, DbError> {
         let orchestrator = DbOrchestartor::try_new(
             connection_urls,
             RetryConfig {
diff --git a/aggregation_mode/gateway/src/main.rs b/aggregation_mode/gateway/src/main.rs
@@ -26,7 +26,7 @@ async fn main() {
     let config = Config::from_file(&config_file_path).expect("Config is valid");
     tracing::info!("Config loaded");
 
-    let db = Db::try_new(&[config.db_connection_url.as_str()])
+    let db = Db::try_new(config.db_connection_urls.as_slice())
         .await
         .expect("db to start");
 
diff --git a/aggregation_mode/payments_poller/src/config.rs b/aggregation_mode/payments_poller/src/config.rs
@@ -4,7 +4,7 @@ use serde::{Deserialize, Serialize};
 
 #[derive(Clone, Debug, Deserialize, Serialize)]
 pub struct Config {
-    pub db_connection_url: String,
+    pub db_connection_urls: Vec<String>,
     pub eth_rpc_url: String,
     pub payment_service_address: String,
 }
diff --git a/aggregation_mode/payments_poller/src/db.rs b/aggregation_mode/payments_poller/src/db.rs
@@ -12,7 +12,7 @@ pub enum DbError {
 }
 
 impl Db {
-    pub async fn try_new(connection_urls: &[&str]) -> Result<Self, DbError> {
+    pub async fn try_new(connection_urls: &[String]) -> Result<Self, DbError> {
         let orchestartor = DbOrchestartor::try_new(
             connection_urls,
             RetryConfig {
diff --git a/aggregation_mode/payments_poller/src/main.rs b/aggregation_mode/payments_poller/src/main.rs
@@ -26,7 +26,7 @@ async fn main() {
     let config = Config::from_file(&config_file_path).expect("Config is valid");
     tracing::info!("Config loaded");
 
-    let db = Db::try_new(&[config.db_connection_url.as_str()])
+    let db = Db::try_new(&config.db_connection_urls.as_slice())
         .await
         .expect("db to start");
 
diff --git a/aggregation_mode/proof_aggregator/src/backend/config.rs b/aggregation_mode/proof_aggregator/src/backend/config.rs
@@ -20,7 +20,7 @@ pub struct Config {
     pub risc0_chunk_aggregator_image_id: String,
     pub sp1_chunk_aggregator_vk_hash: String,
     pub monthly_budget_eth: f64,
-    pub db_connection_url: String,
+    pub db_connection_urls: Vec<String>,
 }
 
 impl Config {
diff --git a/aggregation_mode/proof_aggregator/src/backend/db.rs b/aggregation_mode/proof_aggregator/src/backend/db.rs
@@ -23,7 +23,7 @@ pub enum DbError {
 }
 
 impl Db {
-    pub async fn try_new(connection_urls: &[&str]) -> Result<Self, DbError> {
+    pub async fn try_new(connection_urls: &[String]) -> Result<Self, DbError> {
         let orchestrator = DbOrchestartor::try_new(
             connection_urls,
             RetryConfig {
diff --git a/aggregation_mode/proof_aggregator/src/backend/mod.rs b/aggregation_mode/proof_aggregator/src/backend/mod.rs
@@ -85,7 +85,7 @@ impl ProofAggregator {
         let engine =
             ZKVMEngine::from_env().expect("AGGREGATOR env variable to be set to one of sp1|risc0");
 
-        let db = Db::try_new(&[&config.db_connection_url])
+        let db = Db::try_new(config.db_connection_urls.as_slice())
             .await
             .expect("To connect to db");
 
diff --git a/config-files/config-agg-mode-gateway-ethereum-package.yaml b/config-files/config-agg-mode-gateway-ethereum-package.yaml
@@ -1,5 +1,6 @@
 port: 8089
-db_connection_url: "postgres://postgres:postgres@localhost:5435/"
+db_connection_urls: 
+  - "postgres://postgres:postgres@localhost:5435/"
 eth_rpc_url: "http://localhost:8545"
 payment_service_address: "0x922D6956C99E12DFeB3224DEA977D0939758A1Fe"
 network: "devnet"
diff --git a/config-files/config-agg-mode-gateway.yaml b/config-files/config-agg-mode-gateway.yaml
@@ -1,5 +1,6 @@
 port: 8089
-db_connection_url: "postgres://postgres:postgres@localhost:5435/"
+db_connection_urls: 
+  - "postgres://postgres:postgres@localhost:5435/"
 eth_rpc_url: "http://localhost:8545"
 payment_service_address: "0x922D6956C99E12DFeB3224DEA977D0939758A1Fe"
 network: "devnet"
diff --git a/config-files/config-proof-aggregator-ethereum-package.yaml b/config-files/config-proof-aggregator-ethereum-package.yaml
@@ -12,7 +12,8 @@ proofs_per_chunk: 512 # Amount of proofs to process per chunk
 # Since each proof commitments takes 32 bytes hash
 # We can aggregate as much proofs as 126.976 / 32 = 3968 per blob
 total_proofs_limit: 3968
-db_connection_url: "postgres://postgres:postgres@localhost:5435/"
+db_connection_urls: 
+  - "postgres://postgres:postgres@localhost:5435/"
 
 # Monthly ETH budget for on-chain proof verification. This value must be non-negative.
 # This value will be used to calculate how much gas we can spend per second to send proofs on-chain.
diff --git a/config-files/config-proof-aggregator.yaml b/config-files/config-proof-aggregator.yaml
@@ -12,7 +12,8 @@ proofs_per_chunk: 512 # Amount of proofs to process per chunk
 # Since each proof commitments takes 32 bytes hash
 # We can aggregate as much proofs as 126.976 / 32 = 3968 per blob
 total_proofs_limit: 3968
-db_connection_url: "postgres://postgres:postgres@localhost:5435/"
+db_connection_urls: 
+  - "postgres://postgres:postgres@localhost:5435/"
 
 # Monthly ETH budget for on-chain proof verification. This value must be non-negative.
 # This value will be used to calculate how much gas we can spend per second to send proofs on-chain.

Original file line number	Diff line number	Diff line change
`@@ -5,7 +5,7 @@ use serde::{Deserialize, Serialize};`
`5`	`5`	`#[derive(Clone, Debug, Deserialize, Serialize)]`
`6`	`6`	`pub struct Config {`
`7`	`7`	`pub port: u16,`
`8`		`- pub db_connection_url: String,`
	`8`	`+ pub db_connection_urls: Vec<String>,`
`9`	`9`	`pub network: String,`
`10`	`10`	`pub max_daily_proofs_per_user: i64,`
`11`	`11`	`}`
Original file line number	Diff line number	Diff line change
`@@ -14,7 +14,7 @@ pub enum DbError {`
`14`	`14`	`}`
`15`	`15`
`16`	`16`	`impl Db {`
`17`		`- pub async fn try_new(connection_urls: &[&str]) -> Result<Self, DbError> {`
	`17`	`+ pub async fn try_new(connection_urls: &[String]) -> Result<Self, DbError> {`
`18`	`18`	`let orchestrator = DbOrchestartor::try_new(`
`19`	`19`	`connection_urls,`
`20`	`20`	`RetryConfig {`
Original file line number	Diff line number	Diff line change
`@@ -4,7 +4,7 @@ use serde::{Deserialize, Serialize};`
`4`	`4`
`5`	`5`	`#[derive(Clone, Debug, Deserialize, Serialize)]`
`6`	`6`	`pub struct Config {`
`7`		`- pub db_connection_url: String,`
	`7`	`+ pub db_connection_urls: Vec<String>,`
`8`	`8`	`pub eth_rpc_url: String,`
`9`	`9`	`pub payment_service_address: String,`
`10`	`10`	`}`
Original file line number	Diff line number	Diff line change
`@@ -12,7 +12,7 @@ pub enum DbError {`
`12`	`12`	`}`
`13`	`13`
`14`	`14`	`impl Db {`
`15`		`- pub async fn try_new(connection_urls: &[&str]) -> Result<Self, DbError> {`
	`15`	`+ pub async fn try_new(connection_urls: &[String]) -> Result<Self, DbError> {`
`16`	`16`	`let orchestartor = DbOrchestartor::try_new(`
`17`	`17`	`connection_urls,`
`18`	`18`	`RetryConfig {`
Original file line number	Diff line number	Diff line change
`@@ -20,7 +20,7 @@ pub struct Config {`
`20`	`20`	`pub risc0_chunk_aggregator_image_id: String,`
`21`	`21`	`pub sp1_chunk_aggregator_vk_hash: String,`
`22`	`22`	`pub monthly_budget_eth: f64,`
`23`		`- pub db_connection_url: String,`
	`23`	`+ pub db_connection_urls: Vec<String>,`
`24`	`24`	`}`
`25`	`25`
`26`	`26`	`impl Config {`
Original file line number	Diff line number	Diff line change
`@@ -23,7 +23,7 @@ pub enum DbError {`
`23`	`23`	`}`
`24`	`24`
`25`	`25`	`impl Db {`
`26`		`- pub async fn try_new(connection_urls: &[&str]) -> Result<Self, DbError> {`
	`26`	`+ pub async fn try_new(connection_urls: &[String]) -> Result<Self, DbError> {`
`27`	`27`	`let orchestrator = DbOrchestartor::try_new(`
`28`	`28`	`connection_urls,`
`29`	`29`	`RetryConfig {`