google_cloud_spanner/apiv1/
conn_pool.rs

1use google_cloud_gax::conn::{ConnectionManager as GRPCConnectionManager, ConnectionOptions, Environment, Error};
2use google_cloud_googleapis::spanner::v1::spanner_client::SpannerClient;
3
4use crate::apiv1::spanner_client::Client;
5
6pub const AUDIENCE: &str = "https://spanner.googleapis.com/";
7pub const SPANNER: &str = "spanner.googleapis.com";
8pub const SCOPES: [&str; 2] = [
9    "https://www.googleapis.com/auth/cloud-platform",
10    "https://www.googleapis.com/auth/spanner.data",
11];
12
13pub struct ConnectionManager {
14    inner: GRPCConnectionManager,
15}
16
17impl ConnectionManager {
18    pub async fn new(
19        pool_size: usize,
20        environment: &Environment,
21        domain: &str,
22        conn_options: &ConnectionOptions,
23    ) -> Result<Self, Error> {
24        Ok(ConnectionManager {
25            inner: GRPCConnectionManager::new(pool_size, domain, AUDIENCE, environment, conn_options).await?,
26        })
27    }
28
29    pub fn num(&self) -> usize {
30        self.inner.num()
31    }
32
33    pub fn conn(&self) -> Client {
34        let conn = self.inner.conn();
35        Client::new(SpannerClient::new(conn))
36    }
37}