fix(kms-connector): add more logs

Author: eudelins-zama

kms-connector/simple-connector/src/gw_adapters/decryption/adapter.rs

@@ -65,10 +65,11 @@ impl<P: Provider + Clone> DecryptionAdapter<P> {
 
         let contract = Decryption::new(self.decryption_address, self.provider.clone());
 
-        let mut call = contract
-            .publicDecryptionResponse(id, result, signature.into())
-            .into_transaction_request();
-        call.gas = Some(INFINITE_GAS_LIMIT);
+        let call_builder = contract.publicDecryptionResponse(id, result, signature.into());
+        info!(decryption_id = ?id, "public decryption calldata length {}", call_builder.calldata().len());
+
+        let mut call = call_builder.into_transaction_request();
+        self.estimate_gas(id, &mut call).await;
         let tx = self.send_tx_with_retry(call).await?;
 
         // TODO: optimize for low latency
@@ -77,6 +78,7 @@ impl<P: Provider + Clone> DecryptionAdapter<P> {
             .await
             .map_err(|e| Error::Contract(e.to_string()))?;
         info!(decryption_id = ?id, "🎯 Public Decryption response sent with tx receipt: {:?}", receipt);
+        info!(decryption_id = ?id, "⛽ Gas consumed for Public Decryption: {}", receipt.gas_used);
         Ok(())
     }
 
@@ -110,10 +112,11 @@ impl<P: Provider + Clone> DecryptionAdapter<P> {
         let contract = Decryption::new(self.decryption_address, self.provider.clone());
 
         // Create and send transaction
-        let mut call = contract
-            .userDecryptionResponse(id, result, signature.into())
-            .into_transaction_request();
-        call.gas = Some(INFINITE_GAS_LIMIT);
+        let call_builder = contract.userDecryptionResponse(id, result, signature.into());
+        info!(decryption_id = ?id, "user decryption calldata length {}", call_builder.calldata().len());
+
+        let mut call = call_builder.into_transaction_request();
+        self.estimate_gas(id, &mut call).await;
         let tx = self.send_tx_with_retry(call).await?;
 
         // TODO: optimize for low latency
@@ -122,9 +125,24 @@ impl<P: Provider + Clone> DecryptionAdapter<P> {
             .await
             .map_err(|e| Error::Contract(e.to_string()))?;
         info!(decryption_id = ?id, "🎯 User Decryption response sent with tx receipt: {:?}", receipt);
+        info!(decryption_id = ?id, "⛽ Gas consumed for User Decryption: {}", receipt.gas_used);
         Ok(())
     }
 
+    /// Estimates the `gas_limit` for the upcoming transaction.
+    async fn estimate_gas(&self, id: U256, call: &mut TransactionRequest) {
+        match self.provider.estimate_gas(call.clone()).await {
+            Ok(gas) => info!(decryption_id = ?id, "Initial gas estimation for the tx: {gas}"),
+            Err(e) => warn!(decryption_id = ?id, "Failed to estimate gas for the tx: {e}"),
+        }
+
+        // TODO: temporary workaround for out-of-gas errors
+        // Our automatic estimation fails during gas pikes.
+        // (see https://zama-ai.slack.com/archives/C0915Q59CKG/p1749843623276629?thread_ts=1749828466.079719&cid=C0915Q59CKG)
+        info!(decryption_id = ?id, "Updating `gas_limit` to max value");
+        call.gas = Some(INFINITE_GAS_LIMIT);
+    }
+
     /// Sends the requested transactions with one retry.
     async fn send_tx_with_retry(
         &self,