Enable happy eyeballs when using hickory-dns

Ported from seanmonstar/reqwest#2378

Signed-off-by: Jiahao XU <30436523+NobodyXu@users.noreply.github.com>
This commit is contained in:
Jiahao XU 2024-08-11 16:34:59 +10:00 committed by GitHub
parent 43012ceb2c
commit 651a82a4da
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -1,6 +1,6 @@
use std::{net::SocketAddr, sync::Arc}; use std::{net::SocketAddr, sync::Arc};
use hickory_resolver::TokioAsyncResolver; use hickory_resolver::{system_conf, TokioAsyncResolver};
use once_cell::sync::OnceCell; use once_cell::sync::OnceCell;
use reqwest::dns::{Addrs, Name, Resolve, Resolving}; use reqwest::dns::{Addrs, Name, Resolve, Resolving};
use tracing::{debug, instrument, warn}; use tracing::{debug, instrument, warn};
@ -26,15 +26,14 @@ impl Resolve for TrustDnsResolver {
} }
} }
#[instrument] #[cfg(unix)]
fn new_resolver() -> Result<TokioAsyncResolver, BoxError> { fn get_configs() -> Result<(ResolverConfig, ResolverOpts), BoxError> {
#[cfg(unix)]
{
debug!("Using system DNS resolver configuration"); debug!("Using system DNS resolver configuration");
Ok(TokioAsyncResolver::tokio_from_system_conf()?) system_conf::read_system_conf().map_err(Into::into)
} }
#[cfg(windows)]
{ #[cfg(windows)]
fn get_configs() -> Result<(ResolverConfig, ResolverOpts), BoxError> {
debug!("Using custom DNS resolver configuration"); debug!("Using custom DNS resolver configuration");
let mut config = ResolverConfig::new(); let mut config = ResolverConfig::new();
let opts = ResolverOpts::default(); let opts = ResolverOpts::default();
@ -55,9 +54,16 @@ fn new_resolver() -> Result<TokioAsyncResolver, BoxError> {
} }
}); });
debug!("Resolver configuration complete");
Ok(TokioAsyncResolver::tokio(config, opts)) Ok(TokioAsyncResolver::tokio(config, opts))
} }
#[instrument]
fn new_resolver() -> Result<TokioAsyncResolver, BoxError> {
let (config, mut opts) = get_configs()?;
debug!("Resolver configuration complete");
opts.ip_strategy = LookupIpStrategy::Ipv4AndIpv6;
Ok(TokioAsyncResolver::tokio(config, opts))
} }
#[cfg(windows)] #[cfg(windows)]