use crate::Result;
#[derive(Clone, Debug)]
pub struct StorageControl<T>
where
T: super::stub::StorageControl + std::fmt::Debug + Send + Sync,
{
inner: T,
duration: gaxi::observability::DurationMetric,
}
impl<T> StorageControl<T>
where
T: super::stub::StorageControl + std::fmt::Debug + Send + Sync,
{
pub fn new(inner: T) -> Self {
Self {
inner,
duration: gaxi::observability::DurationMetric::new(&info::INSTRUMENTATION_CLIENT_INFO),
}
}
}
impl<T> super::stub::StorageControl for StorageControl<T>
where
T: super::stub::StorageControl + std::fmt::Debug + Send + Sync,
{
#[tracing::instrument(level = tracing::Level::DEBUG, ret)]
async fn create_folder(
&self,
req: crate::model::CreateFolderRequest,
options: crate::RequestOptions,
) -> Result<crate::Response<crate::model::Folder>> {
let (_span, pending) = gaxi::client_request_signals!(
metric: self.duration.clone(),
info: *info::INSTRUMENTATION_CLIENT_INFO,
method: "client::StorageControl::create_folder",
self.inner.create_folder(req, options));
pending.await
}
#[tracing::instrument(level = tracing::Level::DEBUG, ret)]
async fn delete_folder(
&self,
req: crate::model::DeleteFolderRequest,
options: crate::RequestOptions,
) -> Result<crate::Response<()>> {
let (_span, pending) = gaxi::client_request_signals!(
metric: self.duration.clone(),
info: *info::INSTRUMENTATION_CLIENT_INFO,
method: "client::StorageControl::delete_folder",
self.inner.delete_folder(req, options));
pending.await
}
#[tracing::instrument(level = tracing::Level::DEBUG, ret)]
async fn get_folder(
&self,
req: crate::model::GetFolderRequest,
options: crate::RequestOptions,
) -> Result<crate::Response<crate::model::Folder>> {
let (_span, pending) = gaxi::client_request_signals!(
metric: self.duration.clone(),
info: *info::INSTRUMENTATION_CLIENT_INFO,
method: "client::StorageControl::get_folder",
self.inner.get_folder(req, options));
pending.await
}
#[tracing::instrument(level = tracing::Level::DEBUG, ret)]
async fn list_folders(
&self,
req: crate::model::ListFoldersRequest,
options: crate::RequestOptions,
) -> Result<crate::Response<crate::model::ListFoldersResponse>> {
let (_span, pending) = gaxi::client_request_signals!(
metric: self.duration.clone(),
info: *info::INSTRUMENTATION_CLIENT_INFO,
method: "client::StorageControl::list_folders",
self.inner.list_folders(req, options));
pending.await
}
#[tracing::instrument(level = tracing::Level::DEBUG, ret)]
async fn rename_folder(
&self,
req: crate::model::RenameFolderRequest,
options: crate::RequestOptions,
) -> Result<crate::Response<google_cloud_longrunning::model::Operation>> {
let (_span, pending) = gaxi::client_request_signals!(
metric: self.duration.clone(),
info: *info::INSTRUMENTATION_CLIENT_INFO,
method: "client::StorageControl::rename_folder",
self.inner.rename_folder(req, options));
pending.await
}
#[tracing::instrument(level = tracing::Level::DEBUG, ret)]
async fn delete_folder_recursive(
&self,
req: crate::model::DeleteFolderRecursiveRequest,
options: crate::RequestOptions,
) -> Result<crate::Response<google_cloud_longrunning::model::Operation>> {
let (_span, pending) = gaxi::client_request_signals!(
metric: self.duration.clone(),
info: *info::INSTRUMENTATION_CLIENT_INFO,
method: "client::StorageControl::delete_folder_recursive",
self.inner.delete_folder_recursive(req, options));
pending.await
}
#[tracing::instrument(level = tracing::Level::DEBUG, ret)]
async fn get_storage_layout(
&self,
req: crate::model::GetStorageLayoutRequest,
options: crate::RequestOptions,
) -> Result<crate::Response<crate::model::StorageLayout>> {
let (_span, pending) = gaxi::client_request_signals!(
metric: self.duration.clone(),
info: *info::INSTRUMENTATION_CLIENT_INFO,
method: "client::StorageControl::get_storage_layout",
self.inner.get_storage_layout(req, options));
pending.await
}
#[tracing::instrument(level = tracing::Level::DEBUG, ret)]
async fn create_managed_folder(
&self,
req: crate::model::CreateManagedFolderRequest,
options: crate::RequestOptions,
) -> Result<crate::Response<crate::model::ManagedFolder>> {
let (_span, pending) = gaxi::client_request_signals!(
metric: self.duration.clone(),
info: *info::INSTRUMENTATION_CLIENT_INFO,
method: "client::StorageControl::create_managed_folder",
self.inner.create_managed_folder(req, options));
pending.await
}
#[tracing::instrument(level = tracing::Level::DEBUG, ret)]
async fn delete_managed_folder(
&self,
req: crate::model::DeleteManagedFolderRequest,
options: crate::RequestOptions,
) -> Result<crate::Response<()>> {
let (_span, pending) = gaxi::client_request_signals!(
metric: self.duration.clone(),
info: *info::INSTRUMENTATION_CLIENT_INFO,
method: "client::StorageControl::delete_managed_folder",
self.inner.delete_managed_folder(req, options));
pending.await
}
#[tracing::instrument(level = tracing::Level::DEBUG, ret)]
async fn get_managed_folder(
&self,
req: crate::model::GetManagedFolderRequest,
options: crate::RequestOptions,
) -> Result<crate::Response<crate::model::ManagedFolder>> {
let (_span, pending) = gaxi::client_request_signals!(
metric: self.duration.clone(),
info: *info::INSTRUMENTATION_CLIENT_INFO,
method: "client::StorageControl::get_managed_folder",
self.inner.get_managed_folder(req, options));
pending.await
}
#[tracing::instrument(level = tracing::Level::DEBUG, ret)]
async fn list_managed_folders(
&self,
req: crate::model::ListManagedFoldersRequest,
options: crate::RequestOptions,
) -> Result<crate::Response<crate::model::ListManagedFoldersResponse>> {
let (_span, pending) = gaxi::client_request_signals!(
metric: self.duration.clone(),
info: *info::INSTRUMENTATION_CLIENT_INFO,
method: "client::StorageControl::list_managed_folders",
self.inner.list_managed_folders(req, options));
pending.await
}
#[tracing::instrument(level = tracing::Level::DEBUG, ret)]
async fn create_anywhere_cache(
&self,
req: crate::model::CreateAnywhereCacheRequest,
options: crate::RequestOptions,
) -> Result<crate::Response<google_cloud_longrunning::model::Operation>> {
let (_span, pending) = gaxi::client_request_signals!(
metric: self.duration.clone(),
info: *info::INSTRUMENTATION_CLIENT_INFO,
method: "client::StorageControl::create_anywhere_cache",
self.inner.create_anywhere_cache(req, options));
pending.await
}
#[tracing::instrument(level = tracing::Level::DEBUG, ret)]
async fn update_anywhere_cache(
&self,
req: crate::model::UpdateAnywhereCacheRequest,
options: crate::RequestOptions,
) -> Result<crate::Response<google_cloud_longrunning::model::Operation>> {
let (_span, pending) = gaxi::client_request_signals!(
metric: self.duration.clone(),
info: *info::INSTRUMENTATION_CLIENT_INFO,
method: "client::StorageControl::update_anywhere_cache",
self.inner.update_anywhere_cache(req, options));
pending.await
}
#[tracing::instrument(level = tracing::Level::DEBUG, ret)]
async fn disable_anywhere_cache(
&self,
req: crate::model::DisableAnywhereCacheRequest,
options: crate::RequestOptions,
) -> Result<crate::Response<crate::model::AnywhereCache>> {
let (_span, pending) = gaxi::client_request_signals!(
metric: self.duration.clone(),
info: *info::INSTRUMENTATION_CLIENT_INFO,
method: "client::StorageControl::disable_anywhere_cache",
self.inner.disable_anywhere_cache(req, options));
pending.await
}
#[tracing::instrument(level = tracing::Level::DEBUG, ret)]
async fn pause_anywhere_cache(
&self,
req: crate::model::PauseAnywhereCacheRequest,
options: crate::RequestOptions,
) -> Result<crate::Response<crate::model::AnywhereCache>> {
let (_span, pending) = gaxi::client_request_signals!(
metric: self.duration.clone(),
info: *info::INSTRUMENTATION_CLIENT_INFO,
method: "client::StorageControl::pause_anywhere_cache",
self.inner.pause_anywhere_cache(req, options));
pending.await
}
#[tracing::instrument(level = tracing::Level::DEBUG, ret)]
async fn resume_anywhere_cache(
&self,
req: crate::model::ResumeAnywhereCacheRequest,
options: crate::RequestOptions,
) -> Result<crate::Response<crate::model::AnywhereCache>> {
let (_span, pending) = gaxi::client_request_signals!(
metric: self.duration.clone(),
info: *info::INSTRUMENTATION_CLIENT_INFO,
method: "client::StorageControl::resume_anywhere_cache",
self.inner.resume_anywhere_cache(req, options));
pending.await
}
#[tracing::instrument(level = tracing::Level::DEBUG, ret)]
async fn get_anywhere_cache(
&self,
req: crate::model::GetAnywhereCacheRequest,
options: crate::RequestOptions,
) -> Result<crate::Response<crate::model::AnywhereCache>> {
let (_span, pending) = gaxi::client_request_signals!(
metric: self.duration.clone(),
info: *info::INSTRUMENTATION_CLIENT_INFO,
method: "client::StorageControl::get_anywhere_cache",
self.inner.get_anywhere_cache(req, options));
pending.await
}
#[tracing::instrument(level = tracing::Level::DEBUG, ret)]
async fn list_anywhere_caches(
&self,
req: crate::model::ListAnywhereCachesRequest,
options: crate::RequestOptions,
) -> Result<crate::Response<crate::model::ListAnywhereCachesResponse>> {
let (_span, pending) = gaxi::client_request_signals!(
metric: self.duration.clone(),
info: *info::INSTRUMENTATION_CLIENT_INFO,
method: "client::StorageControl::list_anywhere_caches",
self.inner.list_anywhere_caches(req, options));
pending.await
}
#[tracing::instrument(level = tracing::Level::DEBUG, ret)]
async fn get_project_intelligence_config(
&self,
req: crate::model::GetProjectIntelligenceConfigRequest,
options: crate::RequestOptions,
) -> Result<crate::Response<crate::model::IntelligenceConfig>> {
let (_span, pending) = gaxi::client_request_signals!(
metric: self.duration.clone(),
info: *info::INSTRUMENTATION_CLIENT_INFO,
method: "client::StorageControl::get_project_intelligence_config",
self.inner.get_project_intelligence_config(req, options));
pending.await
}
#[tracing::instrument(level = tracing::Level::DEBUG, ret)]
async fn update_project_intelligence_config(
&self,
req: crate::model::UpdateProjectIntelligenceConfigRequest,
options: crate::RequestOptions,
) -> Result<crate::Response<crate::model::IntelligenceConfig>> {
let (_span, pending) = gaxi::client_request_signals!(
metric: self.duration.clone(),
info: *info::INSTRUMENTATION_CLIENT_INFO,
method: "client::StorageControl::update_project_intelligence_config",
self.inner.update_project_intelligence_config(req, options));
pending.await
}
#[tracing::instrument(level = tracing::Level::DEBUG, ret)]
async fn get_folder_intelligence_config(
&self,
req: crate::model::GetFolderIntelligenceConfigRequest,
options: crate::RequestOptions,
) -> Result<crate::Response<crate::model::IntelligenceConfig>> {
let (_span, pending) = gaxi::client_request_signals!(
metric: self.duration.clone(),
info: *info::INSTRUMENTATION_CLIENT_INFO,
method: "client::StorageControl::get_folder_intelligence_config",
self.inner.get_folder_intelligence_config(req, options));
pending.await
}
#[tracing::instrument(level = tracing::Level::DEBUG, ret)]
async fn update_folder_intelligence_config(
&self,
req: crate::model::UpdateFolderIntelligenceConfigRequest,
options: crate::RequestOptions,
) -> Result<crate::Response<crate::model::IntelligenceConfig>> {
let (_span, pending) = gaxi::client_request_signals!(
metric: self.duration.clone(),
info: *info::INSTRUMENTATION_CLIENT_INFO,
method: "client::StorageControl::update_folder_intelligence_config",
self.inner.update_folder_intelligence_config(req, options));
pending.await
}
#[tracing::instrument(level = tracing::Level::DEBUG, ret)]
async fn get_organization_intelligence_config(
&self,
req: crate::model::GetOrganizationIntelligenceConfigRequest,
options: crate::RequestOptions,
) -> Result<crate::Response<crate::model::IntelligenceConfig>> {
let (_span, pending) = gaxi::client_request_signals!(
metric: self.duration.clone(),
info: *info::INSTRUMENTATION_CLIENT_INFO,
method: "client::StorageControl::get_organization_intelligence_config",
self.inner.get_organization_intelligence_config(req, options));
pending.await
}
#[tracing::instrument(level = tracing::Level::DEBUG, ret)]
async fn update_organization_intelligence_config(
&self,
req: crate::model::UpdateOrganizationIntelligenceConfigRequest,
options: crate::RequestOptions,
) -> Result<crate::Response<crate::model::IntelligenceConfig>> {
let (_span, pending) = gaxi::client_request_signals!(
metric: self.duration.clone(),
info: *info::INSTRUMENTATION_CLIENT_INFO,
method: "client::StorageControl::update_organization_intelligence_config",
self.inner.update_organization_intelligence_config(req, options));
pending.await
}
#[tracing::instrument(level = tracing::Level::DEBUG, ret)]
async fn get_iam_policy(
&self,
req: google_cloud_iam_v1::model::GetIamPolicyRequest,
options: crate::RequestOptions,
) -> Result<crate::Response<google_cloud_iam_v1::model::Policy>> {
let (_span, pending) = gaxi::client_request_signals!(
metric: self.duration.clone(),
info: *info::INSTRUMENTATION_CLIENT_INFO,
method: "client::StorageControl::get_iam_policy",
self.inner.get_iam_policy(req, options));
pending.await
}
#[tracing::instrument(level = tracing::Level::DEBUG, ret)]
async fn set_iam_policy(
&self,
req: google_cloud_iam_v1::model::SetIamPolicyRequest,
options: crate::RequestOptions,
) -> Result<crate::Response<google_cloud_iam_v1::model::Policy>> {
let (_span, pending) = gaxi::client_request_signals!(
metric: self.duration.clone(),
info: *info::INSTRUMENTATION_CLIENT_INFO,
method: "client::StorageControl::set_iam_policy",
self.inner.set_iam_policy(req, options));
pending.await
}
#[tracing::instrument(level = tracing::Level::DEBUG, ret)]
async fn test_iam_permissions(
&self,
req: google_cloud_iam_v1::model::TestIamPermissionsRequest,
options: crate::RequestOptions,
) -> Result<crate::Response<google_cloud_iam_v1::model::TestIamPermissionsResponse>> {
let (_span, pending) = gaxi::client_request_signals!(
metric: self.duration.clone(),
info: *info::INSTRUMENTATION_CLIENT_INFO,
method: "client::StorageControl::test_iam_permissions",
self.inner.test_iam_permissions(req, options));
pending.await
}
#[tracing::instrument(level = tracing::Level::DEBUG, ret)]
async fn get_intelligence_finding(
&self,
req: crate::model::GetIntelligenceFindingRequest,
options: crate::RequestOptions,
) -> Result<crate::Response<crate::model::IntelligenceFinding>> {
let (_span, pending) = gaxi::client_request_signals!(
metric: self.duration.clone(),
info: *info::INSTRUMENTATION_CLIENT_INFO,
method: "client::StorageControl::get_intelligence_finding",
self.inner.get_intelligence_finding(req, options));
pending.await
}
#[tracing::instrument(level = tracing::Level::DEBUG, ret)]
async fn list_intelligence_findings(
&self,
req: crate::model::ListIntelligenceFindingsRequest,
options: crate::RequestOptions,
) -> Result<crate::Response<crate::model::ListIntelligenceFindingsResponse>> {
let (_span, pending) = gaxi::client_request_signals!(
metric: self.duration.clone(),
info: *info::INSTRUMENTATION_CLIENT_INFO,
method: "client::StorageControl::list_intelligence_findings",
self.inner.list_intelligence_findings(req, options));
pending.await
}
#[tracing::instrument(level = tracing::Level::DEBUG, ret)]
async fn summarize_intelligence_findings(
&self,
req: crate::model::SummarizeIntelligenceFindingsRequest,
options: crate::RequestOptions,
) -> Result<crate::Response<crate::model::SummarizeIntelligenceFindingsResponse>> {
let (_span, pending) = gaxi::client_request_signals!(
metric: self.duration.clone(),
info: *info::INSTRUMENTATION_CLIENT_INFO,
method: "client::StorageControl::summarize_intelligence_findings",
self.inner.summarize_intelligence_findings(req, options));
pending.await
}
#[tracing::instrument(level = tracing::Level::DEBUG, ret)]
async fn get_intelligence_finding_revision(
&self,
req: crate::model::GetIntelligenceFindingRevisionRequest,
options: crate::RequestOptions,
) -> Result<crate::Response<crate::model::IntelligenceFindingRevision>> {
let (_span, pending) = gaxi::client_request_signals!(
metric: self.duration.clone(),
info: *info::INSTRUMENTATION_CLIENT_INFO,
method: "client::StorageControl::get_intelligence_finding_revision",
self.inner.get_intelligence_finding_revision(req, options));
pending.await
}
#[tracing::instrument(level = tracing::Level::DEBUG, ret)]
async fn list_intelligence_finding_revisions(
&self,
req: crate::model::ListIntelligenceFindingRevisionsRequest,
options: crate::RequestOptions,
) -> Result<crate::Response<crate::model::ListIntelligenceFindingRevisionsResponse>> {
let (_span, pending) = gaxi::client_request_signals!(
metric: self.duration.clone(),
info: *info::INSTRUMENTATION_CLIENT_INFO,
method: "client::StorageControl::list_intelligence_finding_revisions",
self.inner.list_intelligence_finding_revisions(req, options));
pending.await
}
#[tracing::instrument(level = tracing::Level::DEBUG, ret)]
async fn get_operation(
&self,
req: google_cloud_longrunning::model::GetOperationRequest,
options: crate::RequestOptions,
) -> Result<crate::Response<google_cloud_longrunning::model::Operation>> {
let (_span, pending) = gaxi::client_request_signals!(
metric: self.duration.clone(),
info: *info::INSTRUMENTATION_CLIENT_INFO,
method: "client::StorageControl::get_operation",
self.inner.get_operation(req, options));
#[cfg(google_cloud_unstable_tracing)]
{
if let Ok(attempt) = google_cloud_lro::POLL_ATTEMPT_COUNT.try_with(|c| *c) {
_span.record("gcp.longrunning.poll_attempt_count", attempt);
_span.record("gcp.longrunning.done", false);
}
}
let result = pending.await;
#[cfg(google_cloud_unstable_tracing)]
{
if google_cloud_lro::POLL_ATTEMPT_COUNT
.try_with(|c| *c)
.is_ok()
{
match &result {
Ok(response) => {
let op = response.body();
_span.record("gcp.longrunning.done", op.done);
if op.done {
let code = match &op.result {
Some(
google_cloud_longrunning::model::operation::Result::Error(
status,
),
) => status.code,
_ => 0,
};
_span.record("gcp.longrunning.status_code", code);
if let Some(
google_cloud_longrunning::model::operation::Result::Error(status),
) = &op.result
{
_span.record("otel.status_code", "ERROR");
_span.record("otel.status_description", &status.message);
_span.record("rpc.response.status_code", status.code);
_span.record(
"error.type",
google_cloud_gax::error::rpc::Code::from(status.code)
.to_string(),
);
}
}
}
Err(e) => {
_span.record("otel.status_code", "ERROR");
_span.record("otel.status_description", e.to_string());
}
}
}
}
result
}
fn get_polling_error_policy(
&self,
options: &crate::RequestOptions,
) -> std::sync::Arc<dyn google_cloud_gax::polling_error_policy::PollingErrorPolicy> {
self.inner.get_polling_error_policy(options)
}
fn get_polling_backoff_policy(
&self,
options: &crate::RequestOptions,
) -> std::sync::Arc<dyn google_cloud_gax::polling_backoff_policy::PollingBackoffPolicy> {
self.inner.get_polling_backoff_policy(options)
}
}
pub(crate) mod info {
const NAME: &str = env!("CARGO_PKG_NAME");
const VERSION: &str = env!("CARGO_PKG_VERSION");
pub(crate) static INSTRUMENTATION_CLIENT_INFO: std::sync::LazyLock<
gaxi::options::InstrumentationClientInfo,
> = std::sync::LazyLock::new(|| {
let mut info = gaxi::options::InstrumentationClientInfo::default();
info.service_name = "storage";
info.client_version = VERSION;
info.client_artifact = NAME;
info.default_host = "storage";
info
});
}