use std::time::Duration;
use surf_::{http::headers as SurfHeaders, Client as SurfClient};
use super::thread::TransportThread;
use crate::{sentry_debug, ClientOptions, Envelope, Transport};
#[cfg_attr(doc_cfg, doc(cfg(feature = "surf")))]
pub struct SurfHttpTransport {
thread: TransportThread,
}
impl SurfHttpTransport {
pub fn new(options: &ClientOptions) -> Self {
Self::new_internal(options, None)
}
pub fn with_client(options: &ClientOptions, client: SurfClient) -> Self {
Self::new_internal(options, Some(client))
}
fn new_internal(options: &ClientOptions, client: Option<SurfClient>) -> Self {
let client = client.unwrap_or_else(SurfClient::new);
let dsn = options.dsn.as_ref().unwrap();
let user_agent = options.user_agent.to_owned();
let auth = dsn.to_auth(Some(&user_agent)).to_string();
let url = dsn.envelope_api_url().to_string();
let thread = TransportThread::new(move |envelope, mut rl| {
let mut body = Vec::new();
envelope.to_writer(&mut body).unwrap();
let request = client.post(&url).header("X-Sentry-Auth", &auth).body(body);
async move {
match request.await {
Ok(mut response) => {
if let Some(retry_after) = response
.header(SurfHeaders::RETRY_AFTER)
.map(|x| x.as_str())
{
rl.update_from_retry_after(retry_after);
}
if let Some(sentry_header) =
response.header("x-sentry-rate-limits").map(|x| x.as_str())
{
rl.update_from_retry_after(sentry_header);
}
match response.body_string().await {
Err(err) => {
sentry_debug!("Failed to read sentry response: {}", err);
}
Ok(text) => {
sentry_debug!("Get response: `{}`", text);
}
}
}
Err(err) => {
sentry_debug!("Failed to send envelope: {}", err);
}
}
rl
}
});
Self { thread }
}
}
impl Transport for SurfHttpTransport {
fn send_envelope(&self, envelope: Envelope) {
self.thread.send(envelope)
}
fn flush(&self, timeout: Duration) -> bool {
self.thread.flush(timeout)
}
fn shutdown(&self, timeout: Duration) -> bool {
self.flush(timeout)
}
}