feat: initial structs for db integration

Author: MarcosNicolau

aggregation_mode/Cargo.lock

@@ -1,4 +1,4 @@
-CREATE TYPE task_status AS ENUM ('pending', 'processing', 'verified');
+CREATE TYPE task_status AS ENUM ('pending', 'processing', 'verified', 'submitted');
 
 CREATE TABLE tasks (
     task_id UUID PRIMARY KEY DEFAULT gen_random_uuid(),

aggregation_mode/db/migrations/001_init.sql

@@ -21,6 +21,7 @@ ciborium = "=0.2.2"
 lambdaworks-crypto = { git = "https://github.com/lambdaclass/lambdaworks.git", rev = "5f8f2cfcc8a1a22f77e8dff2d581f1166eefb80b", features = ["serde"]}
 rayon = "1.10.0"
 backon = "1.2.0"
+sqlx = { version = "0.8", features = [ "runtime-tokio", "postgres", "uuid", "bigdecimal" ] }
 
 # zkvms
 sp1-sdk = { workspace = true }

aggregation_mode/proof_aggregator/Cargo.toml

@@ -25,6 +25,7 @@ pub struct Config {
     pub total_proofs_limit: u16,
     pub sp1_chunk_aggregator_vk_hash: String,
     pub risc0_chunk_aggregator_image_id: String,
+    pub db_connection_url: String,
 }
 
 impl Config {

aggregation_mode/proof_aggregator/src/backend/config.rs

@@ -0,0 +1,37 @@
+use sqlx::{
+    postgres::PgPoolOptions,
+    types::{BigDecimal, Uuid},
+    Pool, Postgres,
+};
+
+#[derive(Clone, Debug)]
+pub struct Db {
+    pool: Pool<Postgres>,
+}
+
+#[derive(Debug, Clone)]
+pub enum DbError {
+    ConnectError(String),
+}
+
+impl Db {
+    pub async fn try_new(connection_url: &str) -> Result<Self, DbError> {
+        let pool = PgPoolOptions::new()
+            .max_connections(5)
+            .connect(connection_url)
+            .await
+            .map_err(|e| DbError::ConnectError(e.to_string()))?;
+
+        Ok(Self { pool })
+    }
+
+    pub async fn get_tasks_and_mark_them_as_processed() {}
+
+    pub async fn mark_tasks_as_pending() {}
+
+    pub async fn mark_tasks_as_processing() {}
+
+    pub async fn mark_tasks_as_verified() {}
+
+    pub async fn mark_tasks_as_submitted() {}
+}

aggregation_mode/proof_aggregator/src/backend/db.rs

@@ -1,11 +1,15 @@
 pub mod config;
+mod db;
 pub mod fetcher;
 mod merkle_tree;
 mod retry;
 mod s3;
 mod types;
 
-use crate::aggregators::{AlignedProof, ProofAggregationError, ZKVMEngine};
+use crate::{
+    aggregators::{AlignedProof, ProofAggregationError, ZKVMEngine},
+    backend::db::Db,
+};
 
 use alloy::{
     consensus::{BlobTransactionSidecar, EnvKzgSettings, EthereumTxEnvelope, TxEip4844WithSidecar},
@@ -80,6 +84,8 @@ impl ProofAggregator {
                 .try_into()
                 .expect("Risc0 chunk aggregator image id must be 32 bytes");
 
+        let db = Db::try_new(self.config.db_connection_url).expect("To connect to db");
+
         Self {
             engine,
             proof_aggregation_service,

aggregation_mode/proof_aggregator/src/backend/mod.rs

@@ -12,6 +12,7 @@ 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/"
 
 # These program ids are the ones from the chunk aggregator programs
 # Can be found in the Proof Aggregation Service deployment config

config-files/config-proof-aggregator-ethereum-package.yaml

@@ -12,6 +12,7 @@ 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/"
 
 # These program ids are the ones from the chunk aggregator programs
 # Can be found in the Proof Aggregation Service deployment config