google-cloud-spanner 0.34.1-preview

// Copyright 2026 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//
// Code generated by sidekick. DO NOT EDIT.
use crate::Result;

/// Implements a [Spanner](super::stub::Spanner) decorator for logging and tracing.
#[derive(Clone, Debug)]
pub struct Spanner<T>
where
    T: super::stub::Spanner + std::fmt::Debug + Send + Sync,
{
    inner: T,
    duration: gaxi::observability::DurationMetric,
}

impl<T> Spanner<T>
where
    T: super::stub::Spanner + 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::Spanner for Spanner<T>
where
    T: super::stub::Spanner + std::fmt::Debug + Send + Sync,
{
    #[tracing::instrument(level = tracing::Level::DEBUG, ret)]
    async fn create_session(
        &self,
        req: crate::model::CreateSessionRequest,
        options: crate::RequestOptions,
    ) -> Result<crate::Response<crate::model::Session>> {
        let (_span, pending) = gaxi::client_request_signals!(
            metric: self.duration.clone(),
            info: *info::INSTRUMENTATION_CLIENT_INFO,
            method: "client::Spanner::create_session",
            self.inner.create_session(req, options));
        pending.await
    }

    #[tracing::instrument(level = tracing::Level::DEBUG, ret)]
    async fn batch_create_sessions(
        &self,
        req: crate::model::BatchCreateSessionsRequest,
        options: crate::RequestOptions,
    ) -> Result<crate::Response<crate::model::BatchCreateSessionsResponse>> {
        let (_span, pending) = gaxi::client_request_signals!(
            metric: self.duration.clone(),
            info: *info::INSTRUMENTATION_CLIENT_INFO,
            method: "client::Spanner::batch_create_sessions",
            self.inner.batch_create_sessions(req, options));
        pending.await
    }

    #[tracing::instrument(level = tracing::Level::DEBUG, ret)]
    async fn get_session(
        &self,
        req: crate::model::GetSessionRequest,
        options: crate::RequestOptions,
    ) -> Result<crate::Response<crate::model::Session>> {
        let (_span, pending) = gaxi::client_request_signals!(
            metric: self.duration.clone(),
            info: *info::INSTRUMENTATION_CLIENT_INFO,
            method: "client::Spanner::get_session",
            self.inner.get_session(req, options));
        pending.await
    }

    #[tracing::instrument(level = tracing::Level::DEBUG, ret)]
    async fn list_sessions(
        &self,
        req: crate::model::ListSessionsRequest,
        options: crate::RequestOptions,
    ) -> Result<crate::Response<crate::model::ListSessionsResponse>> {
        let (_span, pending) = gaxi::client_request_signals!(
            metric: self.duration.clone(),
            info: *info::INSTRUMENTATION_CLIENT_INFO,
            method: "client::Spanner::list_sessions",
            self.inner.list_sessions(req, options));
        pending.await
    }

    #[tracing::instrument(level = tracing::Level::DEBUG, ret)]
    async fn delete_session(
        &self,
        req: crate::model::DeleteSessionRequest,
        options: crate::RequestOptions,
    ) -> Result<crate::Response<()>> {
        let (_span, pending) = gaxi::client_request_signals!(
            metric: self.duration.clone(),
            info: *info::INSTRUMENTATION_CLIENT_INFO,
            method: "client::Spanner::delete_session",
            self.inner.delete_session(req, options));
        pending.await
    }

    #[tracing::instrument(level = tracing::Level::DEBUG, ret)]
    async fn execute_sql(
        &self,
        req: crate::model::ExecuteSqlRequest,
        options: crate::RequestOptions,
    ) -> Result<crate::Response<crate::model::ResultSet>> {
        let (_span, pending) = gaxi::client_request_signals!(
            metric: self.duration.clone(),
            info: *info::INSTRUMENTATION_CLIENT_INFO,
            method: "client::Spanner::execute_sql",
            self.inner.execute_sql(req, options));
        pending.await
    }

    #[tracing::instrument(level = tracing::Level::DEBUG, ret)]
    async fn execute_batch_dml(
        &self,
        req: crate::model::ExecuteBatchDmlRequest,
        options: crate::RequestOptions,
    ) -> Result<crate::Response<crate::model::ExecuteBatchDmlResponse>> {
        let (_span, pending) = gaxi::client_request_signals!(
            metric: self.duration.clone(),
            info: *info::INSTRUMENTATION_CLIENT_INFO,
            method: "client::Spanner::execute_batch_dml",
            self.inner.execute_batch_dml(req, options));
        pending.await
    }

    #[tracing::instrument(level = tracing::Level::DEBUG, ret)]
    async fn read(
        &self,
        req: crate::model::ReadRequest,
        options: crate::RequestOptions,
    ) -> Result<crate::Response<crate::model::ResultSet>> {
        let (_span, pending) = gaxi::client_request_signals!(
            metric: self.duration.clone(),
            info: *info::INSTRUMENTATION_CLIENT_INFO,
            method: "client::Spanner::read",
            self.inner.read(req, options));
        pending.await
    }

    #[tracing::instrument(level = tracing::Level::DEBUG, ret)]
    async fn begin_transaction(
        &self,
        req: crate::model::BeginTransactionRequest,
        options: crate::RequestOptions,
    ) -> Result<crate::Response<crate::model::Transaction>> {
        let (_span, pending) = gaxi::client_request_signals!(
            metric: self.duration.clone(),
            info: *info::INSTRUMENTATION_CLIENT_INFO,
            method: "client::Spanner::begin_transaction",
            self.inner.begin_transaction(req, options));
        pending.await
    }

    #[tracing::instrument(level = tracing::Level::DEBUG, ret)]
    async fn commit(
        &self,
        req: crate::model::CommitRequest,
        options: crate::RequestOptions,
    ) -> Result<crate::Response<crate::model::CommitResponse>> {
        let (_span, pending) = gaxi::client_request_signals!(
            metric: self.duration.clone(),
            info: *info::INSTRUMENTATION_CLIENT_INFO,
            method: "client::Spanner::commit",
            self.inner.commit(req, options));
        pending.await
    }

    #[tracing::instrument(level = tracing::Level::DEBUG, ret)]
    async fn rollback(
        &self,
        req: crate::model::RollbackRequest,
        options: crate::RequestOptions,
    ) -> Result<crate::Response<()>> {
        let (_span, pending) = gaxi::client_request_signals!(
            metric: self.duration.clone(),
            info: *info::INSTRUMENTATION_CLIENT_INFO,
            method: "client::Spanner::rollback",
            self.inner.rollback(req, options));
        pending.await
    }

    #[tracing::instrument(level = tracing::Level::DEBUG, ret)]
    async fn partition_query(
        &self,
        req: crate::model::PartitionQueryRequest,
        options: crate::RequestOptions,
    ) -> Result<crate::Response<crate::model::PartitionResponse>> {
        let (_span, pending) = gaxi::client_request_signals!(
            metric: self.duration.clone(),
            info: *info::INSTRUMENTATION_CLIENT_INFO,
            method: "client::Spanner::partition_query",
            self.inner.partition_query(req, options));
        pending.await
    }

    #[tracing::instrument(level = tracing::Level::DEBUG, ret)]
    async fn partition_read(
        &self,
        req: crate::model::PartitionReadRequest,
        options: crate::RequestOptions,
    ) -> Result<crate::Response<crate::model::PartitionResponse>> {
        let (_span, pending) = gaxi::client_request_signals!(
            metric: self.duration.clone(),
            info: *info::INSTRUMENTATION_CLIENT_INFO,
            method: "client::Spanner::partition_read",
            self.inner.partition_read(req, options));
        pending.await
    }
}

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 = "spanner";
        info.client_version = VERSION;
        info.client_artifact = NAME;
        info.default_host = "spanner";
        info
    });
}