feat: db migrations

Author: MarcosNicolau

Makefile

@@ -302,12 +302,19 @@ proof_aggregator_install: ## Install the aggregation mode with proving enabled
 proof_aggregator_write_program_ids: ## Write proof aggregator zkvm programs ids
 	@cd aggregation_mode/proof_aggregator && ./scripts/build_programs.sh
 
-agg_mode_batcher_start_local:
+agg_mode_docker_up:
 	@cd aggregation_mode && docker-compose up -d
+
+agg_mode_docker_down:
+	@cd aggregation_mode && docker-compose down
+
+agg_mode_run_migrations: agg_mode_docker_up
+	cargo run --manifest-path ./aggregation_mode/Cargo.toml --release --bin migrate -- postgres://postgres:postgres@localhost:5435/
+
+agg_mode_batcher_start_local: agg_mode_run_migrations
 	cargo run --manifest-path ./aggregation_mode/Cargo.toml --release --bin agg_mode_batcher -- config-files/config-agg-mode-batcher.yaml
 
-agg_mode_batcher_start_ethereum_package:
-	@cd aggregation_mode && docker-compose up -d
+agg_mode_batcher_start_ethereum_package: agg_mode_run_migrations
 	cargo run --manifest-path ./aggregation_mode/Cargo.toml --release --bin agg_mode_batcher -- config-files/config-agg-mode-batcher-ethereum-package.yaml
 
 __AGGREGATOR__: ## ____

aggregation_mode/Cargo.lock

@@ -1,6 +1,6 @@
 [workspace]
 resolver = "2"
-members = ["./batcher", "./proof_aggregator"]
+members = ["./batcher", "./proof_aggregator", "./db"]
 
 [workspace.package]
 version = "0.1.0"

aggregation_mode/Cargo.toml

@@ -0,0 +1,14 @@
+[package]
+name = "agg_mode_db"
+version = "0.1.0"
+edition = "2021"
+
+[dependencies]
+tokio = { version = "1"}
+# TODO: enable tls
+sqlx = { version = "0.8", features = [ "runtime-tokio", "postgres", "migrate" ] }
+
+
+[[bin]]
+name = "migrate"
+path = "cmd/migrate.rs"

aggregation_mode/db/Cargo.toml

@@ -0,0 +1,27 @@
+use sqlx::postgres::PgPoolOptions;
+
+fn read_db_conn_from_args() -> String {
+    let args: Vec<String> = std::env::args().collect();
+    if args.len() < 2 {
+        panic!(
+            "You mus provide a config file. Usage: {} <config-file-path>",
+            args[0]
+        );
+    }
+
+    args[1].clone()
+}
+
+#[tokio::main]
+async fn main() {
+    let db_connection_url = read_db_conn_from_args();
+    let pool = PgPoolOptions::new()
+        .connect(&db_connection_url)
+        .await
+        .expect("To connect to db");
+
+    sqlx::migrate!("./migrations")
+        .run(&pool)
+        .await
+        .expect("Migrations to run correctly")
+}

aggregation_mode/db/cmd/migrate.rs

@@ -0,0 +1,24 @@
+CREATE TYPE task_status AS ENUM ('pending', 'verified');
+
+CREATE TABLE tasks (
+    task_id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
+    status task_status
+);
+
+CREATE TABLE proofs (
+    proof_id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
+    address CHAR(42), 
+    proving_system_id INT,
+    vm_program_code BYTEA,
+    proof BYTEA,
+    public_inputs BYTEA,
+    proof_commitment BYTEA,
+    task_id UUID REFERENCES tasks(task_id)
+);
+
+CREATE TABLE payment_events (
+    payment_event_id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
+    address CHAR(42),
+    started_at INTEGER,
+    valid_until INTEGER
+);

aggregation_mode/db/migrations/001_init.sql

@@ -20,4 +20,4 @@ services:
     depends_on:
       - postgres
     ports:
-      - 8090:8090
+      - 8090:8080