fix(prover-gateway): Check for batch existence in DB & add tracing for axum (#4270)

## What ❔

<!-- What are the changes this PR brings about? -->
<!-- Example: This PR adds a PR template to the repo. -->
<!-- (For bigger PRs adding more context is appreciated) -->

## Why ❔

<!-- Why are these changes done? What goal do they contribute to? What
are the principles behind them? -->
<!-- The `Why` has to be clear to non-Matter Labs entities running their
own ZK Chain -->
<!-- Example: PR templates ensure PR reviewers, observers, and future
iterators are in context about the evolution of repos. -->

## Is this a breaking change?
- [ ] Yes
- [x] No

## Operational changes
<!-- Any config changes? Any new flags? Any changes to any scripts? -->
<!-- Please add anything that non-Matter Labs entities running their own
ZK Chain may need to know -->

## Checklist

<!-- Check your PR fulfills the following items. -->
<!-- For draft PRs check the boxes as you complete them. -->

- [ ] PR title corresponds to the body of PR (we generate changelog
entries from PRs).
- [ ] Tests for the changes have been added / updated.
- [ ] Documentation comments have been added / updated.
- [ ] Code has been formatted via `zkstack dev fmt` and `zkstack dev
lint`.

Author: Artemka374

prover/Cargo.lock

@@ -60,6 +60,7 @@ tokio = "1"
 tokio-util = "0.7.11"
 tokio-stream = "0.1.16"
 toml_edit = "0.14.4"
+tower-http = "0.5.2"
 tracing = "0.1"
 tracing-subscriber = "0.3"
 tracing-test = "0.2.5"

prover/Cargo.toml

@@ -22,6 +22,7 @@ zksync_vlog.workspace = true
 axum.workspace = true
 anyhow.workspace = true
 tracing.workspace = true
+tower-http = { workspace = true, features = ["trace"] }
 reqwest = { workspace = true, features = ["blocking"] }
 tokio = { workspace = true, features = ["time", "macros"] }
 thiserror.workspace = true

prover/crates/bin/prover_fri_gateway/Cargo.toml

@@ -79,8 +79,8 @@ impl ProofDataManager {
             .connection()
             .await
             .unwrap()
-            .fri_basic_witness_generator_dal()
-            .protocol_version_for_l1_batch(batch_id)
+            .fri_proof_compressor_dal()
+            .protocol_version_for_successful_proof(batch_id)
             .await;
 
         let Some(protocol_version) = protocol_version else {

prover/crates/bin/prover_fri_gateway/src/proof_data_manager.rs

@@ -7,6 +7,7 @@ use axum::{
     Json, Router,
 };
 use tokio::sync::watch;
+use tower_http::trace::TraceLayer;
 use zksync_prover_interface::api::{
     PollGeneratedProofsRequest, PollGeneratedProofsResponse, ProofGenerationData,
     SubmitProofGenerationDataResponse,
@@ -27,6 +28,7 @@ impl Api {
                 "/submit_request_for_proofs",
                 post(Api::save_proof_generation_data),
             )
+            .layer(TraceLayer::new_for_http())
             .layer(DefaultBodyLimit::disable())
             .with_state(processor);
 

prover/crates/bin/prover_fri_gateway/src/server.rs

@@ -228,6 +228,30 @@ impl FriProofCompressorDal<'_, '_> {
         }
     }
 
+    pub async fn protocol_version_for_successful_proof(
+        &mut self,
+        batch_id: L1BatchId,
+    ) -> Option<ProtocolSemanticVersion> {
+        sqlx::query!(
+            r#"
+            SELECT protocol_version, protocol_version_patch
+            FROM proof_compression_jobs_fri
+            WHERE l1_batch_number = $1 AND chain_id = $2 AND status = 'successful'
+            "#,
+            batch_id.batch_number().0 as i64,
+            batch_id.chain_id().inner() as i64,
+        )
+        .fetch_optional(self.storage.conn())
+        .await
+        .unwrap()
+        .map(|row| {
+            ProtocolSemanticVersion::new(
+                ProtocolVersionId::try_from(row.protocol_version.unwrap() as u16).unwrap(),
+                VersionPatch(row.protocol_version_patch as u32),
+            )
+        })
+    }
+
     pub async fn mark_proof_sent_to_server(&mut self, batch_id: L1BatchId) -> Result<(), DalError> {
         sqlx::query!(
             r#"