feat(sidekick/rust): use RequestRecorder (#4883)

In the o11y project, we are adding `RequestRecorder` a new way to
capture the data used in o11y signals (spans, logs, and metrics). This
starts using it in the generated code.

Author: coryan

internal/sidekick/rust/templates/crate/src/tracing.rs.mustache

@@ -79,18 +79,12 @@ where T: super::stub::{{Codec.Name}} + std::fmt::Debug + Send + Sync {
         {{#Codec.DetailedTracingAttributes}}
         #[cfg(google_cloud_unstable_tracing)]
         {
-            use gaxi::observability::ClientSignalsExt as _;
-            let (start, span) = gaxi::client_request_signals!(
-                &info::INSTRUMENTATION_CLIENT_INFO,
-                &options,
-                "client::{{Codec.ServiceNameToPascal}}",
-                "{{Codec.Name}}",
-                Some("{{SourceService.Package}}.{{SourceService.Name}}/{{Name}}")
-            );
-            self.inner
-                .{{Codec.Name}}(req, options)
-                .instrument_client(self.duration.clone(), start, span)
-                .await
+            let (_span, pending) = gaxi::client_request_signals!(
+                metric: self.duration.clone(),
+                info: *info::INSTRUMENTATION_CLIENT_INFO,
+                method: "client::{{Codec.ServiceNameToPascal}}::{{Codec.Name}}",
+                self.inner.{{Codec.Name}}(req, options));
+            pending.await
         }
         #[cfg(not(google_cloud_unstable_tracing))]
         {{/Codec.DetailedTracingAttributes }}

internal/sidekick/rust/templates/crate/src/transport.rs.mustache

@@ -103,10 +103,10 @@ impl super::stub::{{Codec.Name}} for {{Codec.Name}} {
         {{/HasAutoPopulatedFields}}
         {{#Codec.HasResourceNameGeneration}}
         {{#Codec.DetailedTracingAttributes}}
-        let (builder, method, _path_template, resource_name) = None
+        let (builder, method, _path_template, _resource_name) = None
         {{/Codec.DetailedTracingAttributes}}
         {{^Codec.DetailedTracingAttributes}}
-        let (builder, method, resource_name) = None
+        let (builder, method, _resource_name) = None
         {{/Codec.DetailedTracingAttributes}}
         {{/Codec.HasResourceNameGeneration}}
         {{^Codec.HasResourceNameGeneration}}
@@ -202,19 +202,17 @@ impl super::stub::{{Codec.Name}} for {{Codec.Name}} {
         })??;
         {{#Codec.DetailedTracingAttributes}}
         #[cfg(google_cloud_unstable_tracing)]
-        let options = {
-            use google_cloud_gax::options::internal::{RequestOptionsExt, PathTemplate};
-            options.insert_extension(PathTemplate(_path_template))
-        };
+        if let Some(recorder) = gaxi::observability::RequestRecorder::current() {
+            recorder.on_client_request(
+                gaxi::observability::ClientRequestAttributes::default()
+                    .set_rpc_method("{{SourceService.Package}}.{{SourceService.Name}}/{{Name}}")
+                    .set_url_template(_path_template)
+                    {{#Codec.HasResourceNameGeneration}}
+                    .set_resource_name(_resource_name)
+                    {{/Codec.HasResourceNameGeneration}}
+            );
+        }
         {{/Codec.DetailedTracingAttributes}}
-        {{#Codec.HasResourceNameGeneration}}
-        let options = if !resource_name.is_empty() {
-            use google_cloud_gax::options::internal::{RequestOptionsExt, ResourceName};
-            options.insert_extension(ResourceName(resource_name))
-        } else {
-            options
-        };
-        {{/Codec.HasResourceNameGeneration}}
         let options = google_cloud_gax::options::internal::set_default_idempotency(
             options,
             {{! TODO(#4156) - return idempotency from the above closure }}

internal/sidekick/rust/templates/grpc-client/transport.rs.mustache

@@ -165,7 +165,9 @@ impl super::stub::{{Codec.Name}} for {{Codec.Name}} {
         {{/ReturnsEmpty}}
         {{#Codec.DetailedTracingAttributes}}
         #[cfg(google_cloud_unstable_tracing)]
-        let options = {
+        if let Some(recorder) = gaxi::observability::RequestRecorder::current() {
+            let attributes = gaxi::observability::ClientRequestAttributes::default()
+                .set_rpc_method("{{SourceService.Package}}.{{SourceService.Name}}/{{Name}}");
             {{#Codec.HasGrpcResourceNameArgs}}
             let resource_name = (|| {
                 {{#Codec.ResourceNameTemplateGrpc}}
@@ -180,17 +182,14 @@ impl super::stub::{{Codec.Name}} for {{Codec.Name}} {
                 Some(String::new())
                 {{/Codec.ResourceNameTemplateGrpc}}
             })();
-            if let Some(rn) = resource_name.filter(|s| !s.is_empty()) {
-                use google_cloud_gax::options::internal::{RequestOptionsExt, ResourceName};
-                options.insert_extension(ResourceName(rn))
+            let attributes = if let Some(rn) = resource_name.filter(|s| !s.is_empty()) {
+                attributes.set_resource_name(rn)
             } else {
-                options
-            }
-            {{/Codec.HasGrpcResourceNameArgs}}
-            {{^Codec.HasGrpcResourceNameArgs}}
-            options
+                attributes
+            };
             {{/Codec.HasGrpcResourceNameArgs}}
-        };
+            recorder.on_client_request(attributes);
+        }
         {{/Codec.DetailedTracingAttributes}}
         self.inner
             .execute(