Skip to content

Commit c9971be

Browse files
refactor: simplify probeerror
1 parent fd0b2b6 commit c9971be

File tree

1 file changed

+17
-18
lines changed

1 file changed

+17
-18
lines changed

iroh/src/net_report/reportgen.rs

Lines changed: 17 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -418,12 +418,12 @@ impl Actor {
418418
let res = fut.await;
419419
let res = match res {
420420
Some(Ok(Ok(report))) => Ok(report),
421-
Some(Ok(Err(ProbeError::Error(err, _probe)))) => {
421+
Some(Ok(Err(ProbeError::Error(err)))) => {
422422
warn!("probe failed: {:#}", err);
423423
Err(err)
424424
}
425-
Some(Ok(Err(ProbeError::AbortSet(err, probe)))) => {
426-
debug!(?probe, "probe set aborted: {:#}", err);
425+
Some(Ok(Err(ProbeError::AbortSet(err)))) => {
426+
debug!("probe set aborted: {:#}", err);
427427
set_token.cancel();
428428
Err(err)
429429
}
@@ -478,9 +478,9 @@ impl ProbeReport {
478478
#[derive(Debug)]
479479
enum ProbeError {
480480
/// Abort the current set.
481-
AbortSet(anyhow::Error, Probe),
481+
AbortSet(anyhow::Error),
482482
/// Continue the other probes in the set.
483-
Error(anyhow::Error, Probe),
483+
Error(anyhow::Error),
484484
}
485485

486486
/// Pieces needed to do QUIC address discovery.
@@ -532,7 +532,7 @@ async fn run_probe(
532532
}
533533
Ok(report)
534534
}
535-
Err(err) => Err(ProbeError::Error(err, probe)),
535+
Err(err) => Err(ProbeError::Error(err)),
536536
}
537537
}
538538

@@ -550,7 +550,7 @@ async fn run_probe(
550550
_ => unreachable!(),
551551
}
552552
.context("no relay node addr")
553-
.map_err(|e| ProbeError::AbortSet(e, probe.clone()))?;
553+
.map_err(ProbeError::AbortSet)?;
554554

555555
let url = node.url.clone();
556556
let report = run_quic_probe(
@@ -563,10 +563,10 @@ async fn run_probe(
563563
.await?;
564564
Ok(report)
565565
}
566-
None => Err(ProbeError::AbortSet(
567-
anyhow!("No QUIC endpoint for {}", probe.proto()),
568-
probe,
569-
)),
566+
None => Err(ProbeError::AbortSet(anyhow!(
567+
"No QUIC endpoint for {}",
568+
probe.proto()
569+
))),
570570
}
571571
}
572572
}
@@ -602,18 +602,17 @@ async fn run_quic_probe(
602602
let host = match url.host_str() {
603603
Some(host) => host,
604604
None => {
605-
return Err(ProbeError::Error(
606-
anyhow!("URL must have 'host' to use QUIC address discovery probes"),
607-
probe.clone(),
608-
));
605+
return Err(ProbeError::Error(anyhow!(
606+
"URL must have 'host' to use QUIC address discovery probes"
607+
)));
609608
}
610609
};
611610
let quic_client = iroh_relay::quic::QuicClient::new(quic_config.ep, quic_config.client_config)
612-
.map_err(|e| ProbeError::Error(e, probe.clone()))?;
611+
.map_err(ProbeError::Error)?;
613612
let (addr, latency) = quic_client
614613
.get_addr_and_latency(relay_addr, host)
615614
.await
616-
.map_err(|e| ProbeError::Error(e, probe.clone()))?;
615+
.map_err(ProbeError::Error)?;
617616
let mut result = ProbeReport::new(probe.clone());
618617
if matches!(probe, Probe::QadIpv4 { .. }) {
619618
result.ipv4_can_send = true;
@@ -935,7 +934,7 @@ mod tests {
935934
{
936935
Ok(probe) => probe,
937936
Err(e) => match e {
938-
ProbeError::AbortSet(err, _) | ProbeError::Error(err, _) => {
937+
ProbeError::AbortSet(err) | ProbeError::Error(err) => {
939938
return Err(err.into());
940939
}
941940
},

0 commit comments

Comments
 (0)