feat(aggregation-mode): daily proofs limit

Author: maximopalopoli

aggregation_mode/batcher/src/db.rs

@@ -96,6 +96,21 @@ impl Db {
         .await
     }
 
+    pub async fn get_daily_tasks_by_address(
+        &self,
+        address: &str,
+    ) -> Result<Vec<Receipt>, sqlx::Error> {
+        sqlx::query_as::<_, Receipt>(
+            "SELECT status,merkle_path,nonce,address
+            FROM tasks
+            WHERE address = $1
+            AND inserted_at::date = CURRENT_DATE",
+        )
+        .bind(address.to_lowercase())
+        .fetch_all(&self.pool)
+        .await
+    }
+
     pub async fn insert_task(
         &self,
         address: &str,

aggregation_mode/batcher/src/server/http.rs

@@ -101,6 +101,30 @@ impl BatcherServer {
         };
         let state = state.get_ref();
 
+        // Checking if this address has submited more proofs than the ones allowed per day
+        const MAX_PROOFS_PER_DAY: usize = 4;
+
+        let daily_tasks_by_address = match state
+            .db
+            .get_daily_tasks_by_address(&recovered_address)
+            .await
+        {
+            Ok(receipts) => receipts.len(),
+            Err(_) => {
+                return HttpResponse::InternalServerError().json(AppResponse::new_unsucessfull(
+                    format!("Internal server error").as_str(),
+                    500,
+                ))
+            }
+        };
+
+        if daily_tasks_by_address >= MAX_PROOFS_PER_DAY {
+            return HttpResponse::InternalServerError().json(AppResponse::new_unsucessfull(
+                format!("Request denied: Query limit exceeded.").as_str(),
+                400,
+            ));
+        }
+
         let Ok(count) = state.db.count_tasks_by_address(&recovered_address).await else {
             return HttpResponse::InternalServerError()
                 .json(AppResponse::new_unsucessfull("Internal server error", 500));

aggregation_mode/db/migrations/001_init.sql

@@ -8,7 +8,8 @@ CREATE TABLE tasks (
     program_commitment BYTEA,
     merkle_path BYTEA,
     status task_status DEFAULT 'pending',
-    nonce BIGINT NOT NULL
+    nonce BIGINT NOT NULL,
+    inserted_at TIMESTAMPTZ NOT NULL DEFAULT now()
 );
 
 CREATE TABLE payment_events (
@@ -17,5 +18,6 @@ CREATE TABLE payment_events (
     amount BIGINT,
     started_at BIGINT,
     valid_until BIGINT,
-    tx_hash CHAR(66) UNIQUE
+    tx_hash CHAR(66) UNIQUE,
+    inserted_at TIMESTAMPTZ NOT NULL DEFAULT now()
 );