Lucio FrancoandGitHub
eba7ec7b32
This change introduces proper gRPC interceptors that are avilable regardless of the transport used. Each codegen service now produces an additional method called `with_interceptor` that accepts a `Interceptor`. All examples have been updated to use this new style and interop has a custom `tower::Service` middleware to echo the headers. There is also a new `interceptor` example that shows basic usage. BREAKING CHANGE: removed `interceptor_fn` and `intercep_headers_fn` from `transport` in favor of using `tonic::Interceptor`.
57 lines
1.6 KiB
Rust
57 lines
1.6 KiB
Rust
pub mod api {
|
|
tonic::include_proto!("google.pubsub.v1");
|
|
}
|
|
|
|
use api::{publisher_client::PublisherClient, ListTopicsRequest};
|
|
use tonic::{
|
|
metadata::MetadataValue,
|
|
transport::{Certificate, Channel, ClientTlsConfig},
|
|
Request,
|
|
};
|
|
|
|
const ENDPOINT: &str = "https://pubsub.googleapis.com";
|
|
|
|
#[tokio::main]
|
|
async fn main() -> Result<(), Box<dyn std::error::Error>> {
|
|
let token = std::env::var("GCP_AUTH_TOKEN").map_err(|_| {
|
|
"Pass a valid 0Auth bearer token via `GCP_AUTH_TOKEN` environment variable.".to_string()
|
|
})?;
|
|
|
|
let project = std::env::args()
|
|
.skip(1)
|
|
.next()
|
|
.ok_or("Expected a project name as the first argument.".to_string())?;
|
|
|
|
let bearer_token = format!("Bearer {}", token);
|
|
let header_value = MetadataValue::from_str(&bearer_token)?;
|
|
|
|
let certs = tokio::fs::read("examples/data/gcp/roots.pem").await?;
|
|
|
|
let tls_config = ClientTlsConfig::new()
|
|
.ca_certificate(Certificate::from_pem(certs.as_slice()))
|
|
.domain_name("pubsub.googleapis.com");
|
|
|
|
let channel = Channel::from_static(ENDPOINT)
|
|
.tls_config(tls_config)
|
|
.connect()
|
|
.await?;
|
|
|
|
let mut service = PublisherClient::with_interceptor(channel, move |mut req: Request<()>| {
|
|
req.metadata_mut()
|
|
.insert("authorization", header_value.clone());
|
|
Ok(req)
|
|
});
|
|
|
|
let response = service
|
|
.list_topics(Request::new(ListTopicsRequest {
|
|
project: format!("projects/{0}", project),
|
|
page_size: 10,
|
|
..Default::default()
|
|
}))
|
|
.await?;
|
|
|
|
println!("RESPONSE={:?}", response);
|
|
|
|
Ok(())
|
|
}
|