Improve gRPC transport error logging (#4744)

quickwit/quickwit-codegen/example/src/lib.rs

@@ -624,6 +624,8 @@ mod tests {
 
     #[tokio::test]
     async fn test_transport_errors_handling() {
+        quickwit_common::setup_logging_for_tests();
+
         let addr: SocketAddr = "127.0.0.1:9999".parse().unwrap();
         let channel = Endpoint::from_static("http://127.0.0.1:9999")
             .timeout(Duration::from_millis(100))

quickwit/quickwit-codegen/src/codegen.rs

@@ -353,6 +353,16 @@ impl SynMethod {
         }
     }
 
+    fn rpc_name(&self, mock: bool) -> TokenStream {
+        let request_type = &self.request_type;
+
+        if mock {
+            quote! { super::#request_type::rpc_name() }
+        } else {
+            quote! { #request_type::rpc_name() }
+        }
+    }
+
     fn response_type(&self, context: &CodegenContext, mock: bool) -> TokenStream {
         let response_type = if mock {
             let response_type = &self.response_type;
@@ -1158,14 +1168,15 @@ fn generate_grpc_client_adapter_methods(context: &CodegenContext) -> TokenStream
     for syn_method in &context.methods {
         let method_name = syn_method.name.to_token_stream();
         let request_type = syn_method.request_type(false);
+        let rpc_name = syn_method.rpc_name(false);
         let response_type = syn_method.response_type(context, false);
 
         let into_response_type = if syn_method.server_streaming {
             quote! { |response|
                 {
                     let streaming: tonic::Streaming<_> = response.into_inner();
                     let stream = quickwit_common::ServiceStream::from(streaming);
-                    stream.map_err(crate::error::grpc_status_to_service_error)
+                    stream.map_err(|status| crate::error::grpc_status_to_service_error(status, #rpc_name))
                 }
             }
         } else {
@@ -1177,7 +1188,7 @@ fn generate_grpc_client_adapter_methods(context: &CodegenContext) -> TokenStream
                     .#method_name(request)
                     .await
                     .map(#into_response_type)
-                    .map_err(crate::error::grpc_status_to_service_error)
+                    .map_err(|status| crate::error::grpc_status_to_service_error(status, #rpc_name))
             }
         };
         stream.extend(method);

quickwit/quickwit-ingest/build.rs

@@ -30,6 +30,7 @@ fn main() {
         .with_result_type_path("crate::Result")
         .with_error_type_path("crate::IngestServiceError")
         .with_prost_config(prost_config)
+        .generate_rpc_name_impls()
         .run()
         .unwrap();
 }