forked from hyperium/tonic
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathclient.rs
More file actions
56 lines (44 loc) · 1.67 KB
/
client.rs
File metadata and controls
56 lines (44 loc) · 1.67 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
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()
.nth(1)
.ok_or_else(|| "Expected a project name as the first argument.".to_string())?;
let bearer_token = format!("Bearer {token}");
let header_value: MetadataValue<_> = bearer_token.parse()?;
let data_dir = std::path::PathBuf::from_iter([std::env!("CARGO_MANIFEST_DIR"), "data"]);
let certs = std::fs::read_to_string(data_dir.join("gcp/roots.pem"))?;
let tls_config = ClientTlsConfig::new()
.ca_certificate(Certificate::from_pem(certs))
.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/{project}"),
page_size: 10,
..Default::default()
}))
.await?;
println!("RESPONSE={response:?}");
Ok(())
}