From 7e95a3a66ea48be7b6059d34630e22c503399bdf Mon Sep 17 00:00:00 2001
From: Roman Brodetski <rb@matterlabs.dev>
Date: Sat, 25 Nov 2023 14:00:08 +0400
Subject: [PATCH] feat(witness-generator):  add logs to leaf aggregation job
 (#542)
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

## What ❔

Add some log lines to leaf aggreagtion prover job

## Why ❔

It doesn't save anything to the database and I want to understand why
---
 .../witness_generator/src/leaf_aggregation.rs | 36 +++++++++++++++++++
 1 file changed, 36 insertions(+)

diff --git a/prover/witness_generator/src/leaf_aggregation.rs b/prover/witness_generator/src/leaf_aggregation.rs
index 084a7bd2d4a7..af742775c841 100644
--- a/prover/witness_generator/src/leaf_aggregation.rs
+++ b/prover/witness_generator/src/leaf_aggregation.rs
@@ -150,6 +150,11 @@ impl JobProcessor for LeafAggregationWitnessGenerator {
         let block_number = artifacts.block_number;
         let circuit_id = artifacts.circuit_id;
         let blob_urls = save_artifacts(artifacts, &*self.object_store).await;
+        tracing::info!(
+            "Saved leaf aggregation artifacts for block {} with circuit {}",
+            block_number.0,
+            circuit_id,
+        );
         update_database(
             &self.prover_connection_pool,
             started_at,
@@ -266,6 +271,12 @@ async fn update_database(
     blob_urls: BlobUrls,
     circuit_id: u8,
 ) {
+    tracing::info!(
+        "Updating database for job_id {}, block {} with circuit id {}",
+        job_id,
+        block_number.0,
+        circuit_id,
+    );
     let mut prover_connection = prover_connection_pool.access_storage().await.unwrap();
     let mut transaction = prover_connection.start_transaction().await.unwrap();
     let number_of_dependent_jobs = blob_urls.circuit_ids_and_urls.len();
@@ -273,6 +284,13 @@ async fn update_database(
         .fri_witness_generator_dal()
         .protocol_version_for_l1_batch(block_number)
         .await;
+    tracing::info!(
+        "Inserting {} prover jobs for job_id {}, block {} with circuit id {}",
+        blob_urls.circuit_ids_and_urls.len(),
+        job_id,
+        block_number.0,
+        circuit_id,
+    );
     transaction
         .fri_prover_jobs_dal()
         .insert_prover_jobs(
@@ -283,6 +301,12 @@ async fn update_database(
             protocol_version_id,
         )
         .await;
+    tracing::info!(
+        "Updating node aggregation jobs url for job_id {}, block {} with circuit id {}",
+        job_id,
+        block_number.0,
+        circuit_id,
+    );
     transaction
         .fri_witness_generator_dal()
         .update_node_aggregation_jobs_url(
@@ -293,11 +317,23 @@ async fn update_database(
             blob_urls.aggregations_urls,
         )
         .await;
+    tracing::info!(
+        "Marking leaf aggregation job as successful for job id {}, block {} with circuit id {}",
+        job_id,
+        block_number.0,
+        circuit_id,
+    );
     transaction
         .fri_witness_generator_dal()
         .mark_leaf_aggregation_as_successful(job_id, started_at.elapsed())
         .await;
 
+    tracing::info!(
+        "Committing transaction for job_id {}, block {} with circuit id {}",
+        job_id,
+        block_number.0,
+        circuit_id,
+    );
     transaction.commit().await.unwrap();
 }